mutations.py

"""
Mutations for Clubs
"""

import strawberry
from fastapi.encoders import jsonable_encoder

from db import clubsdb
from models import Club, create_utc_time

# import all models and types
from otypes import (
    FullClubInput,
    FullClubType,
    Info,
    SimpleClubInput,
    SimpleClubType,
)
from utils import (
    check_remove_old_file,
    getUser,
    invalidate_active_clubs_cache,
    invalidate_club_cache,
    update_events_members_cid,
    update_role,
)


@strawberry.mutation
async def createClub(clubInput: FullClubInput, info: Info) -> SimpleClubType:
    """
    Mutation for creation of a new club by CC.

    Args:
        clubInput (otypes.FullClubInput): Full details of the club.
        info (otypes.Info): User metadata and cookies.

    Returns:
        (otypes.SimpleClubType): Details of the created club.

    Raises:
        Exception: Not Authenticated
        Exception: A club with this cid already exists
        Exception: A club with this short code already exists
        Exception: Invalid Club ID/Club Email
        Exception: Error in updating the role for the club
        Exception: Not Authenticated to access this API
    """
    user = info.context.user
    if user is None:
        raise Exception("Not Authenticated")

    role = user["role"]
    club_input = jsonable_encoder(clubInput.to_pydantic())

    if role in ["cc"]:
        club_input["cid"] = club_input["email"].split("@")[0]

        cid_exists = await clubsdb.find_one({"cid": club_input["cid"]})
        if cid_exists:
            raise Exception("A club with this cid already exists")

        # Check whether this cid is valid or not
        clubMember = await getUser(club_input["cid"], info.context.cookies)
        if clubMember is None:
            raise Exception("Invalid Club ID/Club Email")

        code_exists = await clubsdb.find_one({"code": club_input["code"]})
        if code_exists:
            raise Exception("A club with this short code already exists")

        created_record = await clubsdb.insert_one(club_input)
        created_sample = Club.model_validate(
            await clubsdb.find_one({"_id": created_record.inserted_id})
        )

        if not await update_role(club_input["cid"], info.context.cookies):
            raise Exception("Error in updating the role for the club")

        await invalidate_active_clubs_cache()

        return SimpleClubType.from_pydantic(created_sample)

    else:
        raise Exception("Not Authenticated to access this API")


@strawberry.mutation
async def editClub(clubInput: FullClubInput, info: Info) -> FullClubType:
    """
    Mutation for editing of the club details either by that specific club or
    the cc

    This method is used for editing the club details.
    CC can edit any club details, but the club can only edit its own details.
    Only CC can change a clubs name/email and category.


    Args:
        clubInput (otypes.FullClubInput): Full details of the club to be updated to.
        info (otypes.Info): User metadata and cookies.

    Returns:
        (otypes.FullClubType): Full Details of the edited club.

    Raises:
        Exception: Not Authenticated.
        Exception: A club with this code does not exist.
        Exception: Invalid Club ID/Club Email.
        Exception: Error in updating the role/cid.
        Exception: Authentication Error! (CLUB ID CHANGED).
        Exception: You dont have permission to change the name/email of the
                   club. Please contact CC for it.
        Exception: Only CC is allowed to change the category of club.
        Exception: Not Authenticated to access this API.
    """  # noqa: E501
    user = info.context.user
    if user is None:
        raise Exception("Not Authenticated")

    role = user["role"]
    uid = user["uid"]

    club_input = jsonable_encoder(clubInput.to_pydantic())

    if role in ["cc"]:
        exists = await clubsdb.find_one({"code": club_input["code"]})
        if not exists:
            raise Exception("A club with this code doesn't exist")

        # Check whether this cid is valid or not
        clubMember = await getUser(club_input["cid"], info.context.cookies)
        if clubMember is None:
            raise Exception("Invalid Club ID/Club Email")

        club_input["state"] = exists["state"]
        club_input["_id"] = exists["_id"]

        await check_remove_old_file(exists, club_input, "logo")
        await check_remove_old_file(exists, club_input, "banner")
        await check_remove_old_file(exists, club_input, "banner_square")

        await clubsdb.replace_one({"code": club_input["code"]}, club_input)
        if "socials" in club_input.keys():
            await clubsdb.update_one(
                {"code": club_input["code"]},
                {
                    "$set": {
                        "socials.website": club_input["socials"]["website"],
                        "socials.instagram": club_input["socials"][
                            "instagram"
                        ],
                        "socials.facebook": club_input["socials"]["facebook"],
                        "socials.youtube": club_input["socials"]["youtube"],
                        "socials.twitter": club_input["socials"]["twitter"],
                        "socials.linkedin": club_input["socials"]["linkedin"],
                        "socials.discord": club_input["socials"]["discord"],
                        "socials.whatsapp": club_input["socials"]["whatsapp"],
                        "socials.other_links": club_input["socials"][
                            "other_links"
                        ],
                    }
                },
            )
        await clubsdb.update_one(
            {"code": club_input["code"]},
            {
                "$set": {
                    "created_time": exists["created_time"],
                    "updated_time": create_utc_time(),
                }
            },
        )

        await invalidate_club_cache(club_input["cid"])
        await invalidate_active_clubs_cache()

        if exists["cid"] != club_input["cid"]:
            await invalidate_club_cache(exists["cid"])
            return1 = await update_role(
                exists["cid"], info.context.cookies, role="public"
            )
            return2 = await update_role(
                club_input["cid"], info.context.cookies, role="club"
            )
            return3 = await update_events_members_cid(
                exists["cid"], club_input["cid"], cookies=info.context.cookies
            )
            if not return1 or not return2 or not return3:
                raise Exception("Error in updating the role/cid.")

        result = Club.model_validate(
            await clubsdb.find_one({"code": club_input["code"]})
        )
        return FullClubType.from_pydantic(result)

    elif role in ["club"]:
        if uid != club_input["cid"]:
            raise Exception("Authentication Error! (CLUB ID CHANGED)")

        exists = await clubsdb.find_one({"cid": club_input["cid"]})
        if not exists:
            raise Exception("A club with this cid doesn't exist")

        if (
            club_input["name"] != exists["name"]
            or club_input["email"] != exists["email"]
        ):
            raise Exception(
                "You don't have permission to change the name/email of the"
                "club. Please contact CC for it"  # noqa: E501
            )

        if club_input["category"] != exists["category"]:
            raise Exception(
                "Only CC is allowed to change the category of club."
            )

        club_input["state"] = exists["state"]
        club_input["_id"] = exists["_id"]

        await check_remove_old_file(exists, club_input, "logo")
        await check_remove_old_file(exists, club_input, "banner")
        await check_remove_old_file(exists, club_input, "banner_square")

        await clubsdb.replace_one({"cid": uid}, club_input)
        if "socials" in club_input.keys():
            await clubsdb.update_one(
                {"cid": club_input["cid"]},
                {
                    "$set": {
                        "socials.website": club_input["socials"]["website"],
                        "socials.instagram": club_input["socials"][
                            "instagram"
                        ],
                        "socials.facebook": club_input["socials"]["facebook"],
                        "socials.youtube": club_input["socials"]["youtube"],
                        "socials.twitter": club_input["socials"]["twitter"],
                        "socials.linkedin": club_input["socials"]["linkedin"],
                        "socials.discord": club_input["socials"]["discord"],
                        "socials.whatsapp": club_input["socials"]["whatsapp"],
                        "socials.other_links": club_input["socials"][
                            "other_links"
                        ],
                    }
                },
            )

        # also autofills the updated time
        await clubsdb.update_one(
            {"cid": club_input["cid"]},
            {
                "$set": {
                    "created_time": exists["created_time"],
                    "updated_time": create_utc_time(),
                }
            },
        )

        await invalidate_club_cache(club_input["cid"])
        await invalidate_active_clubs_cache()

        result = Club.model_validate(
            await clubsdb.find_one({"cid": club_input["cid"]})
        )
        return FullClubType.from_pydantic(result)

    else:
        raise Exception("Not Authenticated to access this API")


