68 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			68 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
/*
 | 
						|
Copyright 2018 Google LLC
 | 
						|
 | 
						|
Licensed 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 commands
 | 
						|
 | 
						|
import (
 | 
						|
	"github.com/GoogleContainerTools/kaniko/pkg/dockerfile"
 | 
						|
	"github.com/GoogleContainerTools/kaniko/pkg/util"
 | 
						|
	v1 "github.com/google/go-containerregistry/pkg/v1"
 | 
						|
	"github.com/moby/buildkit/frontend/dockerfile/instructions"
 | 
						|
)
 | 
						|
 | 
						|
type ArgCommand struct {
 | 
						|
	BaseCommand
 | 
						|
	cmd *instructions.ArgCommand
 | 
						|
}
 | 
						|
 | 
						|
// ExecuteCommand only needs to add this ARG key/value as seen
 | 
						|
func (r *ArgCommand) ExecuteCommand(config *v1.Config, buildArgs *dockerfile.BuildArgs) error {
 | 
						|
	key, val, err := ParseArg(r.cmd.Key, r.cmd.Value, config.Env, buildArgs)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	buildArgs.AddArg(key, val)
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func ParseArg(key string, val *string, env []string, ba *dockerfile.BuildArgs) (string, *string, error) {
 | 
						|
	replacementEnvs := ba.ReplacementEnvs(env)
 | 
						|
	resolvedKey, err := util.ResolveEnvironmentReplacement(key, replacementEnvs, false)
 | 
						|
	if err != nil {
 | 
						|
		return "", nil, err
 | 
						|
	}
 | 
						|
	var resolvedValue *string
 | 
						|
	if val != nil {
 | 
						|
		value, err := util.ResolveEnvironmentReplacement(*val, replacementEnvs, false)
 | 
						|
		if err != nil {
 | 
						|
			return "", nil, err
 | 
						|
		}
 | 
						|
		resolvedValue = &value
 | 
						|
	} else {
 | 
						|
		meta := ba.GetAllMeta()
 | 
						|
		if value, ok := meta[resolvedKey]; ok {
 | 
						|
			resolvedValue = &value
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return resolvedKey, resolvedValue, nil
 | 
						|
}
 | 
						|
 | 
						|
// String returns some information about the command for the image config history
 | 
						|
func (r *ArgCommand) String() string {
 | 
						|
	return r.cmd.String()
 | 
						|
}
 |