getcompanion-ai/computer-host
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/computer-host.git synced 2026-04-15 05:02:05 +00:00
Code Issues 6 Projects Releases Packages Wiki Activity Actions

feat(contracts): add published ports, snapshot request, and storage report types

Browse source
This commit is contained in:
Harivansh Rathi 2026-04-09 14:05:59 +00:00
parent 501ae2abd5
commit 26b5d2966d
20 changed files with 893 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

24
internal/daemon/snapshot.go
View file

@ -2,8 +2,6 @@ package daemon
import (
"context"
"crypto/rand"
"encoding/hex"
"fmt"
"io"
"os"
@ -18,10 +16,14 @@ import (
contracthost "github.com/getcompanion-ai/computer-host/contract"
)
func (d *Daemon) CreateSnapshot(ctx context.Context, machineID contracthost.MachineID) (*contracthost.CreateSnapshotResponse, error) {
func (d *Daemon) CreateSnapshot(ctx context.Context, machineID contracthost.MachineID, req contracthost.CreateSnapshotRequest) (*contracthost.CreateSnapshotResponse, error) {
unlock := d.lockMachine(machineID)
defer unlock()
if err := validateSnapshotID(req.SnapshotID); err != nil {
return nil, err
}
record, err := d.store.GetMachine(ctx, machineID)
if err != nil {
return nil, err
@ -39,7 +41,7 @@ func (d *Daemon) CreateSnapshot(ctx context.Context, machineID contracthost.Mach
}
}
snapshotID := contracthost.SnapshotID(generateID())
snapshotID := req.SnapshotID
if err := d.store.UpsertOperation(ctx, model.OperationRecord{
MachineID: machineID,
@ -356,14 +358,6 @@ func networkAllocationInUse(target firecracker.NetworkAllocation, used []firecra
return false
}
func generateID() string {
b := make([]byte, 16)
if _, err := rand.Read(b); err != nil {
panic(fmt.Sprintf("generate id: %v", err))
}
return hex.EncodeToString(b)
}
// moveFile copies src to dst then removes src. Works across filesystem boundaries
// unlike os.Rename, which is needed when moving files out of /proc/<pid>/root/.
func moveFile(src, dst string) error {
@ -371,7 +365,9 @@ func moveFile(src, dst string) error {
if err != nil {
return err
}
defer in.Close()
defer func() {
_ = in.Close()
}()
out, err := os.Create(dst)
if err != nil {
@ -379,7 +375,7 @@ func moveFile(src, dst string) error {
}
if _, err := io.Copy(out, in); err != nil {
out.Close()
_ = out.Close()
_ = os.Remove(dst)
return err
}