From c3c775786cdedc6f27eb136ddbe176ea6139f67a Mon Sep 17 00:00:00 2001
From: Pablo Lara <larabjj@gmail.com>
Date: Fri, 1 Nov 2024 09:30:43 +0100
Subject: [PATCH] feat: add credentials for kubernetes

---
 actions/providers/providers.ts                |  4 +++
 .../workflow/forms/via-credentials-form.tsx   | 25 ++++++++++++--
 .../workflow/forms/via-credentials/index.ts   |  1 +
 .../via-credentials/k8s-credentials-form.tsx  | 34 +++++++++++++++++++
 types/components.ts                           |  9 ++++-
 types/formSchemas.ts                          |  8 ++++-
 6 files changed, 77 insertions(+), 4 deletions(-)
 create mode 100644 components/providers/workflow/forms/via-credentials/k8s-credentials-form.tsx

diff --git a/actions/providers/providers.ts b/actions/providers/providers.ts
index 58975d7987..c4b654f1bb 100644
--- a/actions/providers/providers.ts
+++ b/actions/providers/providers.ts
@@ -178,6 +178,10 @@ export const addCredentialsProvider = async (formData: FormData) => {
       client_secret: formData.get("client_secret"),
       refresh_token: formData.get("refresh_token"),
     };
+  } else if (providerType === "kubernetes") {
+    secret = {
+      kubeconfig_content: formData.get("kubeconfig_content"),
+    };
   }
 
   const bodyData = {
diff --git a/components/providers/workflow/forms/via-credentials-form.tsx b/components/providers/workflow/forms/via-credentials-form.tsx
index 105962c53c..5a6524ce0a 100644
--- a/components/providers/workflow/forms/via-credentials-form.tsx
+++ b/components/providers/workflow/forms/via-credentials-form.tsx
@@ -16,18 +16,24 @@ import {
   AWSCredentials,
   AzureCredentials,
   GCPCredentials,
+  KubernetesCredentials,
 } from "@/types";
 
 import { AWScredentialsForm } from "./via-credentials/aws-credentials-form";
 import { AzureCredentialsForm } from "./via-credentials/azure-credentials-form";
 import { GCPcredentialsForm } from "./via-credentials/gcp-credentials-form";
+import { KubernetesCredentialsForm } from "./via-credentials/k8s-credentials-form";
 
 type CredentialsFormSchema = z.infer<
   ReturnType<typeof addCredentialsFormSchema>
 >;
 
 // Add this type intersection to include all fields
-type FormType = CredentialsFormSchema & AWSCredentials & AzureCredentials;
+type FormType = CredentialsFormSchema &
+  AWSCredentials &
+  AzureCredentials &
+  GCPCredentials &
+  KubernetesCredentials;
 
 export const ViaCredentialsForm = ({
   searchParams,
@@ -65,7 +71,11 @@ export const ViaCredentialsForm = ({
                 client_secret: "",
                 refresh_token: "",
               }
-            : {}),
+            : providerType === "kubernetes"
+              ? {
+                  kubeconfig_content: "",
+                }
+              : {}),
     },
   });
 
@@ -121,6 +131,12 @@ export const ViaCredentialsForm = ({
               message: errorMessage,
             });
             break;
+          case "/data/attributes/secret/kubeconfig_content":
+            form.setError("kubeconfig_content", {
+              type: "server",
+              message: errorMessage,
+            });
+            break;
           case "/data/attributes/name":
             form.setError("secretName", {
               type: "server",
@@ -164,6 +180,11 @@ export const ViaCredentialsForm = ({
             control={form.control as unknown as Control<GCPCredentials>}
           />
         )}
+        {providerType === "kubernetes" && (
+          <KubernetesCredentialsForm
+            control={form.control as unknown as Control<KubernetesCredentials>}
+          />
+        )}
 
         <span className="text-sm text-default-500">Name (Optional)</span>
         <CustomInput
diff --git a/components/providers/workflow/forms/via-credentials/index.ts b/components/providers/workflow/forms/via-credentials/index.ts
index 8ae3137774..d3198a1d0e 100644
--- a/components/providers/workflow/forms/via-credentials/index.ts
+++ b/components/providers/workflow/forms/via-credentials/index.ts
@@ -1,3 +1,4 @@
 export * from "./aws-credentials-form";
 export * from "./azure-credentials-form";
 export * from "./gcp-credentials-form";
+export * from "./k8s-credentials-form";
diff --git a/components/providers/workflow/forms/via-credentials/k8s-credentials-form.tsx b/components/providers/workflow/forms/via-credentials/k8s-credentials-form.tsx
new file mode 100644
index 0000000000..0725d134eb
--- /dev/null
+++ b/components/providers/workflow/forms/via-credentials/k8s-credentials-form.tsx
@@ -0,0 +1,34 @@
+import { Control } from "react-hook-form";
+
+import { CustomInput } from "@/components/ui/custom";
+import { KubernetesCredentials } from "@/types";
+
+export const KubernetesCredentialsForm = ({
+  control,
+}: {
+  control: Control<KubernetesCredentials>;
+}) => {
+  return (
+    <>
+      <div className="text-left">
+        <div className="text-2xl font-bold leading-9 text-default-foreground">
+          Connect via Credentials
+        </div>
+        <div className="py-2 text-default-500">
+          Please provide the information for your Kubernetes credentials.
+        </div>
+      </div>
+      <CustomInput
+        control={control}
+        name="kubeconfig_content"
+        type="text"
+        label="Kubeconfig Content"
+        labelPlacement="inside"
+        placeholder="Enter the Kubeconfig Content"
+        variant="bordered"
+        isRequired
+        isInvalid={!!control._formState.errors.kubeconfig_content}
+      />
+    </>
+  );
+};
diff --git a/types/components.ts b/types/components.ts
index 6d434f1264..9c84ce65ab 100644
--- a/types/components.ts
+++ b/types/components.ts
@@ -50,10 +50,17 @@ export type GCPCredentials = {
   providerId: string;
 };
 
+export type KubernetesCredentials = {
+  kubeconfig_content: string;
+  secretName: string;
+  providerId: string;
+};
+
 export type CredentialsFormSchema =
   | AWSCredentials
   | AzureCredentials
-  | GCPCredentials;
+  | GCPCredentials
+  | KubernetesCredentials;
 
 export interface SearchParamsProps {
   [key: string]: string | string[] | undefined;
diff --git a/types/formSchemas.ts b/types/formSchemas.ts
index bffffbaf0f..ee15b5a16b 100644
--- a/types/formSchemas.ts
+++ b/types/formSchemas.ts
@@ -95,7 +95,13 @@ export const addCredentialsFormSchema = (providerType: string) =>
               client_secret: z.string().nonempty("Client Secret is required"),
               refresh_token: z.string().nonempty("Refresh Token is required"),
             }
-          : {}),
+          : providerType === "kubernetes"
+            ? {
+                kubeconfig_content: z
+                  .string()
+                  .nonempty("Kubeconfig Content is required"),
+              }
+            : {}),
   });
 
 export const editProviderFormSchema = (currentAlias: string) =>