@strawberry.mutation
async def deleteClub(clubInput: SimpleClubInput, info: Info) -> SimpleClubType:
    """
    Mutation for the cc to move a club to deleted state.

    Args:
        clubInput (otypes.SimpleClubInput): The club cid.
        info (otypes.Info): User metadata and cookies.

    Returns:
        (otypes.SimpleClubType): Details of the deleted club.

    Raises:
        Exception: Not Authenticated.
        Exception: Not Authenticated to access this API.
    """
    user = info.context.user
    if user is None:
        raise Exception("Not Authenticated")

    role = user["role"]
    club_input = jsonable_encoder(clubInput)

    if role not in ["cc"]:
        raise Exception("Not Authenticated to access this API")

    # also autofills the updated time
    await clubsdb.update_one(
        {"cid": club_input["cid"]},
        {"$set": {"state": "deleted", "updated_time": create_utc_time()}},
    )

    await update_role(club_input["cid"], info.context.cookies, "public")

    updated_sample = Club.model_validate(
        await clubsdb.find_one({"cid": club_input["cid"]})
    )

    await invalidate_active_clubs_cache()
    await invalidate_club_cache(club_input["cid"])

    return SimpleClubType.from_pydantic(updated_sample)


@strawberry.mutation
async def restartClub(
    clubInput: SimpleClubInput, info: Info
) -> SimpleClubType:
    """
    Mutation for cc to move a club from deleted state to active state.

    Args:
        clubInput (otypes.SimpleClubInput): The club cid.
        info (otypes.Info): User metadata and cookies.

    Returns:
        (otypes.SimpleClubType): Details of the restarted clubs cid.

    Raises:
        Exception: Not Authenticated.
        Exception: Not Authenticated to access this API.
    """
    user = info.context.user
    if user is None:
        raise Exception("Not Authenticated")

    role = user["role"]
    club_input = jsonable_encoder(clubInput)

    if role not in ["cc"]:
        raise Exception("Not Authenticated to access this API")

    # also autofills the updated time
    await clubsdb.update_one(
        {"cid": club_input["cid"]},
        {"$set": {"state": "active", "updated_time": create_utc_time()}},
    )

    await update_role(club_input["cid"], info.context.cookies, "club")

    updated_sample = Club.model_validate(
        await clubsdb.find_one({"cid": club_input["cid"]})
    )

    await invalidate_active_clubs_cache()
    await invalidate_club_cache(club_input["cid"])

    return SimpleClubType.from_pydantic(updated_sample)


# register all mutations
mutations = [
    createClub,
    editClub,
    deleteClub,
    restartClub,
]