core/resolvers/bookmark.py

from operator import and_

from graphql import GraphQLError
from sqlalchemy import delete, insert

from orm.author import AuthorBookmark
from orm.shout import Shout
from resolvers.feed import apply_options
from resolvers.reader import get_shouts_with_links, query_with_stat
from services.auth import login_required
from services.common_result import CommonResult
from services.db import local_session
from services.schema import mutation, query


@query.field("load_shouts_bookmarked")
@login_required
def load_shouts_bookmarked(_, info, options):
    """
    Load bookmarked shouts for the authenticated user.

    Args:
        limit (int): Maximum number of shouts to return.
        offset (int): Number of shouts to skip.

    Returns:
        list: List of bookmarked shouts.
    """
    author_dict = info.context.get("author", {})
    author_id = author_dict.get("id")
    if not author_id:
        raise GraphQLError("User not authenticated")

    q = query_with_stat(info)
    q = q.join(AuthorBookmark)
    q = q.filter(
        and_(
            Shout.id == AuthorBookmark.shout,
            AuthorBookmark.author == author_id,
        )
    )
    q, limit, offset = apply_options(q, options, author_id)
    return get_shouts_with_links(info, q, limit, offset)


@mutation.field("toggle_bookmark_shout")
def toggle_bookmark_shout(_, info, slug: str) -> CommonResult:
    """
    Toggle bookmark status for a specific shout.

    Args:
        slug (str): Unique identifier of the shout.

    Returns:
        CommonResult: Result of the operation with bookmark status.
    """
    author_dict = info.context.get("author", {})
    author_id = author_dict.get("id")
    if not author_id:
        raise GraphQLError("User not authenticated")

    with local_session() as db:
        shout = db.query(Shout).filter(Shout.slug == slug).first()
        if not shout:
            raise GraphQLError("Shout not found")

        existing_bookmark = (
            db.query(AuthorBookmark)
            .filter(AuthorBookmark.author == author_id, AuthorBookmark.shout == shout.id)
            .first()
        )

        if existing_bookmark:
            db.execute(
                delete(AuthorBookmark).where(AuthorBookmark.author == author_id, AuthorBookmark.shout == shout.id)
            )
            result = False
        else:
            db.execute(insert(AuthorBookmark).values(author=author_id, shout=shout.id))
            result = True

        db.commit()
        return result