Extend SingleStoreDB support (#531)

sql-formatter-org · Nov 22, 2022 · 7d00bec · 7d00bec
2 parents b88d27a + 1445cde
commit 7d00bec
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 2 deletions.
diff --git a/src/languages/singlestoredb/singlestoredb.formatter.ts b/src/languages/singlestoredb/singlestoredb.formatter.ts
@@ -254,10 +254,27 @@ export const singlestoredb: DialectOptions = {
       { quote: '``', prefixes: ['@'], requirePrefix: true },
     ],
     lineCommentTypes: ['--', '#'],
-    operators: [':=', '&', '|', '^', '~', '<<', '>>', '<=>', '&&', '||'],
+    operators: [
+      ':=',
+      '&',
+      '|',
+      '^',
+      '~',
+      '<<',
+      '>>',
+      '<=>',
+      '&&',
+      '||',
+      '::',
+      '::$',
+      '::%',
+      ':>',
+      '!:>',
+    ],
     postProcess,
   },
   formatOptions: {
+    alwaysDenseOperators: ['::', '::$', '::%'],
     onelineClauses,
   },
 };

diff --git a/test/singlestoredb.test.ts b/test/singlestoredb.test.ts
@@ -1,3 +1,4 @@
+import dedent from 'dedent-js';
 import { format as originalFormat, FormatFn } from '../src/sqlFormatter.js';
 import behavesLikeMariaDbFormatter from './behavesLikeMariaDbFormatter.js';
 
@@ -33,7 +34,7 @@ describe('SingleStoreDbFormatter', () => {
   ]);
   supportsOperators(
     format,
-    [':=', '&', '|', '^', '~', '<<', '>>', '<=>', '&&', '||'],
+    [':=', '&', '|', '^', '~', '<<', '>>', '<=>', '&&', '||', ':>', '!:>'],
     ['AND', 'OR']
   );
   supportsLimiting(format, { limit: true, offset: true });
@@ -45,4 +46,37 @@ describe('SingleStoreDbFormatter', () => {
     modify: true,
     renameTo: true,
   });
+
+  describe(`formats traversal of semi structured data`, () => {
+    it(`formats '::' path-operator without spaces`, () => {
+      expect(format(`SELECT * FROM foo WHERE json_foo::bar = 'foobar'`)).toBe(dedent`
+        SELECT
+          *
+        FROM
+          foo
+        WHERE
+          json_foo::bar = 'foobar'
+      `);
+    });
+    it(`formats '::$' conversion path-operator without spaces`, () => {
+      expect(format(`SELECT * FROM foo WHERE json_foo::$bar = 'foobar'`)).toBe(dedent`
+        SELECT
+          *
+        FROM
+          foo
+        WHERE
+          json_foo::$bar = 'foobar'
+      `);
+    });
+    it(`formats '::%' conversion path-operator without spaces`, () => {
+      expect(format(`SELECT * FROM foo WHERE json_foo::%bar = 'foobar'`)).toBe(dedent`
+        SELECT
+          *
+        FROM
+          foo
+        WHERE
+          json_foo::%bar = 'foobar'
+      `);
+    });
+  });
 });