Add the socket server

Author: slimslenderslacks

.gitignore

@@ -22,3 +22,4 @@
 /docs/public/
 /docs/.hugo_build.lock
 /functions/memory/memory.json
+/log/

compose.yaml

@@ -0,0 +1,18 @@
+services:
+  mcp_docker:
+    image: mcp/docker
+    ports:
+      - 8811:8811
+    volumes:
+      - "/var/run/docker.sock:/var/run/docker.sock"
+      - "docker-prompts:/prompts"
+    command:
+      - serve
+      - --mcp
+      - --port
+      - "8811"
+      - --register
+      - github:docker/labs-ai-tools-for-devs?path=prompts/bootstrap.md
+volumes:
+  docker-prompts:
+    external: true

prompts/chrome.md

@@ -5,6 +5,8 @@ arguments:
   - name: query
     description: the question to answer
     required: true
+parameter-values:
+  query: "What is url of the logo for docker.com?"
 tools:
   - name: websocat
     description: A tool to send and receive messages over a websocket.
@@ -108,6 +110,6 @@ It is important that when you are done with your page, you close it. This is imp
 
 The following is the question you are trying to answer:
 
-what is the url of the logo for docker.com?
+{{query}}
 
 (if there is no question, just confirm Chrome is running and ready to answer questions)

runbook.md

@@ -0,0 +1,59 @@
+```sh
+docker build -t mcp/docker .
+```
+
+```sh
+docker extension rm vonwig/labs-ai-tools-for-devs
+```
+
+```sh
+docker extension install vonwig/labs-ai-tools-for-devs:0.1.16
+```
+
+```sh
+docker extension update docker/labs-ai-tools-for-devs:0.1.4
+```
+
+```sh
+# docker:command=build-pre-release
+docker buildx build \
+    --builder hydrobuild \
+    --platform linux/amd64,linux/arm64 \
+    --tag mcp/docker:prerelease \
+    --file Dockerfile \
+    --push .
+docker pull mcp/docker:prerelease
+```
+
+```sh
+# docker:command=build-release
+docker buildx build \
+    --builder hydrobuild \
+    --platform linux/amd64,linux/arm64 \
+    --tag mcp/docker:latest \
+    --file Dockerfile \
+    --push .
+docker pull mcp/docker:latest
+```
+
+```sh
+docker run --rm -t --init -v docker-prompts:/prompts alpine:latest tail -f /prompts/log/docker-mcp-server.out
+```
+
+```sh
+clj -M:main serve --mcp --port 8811 --register "github:docker/labs-ai-tools-for-devs?path=prompts/bootstrap.md"
+```
+
+```sh
+docker run --rm -i --pull always -q --init \
+           -v /var/run/docker.sock:/var/run/docker.sock \
+           --mount type=volume,source=docker-prompts,target=/prompts \
+           -p 8811:8811 \
+           mcp/docker:latest \
+           serve --mcp --port 8811 \
+                 --register "github:docker/labs-ai-tools-for-devs?path=prompts/bootstrap.md"
+```
+
+```sh
+socat STDIO TCP:127.0.0.1:8811
+```

src/docker.clj

@@ -410,19 +410,19 @@
     client))
 
 (defn get-block [x-bytes start]
-  (let [[type _ _ _ n1 n2 n3 n4] (Arrays/copyOfRange ^bytes x-bytes start (+ start 8))]
-    (let [size (.getInt (ByteBuffer/wrap (Arrays/copyOfRange ^bytes x-bytes (+ start 4) (+ start 8))))]
-      [(case type 0 :stdin 1 :stdout 2 :stderr)
+  (let [[type _ _ _ n1 n2 n3 n4] (Arrays/copyOfRange ^bytes x-bytes ^int start ^int (+ start 8))]
+    (let [size (.getInt (ByteBuffer/wrap (Arrays/copyOfRange ^bytes x-bytes ^int (+ start 4) ^int (+ start 8))))]
+      [(case (int type) 0 :stdin 1 :stdout 2 :stderr)
        size
-       (String. (Arrays/copyOfRange ^bytes x-bytes (+ start 8) (+ start 8 size)))])))
+       (String. (Arrays/copyOfRange ^bytes x-bytes ^int (+ start 8) ^int (+ start 8 size)))])))
 
 (defn process-bytes [x-bytes]
   (loop [agg {:stdout "" :stderr "" :stdin ""} start 0]
     (if (> (- (count x-bytes) start) 8)
       (let [[type size s] (get-block x-bytes start)]
         (recur
          (update agg type str s)
-         (+ start size 8)))
+         (long (+ start size 8))))
       agg)))
 
 (defn docker-stream-format->stdout [bytes]

src/docker/main.clj

@@ -71,6 +71,8 @@
                 :default []
                 :update-fn conj]
                [nil "--mcp" "use the mcp jsonrpc protocol"]
+               [nil "--port PORT" "run a socket server"
+                :parse-fn #(Long/parseLong %)]
                [nil "--debug" "add debug logging"]
                [nil "--help" "print option summary"]])
 
@@ -172,7 +174,10 @@
                     {})))
                 opts)))
     "serve" (fn []
-              (let [[producer server-promise] (jsonrpc.server/run-server! opts)]
+              (let [server-opts (jsonrpc.server/server-context opts)
+                    [producer server-promise] (if (:port opts)
+                                                (jsonrpc.server/run-socket-server! opts server-opts)
+                                                (jsonrpc.server/run-server! opts server-opts))]
                 (alter-var-root
                  #'jsonrpc/notify
                  (constantly
@@ -208,6 +213,7 @@
              (try
                (cmd)
                (catch Throwable t
+                 (.printStackTrace t)
                  (jsonrpc/notify :error {:content (str t)})
                  (System/exit 1))))))))
     (catch Throwable t