inital commit

Overlord-Client/cmd/agent/handlers/processes.go

@@ -0,0 +1,52 @@
+package handlers
+
+import (
+	"context"
+	"log"
+
+	"overlord-client/cmd/agent/runtime"
+	"overlord-client/cmd/agent/wire"
+)
+
+func HandleProcessList(ctx context.Context, env *runtime.Env, cmdID string) error {
+	log.Printf("process_list: listing all processes")
+
+	processes := []wire.ProcessInfo{}
+	var errMsg string
+
+	procs, err := listProcesses()
+	if err != nil {
+		errMsg = err.Error()
+		log.Printf("process_list error: %v", err)
+	} else {
+		processes = procs
+	}
+
+	result := wire.ProcessListResult{
+		Type:      "process_list_result",
+		CommandID: cmdID,
+		Processes: processes,
+		Error:     errMsg,
+	}
+
+	return wire.WriteMsg(ctx, env.Conn, result)
+}
+
+func HandleProcessKill(ctx context.Context, env *runtime.Env, cmdID string, pid int32) error {
+	log.Printf("process_kill: %d", pid)
+
+	err := killProcess(pid)
+	ok := err == nil
+	errMsg := ""
+	if err != nil {
+		errMsg = err.Error()
+	}
+
+	result := wire.CommandResult{
+		Type:      "command_result",
+		CommandID: cmdID,
+		OK:        ok,
+		Message:   errMsg,
+	}
+	return wire.WriteMsg(ctx, env.Conn, result)
+}