From 7109d6c66f63c8b3986bc7765400cf33fb536ecd Mon Sep 17 00:00:00 2001
From: sfwn <sfwnlin@gmail.com>
Date: Fri, 5 Jul 2024 15:02:37 +0800
Subject: [PATCH] feat(ios-action): create runner-task with org-id

---
 actions/ios/1.0/internal/conf/conf.go       |  1 +
 actions/ios/1.0/internal/pkg/build/api.go   |  2 +-
 actions/ios/1.0/internal/pkg/build/build.go | 13 +++++++++++--
 3 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/actions/ios/1.0/internal/conf/conf.go b/actions/ios/1.0/internal/conf/conf.go
index 0e74f0a5..614ee39f 100644
--- a/actions/ios/1.0/internal/conf/conf.go
+++ b/actions/ios/1.0/internal/conf/conf.go
@@ -9,6 +9,7 @@ type Conf struct {
 	DiceOpenapiPrefix string `env:"DICE_OPENAPI_ADDR" required:"true"`
 	PipelineTaskID    string `env:"PIPELINE_TASK_ID" required:"true"`
 	PipelineID        string `env:"PIPELINE_ID" required:"true"`
+	DiceOrgID         int64  `env:"DICE_ORG_ID" required:"true"`
 
 	Commands        []string             `env:"ACTION_COMMANDS"`
 	Targets         []string             `env:"ACTION_TARGETS"`
diff --git a/actions/ios/1.0/internal/pkg/build/api.go b/actions/ios/1.0/internal/pkg/build/api.go
index 5cab7e44..5dc0bcee 100644
--- a/actions/ios/1.0/internal/pkg/build/api.go
+++ b/actions/ios/1.0/internal/pkg/build/api.go
@@ -47,7 +47,7 @@ func UpdateTaskStatus(cfg conf.Conf, taskID int64, status string) error {
 	return nil
 }
 
-func CreateTask(cfg conf.Conf, req *apistructs.CreateRunnerTaskRequest) (int64, error) {
+func CreateTask(cfg conf.Conf, req *CreateRunnerTaskRequest) (int64, error) {
 	var resp apistructs.CreateRunnerTaskResponse
 	request := httpclient.New(httpclient.WithCompleteRedirect()).Post(cfg.DiceOpenapiPrefix).
 		Path(fmt.Sprintf("/api/runner/tasks")).
diff --git a/actions/ios/1.0/internal/pkg/build/build.go b/actions/ios/1.0/internal/pkg/build/build.go
index 496ebeed..0995f61a 100644
--- a/actions/ios/1.0/internal/pkg/build/build.go
+++ b/actions/ios/1.0/internal/pkg/build/build.go
@@ -39,6 +39,11 @@ type RunnerTask struct {
 	ResultDataUrl  string `json:"result_data_url"`
 }
 
+type CreateRunnerTaskRequest struct {
+	apistructs.CreateRunnerTaskRequest
+	OrgID int64 `json:"org_id"`
+}
+
 func Execute() error {
 
 	envconf.MustLoad(&cfg)
@@ -115,15 +120,19 @@ func Execute() error {
 	}
 
 	commands = append(commands, cfg.Commands...)
-	createReq := &apistructs.CreateRunnerTaskRequest{
+	createReq := apistructs.CreateRunnerTaskRequest{
 		JobID:          cfg.PipelineTaskLogID,
 		ContextDataUrl: uploadResult.Data.DownloadURL,
 		Commands:       commands,
 		WorkDir:        strings.Replace(cfg.Context, cfg.PipelineContext+"/", "", -1),
 		Targets:        cfg.Targets,
 	}
+	createReqWithOrg := CreateRunnerTaskRequest{
+		CreateRunnerTaskRequest: createReq,
+		OrgID:                   cfg.DiceOrgID,
+	}
 
-	taskID, err := CreateTask(cfg, createReq)
+	taskID, err := CreateTask(cfg, &createReqWithOrg)
 	if err != nil {
 		return err
 	}