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: snapshot upload async allows restore

Browse source
This commit is contained in:
Harivansh Rathi 2026-04-10 22:48:45 +00:00
parent 30282928f5
commit 1e7829a974
3 changed files with 108 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

27
internal/httpapi/handlers.go
View file

@ -23,6 +23,7 @@ type Service interface {
UploadSnapshot(context.Context, contracthost.SnapshotID, contracthost.UploadSnapshotRequest) (*contracthost.UploadSnapshotResponse, error)
ListSnapshots(context.Context, contracthost.MachineID) (*contracthost.ListSnapshotsResponse, error)
GetSnapshot(context.Context, contracthost.SnapshotID) (*contracthost.GetSnapshotResponse, error)
GetSnapshotArtifact(context.Context, contracthost.SnapshotID, string) (*SnapshotArtifactContent, error)
DeleteSnapshotByID(context.Context, contracthost.SnapshotID) error
RestoreSnapshot(context.Context, contracthost.SnapshotID, contracthost.RestoreSnapshotRequest) (*contracthost.RestoreSnapshotResponse, error)
CreatePublishedPort(context.Context, contracthost.MachineID, contracthost.CreatePublishedPortRequest) (*contracthost.CreatePublishedPortResponse, error)
@ -30,6 +31,11 @@ type Service interface {
DeletePublishedPort(context.Context, contracthost.MachineID, contracthost.PublishedPortID) error
}
type SnapshotArtifactContent struct {
Name string
Path string
}
type Handler struct {
service Service
}
@ -298,6 +304,27 @@ func (h *Handler) handleSnapshot(w http.ResponseWriter, r *http.Request) {
return
}
if len(parts) == 3 && parts[1] == "artifacts" {
if r.Method != http.MethodGet {
writeMethodNotAllowed(w)
return
}
artifact, err := h.service.GetSnapshotArtifact(r.Context(), snapshotID, parts[2])
if err != nil {
writeError(w, statusForError(err), err)
return
}
if artifact == nil {
writeError(w, http.StatusNotFound, fmt.Errorf("snapshot artifact %q not found", parts[2]))
return
}
if artifact.Name != "" {
w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%q", artifact.Name))
}
http.ServeFile(w, r, artifact.Path)
return
}
writeError(w, http.StatusNotFound, fmt.Errorf("route not found"))
}