@skyramp/mcp 0.0.1 → 0.0.3

package/build/index.js

@@ -15,6 +15,8 @@ import { registerContractTestTool } from "./tools/generateContractRestTool.js";
 import { registerLoadTestTool } from "./tools/generateLoadRestTool.js";
 import { registerIntegrationTestTool } from "./tools/generateIntegrationRestTool.js";
 import { registerEndtoEndTestTool } from "./tools/generateE2ERestTool.js";
+import { registerLoginTool } from "./tools/loginTool.js";
+import { registerLogoutTool } from "./tools/logoutTool.js";
 const server = new McpServer({
     name: "Skyramp MCP Server",
     version: "1.0.0",
@@ -37,6 +39,8 @@ registerIntegrationTestTool(server);
 registerEndtoEndTestTool(server);
 registerUITestTool(server);
 // Register other Skyramp tools
+registerLoginTool(server);
+registerLogoutTool(server);
 registerExecuteSkyrampTestTool(server);
 registerTraceTool(server);
 registerTraceStopTool(server);

package/build/services/TestGenerationService.js

@@ -24,8 +24,21 @@ export class TestGenerationService {
                         type: "text",
                         text: `${result}
 
-If the test is successfully created, I can go ahead and refactor your test code to make it modular and reusable?
-Reply "yes" to continue, or "no" to skip this step.`,
+If the test was successfully created, I can help refactor your test code to make it more modular and reusable.
+Would you like to proceed with this step?
+
+Reply "yes" to continue or "no" to skip.
+
+If you reply "yes", I will:
+
+Parameterize the test functions into logical groups (e.g., login) to enable reusability for other test scenarios.
+
+NOT alter any existing logic, test scenarios, or test cases.
+
+NOT add any new logic or classes.
+
+Only restructure the code for better modularity and reusability.
+`,
                     },
                 ],
                 isError: false,

package/build/tools/executeSkyrampTestTool.js

@@ -3,13 +3,22 @@ import Docker from "dockerode";
 import path from "path";
 import { Writable } from "stream";
 import { logger } from "../utils/logger.js";
+// Configuration for language-specific file paths
+const LANGUAGE_PATHS = {
+    python: "/tmp",
+    javascript: "/tmp",
+    typescript: "/tmp",
+    java: "/app/src",
+};
 export function registerExecuteSkyrampTestTool(server) {
     server.registerTool("skyramp_execute_test", {
-        description: "Execute a Skyramp test",
+        description: `Execute a Skyramp test to validate your application using isolated containerized environments.
+      Before I can execute the test for you, I need to confirm the following information:
+      Authentication: Please provide me with your authentication token (by default we assume a Bearer Token). If you don't require any token for your service please just say: 'No token required'`,
         inputSchema: {
             language: z
                 .string()
-                .describe("Programming language of the test file to execute (e.g., python, javascript, typescript, java). ONLY SUPPORTS PYTHON FOR NOW."),
+                .describe("Programming language of the test file to execute (e.g., python, javascript, typescript, java)"),
             testType: z
                 .string()
                 .describe("Type of the test to execute (e.g., integration, contract, smoke, fuzz, load, e2e, ui). TEST TYPE MUST BE FROM [integration, contract, smoke, fuzz, load, e2e, ui]."),
@@ -18,19 +27,32 @@ export function registerExecuteSkyrampTestTool(server) {
                 .describe("ALWAYS USE ABSOLUTE PATH to the test file to execute"),
             token: z
                 .string()
-                .describe("Skyramp authentication token for test execution"),
+                .describe("Skyramp authentication token for test execution. Use '' when user confirms 'No token required'"),
         },
         annotations: {
             keywords: ["run test", "execute test"],
         },
     }, async (params) => {
         var docker = new Docker();
-        var image = "public.ecr.aws/skyramp/rampup/runner:mcp-v0.0.1";
-        var command = ["./runner.sh", "python", "/tmp/test.py", params.testType];
+        var image = "public.ecr.aws/skyramp/rampup/runner:mcp-v0.0.3";
+        // Use path.basename for safer filename extraction
+        const filename = path.basename(params.testFile);
+        if (!filename) {
+            throw new Error("Invalid test file path: could not extract filename");
+        }
+        // Use configurable language paths with fallback
+        const basePath = LANGUAGE_PATHS[params.language] || "/tmp";
+        const filePath = path.join(basePath, filename);
+        var command = [
+            "./root/runner.sh",
+            params.language,
+            filePath,
+            params.testType,
+        ];
         const hostConfig = {
             Mounts: [
                 {
-                    Target: "/tmp/test.py",
+                    Target: filePath,
                     Source: path.resolve(params.testFile),
                     Type: "bind",
                 },
@@ -50,6 +72,24 @@ export function registerExecuteSkyrampTestTool(server) {
         var stream = new DockerStream();
         try {
             var statusCode = 0;
+            // check if image exists
+            const images = await docker.listImages();
+            const imageExists = images.some((img) => (img.RepoTags && img.RepoTags.includes(image)) ||
+                (img.RepoDigests && img.RepoDigests.includes(image)));
+            if (!imageExists) {
+                // Pull the image if it does not exist
+                await new Promise((resolve, reject) => {
+                    docker.pull(image, (err, stream) => {
+                        if (err)
+                            return reject(err);
+                        docker.modem.followProgress(stream, (err, res) => {
+                            if (err)
+                                return reject(err);
+                            resolve(res);
+                        });
+                    });
+                });
+            }
             await docker
                 .run(image, command, stream, { Env: env, HostConfig: hostConfig })
                 .then(function (data) {

package/build/tools/loginTool.js

@@ -0,0 +1,43 @@
+import { SkyrampClient } from "@skyramp/skyramp";
+import { logger } from "../utils/logger.js";
+export function registerLoginTool(server) {
+    server.registerTool("skyramp_login", {
+        description: "Login to Skyramp platform",
+        inputSchema: {},
+        annotations: {
+            keywords: ["login", "authenticate", "skyramp login"],
+        },
+    }, async () => {
+        const client = new SkyrampClient();
+        try {
+            logger.info("Attempting to login to Skyramp...");
+            const result = await client.login();
+            // Strip ANSI codes if result contains them
+            const cleanResult = result.replace(/\u001b\[[0-9;]*m/g, "");
+            logger.info("Skyramp login successful");
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: cleanResult,
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            const errorMessage = `Login to Skyramp failed: ${error.message}`;
+            logger.error("Skyramp login failed", {
+                error: error.message,
+            });
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: errorMessage,
+                    },
+                ],
+                isError: true,
+            };
+        }
+    });
+}

package/build/tools/logoutTool.js

@@ -0,0 +1,43 @@
+import { SkyrampClient } from "@skyramp/skyramp";
+import { logger } from "../utils/logger.js";
+export function registerLogoutTool(server) {
+    server.registerTool("skyramp_logout", {
+        description: "Logout from Skyramp platform",
+        inputSchema: {},
+        annotations: {
+            keywords: ["logout", "sign out", "skyramp logout"],
+        },
+    }, async () => {
+        const client = new SkyrampClient();
+        try {
+            logger.info("Attempting to logout from Skyramp...");
+            const result = await client.logout();
+            // Strip ANSI codes if result contains them
+            const cleanResult = result.replace(/\u001b\[[0-9;]*m/g, "");
+            logger.info("Skyramp logout successful");
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: cleanResult,
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            const errorMessage = `Logout from Skyramp failed: ${error.message}`;
+            logger.error("Skyramp logout failed", {
+                error: error.message,
+            });
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: errorMessage,
+                    },
+                ],
+                isError: true,
+            };
+        }
+    });
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skyramp/mcp",
-  "version": "0.0.1",
+  "version": "0.0.3",
   "main": "build/index.js",
   "type": "module",
   "bin": {
@@ -42,7 +42,7 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.11.4",
-    "@skyramp/skyramp": "^1.2.7",
+    "@skyramp/skyramp": "^1.2.9",
     "dockerode": "^4.0.6",
     "zod": "^3.25.3"
   },
@@ -58,4 +58,4 @@
   "engines": {
     "node": ">=18"
   }
-}
+}