Merge pull request #46 from vertical-blank/oracle-quotation-operator

Oracle quotation operator

Author: vertical-blank

src/main/java/com/github/vertical_blank/sqlformatter/languages/PlSqlFormatter.java

@@ -434,6 +434,7 @@ public DialectConfig dialectConfig() {
             Arrays.asList(
                 StringLiteral.DOUBLE_QUOTE,
                 StringLiteral.N_SINGLE_QUOTE,
+                StringLiteral.Q_SINGLE_QUOTE,
                 StringLiteral.SINGLE_QUOTE,
                 StringLiteral.BACK_QUOTE))
         .openParens(Arrays.asList("(", "CASE"))

src/main/java/com/github/vertical_blank/sqlformatter/languages/StringLiteral.java

@@ -12,6 +12,7 @@ public class StringLiteral {
   public static final String U_SINGLE_QUOTE = "U&''";
   public static final String E_SINGLE_QUOTE = "E''";
   public static final String N_SINGLE_QUOTE = "N''";
+  public static final String Q_SINGLE_QUOTE = "Q''";
   public static final String SINGLE_QUOTE = "''";
   public static final String BRACE = "{}";
   public static final String DOLLAR = "$$";
@@ -47,6 +48,10 @@ private enum Preset {
         "(('[^'\\\\]*(?:\\\\.[^'\\\\]*)*('|$))+)"), // "((^'((?:''|[^'])*)')+)"),
     /** N'' */
     N_SINGLE_QUOTE(StringLiteral.N_SINGLE_QUOTE, "((N'[^'\\\\]*(?:\\\\.[^'\\\\]*)*('|$))+)"),
+    /** q'' */
+    Q_SINGLE_QUOTE(
+        StringLiteral.Q_SINGLE_QUOTE,
+        "(?i)((n?q'\\[.*($|\\]'))+)|((n?q'\\{.*($|\\}'))+)|((n?q'\\<.*($|\\>'))+)|((n?q'\\(.*($|\\)'))+)"),
     // single_quote("((^'((?:''|[^'])*)')+)"),
     E_SINGLE_QUOTE(StringLiteral.E_SINGLE_QUOTE, "((E'[^'\\\\]*(?:\\\\.[^'\\\\]*)*('|$))+)"),
     /** U&amp;'' */

src/test/kotlin/com/github/vertical_blank/sqlformatter/PlSqlFormatterTest.kt

@@ -28,7 +28,10 @@ object PlSqlFormatterTest :
           supportsStrings(
               formatter,
               listOf(
-                  StringLiteral.DOUBLE_QUOTE, StringLiteral.SINGLE_QUOTE, StringLiteral.BACK_QUOTE))
+                  StringLiteral.DOUBLE_QUOTE,
+                  StringLiteral.SINGLE_QUOTE,
+                  StringLiteral.BACK_QUOTE,
+                  StringLiteral.Q_SINGLE_QUOTE))
           supportsBetween(formatter)
           supportsSchema(formatter)
           supportsOperators(formatter, listOf("||", "**", "!=", ":="))

src/test/kotlin/com/github/vertical_blank/sqlformatter/features/Strings.kt

@@ -65,5 +65,18 @@ fun Suite.supportsStrings(formatter: SqlFormatter.Formatter, stringTypes: List<S
         expect(format("N'foo \\' JOIN bar'")).toBe("N'foo \\' JOIN bar'")
       }
     }
+
+    if (stringTypes.contains(StringLiteral.Q_SINGLE_QUOTE)) {
+      it("supports Oracle quotation operator") {
+        expect(format("Q'[I'm boy]'")).toBe("Q'[I'm boy]'")
+        expect(format("Q'{I'm boy}'")).toBe("Q'{I'm boy}'")
+        expect(format("Q'<I'm boy>'")).toBe("Q'<I'm boy>'")
+        expect(format("Q'(I'm boy)'")).toBe("Q'(I'm boy)'")
+        expect(format("NQ'[I'm boy]'")).toBe("NQ'[I'm boy]'")
+        expect(format("NQ'{I'm boy}'")).toBe("NQ'{I'm boy}'")
+        expect(format("NQ'<I'm boy>'")).toBe("NQ'<I'm boy>'")
+        expect(format("NQ'(I'm boy)'")).toBe("NQ'(I'm boy)'")
+      }
+    }
   }
 }