Merge pull request #55 from Real-Dev-Squad/feature/discord-roles

Feature/discord roles

Author: DashDeipayan

package.json

@@ -35,9 +35,9 @@
     "wrangler": "^2.5.0"
   },
   "dependencies": {
+    "dotenv": "^16.0.3",
     "@tsndr/cloudflare-worker-jwt": "^2.2.1",
     "discord-interactions": "^3.2.0",
-    "dotenv": "^16.0.3",
     "itty-router": "^3.0.11",
     "node-fetch": "^3.3.0"
   },

src/constants/responses.ts

@@ -21,3 +21,5 @@ export const INTERNAL_SERVER_ERROR =
 
 export const RETRY_COMMAND =
   "Oops, we didn't catch that! Please use the command again.";
+
+export const ROLE_ADDED = "Role added successfully";

src/constants/urls.ts

@@ -1,6 +1,6 @@
 export const RDS_BASE_API_URL = "https://api.realdevsquad.com";
 export const RDS_BASE_STAGING_API_URL = "https://staging-api.realdevsquad.com";
-export const RDS_BASE_DEVELOPMENT_API_URL = "http://localhost:3000"; // If needed, modify the URL to your local API server
+export const RDS_BASE_DEVELOPMENT_API_URL = "YOUR_LOCAL_API_URL"; // If needed, modify the URL to your local API server
 
 export const DISCORD_BASE_URL = "https://discord.com/api/v10";
 export const DISCORD_AVATAR_BASE_URL = "https://cdn.discordapp.com/avatars";

src/controllers/guildRoleHandler.ts

@@ -0,0 +1,41 @@
+import * as response from "../constants/responses";
+import { env } from "../typeDefinitions/default.types";
+import JSONResponse from "../utils/JsonResponse";
+import { IRequest } from "itty-router";
+import { addGroupRole, createGuildRole } from "../utils/guildRole";
+import {
+  createNewRole,
+  memberGroupRole,
+} from "../typeDefinitions/discordMessage.types";
+import { verifyAuthToken } from "../utils/verifyAuthToken";
+
+export async function createGuildRoleHandler(request: IRequest, env: env) {
+  const authHeader = request.headers.get("Authorization");
+  if (!authHeader) {
+    return new JSONResponse(response.BAD_SIGNATURE);
+  }
+  try {
+    await verifyAuthToken(authHeader, env);
+    const body: createNewRole = await request.json();
+
+    const res = await createGuildRole(body, env);
+    return new JSONResponse(res);
+  } catch (err) {
+    return new JSONResponse(response.BAD_SIGNATURE);
+  }
+}
+export async function addGroupRoleHandler(request: IRequest, env: env) {
+  const authHeader = request.headers.get("Authorization");
+  if (!authHeader) {
+    return new JSONResponse(response.BAD_SIGNATURE);
+  }
+  try {
+    await verifyAuthToken(authHeader, env);
+    const body: memberGroupRole = await request.json();
+
+    const res = await addGroupRole(body, env);
+    return new JSONResponse(res);
+  } catch (err) {
+    return new JSONResponse(response.BAD_SIGNATURE);
+  }
+}

src/index.ts

@@ -6,6 +6,10 @@ import { env } from "./typeDefinitions/default.types";
 import { discordMessageRequest } from "./typeDefinitions/discordMessage.types";
 import JSONResponse from "./utils/JsonResponse";
 import { verifyBot } from "./utils/verifyBot";
+import {
+  addGroupRoleHandler,
+  createGuildRoleHandler,
+} from "./controllers/guildRoleHandler";
 import { getMembersInServerHandler } from "./controllers/getMembersInServer";
 
 const router = Router();
@@ -16,6 +20,10 @@ router.get("/", async () => {
   });
 });
 
