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",
 | |
| 	)
 | |
| }
 |