Commit 58097b6

@@ -1,100 +1,15 @@
 package get
 
 import (
-	"fmt"
-	"os"
-	"strings"
-
-	human "github.com/dustin/go-humanize"
-	"github.com/jedib0t/go-pretty/v6/table"
-	"github.com/kahnwong/swissknife/configs/color"
-	"github.com/rs/zerolog/log"
-	"github.com/shirou/gopsutil/v4/disk"
+	"github.com/kahnwong/swissknife/internal/get"
 	"github.com/spf13/cobra"
 )
 
-func listVolumes() {
-	// ref: <https://stackoverflow.com/a/64141403>
-	// layout inspired by `duf`
-
-	// init table
-	t := table.NewWriter()
-	t.SetOutputMirror(os.Stdout)
-	t.AppendHeader(table.Row{"Mounted on", "Size", "Used", "Avail", "Use%", "Type", "Filesystem"})
-
-	// get volumes info
-	partitions, err := disk.Partitions(false)
-	if err != nil {
-		log.Fatal().Msg("Error getting partitions info")
-	}
-
-	for _, partition := range partitions {
-		// linux
-		isSquashFs := partition.Fstype == "squashfs"
-		isSnap := strings.Contains(partition.Mountpoint, "snap")
-		isKubernetes := strings.Contains(partition.Mountpoint, "kubelet")
-		// osx
-		isOsx := strings.Contains(partition.Mountpoint, "System/Volumes")
-		isDevFs := partition.Fstype == "devfs"
-		isOsxNix := strings.Contains(partition.Mountpoint, "/nix")
-		if !isSquashFs && !isSnap && !isKubernetes && !isOsx && !isDevFs && !isOsxNix {
-			device := partition.Mountpoint
-			stats, err := disk.Usage(device)
-			if err != nil {
-				log.Fatal().Msg("Error getting disk info")
-			}
-
-			if stats.Total == 0 {
-				continue
-			}
-
-			// disk available
-			diskAvail := stats.Free
-			diskAvailStr := ""
-			if diskAvail < 50*1024*1024*1024 { // if free space less than 50 GB
-				diskAvailStr = color.Red(human.Bytes(diskAvail))
-			} else if diskAvail < 100*1024*1024*1024 { // if free space less than 100 GB
-				diskAvailStr = color.Yellow(human.Bytes(diskAvail))
-			} else {
-				diskAvailStr = color.Green(human.Bytes(diskAvail))
-			}
-
-			// disk use percent
-			percent := fmt.Sprintf("%2.f%%", stats.UsedPercent)
-			percentRaw := stats.UsedPercent
-			percentStr := ""
-			if percentRaw > 80 {
-				percentStr = color.Red(percent)
-			} else if percentRaw > 70 {
-				percentStr = color.Yellow(percent)
-			} else {
-				percentStr = color.Green(percent)
-			}
-
-			// append info to table
-			t.AppendRows([]table.Row{
-				{
-					color.Blue(partition.Mountpoint),
-					human.Bytes(stats.Total),
-					human.Bytes(stats.Used),
-					diskAvailStr,
-					percentStr,
-					color.Magenta(partition.Fstype),
-					color.Magenta(partition.Device),
-				},
-			})
-		}
-	}
-
-	// render
-	t.Render()
-}
-
 var listVolumesCmd = &cobra.Command{
 	Use:   "volumes",
 	Short: "List volumes",
 	Run: func(cmd *cobra.Command, args []string) {
-		listVolumes()
+		get.Volumes()
 	},
 }
 

@@ -0,0 +1,94 @@
+package get
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	human "github.com/dustin/go-humanize"
+	"github.com/jedib0t/go-pretty/v6/table"
+	"github.com/kahnwong/swissknife/configs/color"
+	"github.com/rs/zerolog/log"
+	"github.com/shirou/gopsutil/v4/disk"
+)
+
+func listVolumes() {
+	// ref: <https://stackoverflow.com/a/64141403>
+	// layout inspired by `duf`
+
+	// init table
+	t := table.NewWriter()
+	t.SetOutputMirror(os.Stdout)
+	t.AppendHeader(table.Row{"Mounted on", "Size", "Used", "Avail", "Use%", "Type", "Filesystem"})
+
+	// get volumes info
+	partitions, err := disk.Partitions(false)
+	if err != nil {
+		log.Fatal().Msg("Error getting partitions info")
+	}
+
+	for _, partition := range partitions {
+		// linux
+		isSquashFs := partition.Fstype == "squashfs"
+		isSnap := strings.Contains(partition.Mountpoint, "snap")
+		isKubernetes := strings.Contains(partition.Mountpoint, "kubelet")
+		// osx
+		isOsx := strings.Contains(partition.Mountpoint, "System/Volumes")
+		isDevFs := partition.Fstype == "devfs"
+		isOsxNix := strings.Contains(partition.Mountpoint, "/nix")
+		if !isSquashFs && !isSnap && !isKubernetes && !isOsx && !isDevFs && !isOsxNix {
+			device := partition.Mountpoint
+			stats, err := disk.Usage(device)
+			if err != nil {
+				log.Fatal().Msg("Error getting disk info")
+			}
+
+			if stats.Total == 0 {
+				continue
+			}
+
+			// disk available
+			diskAvail := stats.Free
+			diskAvailStr := ""
+			if diskAvail < 50*1024*1024*1024 { // if free space less than 50 GB
+				diskAvailStr = color.Red(human.Bytes(diskAvail))
+			} else if diskAvail < 100*1024*1024*1024 { // if free space less than 100 GB
+				diskAvailStr = color.Yellow(human.Bytes(diskAvail))
+			} else {
+				diskAvailStr = color.Green(human.Bytes(diskAvail))
+			}
+
+			// disk use percent
+			percent := fmt.Sprintf("%2.f%%", stats.UsedPercent)
+			percentRaw := stats.UsedPercent
+			percentStr := ""
+			if percentRaw > 80 {
+				percentStr = color.Red(percent)
+			} else if percentRaw > 70 {
+				percentStr = color.Yellow(percent)
+			} else {
+				percentStr = color.Green(percent)
+			}
+
+			// append info to table
+			t.AppendRows([]table.Row{
+				{
+					color.Blue(partition.Mountpoint),
+					human.Bytes(stats.Total),
+					human.Bytes(stats.Used),
+					diskAvailStr,
+					percentStr,
+					color.Magenta(partition.Fstype),
+					color.Magenta(partition.Device),
+				},
+			})
+		}
+	}
+
+	// render
+	t.Render()
+}
+
+func Volumes() {
+	listVolumes()
+}