Updated the logic of signing JWT into seperate file

Author: vikhyat187

src/utils/authTokenGenerator.ts

@@ -0,0 +1,12 @@
+import jwt from "@tsndr/cloudflare-worker-jwt";
+
+export async function generateDiscordAuthToken(
+  name: string,
+  expiry: number,
+  privateKey: string,
+  algorithm: string
+) {
+  return await jwt.sign({ name: name, exp: expiry }, privateKey, {
+    algorithm: algorithm,
+  });
+}

src/utils/awsAccess.ts

@@ -3,29 +3,31 @@ import { env } from "../typeDefinitions/default.types";
 import config from "../../config/config";
 import { discordTextResponse } from "./discordResponse";
 import { DISCORD_BASE_URL, AWS_IAM_SIGNIN_URL } from "../constants/urls";
+import { generateDiscordAuthToken } from "./authTokenGenerator";
 
 export async function processAWSAccessRequest(
   discordUserId: string,
   awsGroupId: string,
   env: env,
   channelId: number
 ): Promise<void> {
-  const authToken = await jwt.sign(
-    { name: "Cloudflare Worker", exp: Math.floor(Date.now() / 1000) + 2 },
+  const authToken = await generateDiscordAuthToken(
+    "Cloudflare Worker",
+    Math.floor(Date.now() / 1000) + 2,
     env.BOT_PRIVATE_KEY,
-    { algorithm: "RS256" }
+    "RS256"
   );
+  const discordReplyUrl = `${DISCORD_BASE_URL}/channels/${channelId}/messages`;
+  const base_url = config(env).RDS_BASE_API_URL;
+  const grantAWSAccessAPIUrl = `${base_url}/aws-access`;
 
   try {
-    const base_url = config(env).RDS_BASE_API_URL;
     const requestData = {
       groupId: awsGroupId,
       userId: discordUserId,
     };
 
-    const url = `${base_url}/aws-access`;
-
-    const response = await fetch(url, {
+    const response = await fetch(grantAWSAccessAPIUrl, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
@@ -42,7 +44,7 @@ export async function processAWSAccessRequest(
     } else {
       content = `AWS access granted successfully <@${discordUserId}>! Please head over to AWS - ${AWS_IAM_SIGNIN_URL}.`;
     }
-    await fetch(`${DISCORD_BASE_URL}/channels/${channelId}/messages`, {
+    await fetch(discordReplyUrl, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
@@ -54,7 +56,7 @@ export async function processAWSAccessRequest(
     });
   } catch (err) {
     const content = `<@${discordUserId}> Error occurred while granting AWS access.`;
-    await fetch(`${DISCORD_BASE_URL}/channels/${channelId}/messages`, {
+    await fetch(discordReplyUrl, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",

src/utils/sendUserDiscordData.ts

@@ -2,6 +2,7 @@ import { env } from "../typeDefinitions/default.types";
 import jwt from "@tsndr/cloudflare-worker-jwt";
 import { DISCORD_AVATAR_BASE_URL } from "../constants/urls";
 import config from "../../config/config";
+import { generateDiscordAuthToken } from "./authTokenGenerator";
 
 export const sendUserDiscordData = async (
   token: string,
@@ -12,10 +13,11 @@ export const sendUserDiscordData = async (
   discordJoinedAt: string,
   env: env
 ) => {
-  const authToken = await jwt.sign(
-    { name: "Cloudflare Worker", exp: Math.floor(Date.now() / 1000) + 2 },
+  const authToken = await generateDiscordAuthToken(
+    "Cloudflare Worker",
+    Math.floor(Date.now() / 1000) + 2,
     env.BOT_PRIVATE_KEY,
-    { algorithm: "RS256" }
+    "RS256"
   );
   const data = {
     type: "discord",