werf/werf

package instruction

import (
    "context"
    "fmt"

    "github.com/moby/buildkit/frontend/dockerfile/instructions"

    "github.com/werf/werf/v2/pkg/buildah"
    "github.com/werf/werf/v2/pkg/container_backend"
)

type Copy struct {
    instructions.CopyCommand
}

func NewCopy(i instructions.CopyCommand) *Copy {
    return &Copy{CopyCommand: i}
}

func (i *Copy) UsesBuildContext() bool {
    return i.From == ""
}

func (i *Copy) Apply(ctx context.Context, containerName string, drv buildah.Buildah, drvOpts buildah.CommonOpts, buildContextArchive container_backend.BuildContextArchiver) error {
    var contextDir string
    if i.UsesBuildContext() {
        var err error
        contextDir, err = buildContextArchive.ExtractOrGetExtractedDir(ctx)
        if err != nil {
            return fmt.Errorf("unable to extract build context: %w", err)
        }
    } else {
        container, err := drv.FromCommand(ctx, "", i.From, buildah.FromCommandOpts{})
        if err != nil {
            return fmt.Errorf("unable to create container from image %q: %w", i.From, err)
        }

        contextDir, err = drv.Mount(ctx, container, buildah.MountOpts{})
        if err != nil {
            return fmt.Errorf("unable to mount container %q: %w", container, err)
        }
    }

    if err := drv.Copy(ctx, containerName, contextDir, i.SourcePaths, i.DestPath, buildah.CopyOpts{
        CommonOpts: drvOpts,
        Chown:      i.Chown,
        Chmod:      i.Chmod,
    }); err != nil {
        return fmt.Errorf("error copying %v to %s for container %s: %w", i.SourcePaths, i.DestPath, containerName, err)
    }

    return nil
}