register basic-memory with mcpjungle via one-shot service

Author: stackptr

hosts/glyph/services/default.nix

@@ -61,5 +61,11 @@
   };
 
   services.basic-memory.enable = true;
-  services.mcpjungle.enable = true;
+  services.mcpjungle = {
+    enable = true;
+    servers.basic-memory = {
+      url = "http://127.0.0.1:8091/mcp";
+      description = "Knowledge management with markdown files";
+    };
+  };
 }

modules/nixos/llm/mcpjungle.nix

@@ -18,6 +18,24 @@ in {
     };
 
     openFirewall = lib.mkEnableOption "opening firewall ports for MCPJungle";
+
+    servers = lib.mkOption {
+      type = lib.types.attrsOf (lib.types.submodule {
+        options = {
+          url = lib.mkOption {
+            type = lib.types.str;
+            description = "Streamable HTTP URL of the MCP server.";
+          };
+          description = lib.mkOption {
+            type = lib.types.str;
+            default = "";
+            description = "Description of the MCP server.";
+          };
+        };
+      });
+      default = {};
+      description = "MCP servers to register with the gateway.";
+    };
   };
 
   config = lib.mkIf cfg.enable {
@@ -55,6 +73,29 @@ in {
       };
     };
 
+    systemd.services.mcpjungle-register = lib.mkIf (cfg.servers != {}) {
+      description = "Register MCP servers with MCPJungle";
+      after = ["mcpjungle.service"];
+      requires = ["mcpjungle.service"];
+      wantedBy = ["multi-user.target"];
+      serviceConfig = {
+        Type = "oneshot";
+        RemainAfterExit = true;
+      };
+      script = let
+        registry = "http://127.0.0.1:${toString cfg.port}";
+        bin = lib.getExe cfg.package;
+        registrations = lib.concatStringsSep "\n" (lib.mapAttrsToList (name: server: ''
+            if ! ${bin} list servers --registry ${registry} 2>/dev/null | grep -q '${name}'; then
+              ${bin} register --name '${name}' --description '${server.description}' --url '${server.url}' --registry ${registry}
+            fi
+          '')
+          cfg.servers);
+      in ''
+        ${registrations}
+      '';
+    };
+
     networking.firewall.allowedTCPPorts = lib.mkIf cfg.openFirewall [cfg.port];
   };
 }