increase timeout

Author: ezynda3

README.md

@@ -42,6 +42,9 @@ codebench-mcp --enabled-modules http,fetch
 # Disable specific modules (enable all others)
 codebench-mcp --disabled-modules timers
 
+# Set custom execution timeout (in seconds)
+codebench-mcp --execution-timeout 600  # 10 minutes
+
 # Show help
 codebench-mcp --help
 ```
@@ -212,6 +215,10 @@ Execute JavaScript code with a modern runtime environment.
 **Parameters:**
 - `code` (required): JavaScript code to execute
 
+**Configuration:**
+- Default execution timeout: 5 minutes
+- Configurable via `--execution-timeout <seconds>` CLI flag
+
 **Example:**
 ```javascript
 console.log("Hello, World!");
@@ -259,6 +266,7 @@ console.log('Pathname:', url.pathname);
 - **Module access varies** - Some modules are global (fetch, http), others may need require()
 - **Each execution creates a fresh VM** - For isolation, each execution starts with a clean state
 - **Module filtering** - Configuration exists but actual runtime filtering not fully implemented
+- **Execution timeout** - JavaScript execution is limited by configurable timeout (default: 5 minutes)
 
 ## Building
 

cmd/root.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"slices"
 	"strings"
+	"time"
 
 	"github.com/mark3labs/codebench-mcp/internal/logger"
 	"github.com/mark3labs/codebench-mcp/server"
@@ -16,6 +17,7 @@ var (
 	enabledModules  []string
 	disabledModules []string
 	debugMode       bool
+	executionTimeout int
 )
 
 // Available modules
@@ -86,6 +88,7 @@ with a modern runtime including http, fetch, timers, buffer, crypto, and other m
 		// Create server with module configuration
 		config := server.ModuleConfig{
 			EnabledModules: modulesToEnable,
+			ExecutionTimeout: time.Duration(executionTimeout) * time.Second,
 		}
 
 		jss, err := server.NewJSServerWithConfig(config)
@@ -119,6 +122,8 @@ func init() {
 			strings.Join(availableModules, ", ")))
 	rootCmd.Flags().BoolVar(&debugMode, "debug", false,
 		"Enable debug logging (outputs to stderr)")
+	rootCmd.Flags().IntVar(&executionTimeout, "execution-timeout", 300,
+		"JavaScript execution timeout in seconds (default: 300 = 5 minutes)")
 
 	rootCmd.MarkFlagsMutuallyExclusive("enabled-modules", "disabled-modules")
 }

server/server.go

@@ -29,8 +29,9 @@ import (
 var Version = "dev"
 
 type ModuleConfig struct {
-	EnabledModules  []string
-	DisabledModules []string
+	EnabledModules   []string
+	DisabledModules  []string
+	ExecutionTimeout time.Duration
 }
 
 type JSHandler struct {
@@ -42,7 +43,8 @@ type JSHandler struct {
 
 func NewJSHandler() *JSHandler {
 	return NewJSHandlerWithConfig(ModuleConfig{
-		EnabledModules: []string{"http", "fetch", "timers", "buffer", "kv", "crypto", "encoding", "url", "cache"},
+		EnabledModules:   []string{"http", "fetch", "timers", "buffer", "kv", "crypto", "encoding", "url", "cache"},
+		ExecutionTimeout: 5 * time.Minute,
 	})
 }
 
@@ -213,8 +215,12 @@ func (h *JSHandler) handleRegularCode(ctx context.Context, code string) (*mcp.Ca
 	consoleModule := console.NewConsoleModule(&output)
 	consoleModule.Setup(vm.Runtime())
 
-	// Execute the JavaScript code with a timeout for regular code
-	execCtx, cancel := context.WithTimeout(ctx, time.Second*10)
+	// Execute the JavaScript code with configurable timeout
+	timeout := h.config.ExecutionTimeout
+	if timeout == 0 {
+		timeout = 5 * time.Minute // Default fallback
+	}
+	execCtx, cancel := context.WithTimeout(ctx, timeout)
 	defer cancel()
 
 	// Execute in a goroutine to respect timeout