-
Notifications
You must be signed in to change notification settings - Fork 173
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: razzle <[email protected]>
- Loading branch information
Showing
4 changed files
with
147 additions
and
135 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,144 @@ | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
| // SPDX-FileCopyrightText: 2021-Present The Zarf Authors | ||
|
|
||
| // Package composer contains functions for composing components within Zarf packages. | ||
| package composer | ||
|
|
||
| import ( | ||
| "context" | ||
| "crypto/sha256" | ||
| "fmt" | ||
| "os" | ||
| "path/filepath" | ||
|
|
||
| "github.com/defenseunicorns/zarf/src/config" | ||
| "github.com/defenseunicorns/zarf/src/pkg/layout" | ||
| "github.com/defenseunicorns/zarf/src/pkg/message" | ||
| "github.com/defenseunicorns/zarf/src/pkg/oci" | ||
| "github.com/defenseunicorns/zarf/src/pkg/utils" | ||
| "github.com/mholt/archiver/v3" | ||
| ocispec "github.com/opencontainers/image-spec/specs-go/v1" | ||
| "oras.land/oras-go/v2/content" | ||
| ocistore "oras.land/oras-go/v2/content/oci" | ||
| ) | ||
|
|
||
| func (ic *ImportChain) getRemote(url string) (*oci.OrasRemote, error) { | ||
| if ic.remote != nil { | ||
| return ic.remote, nil | ||
| } | ||
| var err error | ||
| ic.remote, err = oci.NewOrasRemote(url) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return ic.remote, nil | ||
| } | ||
|
|
||
| // ContainsOCIImport returns true if the import chain contains a remote import | ||
| func (ic *ImportChain) ContainsOCIImport() bool { | ||
| // only the 2nd to last node may have a remote import | ||
| return ic.tail.prev != nil && ic.tail.prev.Import.URL != "" | ||
| } | ||
|
|
||
| // OCIImportDefinition returns the url and name of the remote import | ||
| func (ic *ImportChain) OCIImportDefinition() (string, string) { | ||
| if !ic.ContainsOCIImport() { | ||
| return "", "" | ||
| } | ||
| return ic.tail.prev.Import.URL, ic.tail.prev.ImportName() | ||
| } | ||
|
|
||
| func (ic *ImportChain) fetchOCISkeleton() error { | ||
| if !ic.ContainsOCIImport() { | ||
| return nil | ||
| } | ||
| node := ic.tail.prev | ||
| remote, err := ic.getRemote(node.Import.URL) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| manifest, err := remote.FetchRoot() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| name := node.ImportName() | ||
|
|
||
| componentDesc := manifest.Locate(filepath.Join(layout.ComponentsDir, fmt.Sprintf("%s.tar", name))) | ||
|
|
||
| cache := filepath.Join(config.GetAbsCachePath(), "oci") | ||
| if err := utils.CreateDirectory(cache, 0700); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| var tb, dir string | ||
|
|
||
| // if there is not a tarball to fetch, create a directory named based upon | ||
| // the import url and the component name | ||
| if oci.IsEmptyDescriptor(componentDesc) { | ||
| h := sha256.New() | ||
| h.Write([]byte(node.Import.URL + name)) | ||
| id := fmt.Sprintf("%x", h.Sum(nil)) | ||
|
|
||
| dir = filepath.Join(cache, "dirs", id) | ||
|
|
||
| message.Debug("creating empty directory for remote component:", filepath.Join("<zarf-cache>", "oci", "dirs", id)) | ||
| } else { | ||
| tb = filepath.Join(cache, "blobs", "sha256", componentDesc.Digest.Encoded()) | ||
| dir = filepath.Join(cache, "dirs", componentDesc.Digest.Encoded()) | ||
|
|
||
| store, err := ocistore.New(cache) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| ctx := context.TODO() | ||
| // ensure the tarball is in the cache | ||
| exists, err := store.Exists(ctx, componentDesc) | ||
| if err != nil { | ||
| return err | ||
| } else if !exists { | ||
| copyOpts := remote.CopyOpts | ||
| // TODO: investigate why the default FindSuccessors in CopyWithProgress is not working | ||
| copyOpts.FindSuccessors = content.Successors | ||
| if err := remote.CopyWithProgress([]ocispec.Descriptor{componentDesc}, store, copyOpts, cache); err != nil { | ||
| return err | ||
| } | ||
| exists, err := store.Exists(ctx, componentDesc) | ||
| if err != nil { | ||
| return err | ||
| } else if !exists { | ||
| return fmt.Errorf("failed to fetch remote component: %+v", componentDesc) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| if err := utils.CreateDirectory(dir, 0700); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| cwd, err := os.Getwd() | ||
| if err != nil { | ||
| return err | ||
| } | ||
| rel, err := filepath.Rel(cwd, dir) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| // the tail node is the only node whose relativeToHead is based solely upon cwd<->cache | ||
| // contrary to the other nodes, which are based upon the previous node | ||
| ic.tail.relativeToHead = rel | ||
|
|
||
| if oci.IsEmptyDescriptor(componentDesc) { | ||
| // nothing was fetched, nothing to extract | ||
| return nil | ||
| } | ||
|
|
||
| tu := archiver.Tar{ | ||
| OverwriteExisting: true, | ||
| // removes /<component-name>/ from the paths | ||
| StripComponents: 1, | ||
| } | ||
| return tu.Unarchive(tb, dir) | ||
| } |