#!/usr/bin/env python3
from typing import Any
import uuid

from app import models
from app.database import AsyncSession
from app.activitypub import ME
from app.config import MANUALLY_APPROVES_FOLLOWERS
from app.config import BASE_URL
from app.models import Actor
from app.actor import fetch_actor

import app.activitypub as ap


from sqlalchemy import select
from loguru import logger
from uuid import uuid4



def allocate_outbox_id() -> str:
    return str(uuid.uuid4())


def build_object_id(id) -> str:
    return f"{BASE_URL}/tail/{id}"

async def save_incoming(
        db_session: AsyncSession,
        payload: dict,
) -> models.IncomingActivity | None:
    ap_id: str
    if "@context" not in payload:
        logger.warning(f"invalid object: {payload}")
        return None

    if "id" in payload:
        ap_id = payload["id"]
    else:
        ap_id = str(uuid4())

    incoming_activity = models.IncomingActivity(
        ap_id=ap_id,
        ap_object=payload,
    )
    await process_incoming(db_session, payload)
    if db_session.add(incoming_activity):
        return incoming_activity

    await db_session.commit()
    await db_session.refresh(incoming_activity)
    return incoming_activity


async def process_incoming(
        db_session: AsyncSession,
        ap_object: dict,
) -> bool:
    actor = await fetch_actor(ap_object["actor"], db_session)

    if "Follow" == ap_object["type"]:
        if await _handle_follow(db_session, actor.ap_actor["inbox"], ap_object):
            return True
        return False

    return False


async def _handle_follow(
        db_session : AsyncSession,
        inbox_url : str | Any,
        ap_object : dict,
) -> bool:
    if ME["id"] != ap_object["object"]:
        # await db_session.delete(ap_object)
        logger.warning("no match follow object!" + ap_object["id"])
        return False

    if MANUALLY_APPROVES_FOLLOWERS:
        # TODO
        return False

    await _send_accept(db_session, inbox_url, ap_object)
    return True


async def _send_accept(
        db_session: AsyncSession,
        inbox_url : str | Any,
        ap_object : dict,
) -> None :

    reply_id = allocate_outbox_id()

    url = inbox_url
    out = {
        "@context": ap.AS_CTX,
        "id": build_object_id(reply_id),
        "type": "Accept",
        "actor": ME["id"],
        "object": ap_object["id"],
    }
    #TODO outcoming
    await ap.post(url, out)