add headline to device overview

main.go

@@ -6,6 +6,8 @@ import (
 	"time"
 	"log"
 	"strings"
+	"runtime"
+	"strconv"
 	"github.com/schollz/progressbar/v3"
 	"github.com/fatih/color"
 	"github.com/briandowns/spinner"
@@ -45,6 +47,7 @@ func GetPath() string {
 }
 
 func GetDest() string {
+	PrintAvail()
 	dest, err := ac.Read("[ " + color.YellowString("i") + " ]  Please input destination: ")
 	if err != nil {
 		log.Fatal(err)
@@ -105,6 +108,44 @@ func Sync(image *os.File, target *os.File) error {
 	return nil
 }
 
+func PrintAvail() {
+	if runtime.GOOS == "linux" {
+		block, _ := os.ReadDir("/sys/block")
+		if len(block) == 0 {
+			return
+		}
+		var targets []string
+		for _, device := range block {
+			if strings.HasPrefix(device.Name(), "sd") {
+				targets = append(targets, device.Name())
+			}
+			if strings.HasPrefix(device.Name(), "nvme") {
+				targets = append(targets, device.Name())
+			}
+			if strings.HasPrefix(device.Name(), "vd") {
+				targets = append(targets, device.Name())
+			}
+		}
+		fmt.Println("Available devices:")
+		for _, target := range targets {
+			sizefile, _ := os.Open("/sys/block/" + target + "/size")
+			sizeread, _ := io.ReadAll(sizefile)
+			_ = sizefile.Close()
+			sizestring := strings.ReplaceAll(string(sizeread), "\n", "")
+			size, _ := strconv.Atoi(sizestring)
+			size = size * 512
+			size = size / 1024 / 1024 / 1024
+
+			fmt.Print("     * ", "/dev/" + target)
+			if size > 0 {
+				fmt.Print(" [", size, "GB]\n")
+			} else {
+				fmt.Println("")
+			}
+		}
+	}
+}
+
 func main() {
 	s := spinner.New(spinner.CharSets[14], 100*time.Millisecond)
 
@@ -142,18 +183,11 @@ func main() {
 		s.Stop()
 		fmt.Println("\r[", color.RedString("✘"), "]  Getting file details                     ")
 		log.Fatal(err)
-	} else {
-		s.Stop()
-		fmt.Println("\r[", color.GreenString("✓"), "]  Getting file details                     ")
 	}
-
-	s.Prefix = "[ "
-	s.Suffix = " ]  Opening files"
-	s.Start()
 	image, err := os.Open(input)
 	if err != nil {
 		s.Stop()
-		fmt.Println("\r[", color.RedString("✘"), "]  Opening files                   ")
+		fmt.Println("\r[", color.RedString("✘"), "]  Getting file details                     ")
 		log.Fatal(err)
 	}
 	var inputsize int64
@@ -164,11 +198,12 @@ func main() {
 	} else {
 		inputsize, err = image.Seek(0, io.SeekEnd)
 		inputisblock = true
+		_, _ = image.Seek(0, 0)
 	}
 	target, err := os.OpenFile(device, os.O_RDWR, 0660)
 	if err != nil {
 		s.Stop()
-		fmt.Println("\r[", color.RedString("✘"), "]  Opening files                   ")
+		fmt.Println("\r[", color.RedString("✘"), "]  Getting file details                     ")
 		log.Fatal(err)
 	}
 	var targetsize int64
@@ -179,14 +214,15 @@ func main() {
 	} else {
 		targetsize, err = target.Seek(0, io.SeekEnd)
 		targetisblock = true
+		_, _ = target.Seek(0, 0)
 	}
 	if err != nil {
 		s.Stop()
-		fmt.Println("\r[", color.RedString("✘"), "]  Opening files                   ")
+		fmt.Println("\r[", color.RedString("✘"), "]  Getting file details                     ")
 		log.Fatal(err)
 	} else {
 		s.Stop()
-		fmt.Println("\r[", color.GreenString("✓"), "]  Opening files                 ")
+		fmt.Println("\r[", color.GreenString("✓"), "]  Getting file details                     ")
 	}
 	inputmb := fmt.Sprint("[", inputsize / 1024 / 1024, "MB]")
 	devicemb := fmt.Sprint("[", targetsize / 1024 / 1024, "MB]")
@@ -204,7 +240,8 @@ func main() {
 	}
 	fmt.Println("[", color.BlueString("i"), "]  Input device/file: " + input, inputmb, inputblock)
 	fmt.Println("[", color.BlueString("i"), "]  Output device/file: " + device, devicemb, targetblock)
-	if statinput.Size() > statdevice.Size() {
+	if force == false {
+		if inputsize > targetsize {
 			fmt.Println("[", color.RedString("w"), "]", color.RedString(" Warning:"), "Input file seems to be bigger than the destination!")
 		}
 		fmt.Print(color.HiWhiteString("Do you want to continue? [y/N]: "))
@@ -214,7 +251,7 @@ func main() {
 		if ! (yesno == "y" || yesno == "Y") {
 			log.Fatal("aborted")
 		}
-
+	}
 	written, err := WriteImage(image, target, inputsize)
 	if err != nil {
 		fmt.Println("\r[", color.RedString("✘"), "]  Writing image,", written, "bytes written                                                    ")