#!/usr/bin/env python3
from backend import CalibreDB
from flask import Flask, jsonify, send_from_directory, send_file, render_template, request
from flask_cors import CORS
from werkzeug.routing import BaseConverter
import mimetypes
import os


def send_from_cwd(filename):
    return send_from_directory(os.getcwd(), filename)


app = Flask(__name__, static_folder='frontend/dist/static', template_folder='frontend/dist')
CORS(app)

db = CalibreDB()

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


@app.route('/api/series')
def get_series_list():
    return jsonify(list(db.get_series_list()))


@app.route('/api/series/<int:series_id>/cover')
def get_series_cover(series_id):
    return send_from_cwd(db.get_series_cover(series_id))


@app.route('/api/series/<int:series_id>/cover/thumbnail')
def get_series_cover_thumbnail(series_id):
    thumbnail = db.get_series_cover_thumbnail(series_id)
    return send_file(thumbnail, mimetype='image/jpeg')


@app.route('/api/series/<int:series_id>')
def get_series_info(series_id):
    return jsonify(db.get_series_volumes(series_id))


@app.route('/api/volume/<int:volume_id>/cover')
def get_volume_cover(volume_id):
    return send_from_cwd(db.get_volume_cover(volume_id))


@app.route('/api/volume/<int:volume_id>/cover/thumbnail')
def get_volume_cover_thumbnail(volume_id):
    thumbnail = db.get_volume_cover_thumbnail(volume_id)
    return send_file(thumbnail, mimetype='image/jpeg')


@app.route('/api/volume/<int:volume_id>')
def get_volume_info(volume_id):
    return jsonify(db.get_volume_info(volume_id))


@app.route('/api/volume/<int:volume_id>/page/<int:page_number>')
def get_volume_page(volume_id, page_number):
    page = db.get_volume_page(volume_id, page_number)
    mimetype = mimetypes.types_map[page['extension']]
    return send_file(page['data'], mimetype=mimetype)


if __name__ == '__main__':
    app.run(host='0.0.0.0')