/*
	Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
	Copyright (C) 2023 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 {
	Channel,
	ChannelType,
	Guild,
	Invite,
	VanityUrlSchema,
} from "@fosscord/util";
import { Router, Request, Response } from "express";
import { route } from "@fosscord/api";
import { HTTPError } from "lambert-server";

const router = Router();

const InviteRegex = /\W/g;

router.get(
	"/",
	route({ permission: "MANAGE_GUILD" }),
	async (req: Request, res: Response) => {
		const { guild_id } = req.params;
		const guild = await Guild.findOneOrFail({ where: { id: guild_id } });

		if (!guild.features.includes("ALIASABLE_NAMES")) {
			const invite = await Invite.findOne({
				where: { guild_id: guild_id, vanity_url: true },
			});
			if (!invite) return res.json({ code: null });

			return res.json({ code: invite.code, uses: invite.uses });
		} else {
			const invite = await Invite.find({
				where: { guild_id: guild_id, vanity_url: true },
			});
			if (!invite || invite.length == 0) return res.json({ code: null });

			return res.json(
				invite.map((x) => ({ code: x.code, uses: x.uses })),
			);
		}
	},
);

router.patch(
	"/",
	route({ body: "VanityUrlSchema", permission: "MANAGE_GUILD" }),
	async (req: Request, res: Response) => {
		const { guild_id } = req.params;
		const body = req.body as VanityUrlSchema;
		const code = body.code?.replace(InviteRegex, "");

		const guild = await Guild.findOneOrFail({ where: { id: guild_id } });
		if (!guild.features.includes("VANITY_URL"))
			throw new HTTPError("Your guild doesn't support vanity urls");

		if (!code || code.length === 0)
			throw new HTTPError("Code cannot be null or empty");

		const invite = await Invite.findOne({ where: { code } });
		if (invite) throw new HTTPError("Invite already exists");

		const { id } = await Channel.findOneOrFail({
			where: { guild_id, type: ChannelType.GUILD_TEXT },
		});

		await Invite.create({
			vanity_url: true,
			code: code,
			temporary: false,
			uses: 0,
			max_uses: 0,
			max_age: 0,
			created_at: new Date(),
			expires_at: new Date(),
			guild_id: guild_id,
			channel_id: id,
		}).save();

		return res.json({ code: code });
	},
);

export default router;