feat: cli application with go

Author: sudohainguyen

.gitignore

@@ -1,3 +1,4 @@
 *.log
 .vscode
 .metals
+bin

go/cli/cmd/dirs.go

@@ -0,0 +1,36 @@
+package cmd
+
+import (
+	"cli/common"
+	"cli/dirs"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var Depth int
+var Mindirsize int
+
+// dirsCmd represents the dirs command
+var dirsCmd = &cobra.Command{
+	Use:   "dirs",
+	Short: "Show the largest directories in the given path.",
+	Long:  `Quickly scan a directory and find large directories. Use the flags below to target the output.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		if Debug {
+			common.LogFlags()
+		}
+		dirsFound, _ := dirs.ReadDirDepth(Path)
+		dirs.PrintResults(dirsFound)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(dirsCmd)
+
+	dirsCmd.PersistentFlags().IntVarP(&Depth, "depth", "", 2, "Depth of directory tree to display")
+	viper.BindPFlag("depth", dirsCmd.PersistentFlags().Lookup("depth"))
+
+	dirsCmd.PersistentFlags().IntVarP(&Mindirsize, "mindirsize", "", 100, "Only display directories larger than this threshold in MB.")
+	viper.BindPFlag("mindirsize", dirsCmd.PersistentFlags().Lookup("mindirsize"))
+}

go/cli/cmd/files.go

@@ -0,0 +1,50 @@
+package cmd
+
+import (
+	"fmt"
+
+	"cli/common"
+	"cli/files"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var Path string
+var Filecount int
+var Minfilesize int64
+
+// filesCmd represents the files command
+var filesCmd = &cobra.Command{
+	Use:   "files",
+	Short: "Show the largest files in the given path.",
+	Long:  `Quickly scan a directory and find large files. . Use the flags below to target the output.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		if Debug {
+			common.LogFlags()
+		}
+
+		filesFound, err := files.ReadDirRecursively(Path)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+
+		if Filecount > len(filesFound) {
+			Filecount = len(filesFound)
+		}
+
+		filesFound = filesFound[0:Filecount]
+		files.PrintResults(filesFound)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(filesCmd)
+
+	filesCmd.PersistentFlags().IntVarP(&Filecount, "filecount", "f", 10, "Limit the number of files returned")
+	viper.BindPFlag("filecount", filesCmd.PersistentFlags().Lookup("filecount"))
+
+	filesCmd.PersistentFlags().Int64VarP(&Minfilesize, "minfilesize", "", 50, "Minimum size for files in search in MB.")
+	viper.BindPFlag("minfilesize", filesCmd.PersistentFlags().Lookup("minfilesize"))
+}

go/cli/cmd/root.go

@@ -0,0 +1,43 @@
+package cmd
+
+import (
+	"os"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+// rootCmd represents the base command when called without any subcommands
+var Verbose bool
+var Debug bool
+var Highlight int
+
+var rootCmd = &cobra.Command{
+	Use:   "getsize",
+	Short: "List the size of a local directory.",
+	Long:  `This command will display the size of a directory with several different options.`,
+}
+
+// Execute adds all child commands to the root command and sets flags appropriately.
+// This is called by main.main(). It only needs to happen once to the rootCmd.
+func Execute() {
+	err := rootCmd.Execute()
+	if err != nil {
+		os.Exit(1)
+	}
+}
+
+func init() {
+	rootCmd.PersistentFlags().BoolVarP(&Verbose, "verbose", "v", false, "Display more verbose output in console output. (default: false)")
+	viper.BindPFlag("verbose", rootCmd.PersistentFlags().Lookup("verbose"))
+
+	rootCmd.PersistentFlags().BoolVarP(&Debug, "debug", "d", false, "Display debugging output in the console. (default: false)")
+	viper.BindPFlag("debug", rootCmd.PersistentFlags().Lookup("debug"))
+
+	rootCmd.PersistentFlags().IntVarP(&Highlight, "highlight", "", 500, "Highlight files/directories over this threshold, in MB")
+	viper.BindPFlag("highlight", rootCmd.PersistentFlags().Lookup("highlight"))
+
+	rootCmd.PersistentFlags().StringVarP(&Path, "path", "p", "", "Define the path to scan.")
+	rootCmd.MarkPersistentFlagRequired("path")
+	viper.BindPFlag("path", rootCmd.PersistentFlags().Lookup("path"))
+}