feat: add unescape []

outerbase · Mar 4, 2024 · e48157b · e48157b
1 parent 8baf646
commit e48157b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/src/lib/sql-helper.test.ts b/src/lib/sql-helper.test.ts
@@ -38,6 +38,7 @@ describe("Escape SQL", () => {
   it("unescape identity", () => {
     expect(unescapeIdentity(`"users"`)).toBe("users");
     expect(unescapeIdentity(`"us""ers"`)).toBe(`us"ers`);
+    expect(unescapeIdentity(`[users]`)).toBe(`users`);
   });
 });
 

diff --git a/src/lib/sql-helper.ts b/src/lib/sql-helper.ts
@@ -7,8 +7,8 @@ export function escapeIdentity(str: string) {
 }
 
 export function unescapeIdentity(str: string) {
-  let r = str.replace(/^["`]/g, "");
-  r = r.replace(/["`]$/g, "");
+  let r = str.replace(/^["`[]/g, "");
+  r = r.replace(/["`\]]$/g, "");
   r = r.replace(/""/g, `"`);
   return r;
 }

diff --git a/src/lib/validation.ts b/src/lib/validation.ts
@@ -73,7 +73,11 @@ export function validateConnectionEndpoint(
   try {
     const url = new URL(endpoint);
 
-    if (url.protocol !== "wss:" && url.protocol !== "libsql:") {
+    if (
+      url.protocol !== "wss:" &&
+      url.protocol !== "libsql:" &&
+      url.protocol !== "ws:"
+    ) {
       return [false, "We only support wss:// or libsql:// at the moment."];
     }