from bottle import route, run, error, request, response
import sqlite3
from PIL import Image, ImageFont
from PIL import ImageDraw
from io import BytesIO


PAGES = ['/', 'hello', '404']


def get_counter_image(count):
    img = Image.open("j.png")
    font = ImageFont.truetype("ARIAL.TTF", 100)
    draw = ImageDraw.Draw(img)
    print(count)
    draw.text((666, 400), str(count), (255, 255, 255), font=font)
    buf = BytesIO()
    img.save(buf, 'PNG')
    img.close()
    buf.seek(0)
    return buf


class CounterAPI:
    def __init__(self, memory=False):
        self.conn = sqlite3.connect(':memory:' if memory else 'counters.db')
        with self.conn as conn:
            conn.execute('CREATE TABLE IF NOT EXISTS counters (ip text, page text)')

    def put_current_client(self, visited_page):
        with self.conn as conn:
            conn.execute(
                'INSERT INTO COUNTERS(ip, page) VALUES (?, ?)',
                (request.environ.get('REMOTE_ADDR'), visited_page)
            )


API = CounterAPI()


@route('/')
def index():
    API.put_current_client('/')
    name = 'Anonymous'
    if request.cookies.get('name'):
        name = request.cookies.get('name')
    return '<b>Hello {}</b>'.format(name)


@route('/counts')
def counts():
    API.put_current_client('counts')
    values = {}
    with API.conn as conn:
        cur = conn.execute('SELECT * FROM COUNTERS')
        data = cur.fetchall()
        for ip, page in data:
            key = (ip, page)
            values[key] = values[key] + 1 if key in values else 1
    template = '<table>'
    for visits in values:
        template += '<tr><td>{}</td><td>{}</td><td>{}</td></tr>'.format(
            visits[0], visits[1], values[visits]
        )
    return template + '</table>'


@route('/hello/<name>')
def save_name(name):
    API.put_current_client('hello')
    response.set_cookie('name', name, path='/')
    return 'OK'


@route('/upload', method='post')
def save_file():
    upload = request.files.get('img')
    upload.save(upload.filename)
    return 'OK'


@route('/counter')
def get_counter():
    API.put_current_client('counter')
    values = {}
    with API.conn as conn:
        cur = conn.execute('SELECT * FROM COUNTERS')
        data = cur.fetchall()
        for ip, page in data:
            key = (ip, page)
            values[key] = values[key] + 1 if key in values else 1
    counts = sum(values.values())
    image = get_counter_image(counts)
    response.content_type = 'image/png'
    response.headers['Content-Disposition'] = 'attachment;filename=counter.png'
    with image as file:
        return file.read()


@error(404)
def err404(err):
    API.put_current_client('404')
    return '{}, just for you'.format(err)


run(host='localhost', port=8080, debug=True)