@mablhq/mabl-cli 2.72.12 → 2.72.18

package/commands/mcp/mcp_cmds/start.js

@@ -87,7 +87,6 @@ async function startServer(parsed) {
         throw new Error(`Unknown prompt: ${name}`);
     });
     let mablApiClient;
-    let workspaceId;
     server.setRequestHandler(types_js_1.CallToolRequestSchema, async (request) => {
         const { name } = request.params;
         const tool = tools_1.default.find((t) => t.schema.name === name);
@@ -126,6 +125,7 @@ async function startServer(parsed) {
                     },
                 ],
             };
+            let workspaceId;
             try {
                 workspaceId = await getWorkspaceId(parsed[constants_1.CommandArgWorkspaceId], parsed[constants_1.CommandArgApiKey], mablApiClient);
                 if (!workspaceId) {
@@ -153,7 +153,14 @@ async function getWorkspaceId(workspaceId, apiKey, mablApiClient) {
         return workspaceId;
     }
     if (apiKey) {
-        return (await mablApiClient.getApiKeyDetails()).workspace_id;
+        const apiKeyDetails = await mablApiClient.getApiKeyDetails();
+        return apiKeyDetails.workspace_id;
     }
-    return (await cliConfigProvider_1.CliConfigProvider.getWorkspace()).id;
+    const currentWorkspace = await cliConfigProvider_1.CliConfigProvider.getWorkspace();
+    if (currentWorkspace?.id) {
+        return currentWorkspace.id;
+    }
+    await mablApiClient.getSelf();
+    const workspace = await cliConfigProvider_1.CliConfigProvider.getWorkspace();
+    return workspace?.id;
 }

package/commands/mcp/mcp_cmds/tools/createTest.js

@@ -23,6 +23,14 @@ const runTest = (0, mcpMablTool_1.defineTool)({
           On API tests this will be the endpoint to test.
           Try to validate that the url is correct.
           `),
+            credentialsId: zod_1.z
+                .string()
+                .describe('Optional. The id of the credentials to use for training the test. Cloud type credentials are not valid here. If not provided, the user will be prompted to select from available non-cloud credentials.')
+                .optional(),
+            withoutCredentials: zod_1.z
+                .boolean()
+                .describe('Optional. If true, the test will be created without credentials. If false, the user will be prompted to select from available non-cloud credentials.')
+                .optional(),
             name: zod_1.z.string().describe('The name of the test'),
             intent: zod_1.z.string().describe(`The intent of the test.
           Be as detailed as possible.
@@ -64,6 +72,25 @@ const runTest = (0, mcpMablTool_1.defineTool)({
         if (!params.environmentId) {
             return (0, common_1.stringToToolResult)('environmentId is required when creating a test. Request the environment explicitly.', true);
         }
+        const allCredentials = await mablApiClient.getCredentials(workspaceId, 500);
+        const nonCloudCredentials = allCredentials.filter((credential) => !credential.cloud_only);
+        if (params.credentialsId) {
+            const selectedCredential = allCredentials.find((cred) => cred.id === params.credentialsId);
+            if (!selectedCredential) {
+                return (0, common_1.stringToToolResult)(`Credential with id "${params.credentialsId}" not found.`, true);
+            }
+            if (selectedCredential.cloud_only) {
+                return (0, common_1.stringToToolResult)(`Cannot use Cloud Credentials [${params.credentialsId}] for test training. Cloud credentials are only available in cloud runs. Please use a non-cloud credential instead.`, true);
+            }
+        }
+        if (!params.credentialsId &&
+            nonCloudCredentials.length > 0 &&
+            !params.withoutCredentials) {
+            const credentialsList = nonCloudCredentials
+                .map((cred) => `- ${cred.name} (ID: ${cred.id}, Type: ${cred.type})`)
+                .join('\n');
+            return (0, common_1.stringToToolResult)(`Would you like to add credentials to this test? Here are the available non-cloud credentials:\n\n${credentialsList}\n\nIf you want to use one of these credentials, please specify the credentialsId parameter with the desired credential ID. If you don't want to use any credentials, you can proceed without specifying credentialsId and set withoutCredentials to true.`, false);
+        }
         messaging_1.mablEventEmitter.resetOutputEventChannel();
         let fullIntent = params.intent;
         if (params.steps) {
@@ -81,6 +108,7 @@ const runTest = (0, mcpMablTool_1.defineTool)({
         await (0, trainingSessions_1.trainNewTest)({
             environmentId: params.environmentId,
             applicationId: params.applicationId,
+            credentialsId: params.credentialsId,
             url: params.url,
             testName: params.name ?? 'New Test',
             autoBranch: false,

package/commands/mcp/mcp_cmds/tools/getCredentials.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const mcpMablTool_1 = require("./mcpMablTool");
+const zod_1 = require("zod");
+const getCredentials = (0, mcpMablTool_1.defineTool)({
+    schema: {
+        name: 'get-credentials',
+        title: 'Get all mabl credentials',
+        description: 'Get all mabl credentials',
+        inputSchema: zod_1.z.object({}),
+        outputSchema: zod_1.z.object({
+            credentials: zod_1.z.array(zod_1.z.object({
+                id: zod_1.z.string(),
+                name: zod_1.z.string(),
+                type: zod_1.z.string(),
+                cloud_only: zod_1.z.boolean(),
+            })),
+        }),
+        type: 'readOnly',
+    },
+    handle: async (_, mablApiClient, _server, workspaceId) => {
+        const limit = 500;
+        const credentials = await mablApiClient.getCredentials(workspaceId, limit);
+        const credentialsResponse = credentials.map((credential) => ({
+            id: credential.id,
+            name: credential.name,
+            type: credential.type,
+            cloud_only: credential.cloud_only,
+        }));
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify({ credentials: credentialsResponse }, undefined, 2),
+                },
+            ],
+            structuredContent: { credentials: credentialsResponse },
+        };
+    },
+});
+exports.default = [getCredentials];

package/commands/mcp/mcp_cmds/tools/getWorkspaces.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const mcpMablTool_1 = require("./mcpMablTool");
+const zod_1 = require("zod");
+const getWorkspaces = (0, mcpMablTool_1.defineTool)({
+    schema: {
+        name: 'get-workspaces',
+        title: 'Get all available workspaces',
+        description: 'Get all workspaces that the user has access to',
+        inputSchema: zod_1.z.object({
+            limit: zod_1.z
+                .number()
+                .optional()
+                .describe('The maximum number of workspaces to return (default: 50)'),
+        }),
+        outputSchema: zod_1.z.object({
+            workspaces: zod_1.z.array(zod_1.z.object({
+                id: zod_1.z.string().describe('The workspace ID'),
+                name: zod_1.z.string().describe('The workspace name'),
+            })),
+            currentWorkspaceId: zod_1.z
+                .string()
+                .describe('The currently active workspace ID'),
+        }),
+        type: 'readOnly',
+    },
+    handle: async (args, mablApiClient, _server, workspaceId) => {
+        const limit = args.limit || 50;
+        const workspaces = await mablApiClient.getWorkspaces(limit);
+        const workspaceList = workspaces.map((workspace) => ({
+            id: workspace.id,
+            name: workspace.name,
+        }));
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify({
+                        workspaces: workspaceList,
+                        currentWorkspaceId: workspaceId,
+                    }, undefined, 2),
+                },
+            ],
+            structuredContent: {
+                workspaces: workspaceList,
+                currentWorkspaceId: workspaceId,
+            },
+        };
+    },
+});
+exports.default = [getWorkspaces];

package/commands/mcp/mcp_cmds/tools/index.js

@@ -8,6 +8,7 @@ const getTests_1 = __importDefault(require("./getTests"));
 const getMablDeployment_1 = __importDefault(require("./getMablDeployment"));
 const getTestResults_1 = __importDefault(require("./getTestResults"));
 const getApplications_1 = __importDefault(require("./getApplications"));
+const getCredentials_1 = __importDefault(require("./getCredentials"));
 const analyzeFailure_1 = __importDefault(require("./analyzeFailure"));
 const runTest_1 = __importDefault(require("./runTest"));
 const exportToPlaywright_1 = __importDefault(require("./exportToPlaywright"));
@@ -15,8 +16,11 @@ const createTest_1 = __importDefault(require("./createTest"));
 const getPlans_1 = __importDefault(require("./getPlans"));
 const getPlanRunResult_1 = __importDefault(require("./getPlanRunResult"));
 const getLatestPlanRuns_1 = __importDefault(require("./getLatestPlanRuns"));
+const getWorkspaces_1 = __importDefault(require("./getWorkspaces"));
+const switchWorkspace_1 = __importDefault(require("./switchWorkspace"));
 exports.default = [
     ...getApplications_1.default,
+    ...getCredentials_1.default,
     ...getEnvironments_1.default,
     ...getTests_1.default,
     ...getPlans_1.default,
@@ -28,4 +32,6 @@ exports.default = [
     ...createTest_1.default,
     ...getPlanRunResult_1.default,
     ...getLatestPlanRuns_1.default,
+    ...getWorkspaces_1.default,
+    ...switchWorkspace_1.default,
 ];

package/commands/mcp/mcp_cmds/tools/runTest.js

@@ -21,6 +21,10 @@ const runTest = (0, mcpMablTool_1.defineTool)({
                 .string()
                 .describe('The id of the environment to run the test in'),
             testId: zod_1.z.string().describe('The id of the test to run'),
+            credentialsId: zod_1.z
+                .string()
+                .describe('Optional. The id of the credentials to use for running the test. If not provided, the user will be prompted to select from available non-cloud credentials.')
+                .optional(),
         }),
         type: 'destructive',
         outputSchema: runTestOutputSchema,
@@ -39,7 +43,8 @@ const runTest = (0, mcpMablTool_1.defineTool)({
             .on(messaging_1.EventChannelMessageType.outputLogLine, logListener);
         const result = await (0, run_1.run)({
             id: params.testId,
-            'environment-id': params.environmentId || '',
+            'environment-id': params.environmentId ?? '',
+            'credentials-id': params.credentialsId ?? '',
             branch: 'master',
             'branch-changes-only': false,
             browser: 'chrome',

package/commands/mcp/mcp_cmds/tools/switchWorkspace.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const cliConfigProvider_1 = require("../../../../providers/cliConfigProvider");
+const mcpMablTool_1 = require("./mcpMablTool");
+const zod_1 = require("zod");
+const switchWorkspace = (0, mcpMablTool_1.defineTool)({
+    schema: {
+        name: 'switch-workspace',
+        title: 'Switch the current workspace',
+        description: 'Switch to a different workspace by providing a workspace ID',
+        inputSchema: zod_1.z.object({
+            workspaceId: zod_1.z.string().describe('The ID of the workspace to switch to'),
+        }),
+        type: 'destructive',
+    },
+    handle: async (args, mablApiClient, _server, _currentWorkspaceId) => {
+        const { workspaceId: newWorkspaceId } = args;
+        try {
+            const workspace = await mablApiClient.getWorkspace(newWorkspaceId);
+            await cliConfigProvider_1.CliConfigProvider.setWorkspace(workspace);
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Successfully switched workspace to "${workspace.name}" (${newWorkspaceId})`,
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Failed to switch workspace: ${error instanceof Error ? error.message : 'Unknown error'}`,
+                    },
+                ],
+            };
+        }
+    },
+});
+exports.default = [switchWorkspace];