+router.put("/roles/create", createGuildRoleHandler);
+
+router.put("/roles/add", addGroupRoleHandler);
+
 router.get("/discord-members", getMembersInServerHandler);
 
 router.post("/", async (request, env) => {

src/typeDefinitions/discordMessage.types.d.ts

@@ -30,3 +30,25 @@ export interface messageMember {
 export interface createDmChannel {
   id: number;
 }
+export interface createNewRole {
+  rolename: string;
+  mentionable: boolean;
+}
+export interface memberGroupRole {
+  userid: string;
+  roleid: string;
+}
+
+export interface guildRoleResponse {
+  id: string;
+  name: string;
+  color: number;
+  hoist: boolean;
+  icon?: string;
+  unicode_emoji?: string;
+  position: number;
+  permissions: string;
+  managed: boolean;
+  mentionable: boolean;
+  tags?: object;
+}

src/utils/guildRole.ts

@@ -0,0 +1,57 @@
+import { INTERNAL_SERVER_ERROR, ROLE_ADDED } from "../constants/responses";
+import { DISCORD_BASE_URL } from "../constants/urls";
+import { env } from "../typeDefinitions/default.types";
+import {
+  createNewRole,
+  guildRoleResponse,
+  memberGroupRole,
+} from "../typeDefinitions/discordMessage.types";
+
+export async function createGuildRole(
+  body: createNewRole,
+  env: env
+): Promise<guildRoleResponse | string> {
+  const createGuildRoleUrl = `${DISCORD_BASE_URL}/guilds/${env.DISCORD_GUILD_ID}/roles`;
+  const data = {
+    ...body,
+    name: body.rolename,
+  };
+  try {
+    const response = await fetch(createGuildRoleUrl, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bot ${env.DISCORD_TOKEN}`,
+      },
+      body: JSON.stringify(data),
+    });
+    if (response.ok) {
+      return await response.json();
+    } else {
+      return INTERNAL_SERVER_ERROR;
+    }
+  } catch (err) {
+    return INTERNAL_SERVER_ERROR;
+  }
+}
+
+export async function addGroupRole(body: memberGroupRole, env: env) {
+  const { userid, roleid } = body;
+  const createGuildRoleUrl = `${DISCORD_BASE_URL}/guilds/${env.DISCORD_GUILD_ID}/members/${userid}/roles/${roleid}`;
+  try {
+    const response = await fetch(createGuildRoleUrl, {
+      method: "PUT",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bot ${env.DISCORD_TOKEN}`,
+      },
+    });
+    if (response.ok) {
+      return { message: ROLE_ADDED };
+    } else {
+      return INTERNAL_SERVER_ERROR;
+    }
+  } catch (err) {
+    return INTERNAL_SERVER_ERROR;
+  }
+}

src/utils/verifyAuthToken.ts

@@ -0,0 +1,15 @@
+import { env } from "../typeDefinitions/default.types";
+import jwt from "@tsndr/cloudflare-worker-jwt";
+
+/**
+ *
+ * @param authHeader { string } : the auth header of request
+ * @param env { env }: the ctx (context) which contains the secrets put in as wrangler secrets.
+ */
+
+export async function verifyAuthToken(authHeader: string, env: env) {
+  const authToken = authHeader.split(" ")[1];
+  await jwt.verify(authToken, env.RDS_SERVERLESS_PUBLIC_KEY, {
+    algorithm: "RS256",
+  });
+}

src/utils/verifyBot.ts

@@ -1,6 +1,5 @@
 import { verifyKey } from "discord-interactions";
 import { env } from "../typeDefinitions/default.types";
-
 /**
  *
  * @param request { Request } : request the worker receives
@@ -23,6 +22,5 @@ export async function verifyBot(request: Request, env: env) {
     timestamp,
     env.DISCORD_PUBLIC_KEY
   );
-
   return isValidRequest;
 }

tests/fixtures/fixture.ts

@@ -1,4 +1,8 @@
-import { discordMessageRequest } from "../../src/typeDefinitions/discordMessage.types";
+import {
+  createNewRole,
+  discordMessageRequest,
+  memberGroupRole,
+} from "../../src/typeDefinitions/discordMessage.types";
 import { InteractionType } from "discord-interactions";
 
 export const dummyHelloMessage: discordMessageRequest = {
@@ -32,3 +36,18 @@ export const dummyVerifyMessage: discordMessageRequest = {
   },
   guild_id: 123456,
 };
+
+export const dummyCreateBody: createNewRole = {
+  rolename: "test role",
+  mentionable: true,
+};
+
+export const dummyAddRoleBody: memberGroupRole = {
+  userid: "abcd1234",
+  roleid: "defg5678",
+};
+
+export const guildEnv = {
+  DISCORD_GUILD_ID: "1234",
+  DISCORD_TOKEN: "abcd",
+};