#!/usr/bin/env python3
from flask import Flask, jsonify, render_template, Response
from mimetypes import types_map as mimetypes
from zipfile import ZipFile
import logging
import os
import sys
import webbrowser
import werkzeug.exceptions as exceptions

# disable access log
log = logging.getLogger("werkzeug")
log.setLevel(logging.ERROR)
# disable flask startup message
os.environ["WERKZEUG_RUN_MAIN"] = "true"

mimetypes[".webp"] = "image/webp"
archive = ZipFile(sys.argv[1], "r")
app = Flask(
    __name__, static_folder="frontend/dist/static", template_folder="frontend/dist"
)

# kind of redundant, but avoids returning 200 if page does not exist
@app.route("/")
@app.route("/series/0")
@app.route("/volume/0")
def serve_app(**kwargs):
    return render_template("index.html")


@app.route("/api/volume/0")
def get_volume_info():
    return jsonify({"pages": len(archive.filelist), "series": 0, "title": sys.argv[1]})


@app.route("/api/volume/0/page/<int:page_number>")
def get_volume_page(page_number):
    try:
        page = archive.filelist[page_number - 1]
    except IndexError:
        raise exceptions.NotFound() from None
    mimetype = mimetypes[os.path.splitext(page.filename)[1]]
    return Response(archive.read(page.filename), mimetype=mimetype)


if __name__ == "__main__":
    webbrowser.open("http://localhost:5051/volume/0")
    app.run(host="127.0.0.1", port=5051)