codify-mcp 0.6.0 → 0.7.0

package/lib/mcp/server_setup.js

@@ -9,6 +9,7 @@ const { ApifyClient } = require('apify-client');
 const { randomUUID } = require('node:crypto');
 const Ajv = require('ajv');
 const { performActorCall } = require('./actor_caller');
+const { fetchScript } = require('../script-fetcher');
 
 const setupServer = async ({ tools, apifyToken }) => {
 	// Configure logging to stderr to avoid protocol corruption
@@ -73,13 +74,30 @@ const setupServer = async ({ tools, apifyToken }) => {
 				if (toolData.implementation?.type === 'apify-actor') {
 					const client = new ApifyClient({ token: apifyToken });
 					const actorId = toolData.implementation.actorId || 'cyberfly/apify-agent';
-					const script = toolData.implementation.script;
-
-					if (!script) {
+					let script;
+
+					// Determine script source: scriptUrl (KV store) or inline script
+					if (toolData.implementation.scriptUrl) {
+						// Fetch script from persistent KV store
+						try {
+							script = await fetchScript(toolData.implementation.scriptUrl, apifyToken);
+						} catch (error) {
+							return {
+								content: [{
+									type: 'text',
+									text: `[Script Loading Error] Failed to load automation script.\n\nError: ${error.message}\n\nThis means the script storage URL is invalid or your token doesn't have access.`
+								}],
+								isError: true
+							};
+						}
+					} else if (toolData.implementation.script) {
+						// Use inline script (backward compatibility)
+						script = toolData.implementation.script;
+					} else {
 						return {
 							content: [{
 								type: 'text',
-								text: 'Error: No script provided in tool implementation'
+								text: '[Configuration Error] No automation script found. Tool is missing both \'script\' and \'scriptUrl\'.'
 							}],
 							isError: true
 						};

package/lib/script-fetcher.js

@@ -0,0 +1,72 @@
+/**
+ * Script Fetcher Utility
+ * Loads automation scripts from Apify KV store
+ */
+
+const { ApifyClient } = require('apify-client');
+
+const scriptCache = new Map();
+
+/**
+ * Fetch script from Apify KV store
+ * 
+ * @param {string} scriptUrl - Full URL to KV store record
+ *   Format: https://api.apify.com/v2/key-value-stores/{storeId}/records/{key}
+ * @param {string} apifyToken - Apify API token
+ * @returns {Promise<string>} The automation script
+ */
+const fetchScript = async (scriptUrl, apifyToken) => {
+	if (!scriptUrl || !apifyToken) {
+		throw new Error('scriptUrl and apifyToken are required to fetch script');
+	}
+
+	// Check cache first
+	if (scriptCache.has(scriptUrl)) {
+		return scriptCache.get(scriptUrl);
+	}
+
+	try {
+		// Extract store ID and record key from URL
+		// URL format: https://api.apify.com/v2/key-value-stores/{storeId}/records/{key}
+		const match = scriptUrl.match(/\/key-value-stores\/([^/]+)\/records\/(.+)$/);
+		
+		if (!match) {
+			throw new Error(`Invalid script URL format: ${scriptUrl}`);
+		}
+
+		const [, storeId, recordKey] = match;
+
+		// Create Apify client and fetch from KV store
+		const client = new ApifyClient({ token: apifyToken });
+		const kvStore = client.keyValueStore(storeId);
+		const record = await kvStore.getRecord(recordKey);
+
+		if (!record) {
+			throw new Error(`Script not found at ${scriptUrl}`);
+		}
+
+		// Cache the script for future calls
+		scriptCache.set(scriptUrl, record.value);
+
+		return record.value;
+	} catch (error) {
+		throw new Error(`Failed to fetch script from KV store: ${error.message}`);
+	}
+};
+
+const clearScriptCache = () => {
+	scriptCache.clear();
+};
+
+const getCacheStats = () => {
+	return {
+		cachedScripts: scriptCache.size,
+		urls: Array.from(scriptCache.keys())
+	};
+};
+
+module.exports = {
+	fetchScript,
+	clearScriptCache,
+	getCacheStats
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codify-mcp",
-  "version": "0.6.0",
+  "version": "0.7.0",
   "description": "MCP server for custom automation tools using Apify Agent. Easily create reusable automations from casual browser actions.",
   "main": "lib/index.js",
   "bin": {