feat: add unmount volume command

cmd/mount.go

@@ -75,23 +75,23 @@ var mount = &cobra.Command{
 			return
 		}
 
-		Box := box.New(box.Config{
+		infoBox := box.New(box.Config{
 			Px:    2,
 			Py:    1,
 			Type:  "Classic",
 			Color: "Cyan"})
 
 		fmt.Println()
-		Box.Print(
+		infoBox.Print(
 			"NTFS volume mounted succesfully!",
 			fmt.Sprintf(`Partition mounted on /Volumes/%s. Please do copy-paste there.
 Finder will open automatically. Or You can open the directory using the command below.
 
-	$ ntfs-wr open /Volumes/%s
+	$ ntfs-wr open %s
 
 Use this command to unmount the partition:
 
-	$ sudo ntfs-wr unmount /Volumes/%s
+	$ sudo ntfs-wr unmount %s
 `, dirName, dirName, dirName),
 		)
 	},

cmd/root.go

@@ -10,6 +10,7 @@ func init() {
 	RootCmd.AddCommand(listExternalDisk)
 	RootCmd.AddCommand(mount)
 	RootCmd.AddCommand(openVolume)
+	RootCmd.AddCommand(unmount)
 }
 
 var RootCmd = &cobra.Command{

cmd/unmount.go

@@ -1 +1,59 @@
 package cmd
+
+import (
+	"fmt"
+
+	"github.com/cikupin/ntfs-wr/pkg"
+	zlog "github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+)
+
+var unmount = &cobra.Command{
+	Use:     "unmount",
+	Short:   "Unmount external disk partition",
+	Long:    "Unmount external disk partition",
+	Example: "$ ntfs-wr unmount <volume_dir_name>",
+	PreRun: func(*cobra.Command, []string) {
+		if !pkg.IsMacOS() {
+			zlog.Fatal().Msg("this binary runs only for MacOS")
+		}
+
+		if !pkg.IsRoot() {
+			zlog.Fatal().Msg("this binary runs only on root user")
+		}
+	},
+	Run: func(c *cobra.Command, args []string) {
+		if len(args) == 0 {
+			zlog.Error().Msg("please supply volume dirname as an argument. exiting now...")
+			return
+		}
+
+		dirName := args[0]
+		if !pkg.IsVolumeExist(dirName) {
+			zlog.Error().
+				Str("volume_dir", fmt.Sprintf("/Volumes/%s", dirName)).
+				Msg("volume directory is not exist")
+			return
+		}
+
+		err := pkg.UnmountDisk(dirName)
+		if err != nil {
+			zlog.Fatal().Err(err).
+				Str("full_path", fmt.Sprintf("/Volumes/%s", dirName)).
+				Msg("cannot unmount volume")
+			return
+		}
+
+		err = pkg.RemoveVolumeDir(dirName)
+		if err != nil {
+			zlog.Warn().
+				Str("full_path", fmt.Sprintf("/Volumes/%s", dirName)).
+				Msg("volume unmounted succesfully, but failed to remove temporary volume dir. you can manually delete it.")
+			return
+		}
+
+		zlog.Info().
+			Str("full_path", fmt.Sprintf("/Volumes/%s", dirName)).
+			Msg("volume unmounted succesfully")
+	},
+}

pkg/disk.go

@@ -6,6 +6,7 @@ import (
 	"os/exec"
 	"regexp"
 	"strings"
+	"time"
 
 	zlog "github.com/rs/zerolog/log"
 )
@@ -42,6 +43,7 @@ func getMappedDisk(info string) map[string]string {
 	return mappedDisk
 }
 
+// MountDisk mounts partition to the given volume dir name
 func MountDisk(disk string, volumeDirName string) error {
 	zlog.Info().Msg("mounting volume is in progres...")
 
@@ -53,3 +55,19 @@ func MountDisk(disk string, volumeDirName string) error {
 	}
 	return nil
 }
+
+func UnmountDisk(volumeDirName string) error {
+	zlog.Info().Msg("unmounting volume is in progres...")
+
+	commands := fmt.Sprintf("umount /Volumes/%s", volumeDirName)
+
+	_, err := exec.Command("bash", "-c", commands).Output()
+	if err != nil {
+		return err
+	}
+
+	// giving some time to make sure that the volume unmounted safely
+	time.Sleep(2 * time.Second)
+
+	return nil
+}

pkg/file.go

@@ -38,3 +38,8 @@ func OpenVolumeDir(volumeDirName string) error {
 	}
 	return nil
 }
+
+// RemoveVolumeDir removes the given directory on /Volumes path
+func RemoveVolumeDir(volumeDirName string) error {
+	return os.RemoveAll(fmt.Sprintf("/Volumes/%s", volumeDirName))
+}