from fastapi import FastAPI, Request, HTTPException
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates

from sqlmodel import SQLModel, Session, create_engine, select

import json
from models import ordinateur

app = FastAPI()
#ordi1 = ordinateur()

app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")

#sqlitemodel
sqlite_file_name = "mydb.db"
sqlite_url = f"sqlite:///mydb.db"

engine = create_engine(sqlite_url)

SQLModel.metadata.create_all(engine)

@app.get("/ordi/{ordi_id}", response_class=HTMLResponse)
async def get_ordi1_info(request: Request, ordi_id: int):
   with Session(engine) as session:
      statement= select(ordinateur).where(ordinateur.id == ordi_id)
      this_ordi = session.exec(statement).first()
   return templates.TemplateResponse(
      request=request, name="item.html", context={"ordi": this_ordi}
    )


@app.post("/endpoint")
async def receive_info(request: Request):
   # Lire le body brut
   body = await request.body()
   print(body)


   # Parser le JSON
   try:
      data = json.loads(body)
   except json.JSONDecodeError:
      raise HTTPException(status_code=400, detail="Invalid JSON")


      # Debug
   hardware = data.get('HARDWARE',{})
   software = data.get('SOFTWARE',{})

   with Session(engine) as session:
         statement = select(ordinateur).where(ordinateur.mac_address == hardware.get("mac_address", ""))
         ordi1 = session.exec(statement).first()
   
   is_new = False
   if not ordi1: 
      ordi1 = ordinateur()
      is_new = True

   print("Infos reçues :", data)
   ordi1.mb_serial = hardware.get("mb_serial", "")
   ordi1.hostname = hardware.get("hostname", "")
   ordi1.cpu = hardware.get("cpu", "")
   ordi1.cpu_id = hardware.get("cpu_id", "")
   ordi1.memory_mb = hardware.get("memory_mb", "")
   ordi1.ram_size = hardware.get("ram_size", "")
   ordi1.ram_gen = hardware.get("ram_gen", "")
   ordi1.sizes = hardware.get("sizes", "")
   ordi1.cpu_freq_min = hardware.get("cpu_freq_min", "")
   ordi1.cpu_freq_cur = hardware.get("cpu_freq_cur", "")
   ordi1.cpu_freq_max = hardware.get("cpu_freq_max", "")
   ordi1.gpu_model = hardware.get("gpu_model", "")
   ordi1.chassis_serial = hardware.get("chassis_serial", "")
   ordi1.cpu_cores_nb = hardware.get("cpu_cores_nb", "")
   ordi1.cpu_threads_nb = hardware.get("cpu_threads_nb", "")
   ordi1.ram_number = hardware.get("ram_number", "")
   ordi1.ram_slots_nb = hardware.get("ram_slots_nb", "")
   ordi1.mac_address = hardware.get("mac_address", "")
   
   ordi1.os = software.get("os", "")
   ordi1.arch = software.get("arch", "")
   ordi1.desktop = software.get("desktop", "")
   ordi1.wm = software.get("wm", "")
   ordi1.kernel = software.get("kernel", "")
   
   if is_new:
      session.add(ordi1)
   session.commit()
   #session.close()

   return ({"status": "ok"})