mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-21 23:00:42 -08:00
52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package module
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/google/go-containerregistry/pkg/name"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/aquasecurity/trivy/pkg/fanal/types"
|
|
"github.com/aquasecurity/trivy/pkg/log"
|
|
"github.com/aquasecurity/trivy/pkg/oci"
|
|
)
|
|
|
|
const mediaType = "application/vnd.module.wasm.content.layer.v1+wasm"
|
|
|
|
// Install installs a module
|
|
func Install(ctx context.Context, dir, repo string, quiet bool, opt types.RegistryOptions) error {
|
|
ref, err := name.ParseReference(repo)
|
|
if err != nil {
|
|
return xerrors.Errorf("repository parse error: %w", err)
|
|
}
|
|
|
|
log.Info("Installing the module from the repository...", log.String("repo", repo))
|
|
art := oci.NewArtifact(repo, quiet, opt)
|
|
|
|
dst := filepath.Join(dir, ref.Context().Name())
|
|
log.Debug("Installing the module...", log.String("dst", dst))
|
|
if err = art.Download(ctx, dst, oci.DownloadOption{MediaType: mediaType}); err != nil {
|
|
return xerrors.Errorf("module download error: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Uninstall uninstalls a module
|
|
func Uninstall(_ context.Context, dir, repo string) error {
|
|
ref, err := name.ParseReference(repo)
|
|
if err != nil {
|
|
return xerrors.Errorf("repository parse error: %w", err)
|
|
}
|
|
|
|
log.Info("Uninstalling the module ...", log.String("module", repo))
|
|
dst := filepath.Join(dir, ref.Context().Name())
|
|
if err = os.RemoveAll(dst); err != nil {
|
|
return xerrors.Errorf("remove error: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|