package main

import (

"fmt"

"os"

namepkg "github.com/google/go-containerregistry/pkg/name"

cachepkg "github.com/linuxkit/linuxkit/src/cmd/linuxkit/cache"

"github.com/linuxkit/linuxkit/src/cmd/linuxkit/registry"

log "github.com/sirupsen/logrus"

"github.com/spf13/cobra"

)

func cacheCleanCmd() *cobra.Command {

var (

publishedOnly bool

)

cmd := &cobra.Command{

Use: "clean",

Short: "empty the linuxkit cache",

Long: `Empty the linuxkit cache.`,

RunE: func(cmd *cobra.Command, args []string) error {

// did we limit to published only?

if !publishedOnly {

if err := os.RemoveAll(cacheDir); err != nil {

return fmt.Errorf("uUnable to clean cache %s: %v", cacheDir, err)

}

log.Infof("Cache emptied: %s", cacheDir)

return nil

}

// list all of the images and content in the cache

p, err := cachepkg.NewProvider(cacheDir)

if err != nil {

return fmt.Errorf("unable to read a local cache: %v", err)

}

images, err := p.List()

if err != nil {

return fmt.Errorf("error reading image names: %v", err)

}

removeImagesFromCache(images, p, publishedOnly)

return nil

},

}

cmd.Flags().BoolVar(&publishedOnly, "published-only", false, "Only clean images that linuxkit can confirm at the time of running have been published to the registry")

return cmd

}

// removeImagesFromCache removes images from the cache.

func removeImagesFromCache(images map[string]string, p *cachepkg.Provider, publishedOnly bool) {

// check each image in the registry. If it exists, remove it here.

for name, hash := range images {

if publishedOnly {

ref, err := namepkg.ParseReference(name)

if err != nil {

continue

}

desc, err := registry.GetRemote().Get(ref)

if err != nil {

log.Debugf("image %s not found in remote registry or error, leaving in cache: %v", name, err)

fmt.Fprintf(os.Stderr, "image %s not found in remote registry, leaving in cache", name)

continue

}

if desc == nil {

fmt.Fprintf(os.Stderr, "image %s not found in remote registry, leaving in cache", name)

continue

}

if desc.Digest.String() != hash {

fmt.Fprintf(os.Stderr, "image %s has mismatched hashes, cache %s vs remote registry %s, leaving in cache", name, hash, desc.Digest.String())

continue

}

}

// we have a match, remove it

fmt.Fprintf(os.Stderr, "removing image %s from cache", name)

if err := p.Remove(name); err != nil {

log.Warnf("Unable to remove image %s: %v", name, err)

}

}

}