From 0263188431e9493373d6a1c47acff4daa650623a Mon Sep 17 00:00:00 2001
From: Asher <ash@coder.com>
Date: Wed, 19 Feb 2020 10:54:23 -0600
Subject: [PATCH] Handle --long=value format in the cli parser

---
 src/node/cli.ts  | 19 +++++++++++++------
 test/cli.test.ts | 12 ++++++------
 2 files changed, 19 insertions(+), 12 deletions(-)

diff --git a/src/node/cli.ts b/src/node/cli.ts
index 072495c3..c43dfa90 100644
--- a/src/node/cli.ts
+++ b/src/node/cli.ts
@@ -128,8 +128,11 @@ export const parse = (argv: string[]): Args => {
     // Options start with a dash and require a value if non-boolean.
     if (!ended && arg.startsWith("-")) {
       let key: keyof Args | undefined
+      let value: string | undefined
       if (arg.startsWith("--")) {
-        key = arg.replace(/^--/, "") as keyof Args
+        const split = arg.replace(/^--/, "").split("=", 2)
+        key = split[0] as keyof Args
+        value = split[1]
       } else {
         const short = arg.replace(/^-/, "")
         const pair = Object.entries(options).find(([, v]) => v.short === short)
@@ -148,13 +151,17 @@ export const parse = (argv: string[]): Args => {
         continue
       }
 
-      // A value is only valid if it doesn't look like an option.
-      let value = argv[i + 1] && !argv[i + 1].startsWith("-") ? argv[++i] : undefined
+      // Might already have a value if it was the --long=value format.
+      if (typeof value === "undefined") {
+        // A value is only valid if it doesn't look like an option.
+        value = argv[i + 1] && !argv[i + 1].startsWith("-") ? argv[++i] : undefined
+      }
+
       if (!value && option.type === OptionalString) {
         ;(args[key] as OptionalString) = new OptionalString(value)
         continue
       } else if (!value) {
-        throw new Error(`${arg} requires a value`)
+        throw new Error(`--${key} requires a value`)
       }
 
       if (option.path) {
@@ -174,7 +181,7 @@ export const parse = (argv: string[]): Args => {
         case "number":
           ;(args[key] as number) = parseInt(value, 10)
           if (isNaN(args[key] as number)) {
-            throw new Error(`${arg} must be a number`)
+            throw new Error(`--${key} must be a number`)
           }
           break
         case OptionalString:
@@ -182,7 +189,7 @@ export const parse = (argv: string[]): Args => {
           break
         default: {
           if (!Object.values(option.type).find((v) => v === value)) {
-            throw new Error(`${arg} valid values: [${Object.values(option.type).join(", ")}]`)
+            throw new Error(`--${key} valid values: [${Object.values(option.type).join(", ")}]`)
           }
           ;(args[key] as string) = value
           break
diff --git a/test/cli.test.ts b/test/cli.test.ts
index 04fcd771..86a14c88 100644
--- a/test/cli.test.ts
+++ b/test/cli.test.ts
@@ -36,19 +36,16 @@ describe("cli", () => {
         "error",
         "--help",
         "--open",
-        "--socket",
-        "mumble",
+        "--socket=mumble",
         "3",
         "--user-data-dir",
         "bar",
-        "--cert",
-        "baz",
+        "--cert=baz",
         "--cert-key",
         "qux",
         "--version",
         "--json",
-        "--port",
-        "8081",
+        "--port=8081",
         "--host",
         "0.0.0.0",
         "4",
@@ -117,6 +114,9 @@ describe("cli", () => {
 
   it("should error if value isn't provided", () => {
     assert.throws(() => parse(["--auth"]), /--auth requires a value/)
+    assert.throws(() => parse(["--auth=", "--log=debug"]), /--auth requires a value/)
+    assert.throws(() => parse(["--auth", "--log"]), /--auth requires a value/)
+    assert.throws(() => parse(["--auth", "--invalid"]), /--auth requires a value/)
   })
 
   it("should error if number option is invalid", () => {