feat: restart to apply update, privileged helper, rollback

internal/infrastructure/useCases/StartUpdateDownload.go

@@ -12,6 +12,8 @@ type StartUpdateDownloadUseCase struct {
 	store          update.StatusStore
 	jobs           update.JobStore
 	helper         update.HelperDetector
+	results        update.ResultStore
+	status         updateStatusBuilder
 	fetcher        update.AssetFetcher
 	verifier       update.SignatureVerifier
 	runner         update.StagedBinaryRunner
@@ -29,6 +31,8 @@ type StartUpdateDownloadDependencies struct {
 	Store          update.StatusStore
 	Jobs           update.JobStore
 	Helper         update.HelperDetector
+	Results        update.ResultStore
+	Requests       update.RestartRequestStore
 	Fetcher        update.AssetFetcher
 	Verifier       update.SignatureVerifier
 	Runner         update.StagedBinaryRunner
@@ -52,9 +56,18 @@ func (f jobFailure) Error() string {
 
 func NewStartUpdateDownloadUseCase(dependencies StartUpdateDownloadDependencies) *StartUpdateDownloadUseCase {
 	return &StartUpdateDownloadUseCase{
-		store:          dependencies.Store,
-		jobs:           dependencies.Jobs,
-		helper:         dependencies.Helper,
+		store:   dependencies.Store,
+		jobs:    dependencies.Jobs,
+		helper:  dependencies.Helper,
+		results: dependencies.Results,
+		status: updateStatusBuilder{
+			store:          dependencies.Store,
+			jobs:           dependencies.Jobs,
+			results:        dependencies.Results,
+			requests:       dependencies.Requests,
+			currentVersion: dependencies.CurrentVersion,
+			now:            dependencies.Now,
+		},
 		fetcher:        dependencies.Fetcher,
 		verifier:       dependencies.Verifier,
 		runner:         dependencies.Runner,
@@ -79,6 +92,9 @@ func (s *StartUpdateDownloadUseCase) Start() (update.UpdateStatus, error) {
 	if job, found := s.jobs.Load(); found && job.Occupies() {
 		return update.UpdateStatus{}, update.ErrJobInProgress
 	}
+	if s.status.build(true).State == update.StateRestarting {
+		return update.UpdateStatus{}, update.ErrRestartPending
+	}
 	result, found := s.store.Load()
 	if !found || !update.IsNewer(s.currentVersion, result.Latest) {
 		return update.UpdateStatus{}, update.ErrNothingToUpdate
@@ -95,10 +111,13 @@ func (s *StartUpdateDownloadUseCase) Start() (update.UpdateStatus, error) {
 		StartedAt:  s.now().UTC(),
 		TotalBytes: binary.Size,
 	}
+	if err := s.results.Delete(); err != nil {
+		return update.UpdateStatus{}, err
+	}
 	if err := s.jobs.Save(job); err != nil {
 		return update.UpdateStatus{}, err
 	}
-	status := buildUpdateStatus(s.store, s.jobs, s.currentVersion, true)
+	status := s.status.build(true)
 	go s.run(job, binary, checksums, signature)
 
 	return status, nil