import json
from datetime import datetime

import feedparser
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from fastapi_utils.tasks import repeat_every

from prophet import view
from prophet.config import AppConfig
from prophet.domain.improvement import Improvement
from prophet.domain.improvement_repo import IImprovementRepo
from prophet.domain.original import Original
from prophet.infra.improvement_pickle_repo import ImprovementPickleRepo
from prophet.infra.improvement_supa_repo import ImprovementSupaRepo
from prophet.infra.llm_groq import GroqClient

BEE_FEED = "https://babylonbee.com/feed"
BEE_FEED_TEST = "test/resources/feed_short.atom"  # NOTE: Switch out when done testing

REFRESH_PERIOD = 3600  # between fetching articles, in seconds
NUM_ARTICLES_TO_KEEP = 50

llm: GroqClient = GroqClient()
repo: IImprovementRepo = ImprovementSupaRepo()


def grab_latest_originals() -> list[Original]:
    feed: feedparser.FeedParserDict = feedparser.parse(BEE_FEED)  # noqa: F841
    results: list[Original] = []
    for entry in feed.entries:
        o = Original(
            title=entry.title,
            summary=entry.summary,
            link=entry.link,
            date=datetime.strptime(entry.published, "%a, %d %b %Y %H:%M:%S %z"),
        )
        results.append(o)
    return results


def keep_only_new_originals(
    additional: list[Original], existing: list[Original] | None = None
):
    if not existing:
        existing = [e.original for e in repo.get_all()]

    existing_hashes = set([e.id for e in existing])

    remaining: list[Original] = []
    for new in additional:
        if new.id not in existing_hashes:
            remaining.append(new)

    return remaining


def improve_originals(originals: list[Original]) -> list[Improvement]:
    improvements: list[Improvement] = []
    for orig in originals:
        new_title = llm.rewrite_title(orig.title)
        new_summary = llm.rewrite_summary(orig, new_title)

        improvements.append(
            Improvement(original=orig, title=new_title, summary=new_summary)
        )
    return improvements


def init() -> FastAPI:
    app = FastAPI()
    app.mount("/static", StaticFiles(directory="static"), name="static")

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

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


app = init()


@app.get("/improve-title")
def improve_headline(content: str):
    return llm.rewrite_title(content)


@app.get("/improve-summary")
def improve_summary(original_title: str, new_title: str, original_summary: str):
    o = Original(
        title=original_title, summary=original_summary, link="", date=datetime.now()
    )
    return llm.rewrite_summary(o, new_title)


# TODO: Switch to lifecycle events to avoid deprecated method
@app.on_event("startup")
@repeat_every(seconds=REFRESH_PERIOD)
async def refresh_articles():
    _ = await fetch_update()
    truncate_to(NUM_ARTICLES_TO_KEEP)


def truncate_to(max_num: int = 50):
    all = repo.get_all()
    if len(all) > max_num:
        to_delete = all[max_num:]
        to_delete_ids = [a.id for a in to_delete]
        try:
            _ = repo.remove_all(to_delete_ids)
        except ValueError:
            print(f"Error deleting articles with IDs: {id}")


@app.get("/update")
async def fetch_update(debug_print: bool = True):
    adding = keep_only_new_originals(grab_latest_originals())
    improved = improve_originals(adding)
    repo.add_all(improved)
    if debug_print:
        print(f"Updated articles. Added {len(improved)} new ones.")
    return json.dumps(improved)


def start() -> None:
    from uvicorn import run

    config = AppConfig.from_env()

    run("prophet.app:app", reload=config.DEVMODE, host="0.0.0.0", port=config.PORT)


if __name__ == "__main__":
    # start()

    ## ADD MANUALLY
    # adding = keep_only_new_originals(grab_latest_originals())
    # improved = improve_originals(adding)
    # save_new_improvements(improved)

    ## SHOW ALL
    improved = repo.get_all()
    for imp in improved:
        imp.original.__post_init__()
        print(f"Old Title: {imp.original.title}")
        print(f"Old Summary: {imp.original.summary}")
        print(f"Old picture: {imp.original.image_link}")
        print("\n")
        print(f"Title: {imp.title}")
        print(f"Summary: {imp.summary}")

        print("-" * 50)
    # repo.add_all(improved)

    ## DELETE TOO_MANY
    # truncate_to(48)