exportfs.go

// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.

package buildkit

import (
	"context"
	"io/fs"
	"os"

	"github.com/moby/buildkit/client"
	"namespacelabs.dev/foundation/internal/compute"
	"namespacelabs.dev/foundation/internal/fnfs"
	"namespacelabs.dev/foundation/internal/workspace/dirs"
	"namespacelabs.dev/foundation/std/tasks"
)

func exportToFS() exporter[fs.FS] { return &exportFS{} }

type exportFS struct {
	outputDir string
}

func (e *exportFS) Kind() string { return "local-fs" }

func (e *exportFS) Prepare(ctx context.Context) error {
	dir, err := dirs.CreateUserTempDir("buildkit", "fs")
	if err != nil {
		return err
	}

	e.outputDir = dir

	compute.On(ctx).Cleanup(tasks.Action("buildkit.build-fs.cleanup"), func(ctx context.Context) error {
		return os.RemoveAll(dir)
	})

	return nil
}

func (e *exportFS) Exports() []client.ExportEntry {
	return []client.ExportEntry{{
		Type:      client.ExporterLocal,
		OutputDir: e.outputDir,
	}}
}

func (e *exportFS) Provide(context.Context, *client.SolveResponse, builtkitOpts) (fs.FS, error) {
	return fnfs.Local(e.outputDir), nil
}