Add missing api tools for auth

Author: agtraveso

src/server.ts

@@ -104,6 +104,14 @@ import {
   modulesWithVersionsToolDefinitions,
   modulesToolNames,
 } from "./tools/modules-tool-definitions.js";
+import {
+  GetTokenSchema,
+  GetAllServiceAccountsSchema,
+  CreateServiceAccountSchema,
+  DeleteServiceAccountSchema,
+  authToolDefinitions,
+  authToolNames,
+} from "./tools/auth-tool-definitions.js";
 
 const RUN_PROCESS_TOOL_NAME_PREFIX = "yc_";
 const RUN_PROCESS_TOOL_TAG = "mcp-tool";
@@ -295,6 +303,7 @@ class YepCodeMcpServer extends Server {
       if (this.tools.includes(API_TOOL_TAGS.FULL)) {
         tools.push(...processesWithVersionsToolDefinitions);
         tools.push(...modulesWithVersionsToolDefinitions);
+        tools.push(...authToolDefinitions);
       }
       if (this.tools.includes(RUN_CODE_TOOL_TAG)) {
         const envVars = await this.yepCodeEnv.getEnvVars();
@@ -1215,6 +1224,53 @@ class YepCodeMcpServer extends Server {
             }
           );
 
+        case authToolNames.getToken:
+          return this.handleToolRequest(
+            GetTokenSchema,
+            request,
+            async (data) => {
+              const token = await this.yepCodeApi.getToken(data.apiToken);
+              return token;
+            }
+          );
+
+        case authToolNames.getAllServiceAccounts:
+          return this.handleToolRequest(
+            GetAllServiceAccountsSchema,
+            request,
+            async () => {
+              const serviceAccounts =
+                await this.yepCodeApi.getAllServiceAccounts();
+              return serviceAccounts;
+            }
+          );
+
+        case authToolNames.createServiceAccount:
+          return this.handleToolRequest(
+            CreateServiceAccountSchema,
+            request,
+            async (data) => {
+              const serviceAccount = await this.yepCodeApi.createServiceAccount(
+                {
+                  name: data.name,
+                }
+              );
+              return serviceAccount;
+            }
+          );
+
+        case authToolNames.deleteServiceAccount:
+          return this.handleToolRequest(
+            DeleteServiceAccountSchema,
+            request,
+            async (data) => {
+              await this.yepCodeApi.deleteServiceAccount(data.id);
+              return {
+                result: `Service account ${data.id} deleted successfully`,
+              };
+            }
+          );
+
         default:
           this.logger.error(`Unknown tool requested: ${request.params.name}`);
           throw new McpError(

src/tools/auth-tool-definitions.ts

@@ -0,0 +1,90 @@
+import { z } from "zod";
+
+// Schema for getting a token
+export const GetTokenSchema = z.object({
+  apiToken: z.string().describe("API token"),
+});
+
+// Schema for getting all service accounts
+export const GetAllServiceAccountsSchema = z.object({});
+
+// Schema for creating a service account
+export const CreateServiceAccountSchema = z.object({
+  name: z.string().describe("Service account name"),
+});
+
+// Schema for deleting a service account
+export const DeleteServiceAccountSchema = z.object({
+  id: z
+    .string()
+    .describe("Unique identifier (UUID) of the service account to delete"),
+});
+
+// Tool names
+export const authToolNames = {
+  getToken: "get_token",
+  getAllServiceAccounts: "get_all_service_accounts",
+  createServiceAccount: "create_service_account",
+  deleteServiceAccount: "delete_service_account",
+} as const;
+
+// Tool definitions
+export const authToolDefinitions = [
+  {
+    name: authToolNames.getToken,
+    title: "Get Token",
+    description: "Gets an authentication token using an API token.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        apiToken: {
+          type: "string",
+          description: "API token",
+        },
+      },
+      required: ["apiToken"],
+    },
+  },
+  {
+    name: authToolNames.getAllServiceAccounts,
+    title: "Get All Service Accounts",
+    description: "Retrieves a list of all service accounts.",
+    inputSchema: {
+      type: "object",
+      properties: {},
+    },
+  },
+  {
+    name: authToolNames.createServiceAccount,
+    title: "Create Service Account",
+    description: "Creates a new service account.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        name: {
+          type: "string",
+          description: "Service account name",
+          example: "my-service-account",
+        },
+      },
+      required: ["name"],
+    },
+  },
+  {
+    name: authToolNames.deleteServiceAccount,
+    title: "Delete Service Account",
+    description:
+      "Permanently deletes a service account. This action cannot be undone.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        id: {
+          type: "string",
+          description:
+            "Unique identifier (UUID) of the service account to delete",
+        },
+      },
+      required: ["id"],
+    },
+  },
+];