summary refs log tree commit diff
path: root/src/api/routes/applications/#id/bot/index.ts
blob: 9bc3c571a1b01f0c6d0e5149e2eeb23268df2be8 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*
	Fosscord: A FOSS re-implementation and extension of the Discord.com backend.
	Copyright (C) 2023 Fosscord and Fosscord 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 { Request, Response, Router } from "express";
import { route } from "@fosscord/api";
import {
	Application,
	generateToken,
	User,
	BotModifySchema,
	handleFile,
	DiscordApiErrors,
} from "@fosscord/util";
import { HTTPError } from "lambert-server";
import { verifyToken } from "node-2fa";

const router: Router = Router();

router.post("/", route({}), async (req: Request, res: Response) => {
	const app = await Application.findOneOrFail({
		where: { id: req.params.id },
		relations: ["owner"],
	});

	if (app.owner.id != req.user_id)
		throw DiscordApiErrors.ACTION_NOT_AUTHORIZED_ON_APPLICATION;

	const user = await User.register({
		username: app.name,
		password: undefined,
		id: app.id,
		req,
	});

	user.id = app.id;
	user.premium_since = new Date();
	user.bot = true;

	await user.save();

	// flags is NaN here?
	app.assign({ bot: user, flags: app.flags || 0 });

	await app.save();

	res.send({
		token: await generateToken(user.id),
	}).status(204);
});

router.post("/reset", route({}), async (req: Request, res: Response) => {
	const bot = await User.findOneOrFail({ where: { id: req.params.id } });
	const owner = await User.findOneOrFail({ where: { id: req.user_id } });

	if (owner.id != req.user_id)
		throw DiscordApiErrors.ACTION_NOT_AUTHORIZED_ON_APPLICATION;

	if (
		owner.totp_secret &&
		(!req.body.code || verifyToken(owner.totp_secret, req.body.code))
	)
		throw new HTTPError(req.t("auth:login.INVALID_TOTP_CODE"), 60008);

	bot.data = { hash: undefined, valid_tokens_since: new Date() };

	await bot.save();

	const token = await generateToken(bot.id);

	res.json({ token }).status(200);
});

router.patch(
	"/",
	route({ body: "BotModifySchema" }),
	async (req: Request, res: Response) => {
		const body = req.body as BotModifySchema;
		if (!body.avatar?.trim()) delete body.avatar;

		const app = await Application.findOneOrFail({
			where: { id: req.params.id },
			relations: ["bot", "owner"],
		});

		if (!app.bot) throw DiscordApiErrors.BOT_ONLY_ENDPOINT;

		if (app.owner.id != req.user_id)
			throw DiscordApiErrors.ACTION_NOT_AUTHORIZED_ON_APPLICATION;

		if (body.avatar)
			body.avatar = await handleFile(
				`/avatars/${app.id}`,
				body.avatar as string,
			);

		app.bot.assign(body);

		app.bot.save();

		await app.save();
		res.json(app).status(200);
	},
);

export default router;