from typing import Dict, Optional

import httpx
from pydantic import BaseModel
from sqlmodel import Field, SQLModel

from learn_sql_model.config import config


class HeroBase(SQLModel, table=False):
    name: str
    secret_name: str
    x: int
    y: int
    # size: int
    # age: Optional[int] = None
    # shoe_size: Optional[int] = None

    # pet_id: Optional[int] = Field(default=None, foreign_key="pet.id")
    # pet: Optional[Pet] = Relationship(back_populates="hero")


class Hero(HeroBase, table=True):
    id: Optional[int] = Field(default=None, primary_key=True)


class HeroCreate(HeroBase):
    ...

    def post(self) -> Hero:
        r = httpx.post(
            f"{config.api_client.url}/hero/",
            json=self.dict(),
        )
        if r.status_code != 200:
            raise RuntimeError(f"{r.status_code}:\n {r.text}")

        return Hero.parse_obj(r.json())


class HeroRead(HeroBase):
    id: int

    @classmethod
    def get(
        cls,
        id: int,
    ) -> Hero:
        r = httpx.get(f"{config.api_client.url}/hero/{id}")
        if r.status_code != 200:
            raise RuntimeError(f"{r.status_code}:\n {r.text}")
        return HeroRead.parse_obj(r.json())


class Heros(BaseModel):
    __root__: list[Hero]

    @classmethod
    def list(
        self,
    ) -> Hero:
        r = httpx.get(f"{config.api_client.url}/heros/")
        if r.status_code != 200:
            raise RuntimeError(f"{r.status_code}:\n {r.text}")
        return Heros.parse_obj({"__root__": r.json()})


class HeroUpdate(SQLModel):
    # id is required to update the hero
    id: int

    # all other fields, must match the model, but with Optional default None
    name: Optional[str] = None
    secret_name: Optional[str] = None
    # age: Optional[int] = None
    # shoe_size: Optional[int] = None
    # x: Optional[int]
    # y: Optional[int]

    # pet_id: Optional[int] = Field(default=None, foreign_key="pet.id")
    # pet: Optional[Pet] = Relationship(back_populates="hero")

    def update(self) -> Hero:
        r = httpx.patch(
            f"{config.api_client.url}/hero/",
            json=self.dict(),
        )
        if r.status_code != 200:
            raise RuntimeError(f"{r.status_code}:\n {r.text}")


class HeroDelete(BaseModel):
    id: int

    @classmethod
    def delete(self, id: int) -> Dict[str, bool]:
        r = httpx.delete(
            f"{config.api_client.url}/hero/{id}",
        )
        if r.status_code != 200:
            raise RuntimeError(f"{r.status_code}:\n {r.text}")
        return {"ok": True}