From d76d1379b6bb6e6e456a32b35ac84f687bb742af Mon Sep 17 00:00:00 2001
From: zc <zc2638@qq.com>
Date: Sat, 6 Apr 2024 06:06:01 +0800
Subject: [PATCH] fix docker worker command build

---
 core/worker/hooks/docker/convert.go | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/core/worker/hooks/docker/convert.go b/core/worker/hooks/docker/convert.go
index 94e108a..887c867 100644
--- a/core/worker/hooks/docker/convert.go
+++ b/core/worker/hooks/docker/convert.go
@@ -28,10 +28,6 @@ import (
 )
 
 func toContainerConfig(spec *worker.Workflow, step *worker.Step) *container.Config {
-	env := step.CombineEnv(map[string]string{
-		"INK_SCRIPT": shell.Script(step.Command),
-	})
-
 	cfg := &container.Config{
 		AttachStdin:  false,
 		AttachStdout: true,
@@ -42,13 +38,17 @@ func toContainerConfig(spec *worker.Workflow, step *worker.Step) *container.Conf
 		ArgsEscaped:  false,
 		WorkingDir:   step.WorkingDir,
 		Image:        step.Image,
-		Env:          worker.EnvToSlice(env),
 	}
 
+	var configEnv map[string]string
+
 	if len(step.Entrypoint) > 0 {
 		cfg.Entrypoint = step.Entrypoint
 		cfg.Cmd = step.Command
-	} else {
+	} else if len(step.Command) > 0 {
+		configEnv = map[string]string{
+			"INK_SCRIPT": shell.Script(step.Command),
+		}
 		cmdName, args := shell.Command()
 		if len(step.Shell) > 0 {
 			cmdName = step.Shell[0]
@@ -57,6 +57,8 @@ func toContainerConfig(spec *worker.Workflow, step *worker.Step) *container.Conf
 		cfg.Entrypoint = append([]string{cmdName}, args...)
 		cfg.Cmd = shell.EchoEnvCommand("INK_SCRIPT", cmdName)
 	}
+	env := step.CombineEnv(configEnv)
+	cfg.Env = worker.EnvToSlice(env)
 
 	if len(step.VolumeMounts) != 0 {
 		cfg.Volumes = toVolumeSet(spec, step)