import os
import random
from io import BytesIO

from flask import Flask, Response, render_template, request, send_file
from flask_caching import Cache
from PIL import Image, ImageOps

app = Flask(__name__)

GREY = "G"
COLOR = "RGB"

cache = Cache(config={"CACHE_TYPE": "SimpleCache"})

app = Flask(__name__)
cache.init_app(app)


def get_cropped_image(x, y, grey=False, retries=0):
    """crops a random image from collection"""
    if retries > 10:
        return None
    im_src = random.choice(os.listdir("./images"))
    im = Image.open(f"images/{im_src}")
    out = BytesIO()
    max_x, max_y = im.size
    if x > max_x and y > max_y:
        return get_cropped_image(x, y, grey, retries + 1)
    im = ImageOps.fit(im, (x, y))
    if grey:
        im = ImageOps.grayscale(im)
    im.save(out, "WEBP", quality=80)
    out.seek(0)
    return out


def make_response(x, y, color_mode=COLOR):
    im = get_cropped_image(x, y, color_mode == GREY)
    if not im:
        return Response(status=401)
    return send_file(im, mimetype="image/webp")


@app.route("/")
def hello():
    u = request.host
    return render_template("index.html", url=u)


@app.route("/bookmarklet")
def bookmarklet():
    u = request.host
    return render_template("bookmarklet.html", url=u)


@app.route("/<int:x>/<int:y>")
@cache.cached(1)
def generate(x, y):
    return make_response(x, y, COLOR)


@app.route("/g/<int:x>/<int:y>")
@cache.cached(1)
def generate_grey(x, y):
    return make_response(x, y, GREY)


if __name__ == "__main__":
    port = int(os.environ.get("PORT", 5000))
    app.run(host="0.0.0.0", port=port)