add clean subcommand

kobtea · Dec 31, 2022 · 43f5336 · 43f5336
1 parent a751289
commit 43f5336
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/cmd/clean.go b/cmd/clean.go
@@ -0,0 +1,26 @@
+package cmd
+
+import (
+	"os"
+
+	"github.com/spf13/cobra"
+)
+
+// cleanCmd represents the clean command
+var cleanCmd = &cobra.Command{
+	Use:   "clean",
+	Short: "Remove contents at working directory",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if err := os.RemoveAll(cfg.WorkingDir); err != nil {
+			return err
+		}
+		if err := os.Mkdir(cfg.WorkingDir, 0755); err != nil {
+			return err
+		}
+		return nil
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(cleanCmd)
+}