Untitled

from datetime import datetime
from typing import Annotated, List

from fastapi import Depends
from sqlalchemy import update
from strawberry import mutation, type

from app.dependencies.get_db import get_db
from app.graphql.contetx import context_dependency, get_context
from app.graphql.schema import Info
from app.graphql.types.content_type import (
    ContentType,
    CreateContentInput,
    DeleteContentInput,
    UpdateContentInput,
)
from app.models import User, Content


@type
class UpdateMutationPayload:
    updated_contents: List[ContentType]
    success: bool
    message: str = ""


@type
class CreateMutationPayload:
    created_contents: List[ContentType]
    success: bool
    message: str = ""


@type
class DeleteMutationPayload:
    deleted_contents: List[ContentType]
    success: bool
    message: str = ""


@type
class ContentMutation:
    @mutation
    def update_contents(
        self, info: Info, input: List[UpdateContentInput]
    ) -> UpdateMutationPayload:
        # Use the context manager to manage the database session
        try:
            updated_contents_ids = []
            for content in input:
                stmt = (
                    update(Content)
                    .where(Content.id == content.id)
                    .values(
                        type=content.type,
                        text=content.text,
                        link=content.link,
                        order=content.order,
                        updated_date=datetime.now(),  # Automatically set to the current time
                    )
                )
                info.context.session.execute(stmt)
                updated_contents_ids.append(content.id)
            info.context.session.commit()
            updated_contents = (
                info.context.session.query(Content)
                .filter(Content.id.in_(updated_contents_ids))
                .all()
            )
            updated_contents_graphql = [
                ContentType(
                    id=content.id,
                    type=content.type,
                    text=content.text,
                    link=content.link,
                    order=content.order,
                    created_date=content.created_date,
                    created_by=content.created_by,
                    updated_date=content.updated_date,
                    updated_by=content.updated_by,
                    is_deleted=content.is_deleted,
                )
                for content in updated_contents
            ]
            return UpdateMutationPayload(
                updated_contents=updated_contents_graphql,
                success=True,
                message="Content updated successfully",
            )
        except Exception as e:
            info.context.session.rollback()
            return UpdateMutationPayload(
                updated_contents=[], success=False, message=f"Update failed: {str(e)}"
            )

    @mutation
    def create_contents(
        self, info: Info, input: List[CreateContentInput]
    ) -> CreateMutationPayload:
        try:
            current_user = info.context.current_user.login
            created_content_ids = []
            for content in input:
                new_content = Content(
                    type=content.type,
                    text=content.text,
                    link=content.link,
                    order=content.order,
                    created_date=datetime.now(),
                    created_by=current_user,
                    updated_date=datetime.now(),
                    updated_by=current_user,
                    is_deleted=0,
                )
                info.context.session.add(new_content)
                info.context.session.flush()  # Ensures the new content gets an ID before commit
                created_content_ids.append(new_content.id)
            info.context.session.commit()
            created_contents = (
                info.context.session.query(Content)
                .filter(Content.id.in_(created_content_ids))
                .all()
            )
            created_contents_graphql = [
                ContentType(
                    id=content.id,
                    type=content.type,
                    text=content.text,
                    link=content.link,
                    order=content.order,
                    created_date=content.created_date,
                    created_by=content.created_by,
                    updated_date=content.updated_date,
                    updated_by=content.updated_by,
                    is_deleted=content.is_deleted,
                )
                for content in created_contents
            ]
            return CreateMutationPayload(
                created_contents=created_contents_graphql,
                success=True,
                message="Content created successfully",
            )
        except Exception as e:
            info.context.session.rollback()
            return CreateMutationPayload(
                created_contents=[], success=False, message=f"Creation failed: {str(e)}"
            )

    @mutation
    def delete_contents(
        self, info: Info, input: List[DeleteContentInput]
    ) -> DeleteMutationPayload:
        try:
            deleted_content = []
            total_deleted = 0
            for content in input:
                stmt = (
                    update(Content)
                    .where(Content.id == content.id)
                    .values(is_deleted=True)
                )
                result = info.context.session.execute(stmt)
                total_deleted += result.rowcount
                # Fetch the deleted content object
                deleted_content.append(
                    info.context.session.query(Content).get(content.id)
                )
            info.context.session.commit()

            if total_deleted > 0:
                return DeleteMutationPayload(
                    deleted_contents=deleted_content,
                    success=True,
                    message=f"{total_deleted} content(s) deleted successfully",
                )
            else:
                return DeleteMutationPayload(
                    deleted_contents=[], success=False, message="No content was deleted"
                )
        except Exception as e:
            info.context.session.rollback()
            return DeleteMutationPayload(
                deleted_contents=[], success=False, message=f"Deletion failed: {str(e)}"
            )