From 214160768aef903254dd1db060ad257085993dc9 Mon Sep 17 00:00:00 2001
From: Lumi Pakkanen <lumi.pakkanen@gmail.com>
Date: Sun, 30 Apr 2023 20:27:01 +0300
Subject: [PATCH] Escape invalid lines in the URL

ref #423
---
 src/__tests__/url-encode.spec.ts | 12 ++++++
 src/url-encode.ts                | 72 +++++++++++++++++++++++---------
 2 files changed, 64 insertions(+), 20 deletions(-)

diff --git a/src/__tests__/url-encode.spec.ts b/src/__tests__/url-encode.spec.ts
index 2a88e887..e242baa1 100644
--- a/src/__tests__/url-encode.spec.ts
+++ b/src/__tests__/url-encode.spec.ts
@@ -112,6 +112,18 @@ describe("URL encoder", () => {
       v: "3",
     });
   });
+
+  it("can escape invalid lines", () => {
+    const lines = ["3/2", "foo", "BAR", "2/1"];
+    expect(encodeLines(lines)).toBe("3F2_EfEoEo_EBAER_2F1");
+  });
+
+  it("can decode invalid lines", () => {
+    const expected = ["3/2", "foo", "BAR", "2/1"];
+    expect(
+      arraysEqual(decodeLines("3F2_EfEoEo_EBAER_2F1"), expected)
+    ).toBeTruthy();
+  });
 });
 
 describe("URL decoder", () => {
diff --git a/src/url-encode.ts b/src/url-encode.ts
index a1985f50..a747175b 100644
--- a/src/url-encode.ts
+++ b/src/url-encode.ts
@@ -11,15 +11,7 @@ const LEFT_ANGLE_BRACKET = "L";
 const RIGHT_ANGLE_BRACKET = "R";
 const LEFT_SQUARE_BRACKET = "Q";
 const PLUS = "P";
-
-const FRACTION_RE = new RegExp(FRACTION, "g");
-const COMMA_RE = new RegExp(COMMA, "g");
-const BACKSLASH_RE = new RegExp(BACKSLASH, "g");
-const SPACE_RE = new RegExp(SPACE, "g");
-const LEFT_ANGLE_BRACKET_RE = new RegExp(LEFT_ANGLE_BRACKET, "g");
-const RIGHT_ANGLE_BRACKET_RE = new RegExp(RIGHT_ANGLE_BRACKET, "g");
-const LEFT_SQUARE_BRACKET_RE = new RegExp(LEFT_SQUARE_BRACKET, "g");
-const PLUS_RE = new RegExp(PLUS, "g");
+const ESCAPE = "E";
 
 // Color shorhands
 const BLACK = "-";
@@ -34,6 +26,13 @@ function isBase36Digit(character: string) {
   return isDigit(character) || /[a-z]/.test(character);
 }
 
+function escapeCharacter(character: string) {
+  if (isBase36Digit(character)) {
+    return ESCAPE + character;
+  }
+  return character;
+}
+
 function encodeDigits(digits: string, keepZero: boolean) {
   if (!digits.length) {
     return "";
@@ -80,6 +79,17 @@ function decodeDigits(digits: string) {
 }
 
 function encodeLine(scaleLine: string) {
+  scaleLine = scaleLine
+    .replace(ESCAPE, ESCAPE + ESCAPE)
+    .replace(FRACTION, ESCAPE + FRACTION)
+    .replace(COMMA, ESCAPE + COMMA)
+    .replace(BACKSLASH, ESCAPE + BACKSLASH)
+    .replace(SPACE, ESCAPE + SPACE)
+    .replace(LEFT_ANGLE_BRACKET, ESCAPE + LEFT_ANGLE_BRACKET)
+    .replace(RIGHT_ANGLE_BRACKET, ESCAPE + RIGHT_ANGLE_BRACKET)
+    .replace(LEFT_SQUARE_BRACKET, ESCAPE + LEFT_SQUARE_BRACKET)
+    .replace(PLUS, ESCAPE + PLUS);
+
   scaleLine = scaleLine
     .replace(/\//g, FRACTION)
     .replace(/,/g, COMMA)
@@ -100,7 +110,7 @@ function encodeLine(scaleLine: string) {
       result += encodeDigits(currentNumber, !".,".includes(lastNondigit));
       currentNumber = "";
       lastNondigit = character;
-      result += character;
+      result += escapeCharacter(character);
     }
   });
   return result + encodeDigits(currentNumber, !".,".includes(lastNondigit));
@@ -109,26 +119,48 @@ function encodeLine(scaleLine: string) {
 function decodeLine(encoded: string) {
   let result = "";
   let currentNumber = "";
+  let passNext = false;
   [...encoded].forEach((character) => {
+    if (passNext) {
+      result += character;
+      passNext = false;
+      return;
+    }
+    if (character === ESCAPE) {
+      passNext = true;
+      return;
+    }
+
     if (isBase36Digit(character)) {
       currentNumber += character;
     } else {
       result += decodeDigits(currentNumber);
       currentNumber = "";
-      result += character;
+
+      if (character === FRACTION) {
+        result += "/";
+      } else if (character === COMMA) {
+        result += ",";
+      } else if (character === BACKSLASH) {
+        result += "\\";
+      } else if (character === SPACE) {
+        result += " ";
+      } else if (character === LEFT_ANGLE_BRACKET) {
+        result += "<";
+      } else if (character === RIGHT_ANGLE_BRACKET) {
+        result += ">";
+      } else if (character === LEFT_SQUARE_BRACKET) {
+        result += "[";
+      } else if (character === PLUS) {
+        result += "+";
+      } else {
+        result += character;
+      }
     }
   });
   result += decodeDigits(currentNumber);
 
-  return result
-    .replace(FRACTION_RE, "/")
-    .replace(COMMA_RE, ",")
-    .replace(BACKSLASH_RE, "\\")
-    .replace(SPACE_RE, " ")
-    .replace(LEFT_ANGLE_BRACKET_RE, "<")
-    .replace(RIGHT_ANGLE_BRACKET_RE, ">")
-    .replace(LEFT_SQUARE_BRACKET_RE, "[")
-    .replace(PLUS_RE, "+");
+  return result;
 }
 
 export function encodeLines(scaleLines: string[]) {