from datetime import datetime, timezone
from typing import override

from supabase import Client

from prophet.config import SupaConfig
from prophet.domain.improvement import Improvement
from prophet.domain.improvement_repo import IImprovementRepo
from prophet.domain.original import Original


class ImprovementSupaRepo(IImprovementRepo):
    config: SupaConfig
    client: Client

    def __init__(self, config: SupaConfig | None = None) -> None:
        self.config = config if config else SupaConfig.from_env()
        self.client = Client(self.config.URL, self.config.KEY)
        try:
            _ = self.client.table(self.config.TABLE).select("*").limit(1)
        except Exception as e:
            print(e)

    @override
    def add(self, improvement: Improvement) -> None:
        _ = (
            self.client.table(self.config.TABLE)
            .insert(self._to_tbl_row(improvement))
            .execute()
        )

    @override
    def add_all(self, improvements: list[Improvement]) -> None:
        _ = (
            self.client.table(self.config.TABLE)
            .insert([self._to_tbl_row(i) for i in improvements])
            .execute()
        )

    @override
    def get(self, id: str) -> Improvement:
        return self._from_tbl_row(
            self.client.table(self.config.TABLE)
            .select("*")
            .eq("uuid", id)
            .execute()
            .data[0]
        )

    @override
    def get_all(self, last_n: int | None = None) -> list[Improvement]:
        if not last_n:
            sql = (
                self.client.table(self.config.TABLE)
                .select("*")
                .order("date_orig_ts", desc=True)
            )
        else:
            sql = (
                self.client.table(self.config.TABLE)
                .select("*")
                .order("date_orig_ts", desc=True)
                .limit(last_n)
            )

        return [self._from_tbl_row(row) for row in sql.execute().data]

    @override
    def remove(self, id: str) -> Improvement:
        resp = (
            self.client.table(self.config.TABLE).delete().eq("uuid", id).execute().data
        )
        if not resp:
            raise ValueError
        return self._from_tbl_row(resp[0])

    @override
    def remove_all(self, ids: list[str]) -> list[Improvement]:
        resp = (
            self.client.table(self.config.TABLE)
            .delete()
            .in_("uuid", ids)
            .execute()
            .data
        )
        if not resp:
            raise ValueError
        return [self._from_tbl_row(item) for item in resp]

    def _to_tbl_row(self, imp: Improvement) -> dict[str, str | int]:
        return {
            "uuid": imp.id,
            "title": imp.title,
            "summary": imp.summary,
            "title_orig": imp.original.title,
            "summary_orig": imp.original.summary,
            "link_orig": imp.original.link,
            "image_link_orig": imp.original.image_link or "",
            "date_orig_ts": int(imp.original.date.astimezone(timezone.utc).timestamp()),
        }

    def _from_tbl_row(self, row: dict[str, str | int]) -> Improvement:
        return Improvement(
            id=str(row["uuid"]),
            title=str(row["title"]),
            summary=str(row["summary"]),
            original=Original(
                title=str(row["title_orig"]),
                summary=str(row["summary_orig"]),
                link=str(row["link_orig"]),
                date=datetime.fromtimestamp(int(row["date_orig_ts"]), tz=timezone.utc),
                image_link=str(row["image_link_orig"]),
            ),
        )


if __name__ == "__main__":
    # response = supabase.table("improvements").select("*").execute()
    repo = ImprovementSupaRepo()
    print("latest entries:\n- ", "\n- ".join([imp.title for imp in repo.get_all(3)]))

    # from prophet.app import grab_latest_originals
    # latest = grab_latest_originals()
    # print("LATEST", latest)

    # improvements = [
    #     Improvement(
    #         title=f"{o.title} but cool",
    #         summary=f"{o.summary} but really cool",
    #         original=o,
    #     )
    #     for o in latest
    # ]
    # repo.add_all(improvements)
    # print(repo.get_all())
    # print(repo.get("6587d90e-952b-4866-85cc-836cebafcca2"))