CloudMigrations: Implement snapshot management apis (#89296)

* add new apis

* add payloads

* create snapshot status type

* add some impl

* finish implementing update

* start implementing build snapshot func

* add more fake build logic

* add cancel endpoint. do some cleanup

* implement GetSnapshot

* implement upload snapshot

* merge onprem status with gms result

* get it working

* update comment

* rename list endpoint

* add query limit and offset

* add helper method to snapshot

* little bit of cleanup

* work on swagger annotations

* manual merge

* generate swagger specs

* clean up curl commands

* fix bugs found during final testing

* fix linter issue

* fix unit test

pkg/services/cloudmigration/gmsclient/inmemory_client.go

@@ -2,9 +2,12 @@ package gmsclient
 
 import (
 	"context"
+	"encoding/json"
 	"math/rand"
+	"time"
 
 	"github.com/grafana/grafana/pkg/services/cloudmigration"
+	"github.com/grafana/grafana/pkg/util"
 )
 
 // NewInMemoryClient returns an implementation of Client that returns canned responses
@@ -12,7 +15,9 @@ func NewInMemoryClient() Client {
 	return &memoryClientImpl{}
 }
 
-type memoryClientImpl struct{}
+type memoryClientImpl struct {
+	snapshot *cloudmigration.InitializeSnapshotResponse
+}
 
 func (c *memoryClientImpl) ValidateKey(ctx context.Context, cm cloudmigration.CloudMigrationSession) error {
 	return nil
@@ -43,3 +48,50 @@ func (c *memoryClientImpl) MigrateData(
 
 	return &result, nil
 }
+
+func (c *memoryClientImpl) InitializeSnapshot(context.Context, cloudmigration.CloudMigrationSession) (*cloudmigration.InitializeSnapshotResponse, error) {
+	c.snapshot = &cloudmigration.InitializeSnapshotResponse{
+		EncryptionKey:  util.GenerateShortUID(),
+		GMSSnapshotUID: util.GenerateShortUID(),
+		UploadURL:      "localhost:3000",
+	}
+
+	return c.snapshot, nil
+}
+
+func (c *memoryClientImpl) GetSnapshotStatus(ctx context.Context, session cloudmigration.CloudMigrationSession, snapshot cloudmigration.CloudMigrationSnapshot) (*cloudmigration.CloudMigrationSnapshot, error) {
+	// just fake an entire response
+	gmsSnapshot := cloudmigration.CloudMigrationSnapshot{
+		Status:         cloudmigration.SnapshotStatusFinished,
+		GMSSnapshotUID: util.GenerateShortUID(),
+		Result:         []byte{},
+		Finished:       time.Now(),
+	}
+
+	result := []cloudmigration.MigrateDataResponseItem{
+		{
+			Type:   cloudmigration.DashboardDataType,
+			RefID:  util.GenerateShortUID(),
+			Status: cloudmigration.ItemStatusOK,
+		},
+		{
+			Type:   cloudmigration.DatasourceDataType,
+			RefID:  util.GenerateShortUID(),
+			Status: cloudmigration.ItemStatusError,
+			Error:  "fake error",
+		},
+		{
+			Type:   cloudmigration.FolderDataType,
+			RefID:  util.GenerateShortUID(),
+			Status: cloudmigration.ItemStatusOK,
+		},
+	}
+
+	b, err := json.Marshal(result)
+	if err != nil {
+		return nil, err
+	}
+	gmsSnapshot.Result = b
+
+	return &gmsSnapshot, nil
+}