Implement BuildPipelineSync with SDK (#5672)

Signed-off-by: Yoshiki Fujikane <[email protected]>

Author: ffjlabo

pkg/app/pipedv1/plugin/kubernetes/deployment/pipeline.go

@@ -154,6 +154,7 @@ func BuildQuickSyncPipeline(autoRollback bool) []sdk.QuickSyncStage {
 	return out
 }
 
+// TODO: Rename this function
 func buildPipelineStages(stages []*deployment.BuildPipelineSyncStagesRequest_StageConfig, autoRollback bool, now time.Time) []*model.PipelineStage {
 	out := make([]*model.PipelineStage, 0, len(stages)+1)
 
@@ -197,3 +198,36 @@ func buildPipelineStages(stages []*deployment.BuildPipelineSyncStagesRequest_Sta
 
 	return out
 }
+
+// buildPipelineStagesWithSDK builds the pipeline stages with the given SDK stages.
+// TODO: Rename this function to buildPipelineStages after removing the old one.
+func buildPipelineStagesWithSDK(stages []sdk.StageConfig, autoRollback bool) []sdk.PipelineStage {
+	out := make([]sdk.PipelineStage, 0, len(stages)+1)
+
+	for _, s := range stages {
+		out = append(out, sdk.PipelineStage{
+			Name:               s.Name,
+			Index:              s.Index,
+			Rollback:           false,
+			Metadata:           make(map[string]string, 0),
+			AvailableOperation: sdk.ManualOperationNone,
+		})
+	}
+
+	if autoRollback {
+		// we set the index of the rollback stage to the minimum index of all stages.
+		minIndex := slices.MinFunc(stages, func(a, b sdk.StageConfig) int {
+			return a.Index - b.Index
+		}).Index
+		// we copy the predefined stage to avoid modifying the original one.
+		out = append(out, sdk.PipelineStage{
+			Name:               StageK8sRollback.String(),
+			Index:              minIndex,
+			Rollback:           true,
+			Metadata:           make(map[string]string, 0),
+			AvailableOperation: sdk.ManualOperationNone,
+		})
+	}
+
+	return out
+}

pkg/app/pipedv1/plugin/kubernetes/deployment/pipeline_test.go

@@ -251,3 +251,91 @@ func TestBuildQuickSyncPipeline(t *testing.T) {
 		})
 	}
 }
+
+func Test_buildPipelineStagesWithSDK(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name         string
+		stages       []sdk.StageConfig
+		autoRollback bool
+		expected     []sdk.PipelineStage
+	}{
+		{
+			name: "without auto rollback",
+			stages: []sdk.StageConfig{
+				{
+					Name:  "Stage 1",
+					Index: 0,
+				},
+				{
+					Name:  "Stage 2",
+					Index: 1,
+				},
+			},
+			autoRollback: false,
+			expected: []sdk.PipelineStage{
+				{
+					Name:               "Stage 1",
+					Index:              0,
+					Rollback:           false,
+					Metadata:           make(map[string]string, 0),
+					AvailableOperation: sdk.ManualOperationNone,
+				},
+				{
+					Name:               "Stage 2",
+					Index:              1,
+					Rollback:           false,
+					Metadata:           make(map[string]string, 0),
+					AvailableOperation: sdk.ManualOperationNone,
+				},
+			},
+		},
+		{
+			name: "with auto rollback",
+			stages: []sdk.StageConfig{
+				{
+					Name:  "Stage 1",
+					Index: 0,
+				},
+				{
+					Name:  "Stage 2",
+					Index: 1,
+				},
+			},
+			autoRollback: true,
+			expected: []sdk.PipelineStage{
+				{
+					Name:               "Stage 1",
+					Index:              0,
+					Rollback:           false,
+					Metadata:           make(map[string]string, 0),
+					AvailableOperation: sdk.ManualOperationNone,
+				},
+				{
+					Name:               "Stage 2",
+					Index:              1,
+					Rollback:           false,
+					Metadata:           make(map[string]string, 0),
+					AvailableOperation: sdk.ManualOperationNone,
+				},
+				{
+					Name:               StageK8sRollback.String(),
+					Index:              0,
+					Rollback:           true,
+					Metadata:           make(map[string]string, 0),
+					AvailableOperation: sdk.ManualOperationNone,
+				},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+
+			actual := buildPipelineStagesWithSDK(tt.stages, tt.autoRollback)
+			assert.Equal(t, tt.expected, actual)
+		})
+	}
+}

pkg/app/pipedv1/plugin/kubernetes/deployment/plugin.go

@@ -63,9 +63,10 @@ func (p *Plugin) FetchDefinedStages() []string {
 	return stages
 }
 
-// FIXME
 func (p *Plugin) BuildPipelineSyncStages(ctx context.Context, _ *sdk.ConfigNone, input *sdk.BuildPipelineSyncStagesInput) (*sdk.BuildPipelineSyncStagesResponse, error) {
-	return nil, nil
+	return &sdk.BuildPipelineSyncStagesResponse{
+		Stages: buildPipelineStagesWithSDK(input.Request.Stages, input.Request.Rollback),
+	}, nil
 }
 
 // FIXME