mage-remote-run 0.28.0 → 0.29.0

package/bin/mage-remote-run.js

@@ -57,6 +57,7 @@ program.command('mcp [args...]')
   .option('--transport <type>', 'Transport type (stdio, http)', 'stdio')
   .option('--host <host>', 'HTTP Host', '127.0.0.1')
   .option('--port <port>', 'HTTP Port', '18098')
+  .option('--token <token>', 'Authentication token')
   .allowExcessArguments(true)
   .allowUnknownOption(true)
   .action(async (args, options) => {

package/lib/api/spec-loader.js

@@ -7,6 +7,10 @@ import { fileURLToPath } from 'url';
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const SPEC_DIR = path.resolve(__dirname, '../../api-specs/2.4.8');
 
+// Simple in-memory cache for parsed specs
+const specCache = {};
+const shouldUseCache = process.env.NODE_ENV !== 'test';
+
 export function loadSpec(type) {
     let specFile;
     if (type === 'ac-saas' || type === 'saas') {
@@ -16,10 +20,18 @@ export function loadSpec(type) {
         specFile = 'swagger-paas.json';
     }
 
+    if (shouldUseCache && specCache[specFile]) {
+        return specCache[specFile];
+    }
+
     const filePath = path.join(SPEC_DIR, specFile);
     if (!fs.existsSync(filePath)) {
         throw new Error(`OpenAPI spec not found at: ${filePath}`);
     }
 
-    return JSON.parse(fs.readFileSync(filePath, 'utf-8'));
+    const spec = JSON.parse(fs.readFileSync(filePath, 'utf-8'));
+    if (shouldUseCache) {
+        specCache[specFile] = spec;
+    }
+    return spec;
 }

package/lib/commands/company.js

@@ -1,5 +1,6 @@
 import { createClient } from '../api/factory.js';
 import { printTable, handleError } from '../utils.js';
+import { COMPANY_CREATE_QUESTIONS, getAddressQuestions, getCompanyUpdateQuestions } from '../prompts/company.js';
 import chalk from 'chalk';
 
 export function registerCompanyCommands(program) {
@@ -148,28 +149,11 @@ Examples:
                 const { default: inquirer } = await import('inquirer');
 
                 // Prompt for basic info
-                const answers = await inquirer.prompt([
-                    { name: 'company_name', message: 'Company Name:' },
-                    { name: 'company_email', message: 'Company Email:' },
-                    { name: 'legal_name', message: 'Legal Name (optional):' },
-                    { name: 'vat_tax_id', message: 'VAT/Tax ID (optional):' },
-                    { name: 'reseller_id', message: 'Reseller ID (optional):' },
-                    { name: 'customer_group_id', message: 'Customer Group ID:', default: '1' },
-                    { name: 'sales_representative_id', message: 'Sales Representative ID (Admin User ID):', default: '1' },
-                    { name: 'email', message: 'Super Admin Email:' },
-                    { name: 'firstname', message: 'Super Admin First Name:' },
-                    { name: 'lastname', message: 'Super Admin Last Name:' }
-                ]);
+                const answers = await inquirer.prompt(COMPANY_CREATE_QUESTIONS);
 
                 // Address is required usually
-                const address = await inquirer.prompt([
-                    { name: 'street', message: 'Street:' },
-                    { name: 'city', message: 'City:' },
-                    { name: 'country_id', message: 'Country ID:', default: 'US' },
-                    { name: 'region', message: 'Region/State:' },
-                    { name: 'postcode', message: 'Postcode:' },
-                    { name: 'telephone', message: 'Telephone:' }
-                ]);
+                const defaultCountry = process.env.MAGE_REMOTE_RUN_DEFAULT_COUNTRY || 'US';
+                const address = await inquirer.prompt(getAddressQuestions(defaultCountry));
 
                 const payload = {
                     company: {
@@ -223,12 +207,7 @@ Examples:
                     throw new Error(`Company ${companyId} not found.`);
                 }
 
-                const answers = await inquirer.prompt([
-                    { name: 'company_name', message: 'Company Name:', default: current.company_name },
-                    { name: 'company_email', message: 'Company Email:', default: current.company_email },
-                    { name: 'sales_representative_id', message: 'Sales Rep ID:', default: String(current.sales_representative_id) },
-                    { name: 'customer_group_id', message: 'Customer Group ID:', default: String(current.customer_group_id) }
-                ]);
+                const answers = await inquirer.prompt(getCompanyUpdateQuestions(current));
 
                 // Merge
                 const payload = {

package/lib/commands/connections.js

@@ -6,6 +6,7 @@ import { input, confirm, select } from '@inquirer/prompts';
 import inquirer from 'inquirer';
 import chalk from 'chalk';
 import { getMissingB2BModules } from '../b2b.js';
+import { getLogoForType } from '../logos.js';
 
 // Helper to test connection (non-interactive or one-shot)
 async function testConnection(name, settings) {
@@ -222,84 +223,7 @@ async function printConnectionStatus(config, options = {}) {
             return;
         }
 
-        // ASCII Logos
-        const logos = {
-            adobe: chalk.red(`
-   @@@@@@@@@@@@@@@         @@@@@@@@@@@@@@@   
-   @@@@@@@@@@@@@@           @@@@@@@@@@@@@@   
-   @@@@@@@@@@@@@             @@@@@@@@@@@@@   
-   @@@@@@@@@@@@@              @@@@@@@@@@@@   
-   @@@@@@@@@@@@               @@@@@@@@@@@@   
-   @@@@@@@@@@@                 @@@@@@@@@@@   
-   @@@@@@@@@@                   @@@@@@@@@@   
-   @@@@@@@@@         @@@         @@@@@@@@@   
-   @@@@@@@@@         @@@         @@@@@@@@@   
-   @@@@@@@@         @@@@@         @@@@@@@@   
-   @@@@@@@         @@@@@@@         @@@@@@@   
-   @@@@@@         @@@@@@@@@         @@@@@@   
-   @@@@@@        @@@@@@@@@@@        @@@@@@   
-   @@@@@        @@@@@@@@@@@@         @@@@@   
-   @@@@         @@@@@@@@@@@@@         @@@@   
-   @@@                 @@@@@@@         @@@   
-   @@                   @@@@@@@         @@   
-   @@                    @@@@@@         @@   
-                                             
-                                             
-                                             `),
-            magento: chalk.hex('#FFA500')(`                        
-                         @@                        
-                      @@@@@@@@                     
-                   @@@@@@@@@@@@@@                  
-                @@@@@@@@@@@@@@@@@@@@               
-             @@@@@@@@@@@@@@@@@@@@@@@@@@            
-          @@@@@@@@@@@@@@    @@@@@@@@@@@@@@         
-       @@@@@@@@@@@@@@          @@@@@@@@@@@@@@      
-     @@@@@@@@@@@@@                @@@@@@@@@@@@@    
-     @@@@@@@@@@                      @@@@@@@@@@    
-     @@@@@@@        @@@@    @@@@        @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@    
-       @@@@@     @@@@@@@    @@@@@@@     @@@@@      
-         @@@     @@@@@@@    @@@@@@@     @@@        
-                 @@@@@@@@@@@@@@@@@@                
-                 @@@@@@@@@@@@@@@@@@                
-                   @@@@@@@@@@@@@@                  
-                      @@@@@@@@                     
-                         @@                        `),
-            mageos: chalk.hex('#FFA500')(`
-               ======          ======              
-            ============    ============           
-        ====================================       
-     ==========================================    
-  ================================================ 
-================================-===============--
-=============----============----============-----
-=========--------=========-------=========--------
-========---------========--------========---------
-========---------========--------========---------
-========---------========--------========---------
-   =====------     ======------     =====------   
-     ==---           ===---           ==---        
-`)
-        };
-
-        let logo = '';
-        if (profile.type && profile.type.startsWith('ac-')) {
-            logo = logos.adobe;
-        } else if (profile.type === 'mage-os') {
-            logo = logos.mageos;
-        } else if (profile.type === 'magento-os') {
-            logo = logos.magento;
-        }
-
+        const logo = getLogoForType(profile.type);
         if (logo) {
             console.log(logo);
         }
@@ -542,10 +466,12 @@ Examples:
         .action(async (options) => {
             try {
                 const config = await loadConfig();
-                let updated = false;
-                for (const [name, profile] of Object.entries(config.profiles || {})) {
-                    updated = (await ensureProfileCapabilities(name, profile, config)) || updated;
-                }
+                const results = await Promise.all(
+                    Object.entries(config.profiles || {}).map(([name, profile]) =>
+                        ensureProfileCapabilities(name, profile, config)
+                    )
+                );
+                const updated = results.some(result => result);
                 if (updated) {
                     await saveConfig(config);
                 }
@@ -765,8 +691,7 @@ Examples:
                     }
                     console.log(chalk.blue(`Testing ${profiles.length} connections...\n`));
 
-                    const results = [];
-                    for (const name of profiles) {
+                    const promises = profiles.map(async (name) => {
                         try {
                             const profileConfig = config.profiles[name];
                             const client = await createClient(profileConfig);
@@ -775,11 +700,13 @@ Examples:
                             await client.get('V1/store/storeViews');
                             const duration = Date.now() - start;
 
-                            results.push([name, chalk.green('SUCCESS'), `${duration}ms`]);
+                            return [name, chalk.green('SUCCESS'), `${duration}ms`];
                         } catch (e) {
-                            results.push([name, chalk.red('FAILED'), e.message]);
+                            return [name, chalk.red('FAILED'), e.message];
                         }
-                    }
+                    });
+
+                    const results = await Promise.all(promises);
 
                     console.log(chalk.bold('Connection Test Results:'));
                     printTable(['Profile', 'Status', 'Details'], results);

package/lib/config.js

@@ -59,7 +59,8 @@ export async function loadConfig() {
 export async function saveConfig(config) {
     try {
         await mkdirp(CONFIG_DIR);
-        fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
+        fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2), { mode: 0o600 });
+        fs.chmodSync(CONFIG_FILE, 0o600);
         if (process.env.DEBUG) {
             console.log(`Configuration saved to ${CONFIG_FILE}`);
         }
@@ -108,7 +109,8 @@ export async function loadTokenCache() {
 export async function saveTokenCache(cache) {
     try {
         await mkdirp(CACHE_DIR);
-        fs.writeFileSync(TOKEN_CACHE_FILE, JSON.stringify(cache, null, 2));
+        fs.writeFileSync(TOKEN_CACHE_FILE, JSON.stringify(cache, null, 2), { mode: 0o600 });
+        fs.chmodSync(TOKEN_CACHE_FILE, 0o600);
     } catch (e) {
         console.error("Error saving token cache:", e.message);
         throw e;

package/lib/logos.js

@@ -0,0 +1,80 @@
+import chalk from 'chalk';
+
+// ASCII Logos
+const logos = {
+    adobe: chalk.red(`
+   @@@@@@@@@@@@@@@         @@@@@@@@@@@@@@@
+   @@@@@@@@@@@@@@           @@@@@@@@@@@@@@
+   @@@@@@@@@@@@@             @@@@@@@@@@@@@
+   @@@@@@@@@@@@@              @@@@@@@@@@@@
+   @@@@@@@@@@@@               @@@@@@@@@@@@
+   @@@@@@@@@@@                 @@@@@@@@@@@
+   @@@@@@@@@@                   @@@@@@@@@@
+   @@@@@@@@@         @@@         @@@@@@@@@
+   @@@@@@@@@         @@@         @@@@@@@@@
+   @@@@@@@@         @@@@@         @@@@@@@@
+   @@@@@@@         @@@@@@@         @@@@@@@
+   @@@@@@         @@@@@@@@@         @@@@@@
+   @@@@@@        @@@@@@@@@@@        @@@@@@
+   @@@@@        @@@@@@@@@@@@         @@@@@
+   @@@@         @@@@@@@@@@@@@         @@@@
+   @@@                 @@@@@@@         @@@
+   @@                   @@@@@@@         @@
+   @@                    @@@@@@         @@
+
+
+                                             `),
+    magento: chalk.hex('#FFA500')(`
+                         @@
+                      @@@@@@@@
+                   @@@@@@@@@@@@@@
+                @@@@@@@@@@@@@@@@@@@@
+             @@@@@@@@@@@@@@@@@@@@@@@@@@
+          @@@@@@@@@@@@@@    @@@@@@@@@@@@@@
+       @@@@@@@@@@@@@@          @@@@@@@@@@@@@@
+     @@@@@@@@@@@@@                @@@@@@@@@@@@@
+     @@@@@@@@@@                      @@@@@@@@@@
+     @@@@@@@        @@@@    @@@@        @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+     @@@@@@@     @@@@@@@    @@@@@@@     @@@@@@@
+       @@@@@     @@@@@@@    @@@@@@@     @@@@@
+         @@@     @@@@@@@    @@@@@@@     @@@
+                 @@@@@@@@@@@@@@@@@@
+                 @@@@@@@@@@@@@@@@@@
+                   @@@@@@@@@@@@@@
+                      @@@@@@@@
+                         @@                        `),
+    mageos: chalk.hex('#FFA500')(`
+               ======          ======
+            ============    ============
+        ====================================
+     ==========================================
+  ================================================
+================================-===============--
+=============----============----============-----
+=========--------=========-------=========--------
+========---------========--------========---------
+========---------========--------========---------
+========---------========--------========---------
+   =====------     ======------     =====------
+     ==---           ===---           ==---
+`)
+};
+
+export function getLogoForType(type) {
+    if (type && type.startsWith('ac-')) {
+        return logos.adobe;
+    } else if (type === 'mage-os') {
+        return logos.mageos;
+    } else if (type === 'magento-os') {
+        return logos.magento;
+    }
+    return '';
+}

package/lib/mcp.js

@@ -3,6 +3,7 @@ import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
 import { z } from "zod";
 import http from "http";
+import crypto from "crypto";
 import chalk from "chalk";
 import { Command } from "commander";
 import { readFileSync } from "fs";
@@ -46,10 +47,51 @@ export async function startMcpServer(options) {
         const host = options.host || '127.0.0.1';
         const port = options.port || 18098;
 
+        // Authentication logic
+        let token = options.token || process.env.MAGE_REMOTE_RUN_MCP_TOKEN;
+
+        if (!token) {
+            token = crypto.randomBytes(16).toString('hex');
+            console.error(chalk.yellow(`--------------------------------------------------------------------------------`));
+            console.error(chalk.yellow(`MCP Server Authentication Token: `) + chalk.green.bold(token));
+            console.error(chalk.yellow(`Use this token to connect to the MCP server via ?token=${token}`));
+            console.error(chalk.yellow(`--------------------------------------------------------------------------------`));
+        }
+
         const transport = new StreamableHTTPServerTransport();
 
         const httpServer = http.createServer(async (req, res) => {
-            if (req.url === '/sse' || (req.url === '/messages' && req.method === 'POST')) {
+            const requestUrl = new URL(req.url, `http://${req.headers.host || 'localhost'}`);
+            const pathname = requestUrl.pathname;
+
+            // Check authentication
+            const queryToken = requestUrl.searchParams.get('token');
+            const authHeader = req.headers.authorization;
+
+            let authenticated = false;
+
+            try {
+                if (queryToken && queryToken.length === token.length && crypto.timingSafeEqual(Buffer.from(queryToken), Buffer.from(token))) {
+                    authenticated = true;
+                } else if (authHeader && authHeader.startsWith('Bearer ')) {
+                    const headerToken = authHeader.substring(7);
+                    if (headerToken.length === token.length && crypto.timingSafeEqual(Buffer.from(headerToken), Buffer.from(token))) {
+                        authenticated = true;
+                    }
+                }
+            } catch (e) {
+                // Ignore crypto errors (e.g. encoding issues)
+            }
+
+            if (!authenticated) {
+                res.writeHead(401, { 'Content-Type': 'application/json' });
+                res.end(JSON.stringify({ error: 'Unauthorized', message: 'Invalid or missing authentication token' }));
+                return;
+            }
+
+            if (pathname === '/sse' || (pathname === '/messages' && req.method === 'POST')) {
+                // Ensure query parameters don't interfere with transport handling if it expects exact path
+                req.url = pathname;
                 await transport.handleRequest(req, res);
             } else {
                 res.writeHead(404);

package/lib/prompts/company.js

@@ -0,0 +1,33 @@
+
+export const COMPANY_CREATE_QUESTIONS = [
+    { name: 'company_name', message: 'Company Name:' },
+    { name: 'company_email', message: 'Company Email:' },
+    { name: 'legal_name', message: 'Legal Name (optional):' },
+    { name: 'vat_tax_id', message: 'VAT/Tax ID (optional):' },
+    { name: 'reseller_id', message: 'Reseller ID (optional):' },
+    { name: 'customer_group_id', message: 'Customer Group ID:', default: '1' },
+    { name: 'sales_representative_id', message: 'Sales Representative ID (Admin User ID):', default: '1' },
+    { name: 'email', message: 'Super Admin Email:' },
+    { name: 'firstname', message: 'Super Admin First Name:' },
+    { name: 'lastname', message: 'Super Admin Last Name:' }
+];
+
+export function getAddressQuestions(defaultCountry = 'US') {
+    return [
+        { name: 'street', message: 'Street:' },
+        { name: 'city', message: 'City:' },
+        { name: 'country_id', message: 'Country ID:', default: defaultCountry },
+        { name: 'region', message: 'Region/State:' },
+        { name: 'postcode', message: 'Postcode:' },
+        { name: 'telephone', message: 'Telephone:' }
+    ];
+}
+
+export function getCompanyUpdateQuestions(current) {
+    return [
+        { name: 'company_name', message: 'Company Name:', default: current.company_name },
+        { name: 'company_email', message: 'Company Email:', default: current.company_email },
+        { name: 'sales_representative_id', message: 'Sales Rep ID:', default: String(current.sales_representative_id) },
+        { name: 'customer_group_id', message: 'Customer Group ID:', default: String(current.customer_group_id) }
+    ];
+}

package/lib/utils.js

@@ -4,12 +4,16 @@ import fs from 'fs';
 import os from 'os';
 import { loadConfig } from './config.js';
 
+const TABLE_OPTIONS = {
+    style: { head: [] }
+};
+
 export function printTable(headers, data) {
     const table = new Table({
+        ...TABLE_OPTIONS,
         head: headers.map(h => chalk.cyan(h)),
-        style: { head: [] }
     });
-    data.forEach(row => table.push(row));
+    table.push(...data);
     console.log(table.toString());
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mage-remote-run",
-  "version": "0.28.0",
+  "version": "0.29.0",
   "description": "The remote swiss army knife for Magento Open Source, Mage-OS, Adobe Commerce",
   "main": "index.js",
   "scripts": {