import sys
import atexit
import signal
from typing import Union

from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware

from pydantic import BaseModel

import dbHandler
import userHandler
import securityHandler

dbConnection = None

def apiInit():
    dbConnection = dbHandler.connect("blorgdb",
                                     "172.20.0.10",
                                     "dev",
                                     "dev",
                                     "5432")
    dbHandler.initTable(dbConnection, "Users", """
    ID SERIAL PRIMARY KEY,
    Username VARCHAR(255),
    Email VARCHAR(255),
    FirstName VARCHAR(255),
    LastName VARCHAR(255),
    Description VARCHAR(255),
    Country VARCHAR(255),
    Theme VARCHAR(255),
    AccentColor VARCHAR(255),
    PasswordHash VARCHAR(255)
    """)
    dbHandler.initTable(dbConnection, "SignOns", """
    ID SERIAL PRIMARY KEY,
    UserID INTEGER,
    LoginSuccess BOOLEAN,
    DateAttempted TIMESTAMP,
    IPLocationAttempted VARCHAR(255)
    """)
    dbHandler.initTable(dbConnection, "AuthTokens", """
    ID SERIAL PRIMARY KEY,
    Token VARCHAR(255),
    OwnerID INTEGER,
    DateCreated TIMESTAMP,
    DateExpiry TIMESTAMP,
    IPLocationCreated VARCHAR(255)
    """)
    dbHandler.initTable(dbConnection, "Blogs", """
    ID SERIAL PRIMARY KEY,
    AuthorID INTEGER,
    CategoryID INTEGER,
    DatePosted TIMESTAMP,
    Description VARCHAR(255)
    """)
    dbHandler.initTable(dbConnection, "Categories", """
    ID SERIAL PRIMARY KEY,
    Name VARCHAR(255)
    """)
    userHandler.createUser(dbConnection, "testuser", "Test", "User", "A test user", "TestCountry", "TestTheme", "TestColor", "testuser")

def apiCleanup():
    dbConnection.close()

@asynccontextmanager
async def apiLifespan(app: FastAPI):
    # API Init 
    apiInit()
    
    # API Clean up
    yield
    apiCleanup()

app = FastAPI(lifespan=apiLifespan)

origins = [
    "http://localhost",
    "http://localhost:8080",
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

@app.get("/")
def getroot():
    return {"Hello": "World"}

class ApiBody(BaseModel):
    username: str
    password: str
@app.post("/api")
def postapi(body: ApiBody):
    print(body.username)
    print(body.password)
    return body

class loginBody(BaseModel):
    username: str
    password: str
    rememberMe: bool
@app.post("/api/login")
def postlogin(body: loginBody, request: Request):
    try:
        if userHandler.checkUserExistence(dbConnection, loginBody.username):
            userID = userHandler.getIDByUsername(dbConnection, loginBody.username)
            if securityHandler.handlePassword(dbConnection, loginBody.password, userID):
                return {"success": True, "authToken": tokenHandler.createToken(dbConnection, userID, loginBody.rememberMe, request.client.host), "message": "User login success!"}
            else:
                return {"success": False, "authToken": "none", "message": "User login failed! Please check your password."}
        else:
            return {"success": False, "authToken": "none", "message": "User login failed! User does not exist."}
    except:
        return {"success": False, "authToken": "none", "message": "User login failed! Unexpected server error."}

@app.get("/api")
def getapi():
    return {"Hello": "API!"}