/*
        Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
        Copyright (C) 2025 Spacebar and Spacebar Contributors

        This program is free software: you can redistribute it and/or modify
        it under the terms of the GNU Affero 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 Affero General Public License for more details.

        You should have received a copy of the GNU Affero General Public License
        along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/
import { route } from "@spacebar/api";
import { createHash } from "node:crypto";
import { Session, Snowflake } from "@spacebar/util";
import { Request, Response, Router } from "express";
import { SessionsLogoutSchema } from "../../../schemas/api/users/SessionsSchemas";
import { In } from "typeorm";
const router = Router({ mergeParams: true });
router.get(
	"/",
	route({
		responses: {
			200: {
				body: "GetSessionsResponse",
			},
		},
	}),
	async (req: Request, res: Response) => {
		const { extended = false } = req.query;
		const sessions = (await Session.find({ where: { user_id: req.user_id, is_admin_session: false } })) as Session[];

		res.json({
			user_sessions: sessions.map((session) => (extended ? session.getExtendedDeviceInfo() : session.getDiscordDeviceInfo())),
		});
	},
);

router.post(
	"/logout",
	route({
		requestBody: "SessionsLogoutSchema",
		responses: {
			204: {},
		},
	}),
	async (req: Request, res: Response) => {
		const body = req.body as SessionsLogoutSchema;

		let sessions: Session[] = [];
		if ("session_ids" in body) {
			sessions = (await Session.find({ where: { user_id: req.user_id, session_id: In(body.session_ids!) } })) as Session[];
		}

		if ("session_id_hashes" in body) {
			const allSessions = (await Session.find({ where: { user_id: req.user_id } })) as Session[];
			const hashSet = new Set(body.session_id_hashes);
			const matchingSessions = allSessions.filter((session) => {
				const hash = createHash("sha256").update(session.session_id).digest("hex");
				return hashSet.has(hash);
			});
			sessions.push(...matchingSessions);
		}

		for (const session of sessions) {
			await session.remove();
		}
		res.status(204).send();
	},
);
export default router;