from fastapi import FastAPI, File, UploadFile, HTTPException
from fastapi.responses import HTMLResponse
from typing import Union

from .epub import epub2html
from .fb2 import fb22html
from .utils import HTMLBook

app = FastAPI()


@app.get("/")
def root():
    return "Hello, World!"


@app.post("/uploadfile/", response_model=HTMLBook)
async def create_upload_file(file: UploadFile = File(...)):
    if file.filename.endswith(".epub"):
        content = await epub2html(file.file)
    elif file.filename.endswith(".fb2"):
        content = await fb22html(file.file)
    else:
        raise HTTPException(status_code=415, detail="Error! Unsupported file type")
    return content