/* Financer - Pocket Money Tracker
 * Copyright (C) 2025  Vidhu Kant Sharma <vidhukant@vidhukant.com>

 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.

 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

import prisma from "../util/prisma";
import { readById } from "./user";

const friend = async (userId: Number, friendId: Number) => {
    // will throw error P2002 if already friends
    await prisma.friend.createMany({
        data: [
            // @ts-ignore
            { userId: userId, friendId: friendId },

            // @ts-ignore
            { userId: friendId, friendId: userId }
        ]
    })

    return readById(userId);
}

const unfriend = async (userId: Number, friendId: Number) => {
    return prisma.friend.deleteMany({
        where: {
            OR: [
                // @ts-ignore
                { userId: userId },

                // @ts-ignore
                { userId: friendId },
            ]
        }
    })
}

const list = async (userId: Number) => {
    const friendsRaw = await prisma.friend.findMany({
        // @ts-ignore
        where: { userId: userId },
        include: { friend: true }
    })

    // @ts-ignore
    return friendsRaw.map((x: any) => {
        const friend = x.friend;
        delete friend["password"];
        delete friend["refreshTokenVersion"];
        delete friend["createdAt"];
        delete friend["updatedAt"];

        return friend;
    })
}

const get = async (userId: Number, friendId: Number) => {
    // Verify that users are friends. Seeing a non-friend's profile isn't allowed
    const friendship = await prisma.friend.findFirst({
        // @ts-ignore
        where: {
            AND: [
                // @ts-ignore
                { userId: userId },
                // @ts-ignore
                { friendId: friendId },
            ]
        }
    })

    if (friendship == null) {
        // user isn't friends with them
        throw new Error("forbidden");
    }

    const friend = await prisma.user.findUnique({
        // @ts-ignore
        where: { id: friendId }
    })
    // @ts-ignore
    delete friend["password"];
    // @ts-ignore
    delete friend["refreshTokenVersion"];
    // @ts-ignore
    delete friend["createdAt"];
    // @ts-ignore
    delete friend["updatedAt"];

    return friend;
}

export {
    friend,
    unfriend,
    list,
    get
}