From 396a02c50bbe2b76b9338d1ba8c5970d8abff925 Mon Sep 17 00:00:00 2001
From: Nddtfjiang <jzc001@qq.com>
Date: Thu, 4 Aug 2022 14:34:26 +0000
Subject: [PATCH] feat: gitlab pipeline convert

Add gitlab ConvertPipelines.

Nddtfjiang <zhicheng.jiang@merico.dev>
---
 plugins/gitlab/impl/impl.go                |   1 +
 plugins/gitlab/tasks/pipeline_convertor.go | 107 +++++++++++++++++++++
 2 files changed, 108 insertions(+)
 create mode 100644 plugins/gitlab/tasks/pipeline_convertor.go

diff --git a/plugins/gitlab/impl/impl.go b/plugins/gitlab/impl/impl.go
index 8b2401a1c4e..07859ad01ea 100644
--- a/plugins/gitlab/impl/impl.go
+++ b/plugins/gitlab/impl/impl.go
@@ -78,6 +78,7 @@ func (plugin Gitlab) SubTaskMetas() []core.SubTaskMeta {
 		tasks.ConvertIssueLabelsMeta,
 		tasks.ConvertMrLabelsMeta,
 		tasks.ConvertCommitsMeta,
+		tasks.ConvertPipelineMeta,
 	}
 }
 
diff --git a/plugins/gitlab/tasks/pipeline_convertor.go b/plugins/gitlab/tasks/pipeline_convertor.go
new file mode 100644
index 00000000000..865bee79b36
--- /dev/null
+++ b/plugins/gitlab/tasks/pipeline_convertor.go
@@ -0,0 +1,107 @@
+/*
+Licensed to the Apache Software Foundation (ASF) under one or more
+contributor license agreements.  See the NOTICE file distributed with
+this work for additional information regarding copyright ownership.
+The ASF licenses this file to You under the Apache License, Version 2.0
+(the "License"); you may not use this file except in compliance with
+the License.  You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package tasks
+
+import (
+	"fmt"
+	"reflect"
+	"time"
+
+	"github.com/apache/incubator-devlake/models/domainlayer"
+	"github.com/apache/incubator-devlake/models/domainlayer/devops"
+	"github.com/apache/incubator-devlake/models/domainlayer/didgen"
+	"github.com/apache/incubator-devlake/plugins/core"
+	"github.com/apache/incubator-devlake/plugins/core/dal"
+	gitlabModels "github.com/apache/incubator-devlake/plugins/gitlab/models"
+	"github.com/apache/incubator-devlake/plugins/helper"
+)
+
+var ConvertPipelineMeta = core.SubTaskMeta{
+	Name:             "convertPipelines",
+	EntryPoint:       ConvertPipelines,
+	EnabledByDefault: true,
+	Description:      "Convert tool layer table gitlab_pipeline into domain layer table pipeline",
+	DomainTypes:      []string{core.DOMAIN_TYPE_CROSS},
+}
+
+func ConvertPipelines(taskCtx core.SubTaskContext) error {
+	db := taskCtx.GetDal()
+	data := taskCtx.GetData().(*GitlabTaskData)
+
+	cursor, err := db.Cursor(dal.From(gitlabModels.GitlabPipeline{}))
+	if err != nil {
+		return err
+	}
+	defer cursor.Close()
+
+	pipelineIdGen := didgen.NewDomainIdGenerator(&gitlabModels.GitlabPipeline{})
+
+	converter, err := helper.NewDataConverter(helper.DataConverterArgs{
+		InputRowType: reflect.TypeOf(gitlabModels.GitlabPipeline{}),
+		Input:        cursor,
+		RawDataSubTaskArgs: helper.RawDataSubTaskArgs{
+			Ctx: taskCtx,
+			Params: GitlabApiParams{
+				ConnectionId: data.Options.ConnectionId,
+				ProjectId:    data.Options.ProjectId,
+			},
+			Table: RAW_USER_TABLE,
+		},
+		Convert: func(inputRow interface{}) ([]interface{}, error) {
+			gitlabPipeline := inputRow.(*gitlabModels.GitlabPipeline)
+
+			createdAt := time.Now()
+			if gitlabPipeline.GitlabCreatedAt != nil {
+				createdAt = *gitlabPipeline.GitlabCreatedAt
+			}
+			finishedAt := time.Now()
+			if gitlabPipeline.Status == "success" && gitlabPipeline.GitlabUpdatedAt != nil {
+				finishedAt = *gitlabPipeline.GitlabUpdatedAt
+			}
+
+			durationTime := finishedAt.Sub(createdAt)
+			domainPipeline := &devops.CICDPipeline{
+				DomainEntity: domainlayer.DomainEntity{
+					Id: pipelineIdGen.Generate(data.Options.ConnectionId, gitlabPipeline.GitlabId),
+				},
+
+				Name:      fmt.Sprintf("%d", gitlabPipeline.GitlabId),
+				CommitSha: gitlabPipeline.Sha,
+				Branch:    gitlabPipeline.Ref,
+				Repo:      fmt.Sprintf("%d", gitlabPipeline.ProjectId),
+				Result:    gitlabPipeline.Status,
+				Status:    gitlabPipeline.Status,
+				Type:      "CI/CD",
+
+				DurationSec:  uint64(durationTime.Seconds()),
+				CreatedDate:  createdAt,
+				FinishedDate: finishedAt,
+			}
+
+			return []interface{}{
+				domainPipeline,
+			}, nil
+		},
+	})
+
+	if err != nil {
+		return err
+	}
+
+	return converter.Execute()
+}