Function GenerateMCPScriptJavaScriptToolScript

pkg/workflow/mcp_scripts_generator.go:188–235

GenerateMCPScriptJavaScriptToolScript generates the JavaScript tool file for a mcp-script tool The user's script code is automatically wrapped in a function with module.exports, so users can write simple code without worrying about exports. Input parameters are destructured and available as local va

(toolConfig *MCPScriptToolConfig)

// so users can write simple code without worrying about exports.
// Input parameters are destructured and available as local variables.
func GenerateMCPScriptJavaScriptToolScript(toolConfig *MCPScriptToolConfig) string {
	mcpScriptsLog.Printf("Generating JavaScript tool script: tool=%s, input_count=%d", toolConfig.Name, len(toolConfig.Inputs))
	var sb strings.Builder

	sb.WriteString("// @ts-check\n")
	sb.WriteString("// Auto-generated mcp-script tool: " + toolConfig.Name + "\n\n")
	sb.WriteString("/**\n")
	sb.WriteString(" * " + toolConfig.Description + "\n")
	sb.WriteString(" * @param {Object} inputs - Input parameters\n")
	// Sort input names for stable code generation in JSDoc
	inputNamesForDoc := sliceutil.SortedKeys(toolConfig.Inputs)
	for _, paramName := range inputNamesForDoc {
		param := toolConfig.Inputs[paramName]
		fmt.Fprintf(&sb, " * @param {%s} inputs.%s - %s\n", param.Type, paramName, param.Description)
	}
	sb.WriteString(" * @returns {Promise<any>} Tool result\n")
	sb.WriteString(" */\n")
	sb.WriteString("async function execute(inputs) {\n")

	// Destructure inputs to make parameters available as local variables
	if len(toolConfig.Inputs) > 0 {
		var paramNames []string
		for paramName := range toolConfig.Inputs {
			safeName := stringutil.SanitizeParameterName(paramName)
			if safeName != paramName {
				// If sanitized, use alias
				paramNames = append(paramNames, fmt.Sprintf("%s: %s", paramName, safeName))
			} else {
				paramNames = append(paramNames, paramName)
			}
		}
		sort.Strings(paramNames)
		fmt.Fprintf(&sb, "  const { %s } = inputs || {};\n\n", strings.Join(paramNames, ", "))
	}

	// Indent the user's script code
	sb.WriteString("  " + strings.ReplaceAll(toolConfig.Script, "\n", "\n  ") + "\n")
	sb.WriteString("}\n\n")
	sb.WriteString("module.exports = { execute };\n\n")

	// Delegate subprocess execution to the shared runner module
	sb.WriteString("// Run when executed directly (as a subprocess by the MCP handler)\n")
	sb.WriteString("if (require.main === module) {\n")
	sb.WriteString("  require(\"./mcp-scripts-runner.cjs\")(execute);\n")
	sb.WriteString("}\n")

	return sb.String()
}

// GenerateMCPScriptShellToolScript generates the shell script for a mcp-script tool
func GenerateMCPScriptShellToolScript(toolConfig *MCPScriptToolConfig) string {