porter/cli/cmd/v2/build.go

package v2

import (
	"context"
	"errors"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/porter-dev/porter/api/types"
	"github.com/porter-dev/porter/cli/cmd/pack"

	"github.com/porter-dev/porter/cli/cmd/docker"

	api "github.com/porter-dev/porter/api/client"
)

const (
	buildMethodPack   = "pack"
	buildMethodDocker = "docker"
	buildLogFilename  = "PORTER_BUILD_LOGS"
)

// buildInput is the input struct for the build method
type buildInput struct {
	ProjectID uint
	// AppName is the name of the application being built and is used to name the repository
	AppName      string
	BuildContext string
	Dockerfile   string
	BuildMethod  string
	// Builder is the image containing the components necessary to build the application in a pack build
	Builder    string
	BuildPacks []string
	// ImageTag is the tag to apply to the new image
	ImageTag string
	// CurrentImageTag is used in docker build to cache from
	CurrentImageTag string
	RepositoryURL   string
	// PullImageBeforeBuild is used to pull the docker image before building
	PullImageBeforeBuild bool

	Env map[string]string
	// SkipPush is used to skip pushing the image to the registry
	SkipPush bool
}

type buildOutput struct {
	Error error
	Logs  string
}

// build will create an image repository if it does not exist, and then build and push the image
func build(ctx context.Context, client api.Client, inp buildInput) buildOutput {
	output := buildOutput{}

	if inp.ProjectID == 0 {
		output.Error = errors.New("must specify a project id")
		return output
	}
	projectID := inp.ProjectID

	if inp.ImageTag == "" {
		output.Error = errors.New("must specify an image tag")
		return output
	}
	tag := inp.ImageTag

	if inp.RepositoryURL == "" {
		output.Error = errors.New("must specify a registry url")
		return output
	}
	repositoryURL := strings.TrimPrefix(inp.RepositoryURL, "https://")

	err := createImageRepositoryIfNotExists(ctx, client, projectID, repositoryURL)
	if err != nil {
		output.Error = fmt.Errorf("error creating image repository: %w", err)
		return output
	}

	dockerAgent, err := docker.NewAgentWithAuthGetter(ctx, client, projectID)
	if err != nil {
		output.Error = fmt.Errorf("error getting docker agent: %w", err)
		return output
	}

	// create a temp file which build logs will be written to
	// temp file gets cleaned up when os exits (i.e. when the GHA completes), so no need to remove it manually
	logFile, _ := os.CreateTemp("", buildLogFilename)

	switch inp.BuildMethod {
	case buildMethodDocker:
		basePath, err := filepath.Abs(".")
		if err != nil {
			output.Error = fmt.Errorf("error getting absolute path: %w", err)
			return output
		}

		buildCtx, dockerfilePath, isDockerfileInCtx, err := resolveDockerPaths(
			basePath,
			inp.BuildContext,
			inp.Dockerfile,
		)
		if err != nil {
			output.Error = fmt.Errorf("error resolving docker paths: %w", err)
			return output
		}

		opts := &docker.BuildOpts{
			ImageRepo:         repositoryURL,
			Tag:               tag,
			CurrentTag:        inp.CurrentImageTag,
			BuildContext:      buildCtx,
			DockerfilePath:    dockerfilePath,
			IsDockerfileInCtx: isDockerfileInCtx,
			Env:               inp.Env,
			LogFile:           logFile,
			UseCache:          inp.PullImageBeforeBuild,
		}

		err = dockerAgent.BuildLocal(
			ctx,
			opts,
		)
		if err != nil {
			output.Error = fmt.Errorf("error building image with docker: %w", err)
			logString := "Error reading contents of build log file"

			if logFile != nil {
				content, err := os.ReadFile(logFile.Name())
				// only continue if we can read the file. if we cannot, logString will be the default
				if err == nil {
					logString = string(content)
				}
			}
			output.Logs = logString
			return output
		}
	case buildMethodPack:
		packAgent := &pack.Agent{}

		opts := &docker.BuildOpts{
			ImageRepo:    repositoryURL,
			Tag:          tag,
			BuildContext: inp.BuildContext,
			Env:          inp.Env,
			LogFile:      logFile,
		}

		buildConfig := &types.BuildConfig{
			Builder:    inp.Builder,
			Buildpacks: inp.BuildPacks,
		}

		if buildConfig.Builder == "heroku/buildpacks:20" {
			if opts.Env == nil {
				opts.Env = map[string]string{}
			}
			opts.Env["ALLOW_EOL_SHIMMED_BUILDER"] = "1"
		}

		err := packAgent.Build(ctx, opts, buildConfig, "")
		if err != nil {
			output.Error = fmt.Errorf("error building image with pack: %w", err)
			logString := "Error reading contents of build log file"

			if logFile != nil {
				content, err := os.ReadFile(logFile.Name())
				// only continue if we can read the file. if we cannot, logString will be the default
				if err == nil {
					logString = string(content)
				}
			}
			output.Logs = logString
			return output
		}
	default:
		output.Error = fmt.Errorf("invalid build method: %s", inp.BuildMethod)
		return output
	}

	if !inp.SkipPush {
		err = dockerAgent.PushImage(ctx, fmt.Sprintf("%s:%s", repositoryURL, tag))
		if err != nil {
			output.Error = fmt.Errorf("error pushing image: %w", err)
			return output
		}
	}

	return output
}

