Auto-generate unique socket path in /tmp

- --socket flag now creates unique socket with timestamp and PID
- Socket path is printed to stdout for clients to use
- Automatic cleanup on server exit
- Add Dates dependency for timestamp generation
- Update documentation and test script

Author: Keno

Manifest.toml

@@ -2,11 +2,11 @@
 
 julia_version = "1.13.0-DEV"
 manifest_format = "2.0"
-project_hash = "b80e681f4ebf05f6eba3c868de5fc47aa2cc335c"
+project_hash = "69b4d7cbdacd0be0af9965d9032956923b35b619"
 
 [[deps.ClaudeMCPTools]]
-deps = ["JSON"]
-git-tree-sha1 = "9b896351226ba85681ccc8769549eab32812337e"
+deps = ["JSON", "Sockets"]
+git-tree-sha1 = "9081fe72fb82c35e8a7aa59d346293b9367218a9"
 repo-rev = "master"
 repo-url = "git@github.com:JuliaComputing/ClaudeMCPTools.jl.git"
 uuid = "b9bb1685-6a70-41d7-9793-2f9fb633d966"
@@ -24,7 +24,7 @@ uuid = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
 version = "0.21.4"
 
 [[deps.LLMBenchMCPServer]]
-deps = ["ClaudeMCPTools", "JSON", "LLMBenchSimple"]
+deps = ["ClaudeMCPTools", "Dates", "JSON", "LLMBenchSimple"]
 path = "."
 uuid = "78754875-2f68-464a-ac60-858ac662827f"
 version = "0.1.0"
@@ -72,6 +72,10 @@ version = "1.11.0"
 uuid = "ea8e919c-243c-51af-8825-aaa63cd721ce"
 version = "0.7.0"
 
+[[deps.Sockets]]
+uuid = "6462fe0b-24de-5631-8697-dd941f90decc"
+version = "1.11.0"
+
 [[deps.TOML]]
 deps = ["Dates"]
 uuid = "fa267f1f-6049-4f14-aa54-33bafae1ed76"

Project.toml

@@ -5,6 +5,7 @@ authors = ["JuliaComputing"]
 
 [deps]
 ClaudeMCPTools = "b9bb1685-6a70-41d7-9793-2f9fb633d966"
+Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
 LLMBenchSimple = "a7b3c4d5-e6f7-8901-2345-678901234567"
 

README.md

@@ -65,7 +65,7 @@ julia --project -e 'using LLMBenchMCPServer; LLMBenchMCPServer.main()' -- MyBenc
 
 Options:
 - `--workdir PATH`: Set the working directory (default: current directory)
-- `--socket PATH`: Run server on Unix domain socket instead of stdio
+- `--socket`: Run server on Unix domain socket (creates unique socket in /tmp)
 - `--no-basic-tools`: Disable basic tools (bash, str_replace_editor)
 - `--verbose`: Enable verbose output
 - `--help, -h`: Show help message
@@ -75,11 +75,18 @@ Options:
 Run the server on a Unix socket for integration with other tools:
 
 ```bash
-# Start server on Unix socket
-julia --project -m LLMBenchMCPServer MyBenchmark --socket /tmp/mcp.sock
-
-# The server will listen on the socket for MCP protocol connections
-# Multiple clients can connect simultaneously
+# Start server on Unix socket (automatically creates unique socket in /tmp)
+julia --project -m LLMBenchMCPServer MyBenchmark --socket
+
+# Output will show the socket path:
+# Socket path: /tmp/mcp_MyBenchmark_20240820_143022_12345.sock
+
+# The server will:
+# - Create a unique socket file in /tmp with timestamp and PID
+# - Print the socket path for clients to connect to
+# - Listen for MCP protocol connections
+# - Support multiple concurrent client connections
+# - Automatically clean up the socket file on exit
 ```
 
 #### Programmatic Usage

src/LLMBenchMCPServer.jl

@@ -2,6 +2,7 @@ module LLMBenchMCPServer
 
 using ClaudeMCPTools
 using JSON
+using Dates
 
 export LLMBenchServer, SetupProblemTool, GradeProblemTool
 export main

src/server.jl

@@ -59,7 +59,7 @@ function @main(args)
 
         Options:
             --workdir PATH      Working directory (default: current directory)
-            --socket PATH       Run server on Unix domain socket instead of stdio
+            --socket            Run server on Unix domain socket (creates socket in /tmp)
             --no-basic-tools    Disable basic tools (bash, str_replace_editor)
             --verbose           Enable verbose output
             --help, -h          Show this help message
@@ -73,15 +73,15 @@ function @main(args)
 
         Examples:
             julia --project -m LLMBenchMCPServer MyBenchmark
-            julia --project -m LLMBenchMCPServer MyBenchmark --socket /tmp/mcp.sock
+            julia --project -m LLMBenchMCPServer MyBenchmark --socket
         """)
         return 0
     end
 
     # Parse arguments
     module_name = args[1]
     working_dir = pwd()
-    socket_path = nothing
+    use_socket = false
     include_basic_tools = true
     verbose = false
 
@@ -90,9 +90,9 @@ function @main(args)
         if args[i] == "--workdir" && i + 1 <= length(args)
             working_dir = args[i + 1]
             i += 2
-        elseif args[i] == "--socket" && i + 1 <= length(args)
-            socket_path = args[i + 1]
-            i += 2
+        elseif args[i] == "--socket"
+            use_socket = true
+            i += 1
         elseif args[i] == "--no-basic-tools"
             include_basic_tools = false
             i += 1
@@ -152,14 +152,29 @@ function @main(args)
             println("Starting MCP server for $module_name")
             println("Working directory: $working_dir")
             println("Tools registered: $(keys(server.tools))")
-            if socket_path !== nothing
-                println("Socket path: $socket_path")
-            end
         end
 
         # Run the server in appropriate mode
-        if socket_path !== nothing
-            ClaudeMCPTools.run_unix_socket_server(server, socket_path, verbose=verbose)
+        if use_socket
+            # Generate a unique socket path in /tmp
+            timestamp = Dates.format(Dates.now(), "yyyymmdd_HHMMSS")
+            pid = getpid()
+            socket_path = "/tmp/mcp_$(module_name)_$(timestamp)_$(pid).sock"
+            
+            println("Socket path: $socket_path")
+            
+            # Run server and ensure cleanup on exit
+            try
+                ClaudeMCPTools.run_unix_socket_server(server, socket_path, verbose=verbose, cleanup=true)
+            finally
+                # Ensure socket is cleaned up even on error
+                if isfile(socket_path)
+                    rm(socket_path)
+                    if verbose
+                        println("Cleaned up socket: $socket_path")
+                    end
+                end
+            end
         else
             ClaudeMCPTools.run_stdio_server(server, verbose=verbose)
         end
@@ -172,8 +187,6 @@ function @main(args)
     return 0
 end
 
-# Compatibility function for direct module usage
-function main(args=ARGS)
-    # Delegate to @main function
-    (@main)(args)
-end
+# Export a regular main function for programmatic use
+const main = @main
+export main