From 4add95c8491e03338da66d3c8dc1829f613fc375 Mon Sep 17 00:00:00 2001
From: gsquared94 <gaghosh@google.com>
Date: Thu, 10 Sep 2020 11:18:55 +0530
Subject: [PATCH] Pass correct build args to `CreateDockerTarContext`

---
 pkg/skaffold/build/cluster/kaniko.go | 5 +----
 pkg/skaffold/diagnose/diagnose.go    | 2 +-
 pkg/skaffold/docker/context.go       | 5 ++---
 pkg/skaffold/docker/context_test.go  | 2 +-
 pkg/skaffold/docker/image.go         | 2 +-
 5 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/pkg/skaffold/build/cluster/kaniko.go b/pkg/skaffold/build/cluster/kaniko.go
index ec1c63869ae..ba3d401ac95 100644
--- a/pkg/skaffold/build/cluster/kaniko.go
+++ b/pkg/skaffold/build/cluster/kaniko.go
@@ -86,10 +86,7 @@ func (b *Builder) copyKanikoBuildContext(ctx context.Context, workspace string,
 
 	buildCtx, buildCtxWriter := io.Pipe()
 	go func() {
-		err := docker.CreateDockerTarContext(ctx, buildCtxWriter, workspace, &latest.DockerArtifact{
-			BuildArgs:      artifact.BuildArgs,
-			DockerfilePath: artifact.DockerfilePath,
-		}, b.insecureRegistries)
+		err := docker.CreateDockerTarContext(ctx, buildCtxWriter, workspace, artifact.DockerfilePath, artifact.BuildArgs, b.insecureRegistries)
 		if err != nil {
 			buildCtxWriter.CloseWithError(fmt.Errorf("creating docker context: %w", err))
 			return
diff --git a/pkg/skaffold/diagnose/diagnose.go b/pkg/skaffold/diagnose/diagnose.go
index 7d2348550bc..dd24ea89768 100644
--- a/pkg/skaffold/diagnose/diagnose.go
+++ b/pkg/skaffold/diagnose/diagnose.go
@@ -136,7 +136,7 @@ func timeToComputeMTimes(deps []string) (time.Duration, error) {
 func sizeOfDockerContext(ctx context.Context, a *latest.Artifact, insecureRegistries map[string]bool) (int64, error) {
 	buildCtx, buildCtxWriter := io.Pipe()
 	go func() {
-		err := docker.CreateDockerTarContext(ctx, buildCtxWriter, a.Workspace, a.DockerArtifact, insecureRegistries)
+		err := docker.CreateDockerTarContext(ctx, buildCtxWriter, a.Workspace, a.DockerArtifact.DockerfilePath, a.DockerArtifact.BuildArgs, insecureRegistries)
 		if err != nil {
 			buildCtxWriter.CloseWithError(fmt.Errorf("creating docker context: %w", err))
 			return
diff --git a/pkg/skaffold/docker/context.go b/pkg/skaffold/docker/context.go
index dc581ad8d23..8c228391bf9 100644
--- a/pkg/skaffold/docker/context.go
+++ b/pkg/skaffold/docker/context.go
@@ -22,12 +22,11 @@ import (
 	"io"
 	"path/filepath"
 
-	"github.com/GoogleContainerTools/skaffold/pkg/skaffold/schema/latest"
 	"github.com/GoogleContainerTools/skaffold/pkg/skaffold/util"
 )
 
-func CreateDockerTarContext(ctx context.Context, w io.Writer, workspace string, a *latest.DockerArtifact, insecureRegistries map[string]bool) error {
-	paths, err := GetDependencies(ctx, workspace, a.DockerfilePath, a.BuildArgs, insecureRegistries)
+func CreateDockerTarContext(ctx context.Context, w io.Writer, workspace string, dockerfilePath string, buildArgs map[string]*string, insecureRegistries map[string]bool) error {
+	paths, err := GetDependencies(ctx, workspace, dockerfilePath, buildArgs, insecureRegistries)
 	if err != nil {
 		return fmt.Errorf("getting relative tar paths: %w", err)
 	}
diff --git a/pkg/skaffold/docker/context_test.go b/pkg/skaffold/docker/context_test.go
index 6c30c84c212..aef027d5a2b 100644
--- a/pkg/skaffold/docker/context_test.go
+++ b/pkg/skaffold/docker/context_test.go
@@ -46,7 +46,7 @@ func TestDockerContext(t *testing.T) {
 
 			reader, writer := io.Pipe()
 			go func() {
-				err := CreateDockerTarContext(context.Background(), writer, dir, artifact, nil)
+				err := CreateDockerTarContext(context.Background(), writer, dir, artifact.DockerfilePath, artifact.BuildArgs, nil)
 				if err != nil {
 					writer.CloseWithError(err)
 				} else {
diff --git a/pkg/skaffold/docker/image.go b/pkg/skaffold/docker/image.go
index 3d925c45168..ae75d1df892 100644
--- a/pkg/skaffold/docker/image.go
+++ b/pkg/skaffold/docker/image.go
@@ -171,7 +171,7 @@ func (l *localDaemon) Build(ctx context.Context, out io.Writer, workspace string
 
 	buildCtx, buildCtxWriter := io.Pipe()
 	go func() {
-		err := CreateDockerTarContext(ctx, buildCtxWriter, workspace, a, l.insecureRegistries)
+		err := CreateDockerTarContext(ctx, buildCtxWriter, workspace, a.DockerfilePath, buildArgs, l.insecureRegistries)
 		if err != nil {
 			buildCtxWriter.CloseWithError(fmt.Errorf("creating docker context: %w", err))
 			return