mirror of
https://github.com/getcompanion-ai/computer-host.git
synced 2026-04-19 13:01:43 +00:00
feat: add guest config injection and host nat wiring
This commit is contained in:
parent
28ca0219d9
commit
a12f54ba5d
8 changed files with 332 additions and 6 deletions
97
internal/daemon/guest_config_test.go
Normal file
97
internal/daemon/guest_config_test.go
Normal file
|
|
@ -0,0 +1,97 @@
|
|||
package daemon
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
contracthost "github.com/getcompanion-ai/computer-host/contract"
|
||||
)
|
||||
|
||||
func TestInjectGuestConfigWritesAuthorizedKeysAndWebhook(t *testing.T) {
|
||||
root := t.TempDir()
|
||||
imagePath := filepath.Join(root, "rootfs.ext4")
|
||||
if err := buildTestExt4Image(root, imagePath); err != nil {
|
||||
t.Fatalf("build ext4 image: %v", err)
|
||||
}
|
||||
|
||||
config := &contracthost.GuestConfig{
|
||||
AuthorizedKeys: []string{
|
||||
"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIGuestKeyOne test-1",
|
||||
"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIGuestKeyTwo test-2",
|
||||
},
|
||||
LoginWebhook: &contracthost.GuestLoginWebhook{
|
||||
URL: "https://example.com/login",
|
||||
BearerToken: "secret-token",
|
||||
},
|
||||
}
|
||||
|
||||
if err := injectGuestConfig(context.Background(), imagePath, config); err != nil {
|
||||
t.Fatalf("inject guest config: %v", err)
|
||||
}
|
||||
|
||||
authorizedKeys, err := readExt4File(imagePath, "/etc/microagent/authorized_keys")
|
||||
if err != nil {
|
||||
t.Fatalf("read authorized_keys: %v", err)
|
||||
}
|
||||
wantKeys := strings.Join(config.AuthorizedKeys, "\n") + "\n"
|
||||
if authorizedKeys != wantKeys {
|
||||
t.Fatalf("authorized_keys mismatch: got %q want %q", authorizedKeys, wantKeys)
|
||||
}
|
||||
|
||||
guestConfigPayload, err := readExt4File(imagePath, "/etc/microagent/guest-config.json")
|
||||
if err != nil {
|
||||
t.Fatalf("read guest-config.json: %v", err)
|
||||
}
|
||||
|
||||
var guestConfig contracthost.GuestConfig
|
||||
if err := json.Unmarshal([]byte(guestConfigPayload), &guestConfig); err != nil {
|
||||
t.Fatalf("unmarshal guest-config.json: %v", err)
|
||||
}
|
||||
if guestConfig.LoginWebhook == nil || guestConfig.LoginWebhook.URL != config.LoginWebhook.URL || guestConfig.LoginWebhook.BearerToken != config.LoginWebhook.BearerToken {
|
||||
t.Fatalf("login webhook mismatch: got %#v want %#v", guestConfig.LoginWebhook, config.LoginWebhook)
|
||||
}
|
||||
}
|
||||
|
||||
func buildTestExt4Image(root string, imagePath string) error {
|
||||
sourceDir := filepath.Join(root, "source")
|
||||
if err := os.MkdirAll(filepath.Join(sourceDir, "etc", "microagent"), 0o755); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := os.WriteFile(imagePath, nil, 0o644); err != nil {
|
||||
return err
|
||||
}
|
||||
command := exec.Command("truncate", "-s", "16M", imagePath)
|
||||
output, err := command.CombinedOutput()
|
||||
if err != nil {
|
||||
return fmt.Errorf("truncate: %w: %s", err, strings.TrimSpace(string(output)))
|
||||
}
|
||||
command = exec.Command("mkfs.ext4", "-q", "-d", sourceDir, "-L", "microagent-root", "-F", imagePath)
|
||||
output, err = command.CombinedOutput()
|
||||
if err != nil {
|
||||
return fmt.Errorf("mkfs.ext4: %w: %s", err, strings.TrimSpace(string(output)))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func readExt4File(imagePath string, targetPath string) (string, error) {
|
||||
command := exec.Command("debugfs", "-R", "cat "+targetPath, imagePath)
|
||||
output, err := command.CombinedOutput()
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("debugfs cat %q: %w: %s", targetPath, err, strings.TrimSpace(string(output)))
|
||||
}
|
||||
lines := strings.Split(string(output), "\n")
|
||||
filtered := make([]string, 0, len(lines))
|
||||
for _, line := range lines {
|
||||
if strings.HasPrefix(line, "debugfs ") {
|
||||
continue
|
||||
}
|
||||
filtered = append(filtered, line)
|
||||
}
|
||||
return strings.TrimPrefix(strings.Join(filtered, "\n"), "\n"), nil
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue