mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-22 23:26:39 -08:00
2f2d1a908b
* refactor: embed config * refactor: replace image and layer with artifact and blob * feat(config): add ArtifactConfig * fix(scanner): use Artifact * test(scanner): update mocks * feat: add repo and fs subcommands * chore(mod): update * refactor: fix warn message * feat(cli): add --no-progress to repo and fs * mod: Update fanal dependency Signed-off-by: Simarpreet Singh <simar@linux.com> Co-authored-by: Simarpreet Singh <simar@linux.com>
58 lines
2.0 KiB
Go
58 lines
2.0 KiB
Go
// Code generated by Wire. DO NOT EDIT.
|
|
|
|
//go:generate wire
|
|
//+build !wireinject
|
|
|
|
package client
|
|
|
|
import (
|
|
"context"
|
|
image2 "github.com/aquasecurity/fanal/artifact/image"
|
|
"github.com/aquasecurity/fanal/cache"
|
|
"github.com/aquasecurity/fanal/image"
|
|
"github.com/aquasecurity/trivy-db/pkg/db"
|
|
"github.com/aquasecurity/trivy/pkg/rpc/client"
|
|
"github.com/aquasecurity/trivy/pkg/scanner"
|
|
"github.com/aquasecurity/trivy/pkg/types"
|
|
"github.com/aquasecurity/trivy/pkg/vulnerability"
|
|
"time"
|
|
)
|
|
|
|
// Injectors from inject.go:
|
|
|
|
func initializeDockerScanner(ctx context.Context, imageName string, artifactCache cache.ArtifactCache, customHeaders client.CustomHeaders, url client.RemoteURL, timeout time.Duration) (scanner.Scanner, func(), error) {
|
|
scannerScanner := client.NewProtobufClient(url)
|
|
clientScanner := client.NewScanner(customHeaders, scannerScanner)
|
|
dockerOption, err := types.GetDockerOption(timeout)
|
|
if err != nil {
|
|
return scanner.Scanner{}, nil, err
|
|
}
|
|
imageImage, cleanup, err := image.NewDockerImage(ctx, imageName, dockerOption)
|
|
if err != nil {
|
|
return scanner.Scanner{}, nil, err
|
|
}
|
|
artifact := image2.NewArtifact(imageImage, artifactCache)
|
|
scanner2 := scanner.NewScanner(clientScanner, artifact)
|
|
return scanner2, func() {
|
|
cleanup()
|
|
}, nil
|
|
}
|
|
|
|
func initializeArchiveScanner(ctx context.Context, filePath string, artifactCache cache.ArtifactCache, customHeaders client.CustomHeaders, url client.RemoteURL, timeout time.Duration) (scanner.Scanner, error) {
|
|
scannerScanner := client.NewProtobufClient(url)
|
|
clientScanner := client.NewScanner(customHeaders, scannerScanner)
|
|
imageImage, err := image.NewArchiveImage(filePath)
|
|
if err != nil {
|
|
return scanner.Scanner{}, err
|
|
}
|
|
artifact := image2.NewArtifact(imageImage, artifactCache)
|
|
scanner2 := scanner.NewScanner(clientScanner, artifact)
|
|
return scanner2, nil
|
|
}
|
|
|
|
func initializeVulnerabilityClient() vulnerability.Client {
|
|
config := db.Config{}
|
|
vulnerabilityClient := vulnerability.NewClient(config)
|
|
return vulnerabilityClient
|
|
}
|