forked from pinks/eris
41 lines
1.4 KiB
TypeScript
41 lines
1.4 KiB
TypeScript
import { Chat } from "grammy_types";
|
|
import { Model } from "indexed_kv";
|
|
import { Output } from "t_rest/client";
|
|
import { Admin, adminStore } from "../app/adminStore.ts";
|
|
import { bot } from "../bot/mod.ts";
|
|
import { sessions } from "./sessionsRoute.ts";
|
|
|
|
export async function withUser<O extends Output>(
|
|
query: { sessionId: string },
|
|
cb: (user: Chat.PrivateGetChat) => Promise<O>,
|
|
) {
|
|
const session = sessions.get(query.sessionId);
|
|
if (!session?.userId) {
|
|
return { status: 401, body: { type: "text/plain", data: "Must be logged in" } } as const;
|
|
}
|
|
const user = await getUser(session.userId);
|
|
return cb(user);
|
|
}
|
|
|
|
export async function withAdmin<O extends Output>(
|
|
query: { sessionId: string },
|
|
cb: (user: Chat.PrivateGetChat, admin: Model<Admin>) => Promise<O>,
|
|
) {
|
|
const session = sessions.get(query.sessionId);
|
|
if (!session?.userId) {
|
|
return { status: 401, body: { type: "text/plain", data: "Must be logged in" } } as const;
|
|
}
|
|
const user = await getUser(session.userId);
|
|
const [admin] = await adminStore.getBy("tgUserId", { value: session.userId });
|
|
if (!admin) {
|
|
return { status: 403, body: { type: "text/plain", data: "Must be an admin" } } as const;
|
|
}
|
|
return cb(user, admin);
|
|
}
|
|
|
|
export async function getUser(userId: number): Promise<Chat.PrivateGetChat> {
|
|
const chat = await bot.api.getChat(userId);
|
|
if (chat.type !== "private") throw new Error("Chat is not private");
|
|
return chat;
|
|
}
|