POC BB Extension

Signed-off-by: Austin Abro <[email protected]>
zarf-dev · Aug 27, 2024 · 148e95c · 148e95c
1 parent f6e06da
commit 148e95c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/src/api/v1beta1/package.go b/src/api/v1beta1/package.go
@@ -69,6 +69,13 @@ func (pkg ZarfPackage) HasImages() bool {
 	return false
 }
 
+func (pkg ZarfPackage) IsAirGap() bool {
+	if pkg.Metadata.Airgap == nil {
+		return true
+	}
+	return *pkg.Metadata.Airgap
+}
+
 // IsSBOMAble checks if a package has contents that an SBOM can be created on (i.e. images, files, or data injections).
 func (pkg ZarfPackage) IsSBOMAble() bool {
 	for _, c := range pkg.Components {

diff --git a/src/pkg/packager/creator/normal.go b/src/pkg/packager/creator/normal.go
@@ -330,7 +330,7 @@ func (pc *PackageCreator) Output(ctx context.Context, dst *layout.PackagePaths,
 	return nil
 }
 
-func (pc *PackageCreator) processExtensions(ctx context.Context, pkg v1beta1.ZarfPackage, layout *layout.PackagePaths) (processedComponents []v1beta1.ZarfComponent, err error) {
+func (pc *PackageCreator) processExtensions(ctx context.Context, pkg types.ZarfPackage, layout *layout.PackagePaths) (processedComponents []types.ZarfComponent, err error) {
 	// Create component paths and process extensions for each component.
 	for _, c := range pkg.Components {
 		componentPaths, err := layout.Components.Create(c)

diff --git a/src/types/package.go b/src/types/package.go
@@ -69,6 +69,13 @@ func (pkg ZarfPackage) HasImages() bool {
 	return false
 }
 
+func (pkg ZarfPackage) IsAirGap() bool {
+	if pkg.Metadata.Airgap == nil {
+		return true
+	}
+	return *pkg.Metadata.Airgap
+}
+
 // IsSBOMAble checks if a package has contents that an SBOM can be created on (i.e. images, files, or data injections).
 func (pkg ZarfPackage) IsSBOMAble() bool {
 	for _, c := range pkg.Components {