Allow for incorrect and missing schemes.

Fixes hashrocket#12

Signed-off-by: Brian Brazil <[email protected]>

main.go

@@ -50,6 +50,22 @@ func root(cmd *cobra.Command, args []string) {
 		os.Exit(1)
 	}
 
+  // Correct and add missing schemes.
+  switch dest.Scheme {
+  case "ws",  "wss":
+  case "http":
+    dest.Scheme = "ws"
+  case "https":
+    dest.Scheme = "wss"
+  default:
+    // Likely no scheme at all, e.g. "localhost:8000".
+    dest, err = url.Parse("ws://" + args[0])
+    if err != nil {
+      fmt.Fprintln(os.Stderr, err)
+      os.Exit(1)
+    }
+  }
+
 	var origin string
 	if options.origin != "" {
 		origin = options.origin