From 59523563be75c43ef57bdabcc1c452a9aeb2db16 Mon Sep 17 00:00:00 2001
From: Muhamad Aji Wibisono <muhamadajiw71@gmail.com>
Date: Mon, 2 Jun 2025 19:54:22 +0700
Subject: [PATCH] feat: added edit on discord connector

---
 .../connectors/[connector_id]/edit/page.tsx        | 12 ++++++++++++
 .../connectors/[connector_id]/page.tsx             |  2 ++
 surfsense_web/components/editConnector/types.ts    |  1 +
 surfsense_web/hooks/useConnectorEditPage.ts        | 14 ++++++++++++--
 surfsense_web/lib/connectors/utils.ts              |  1 +
 5 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/edit/page.tsx b/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/edit/page.tsx
index 644dbc981..34db58f67 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/edit/page.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/edit/page.tsx
@@ -191,6 +191,18 @@ export default function EditConnectorPage() {
 										placeholder="Begins with linkup_..."
 									/>
 								)}
+
+								{/* == Discord == */}
+								{connector.connector_type === "DISCORD_CONNECTOR" && (
+									<EditSimpleTokenForm
+										control={editForm.control}
+										fieldName="DISCORD_BOT_TOKEN"
+										fieldLabel="Discord Bot Token"
+										fieldDescription="Update the Discord Bot Token if needed."
+										placeholder="Bot token..."
+									/>
+								)}
+
 							</CardContent>
 							<CardFooter className="border-t pt-6">
 								<Button
diff --git a/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/page.tsx b/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/page.tsx
index c2726a837..fb3ecb8b5 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/page.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/page.tsx
@@ -52,6 +52,7 @@ const getConnectorTypeDisplay = (type: string): string => {
     "SLACK_CONNECTOR": "Slack Connector",
     "NOTION_CONNECTOR": "Notion Connector",
     "GITHUB_CONNECTOR": "GitHub Connector",
+    "DISCORD_CONNECTOR": "Discord Connector",
     "LINKUP_API": "Linkup",
     // Add other connector types here as needed
   };
@@ -89,6 +90,7 @@ export default function EditConnectorPage() {
       "SLACK_CONNECTOR": "SLACK_BOT_TOKEN",
       "NOTION_CONNECTOR": "NOTION_INTEGRATION_TOKEN",
       "GITHUB_CONNECTOR": "GITHUB_PAT",
+      "DISCORD_CONNECTOR": "DISCORD_BOT_TOKEN",
       "LINKUP_API": "LINKUP_API_KEY"
     };
     return fieldMap[connectorType] || "";
diff --git a/surfsense_web/components/editConnector/types.ts b/surfsense_web/components/editConnector/types.ts
index cc43e1c81..ad16010bc 100644
--- a/surfsense_web/components/editConnector/types.ts
+++ b/surfsense_web/components/editConnector/types.ts
@@ -31,5 +31,6 @@ export const editConnectorSchema = z.object({
     TAVILY_API_KEY: z.string().optional(),
     LINEAR_API_KEY: z.string().optional(),
     LINKUP_API_KEY: z.string().optional(),
+    DISCORD_BOT_TOKEN: z.string().optional(),
 });
 export type EditConnectorFormValues = z.infer<typeof editConnectorSchema>; 
diff --git a/surfsense_web/hooks/useConnectorEditPage.ts b/surfsense_web/hooks/useConnectorEditPage.ts
index 7e81c5524..ad0992aea 100644
--- a/surfsense_web/hooks/useConnectorEditPage.ts
+++ b/surfsense_web/hooks/useConnectorEditPage.ts
@@ -41,7 +41,8 @@ export function useConnectorEditPage(connectorId: number, searchSpaceId: string)
             NOTION_INTEGRATION_TOKEN: "", 
             SERPER_API_KEY: "", 
             TAVILY_API_KEY: "",
-            LINEAR_API_KEY: ""
+            LINEAR_API_KEY: "",
+            DISCORD_BOT_TOKEN: "",
         }, 
     });
 
@@ -60,7 +61,8 @@ export function useConnectorEditPage(connectorId: number, searchSpaceId: string)
                     SERPER_API_KEY: config.SERPER_API_KEY || "",
                     TAVILY_API_KEY: config.TAVILY_API_KEY || "",
                     LINEAR_API_KEY: config.LINEAR_API_KEY || "",
-                    LINKUP_API_KEY: config.LINKUP_API_KEY || ""
+                    LINKUP_API_KEY: config.LINKUP_API_KEY || "",
+                    DISCORD_BOT_TOKEN: config.DISCORD_BOT_TOKEN || "",
                 });
                 if (currentConnector.connector_type === 'GITHUB_CONNECTOR') {
                     const savedRepos = config.repo_full_names || [];
@@ -171,6 +173,12 @@ export function useConnectorEditPage(connectorId: number, searchSpaceId: string)
                     newConfig = { LINKUP_API_KEY: formData.LINKUP_API_KEY };
                 }
                 break;
+            case 'DISCORD_CONNECTOR':
+                if (formData.DISCORD_BOT_TOKEN !== originalConfig.DISCORD_BOT_TOKEN) {
+                    if (!formData.DISCORD_BOT_TOKEN) { toast.error("Discord Bot Token cannot be empty."); setIsSaving(false); return; }
+                    newConfig = { DISCORD_BOT_TOKEN: formData.DISCORD_BOT_TOKEN };
+                }
+                break;
         }
 
         if (newConfig !== null) {
@@ -212,6 +220,8 @@ export function useConnectorEditPage(connectorId: number, searchSpaceId: string)
                     editForm.setValue('LINEAR_API_KEY', newlySavedConfig.LINEAR_API_KEY || "");
                  } else if(connector.connector_type === 'LINKUP_API') {
                     editForm.setValue('LINKUP_API_KEY', newlySavedConfig.LINKUP_API_KEY || "");
+                 } else if(connector.connector_type === 'DISCORD_CONNECTOR') {
+                    editForm.setValue('DISCORD_BOT_TOKEN', newlySavedConfig.DISCORD_BOT_TOKEN || "");
                  }
              }
             if (connector.connector_type === 'GITHUB_CONNECTOR') {
diff --git a/surfsense_web/lib/connectors/utils.ts b/surfsense_web/lib/connectors/utils.ts
index f93bd3f82..022459be8 100644
--- a/surfsense_web/lib/connectors/utils.ts
+++ b/surfsense_web/lib/connectors/utils.ts
@@ -7,6 +7,7 @@ export const getConnectorTypeDisplay = (type: string): string => {
         "NOTION_CONNECTOR": "Notion",
         "GITHUB_CONNECTOR": "GitHub",
         "LINEAR_CONNECTOR": "Linear",
+        "DISCORD_CONNECTOR": "Discord",
         "LINKUP_API": "Linkup",
     };
     return typeMap[type] || type;