type pushInput struct {
	ProjectID     uint
	ImageTag      string
	RepositoryURL string
}

func push(ctx context.Context, client api.Client, inp pushInput) error {
	if inp.ProjectID == 0 {
		return errors.New("must specify a project id")
	}
	projectID := inp.ProjectID

	if inp.ImageTag == "" {
		return errors.New("must specify an image tag")
	}
	tag := inp.ImageTag

	if inp.RepositoryURL == "" {
		return errors.New("must specify a registry url")
	}
	repositoryURL := strings.TrimPrefix(inp.RepositoryURL, "https://")

	dockerAgent, err := docker.NewAgentWithAuthGetter(ctx, client, projectID)
	if err != nil {
		return fmt.Errorf("error getting docker agent: %w", err)
	}

	err = dockerAgent.PushImage(ctx, fmt.Sprintf("%s:%s", repositoryURL, tag))
	if err != nil {
		return fmt.Errorf("error pushing image: %w", err)
	}

	return nil
}

func createImageRepositoryIfNotExists(ctx context.Context, client api.Client, projectID uint, imageURL string) error {
	if projectID == 0 {
		return errors.New("must specify a project id")
	}

	if imageURL == "" {
		return errors.New("must specify an image url")
	}

	regList, err := client.ListRegistries(ctx, projectID)
	if err != nil {
		return fmt.Errorf("error calling list registries: %w", err)
	}

	if regList == nil {
		return errors.New("registry list is nil")
	}

	if len(*regList) == 0 {
		return errors.New("no registries found for project")
	}

	var registryID uint
	for _, registry := range *regList {
		if strings.Contains(strings.TrimPrefix(imageURL, "https://"), strings.TrimPrefix(registry.URL, "https://")) {
			registryID = registry.ID
			break
		}
	}

	if registryID == 0 {
		return errors.New("no registries match url")
	}

	err = client.CreateRepository(
		ctx,
		projectID,
		registryID,
		&types.CreateRegistryRepositoryRequest{
			ImageRepoURI: imageURL,
		},
	)
	if err != nil {
		return fmt.Errorf("error creating repository: %w", err)
	}

	return nil
}

// resolveDockerPaths returns a path to the dockerfile that is either relative or absolute, and a path
// to the build context that is absolute.
//
// The return value will be relative if the dockerfile exists within the build context, absolute
// otherwise. The second return value is true if the dockerfile exists within the build context,
// false otherwise.
func resolveDockerPaths(basePath string, buildContextPath string, dockerfilePath string) (
	absoluteBuildContextPath string,
	outputDockerfilePath string,
	isDockerfileRelative bool,
	err error,
) {
	absoluteBuildContextPath, err = filepath.Abs(buildContextPath)
	if err != nil {
		return "", "", false, fmt.Errorf("error getting absolute path: %w", err)
	}
	outputDockerfilePath = dockerfilePath

	if !filepath.IsAbs(dockerfilePath) {
		outputDockerfilePath = filepath.Join(basePath, dockerfilePath)
	}

	pathComp, err := filepath.Rel(absoluteBuildContextPath, outputDockerfilePath)
	if err != nil {
		return "", "", false, fmt.Errorf("error getting relative path: %w", err)
	}

	if !strings.HasPrefix(pathComp, ".."+string(os.PathSeparator)) {
		isDockerfileRelative = true
		return absoluteBuildContextPath, pathComp, isDockerfileRelative, nil
	}
	isDockerfileRelative = false

	outputDockerfilePath, err = filepath.Abs(outputDockerfilePath)
	if err != nil {
		return "", "", false, err
	}

	return absoluteBuildContextPath, outputDockerfilePath, isDockerfileRelative, nil
}