trivy/internal/operation/operation.go

package operation

import (
	"context"
	"os"

	"golang.org/x/xerrors"

	"github.com/aquasecurity/fanal/cache"
	"github.com/aquasecurity/trivy-db/pkg/db"
	"github.com/aquasecurity/trivy/pkg/log"
	"github.com/aquasecurity/trivy/pkg/utils"
)

func Reset() (err error) {
	log.Logger.Info("Resetting...")
	cache := cache.Initialize(utils.CacheDir())
	if err = cache.Clear(); err != nil {
		return xerrors.New("failed to remove image layer cache")
	}
	if err = os.RemoveAll(utils.CacheDir()); err != nil {
		return xerrors.New("failed to remove cache")
	}
	return nil
}

func ClearCache() error {
	log.Logger.Info("Removing image caches...")
	cache := cache.Initialize(utils.CacheDir())
	if err := cache.Clear(); err != nil {
		return xerrors.New("failed to remove image layer cache")
	}
	return nil
}

func DownloadDB(appVersion, cacheDir string, light, skipUpdate bool) error {
	client := initializeDBClient()
	ctx := context.Background()
	needsUpdate, err := client.NeedsUpdate(ctx, appVersion, light, skipUpdate)
	if err != nil {
		return xerrors.Errorf("database error: %w", err)
	}

	if needsUpdate {
		log.Logger.Info("Need to update DB")
		if err = db.Close(); err != nil {
			return xerrors.Errorf("failed db close: %w", err)
		}
		if err := client.Download(ctx, cacheDir, light); err != nil {
			return xerrors.Errorf("failed to download vulnerability DB: %w", err)
		}

		log.Logger.Info("Reopening DB...")
		if err = db.Init(cacheDir); err != nil {
			return xerrors.Errorf("failed db close: %w", err)
		}
	}

	// for debug
	if err := showDBInfo(); err != nil {
		return xerrors.Errorf("failed to show database info")
	}
	return nil
}

func showDBInfo() error {
	metadata, err := db.Config{}.GetMetadata()
	if err != nil {
		return xerrors.Errorf("something wrong with DB: %w", err)
	}
	log.Logger.Debugf("DB Schema: %d, Type: %d, UpdatedAt: %s, NextUpdate: %s",
		metadata.Version, metadata.Type, metadata.UpdatedAt, metadata.NextUpdate)
	return nil
}