Support host network_mode

Author: slimslenderslacks

prompts/chrome.md

@@ -23,6 +23,7 @@ tools:
         - websocat_args      
     container:
       image: vonwig/websocat:latest
+      network_mode: host
       stdin: 
         content: "{{message|safe}}"
       command:
@@ -38,6 +39,7 @@ tools:
           description: The arguments to pass to curl
     container:
       image: vonwig/curl:latest
+      network_mode: host
       command: 
         - "{{raw|safe}}"
   - name: chrome
@@ -52,9 +54,8 @@ tools:
         - url
     container:
       image: zenika/alpine-chrome
+      network_mode: host
       background: true
-      ports:
-        - "9222:9222"
       command:
         - "--no-sandbox" 
         - "--remote-debugging-address=0.0.0.0"
@@ -81,13 +82,13 @@ Examples:
 
 ```sh
 # Get the websocket url
-curl -X PUT -H "HOST: localhost:9222" -sg http://host.docker.internal:9222/json/new 
+curl -X PUT -sg http://localhost:9222/json/new 
 
 # Navigate to a page
 
 $MESSAGE='Page.navigate {"url":"https://www.docker.com"}' # This format works with --jsonrpc where the first word is the method name and the rest is the arguments.
 
-$MESSAGE | websocat -n1 --jsonrpc --jsonrpc-omit-jsonrpc ws://host.docker.internal:9222/devtools/page/<PAGE_ID>
+$MESSAGE | websocat -n1 --jsonrpc --jsonrpc-omit-jsonrpc ws://localhost:9222/devtools/page/<PAGE_ID>
 
 {"id":2,"result":{"frameId":"A331E56CCB8615EB4FCB720425A82259","loaderId":"EF5AAD19F2F8BB27FAF55F94FFB27DF9"}}
 ```

src/docker.clj

@@ -105,14 +105,44 @@
     :throw false
     :as :stream}))
 
+(defn parse-port [s]
+  (let [[_ host container] (re-find #"(.*):(.*)" s)]
+    {:host {:port host}
+     :container {:port container
+                 :protocol "tcp"}}))
+
+(defn exposed-ports
+  "  params
+       - coll of strings (e.g. [\"9222:9222\"])"
+  [coll]
+  (->> (for [s coll :let [{:keys [container]} (parse-port s)]]
+         [(format "%s/%s" (:port container) (:protocol container)) {}])
+       (into {})))
+
+(defn port-bindings
+  "  params
+       - coll of strings (e.g. [\"9222:9222\"])"
+  [coll]
+  (->>
+    (for [s coll :let [{:keys [container host]} (parse-port s)]]
+      [(format "%s/%s" (:port container) (:protocol container)) [{:HostPort (:port host)}]])
+    (into {})))
+
+(comment
+  (port-bindings ["9222:9222"])
+  (exposed-ports ["9222:9222"]))
+
 ;; check for 201
 ;; entrypoint is an array of strings
 ;; env is a map
 ;; Env is an array of name=value strings
+;;
+;; opts
+;;
 ;; Tty wraps the process in a pseudo terminal
-{:StdinOnce true
- :OpenStdin true}
-(defn create-container [{:keys [image entrypoint workdir command host-dir env thread-id opts mounts volumes ports] 
+;; StdinOnce closes the stdin after the first client detaches
+;; OpenStdin just opens stdin
+(defn create-container [{:keys [image entrypoint workdir command host-dir env thread-id opts mounts volumes ports network_mode]
                          :or {opts {:Tty true}}}]
   (let [payload (json/generate-string
                  (merge
@@ -123,16 +153,18 @@
                                        (into []))})
                   {:HostConfig
                    (merge
-                     {:Binds
-                      (concat ["docker-lsp:/docker-lsp"
-                               "/var/run/docker.sock:/var/run/docker.sock"]
-                              (when host-dir [(format "%s:/project:rw" host-dir)])
-                              (when thread-id [(format "%s:/thread:rw" thread-id)])
-                              (or volumes mounts))}
-                     (when ports 
-                       {:PortBindings {"9222/tcp" [{:HostPort "9222"}]}}))
+                    {:Binds
+                     (concat ["docker-lsp:/docker-lsp"
+                              "/var/run/docker.sock:/var/run/docker.sock"]
+                             (when host-dir [(format "%s:/project:rw" host-dir)])
+                             (when thread-id [(format "%s:/thread:rw" thread-id)])
+                             (or volumes mounts))}
+                    (when network_mode
+                      {:NetworkMode network_mode})
+                    (when ports
+                      {:PortBindings (port-bindings ports)}))
                    :WorkingDir (or workdir "/project")}
-                  (when ports {:ExposedPorts {"9222/tcp" {}}})
+                  (when ports {:ExposedPorts (exposed-ports ports)})
                   (when entrypoint {:Entrypoint entrypoint})
                   (when command {:Cmd command})))
         ascii-payload (String. (.getBytes payload "ASCII"))]