adding monkeytype

backend/src/utils/discord.ts

@@ -0,0 +1,65 @@
+import { isDevEnvironment } from "./misc";
+import * as RedisClient from "../init/redis";
+import { randomBytes } from "crypto";
+import MonkeyError from "./error";
+import { z } from "zod";
+import { parseWithSchema as parseJsonWithSchema } from "@monkeytype/util/json";
+
+const BASE_URL = "https://discord.com/api";
+
+const DiscordIdAndAvatarSchema = z.object({
+  id: z.string(),
+  avatar: z
+    .string()
+    .optional()
+    .or(z.null().transform(() => undefined)),
+});
+type DiscordIdAndAvatar = z.infer<typeof DiscordIdAndAvatarSchema>;
+
+export async function getDiscordUser(
+  tokenType: string,
+  accessToken: string,
+): Promise<DiscordIdAndAvatar> {
+  const response = await fetch(`${BASE_URL}/users/@me`, {
+    headers: {
+      authorization: `${tokenType} ${accessToken}`,
+    },
+  });
+
+  const parsed = parseJsonWithSchema(
+    await response.text(),
+    DiscordIdAndAvatarSchema,
+  );
+
+  return parsed;
+}
+
+export async function getOauthLink(uid: string): Promise<string> {
+  const connection = RedisClient.getConnection();
+  if (!connection) {
+    throw new MonkeyError(500, "Redis connection not found");
+  }
+  const token = randomBytes(10).toString("hex");
+
+  //add the token uid pair to reids
+  await connection.setex(`discordoauth:${uid}`, 60, token);
+
+  return `${BASE_URL}/oauth2/authorize?client_id=798272335035498557&redirect_uri=${
+    isDevEnvironment()
+      ? `http%3A%2F%2Flocalhost%3A3000%2Fverify`
+      : `https%3A%2F%2Fmonkeytype.com%2Fverify`
+  }&response_type=token&scope=identify&state=${token}`;
+}
+
+export async function iStateValidForUser(
+  state: string,
+  uid: string,
+): Promise<boolean> {
+  const connection = RedisClient.getConnection();
+  if (!connection) {
+    throw new MonkeyError(500, "Redis connection not found");
+  }
+  const redisToken = await connection.getdel(`discordoauth:${uid}`);
+
+  return redisToken === state;
+}