package image

import (
	"math/rand"

	"github.com/osbuild/images/internal/environment"
	"github.com/osbuild/images/pkg/artifact"
	"github.com/osbuild/images/pkg/manifest"
	"github.com/osbuild/images/pkg/platform"
	"github.com/osbuild/images/pkg/rpmmd"
	"github.com/osbuild/images/pkg/runner"
)

type BaseContainer struct {
	Base
	Platform              platform.Platform
	OSCustomizations      manifest.OSCustomizations
	Environment           environment.Environment
	ImgTypeCustomizations manifest.OSCustomizations
	Filename              string
}

func NewBaseContainer() *BaseContainer {
	return &BaseContainer{
		Base: NewBase("base-container"),
	}
}

func (img *BaseContainer) InstantiateManifest(m *manifest.Manifest,
	repos []rpmmd.RepoConfig,
	runner runner.Runner,
	rng *rand.Rand) (*artifact.Artifact, error) {
	buildPipeline := addBuildBootstrapPipelines(m, runner, repos, nil)
	buildPipeline.Checkpoint()

	osPipeline := manifest.NewOS(buildPipeline, img.Platform, repos)
	osPipeline.OSCustomizations = img.OSCustomizations
	osPipeline.Environment = img.Environment
	osPipeline.ImgTypeCustomizations = img.ImgTypeCustomizations

	ociPipeline := manifest.NewOCIContainer(buildPipeline, osPipeline)
	ociPipeline.SetFilename(img.Filename)
	artifact := ociPipeline.Export()

	return artifact, nil
}