from typing import Annotated
from fastapi import APIRouter, Depends, HTTPException, Path
from sqlalchemy.orm import Session
from pydantic import BaseModel, Field
from starlette import status

from database import SessionLocal
from models import Post, Thread

router = APIRouter(
    tags=['forum']
)


def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()


db_dependency = Annotated[Session, Depends(get_db)]


class PostCreate(BaseModel):
    author: str = Field('anon')
    title: str = Field('')
    content: str = Field('')


@router.get('/catalog', status_code=status.HTTP_200_OK)
async def get_catalog(db: db_dependency):
    return db.query(Thread).all()


@router.get('/', status_code=status.HTTP_200_OK)
async def get_posts(db: db_dependency):
    return db.query(Post).all()


@router.get('/{thread_id}', status_code=status.HTTP_200_OK)
async def get_thread_by_id(db: db_dependency, thread_id: int = Path(gt=0)):
    posts = db.query(Post).filter(Post.thread_id == thread_id).all()
    if posts:
        return posts

    raise HTTPException(404, f'Could not find thread')

@router.post('/', status_code=status.HTTP_201_CREATED)
async def create_thread(db: db_dependency, data: PostCreate):
    try:
        # Create the post
        post = Post(
            author=data.author,
            title=data.title,
            content=data.content
        )
        db.add(post)
        db.flush()

        # Create the thread
        thread = Thread(
            author=post.author,
            title=post.title,
            content=post.content
        )
        db.add(thread)
        db.flush()

        # Update the Post with thread_id
        post.thread_id = thread.id
        db.commit()

        return {
            'id': post.id,
            'author': post.author,
            'title': post.title,
            'content': post.content
        }
    except Exception as e:
        db.rollback()
        raise HTTPException(status_code=400, detail=str(e))


@router.post('/{thread_id}', status_code=status.HTTP_201_CREATED)
async def create_reply(db: db_dependency, data: PostCreate, thread_id: int = Path(gt=0)):
    thread = db.query(Thread).filter(Thread.id == thread_id).first()
    if thread:
        post = Post(
            thread_id=thread_id,
            author=data.author,
            title=data.title,
            content=data.content
        )
        db.add(post)
        db.commit()

        return {
            'id': post.id,
            'author': post.author,
            'title': post.title,
            'content': post.content
        }

    raise HTTPException(status_code=404, detail='Could not find thread')