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

fix: fix code review issues

Browse source
This commit is contained in:
Harivansh Rathi 2026-04-09 14:42:34 +00:00
parent 26b5d2966d
commit 2db57d0507
5 changed files with 310 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
internal/daemon/lifecycle.go
View file

@ -2,6 +2,7 @@ package daemon
import (
"context"
"errors"
"fmt"
"os"
"path/filepath"
@ -464,6 +465,9 @@ func (d *Daemon) reconcileSnapshot(ctx context.Context, operation model.Operatio
// Snapshot completed successfully, just clear the journal
return d.store.DeleteOperation(ctx, operation.MachineID)
}
if !errors.Is(err, store.ErrNotFound) {
return fmt.Errorf("get snapshot %q during reconciliation: %w", *operation.SnapshotID, err)
}
// Snapshot did not complete: clean up partial snapshot directory and resume the machine
snapshotDir := filepath.Join(d.config.SnapshotsDir, string(*operation.SnapshotID))
_ = os.RemoveAll(snapshotDir)
@ -482,6 +486,9 @@ func (d *Daemon) reconcileRestore(ctx context.Context, operation model.Operation
// Restore completed, clear journal
return d.store.DeleteOperation(ctx, operation.MachineID)
}
if !errors.Is(err, store.ErrNotFound) {
return fmt.Errorf("get machine %q during restore reconciliation: %w", operation.MachineID, err)
}
// Restore did not complete: clean up partial machine directory and disk
_ = os.RemoveAll(filepath.Dir(d.systemVolumePath(operation.MachineID)))
_ = os.RemoveAll(d.machineRuntimeBaseDir(operation.MachineID))