forked from toolshed/abra
		
	
		
			
				
	
	
		
			103 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package server
 | 
						|
 | 
						|
import (
 | 
						|
	"coopcloud.tech/abra/cli/internal"
 | 
						|
	"coopcloud.tech/abra/pkg/autocomplete"
 | 
						|
	"coopcloud.tech/abra/pkg/client"
 | 
						|
	"coopcloud.tech/abra/pkg/formatter"
 | 
						|
	"coopcloud.tech/abra/pkg/log"
 | 
						|
	"github.com/docker/docker/api/types/filters"
 | 
						|
	"github.com/spf13/cobra"
 | 
						|
)
 | 
						|
 | 
						|
var ServerPruneCommand = &cobra.Command{
 | 
						|
	Use:     "prune <server> [flags]",
 | 
						|
	Aliases: []string{"p"},
 | 
						|
	Short:   "Prune resources on a server",
 | 
						|
	Long: `Prunes unused containers, networks, and dangling images.
 | 
						|
 | 
						|
Use "--volumes/-v" to remove volumes that are not associated with a deployed
 | 
						|
app. This can result in unwanted data loss if not used carefully.`,
 | 
						|
	Args: cobra.ExactArgs(1),
 | 
						|
	ValidArgsFunction: func(
 | 
						|
		cmd *cobra.Command,
 | 
						|
		args []string,
 | 
						|
		toComplete string) ([]string, cobra.ShellCompDirective) {
 | 
						|
		return autocomplete.ServerNameComplete()
 | 
						|
	},
 | 
						|
	Run: func(cmd *cobra.Command, args []string) {
 | 
						|
		serverName := internal.ValidateServer(args)
 | 
						|
 | 
						|
		cl, err := client.New(serverName)
 | 
						|
		if err != nil {
 | 
						|
			log.Fatal(err)
 | 
						|
		}
 | 
						|
 | 
						|
		var filterArgs filters.Args
 | 
						|
 | 
						|
		cr, err := cl.ContainersPrune(cmd.Context(), filterArgs)
 | 
						|
		if err != nil {
 | 
						|
			log.Fatal(err)
 | 
						|
		}
 | 
						|
 | 
						|
		cntSpaceReclaimed := formatter.ByteCountSI(cr.SpaceReclaimed)
 | 
						|
		log.Infof("containers pruned: %d; space reclaimed: %s", len(cr.ContainersDeleted), cntSpaceReclaimed)
 | 
						|
 | 
						|
		nr, err := cl.NetworksPrune(cmd.Context(), filterArgs)
 | 
						|
		if err != nil {
 | 
						|
			log.Fatal(err)
 | 
						|
		}
 | 
						|
 | 
						|
		log.Infof("networks pruned: %d", len(nr.NetworksDeleted))
 | 
						|
 | 
						|
		pruneFilters := filters.NewArgs()
 | 
						|
		if allFilter {
 | 
						|
			log.Debugf("removing all images, not only dangling ones")
 | 
						|
			pruneFilters.Add("dangling", "false")
 | 
						|
		}
 | 
						|
 | 
						|
		ir, err := cl.ImagesPrune(cmd.Context(), pruneFilters)
 | 
						|
		if err != nil {
 | 
						|
			log.Fatal(err)
 | 
						|
		}
 | 
						|
 | 
						|
		imgSpaceReclaimed := formatter.ByteCountSI(ir.SpaceReclaimed)
 | 
						|
		log.Infof("images pruned: %d; space reclaimed: %s", len(ir.ImagesDeleted), imgSpaceReclaimed)
 | 
						|
 | 
						|
		if volumesFilter {
 | 
						|
			vr, err := cl.VolumesPrune(cmd.Context(), filterArgs)
 | 
						|
			if err != nil {
 | 
						|
				log.Fatal(err)
 | 
						|
			}
 | 
						|
 | 
						|
			volSpaceReclaimed := formatter.ByteCountSI(vr.SpaceReclaimed)
 | 
						|
			log.Infof("volumes pruned: %d; space reclaimed: %s", len(vr.VolumesDeleted), volSpaceReclaimed)
 | 
						|
		}
 | 
						|
 | 
						|
		return
 | 
						|
	},
 | 
						|
}
 | 
						|
 | 
						|
var (
 | 
						|
	allFilter     bool
 | 
						|
	volumesFilter bool
 | 
						|
)
 | 
						|
 | 
						|
func init() {
 | 
						|
	ServerPruneCommand.Flags().BoolVarP(
 | 
						|
		&allFilter,
 | 
						|
		"all",
 | 
						|
		"a",
 | 
						|
		false,
 | 
						|
		"remove all unused images",
 | 
						|
	)
 | 
						|
 | 
						|
	ServerPruneCommand.Flags().BoolVarP(
 | 
						|
		&volumesFilter,
 | 
						|
		"volumes",
 | 
						|
		"v",
 | 
						|
		false,
 | 
						|
		"remove volumes",
 | 
						|
	)
 | 
						|
}
 |