Allow restore post

launchdarkly · Feb 6, 2025 · 00b42a2 · 00b42a2
1 parent f440629
commit 00b42a2
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 6 deletions.
diff --git a/internal/dev_server/api/restore_backup.go b/internal/dev_server/api/restore_backup.go
@@ -2,9 +2,14 @@ package api
 
 import (
 	"context"
+	"github.com/launchdarkly/ldcli/internal/dev_server/model"
 )
 
 func (s server) RestoreBackup(ctx context.Context, request RestoreBackupRequestObject) (RestoreBackupResponseObject, error) {
-	request.Body
-	panic("implement me")
+	store := model.StoreFromContext(ctx)
+	_, err := store.RestoreBackup(ctx, request.Body)
+	if err != nil {
+		return nil, err
+	}
+	return RestoreBackup200Response{}, nil
 }
diff --git a/internal/dev_server/db/backup/sqllite_backup.go b/internal/dev_server/db/backup/sqllite_backup.go
@@ -74,7 +74,7 @@ func (m *Manager) connectToDb(ctx context.Context, path string) (*sql.DB, error)
 }
 
 // RestoreToFile returns a string path of the sqlite database restored from the stream
-func (m *Manager) RestoreToFile(ctx context.Context, stream io.ReadCloser) (string, error) {
+func (m *Manager) RestoreToFile(ctx context.Context, stream io.Reader) (string, error) {
 	m.mutex.Lock()
 	defer m.mutex.Unlock()
 

diff --git a/internal/dev_server/db/sqlite.go b/internal/dev_server/db/sqlite.go
@@ -349,7 +349,7 @@ func (s Sqlite) DeactivateOverride(ctx context.Context, projectKey, flagKey stri
 	return version, nil
 }
 
-func (s Sqlite) RestoreBackup(ctx context.Context, stream io.ReadCloser) (string, error) {
+func (s Sqlite) RestoreBackup(ctx context.Context, stream io.Reader) (string, error) {
 	filepath, err := s.backupManager.RestoreToFile(ctx, stream)
 
 	return filepath, err

diff --git a/internal/dev_server/model/mocks/store.go b/internal/dev_server/model/mocks/store.go
diff --git a/internal/dev_server/model/store.go b/internal/dev_server/model/store.go
@@ -31,7 +31,7 @@ type Store interface {
 	GetAvailableVariationsForProject(ctx context.Context, projectKey string) (map[string][]Variation, error)
 
 	CreateBackup(ctx context.Context) (io.ReadCloser, int64, error)
-	RestoreBackup(ctx context.Context, stream io.ReadCloser) (string, error)
+	RestoreBackup(ctx context.Context, stream io.Reader) (string, error)
 }
 
 func ContextWithStore(ctx context.Context, store Store) context.Context {