ae37ebb286
Apply "xz" compression to any artifact pipeline if it's specified in the image type. The image filename should be applied to the pipeline only if it's the last one, so we need to skip this assignment if we're going to add a compression pipeline at the end. This is a bit dirty but the plan is to remove artifact compression from manifests and perform it during the export in osbuild-composer, so this should go away soon.
91 lines
2.7 KiB
Go
91 lines
2.7 KiB
Go
package image
|
|
|
|
import (
|
|
"math/rand"
|
|
|
|
"github.com/osbuild/osbuild-composer/internal/artifact"
|
|
"github.com/osbuild/osbuild-composer/internal/disk"
|
|
"github.com/osbuild/osbuild-composer/internal/environment"
|
|
"github.com/osbuild/osbuild-composer/internal/manifest"
|
|
"github.com/osbuild/osbuild-composer/internal/platform"
|
|
"github.com/osbuild/osbuild-composer/internal/rpmmd"
|
|
"github.com/osbuild/osbuild-composer/internal/runner"
|
|
"github.com/osbuild/osbuild-composer/internal/workload"
|
|
)
|
|
|
|
type LiveImage struct {
|
|
Base
|
|
Platform platform.Platform
|
|
PartitionTable *disk.PartitionTable
|
|
OSCustomizations manifest.OSCustomizations
|
|
Environment environment.Environment
|
|
Workload workload.Workload
|
|
Filename string
|
|
Compression string
|
|
}
|
|
|
|
func NewLiveImage() *LiveImage {
|
|
return &LiveImage{
|
|
Base: NewBase("live-image"),
|
|
}
|
|
}
|
|
|
|
func (img *LiveImage) InstantiateManifest(m *manifest.Manifest,
|
|
repos []rpmmd.RepoConfig,
|
|
runner runner.Runner,
|
|
rng *rand.Rand) (*artifact.Artifact, error) {
|
|
buildPipeline := manifest.NewBuild(m, runner, repos)
|
|
buildPipeline.Checkpoint()
|
|
|
|
osPipeline := manifest.NewOS(m, buildPipeline, img.Platform, repos)
|
|
osPipeline.PartitionTable = img.PartitionTable
|
|
osPipeline.OSCustomizations = img.OSCustomizations
|
|
osPipeline.Environment = img.Environment
|
|
osPipeline.Workload = img.Workload
|
|
|
|
imagePipeline := manifest.NewRawImage(m, buildPipeline, osPipeline)
|
|
|
|
var artifact *artifact.Artifact
|
|
var artifactPipeline manifest.Pipeline
|
|
switch img.Platform.GetImageFormat() {
|
|
case platform.FORMAT_RAW:
|
|
if img.Compression == "" {
|
|
imagePipeline.Filename = img.Filename
|
|
}
|
|
artifactPipeline = imagePipeline
|
|
artifact = imagePipeline.Export()
|
|
case platform.FORMAT_QCOW2:
|
|
qcow2Pipeline := manifest.NewQCOW2(m, buildPipeline, imagePipeline)
|
|
if img.Compression == "" {
|
|
qcow2Pipeline.Filename = img.Filename
|
|
}
|
|
qcow2Pipeline.Compat = img.Platform.GetQCOW2Compat()
|
|
artifactPipeline = qcow2Pipeline
|
|
artifact = qcow2Pipeline.Export()
|
|
case platform.FORMAT_VHD:
|
|
vpcPipeline := manifest.NewVPC(m, buildPipeline, imagePipeline)
|
|
if img.Compression == "" {
|
|
vpcPipeline.Filename = img.Filename
|
|
}
|
|
artifactPipeline = vpcPipeline
|
|
artifact = vpcPipeline.Export()
|
|
case platform.FORMAT_VMDK:
|
|
vmdkPipeline := manifest.NewVMDK(m, buildPipeline, imagePipeline)
|
|
if img.Compression == "" {
|
|
vmdkPipeline.Filename = img.Filename
|
|
}
|
|
artifactPipeline = vmdkPipeline
|
|
artifact = vmdkPipeline.Export()
|
|
default:
|
|
panic("invalid image format for image kind")
|
|
}
|
|
|
|
switch img.Compression {
|
|
case "xz":
|
|
xzPipeline := manifest.NewXZ(m, buildPipeline, artifactPipeline)
|
|
xzPipeline.Filename = img.Filename
|
|
artifact = xzPipeline.Export()
|
|
}
|
|
|
|
return artifact, nil
|
|
}
|