diff --git a/main.go b/main.go index fd401ff..fafe7eb 100644 --- a/main.go +++ b/main.go @@ -37,6 +37,10 @@ func GetPath() string { } return homedir + path[1:] } + if path == "" { + fmt.Println("[", color.RedString("!"), "] No image given, retrying.") + path = GetPath() + } return path } @@ -53,6 +57,10 @@ func GetDest() string { } return homedir + dest[1:] } + if dest == "" { + fmt.Println("[", color.RedString("!"), "] No destination given, retrying.") + dest = GetDest() + } return dest }