Fix issue 5298: allow keywords to be used as table names

sqldelight-compiler/src/main/kotlin/app/cash/sqldelight/core/lang/util/TreeUtil.kt

@@ -183,7 +183,7 @@ private fun PsiElement.rangesToReplace(): List<Pair<IntRange, String>> {
             first = parent!!.tableName.range,
             second = parent!!.columns.joinToString(
               separator = ", ",
-              prefix = "${parent!!.tableName.text} (",
+              prefix = "${parent!!.tableName.node.text} (",
               postfix = ")",
             ) { it.name },
           ),

sqldelight-compiler/src/test/kotlin/app/cash/sqldelight/core/QueriesTypeTest.kt

@@ -10,9 +10,11 @@ import org.junit.Test
 import org.junit.rules.TemporaryFolder
 
 class QueriesTypeTest {
-  @get:Rule val temporaryFolder = TemporaryFolder()
+  @get:Rule
+  val temporaryFolder = TemporaryFolder()
 
-  @Test fun `queries file is generated properly via compilation`() {
+  @Test
+  fun `queries file is generated properly via compilation`() {
     val result = FixtureCompiler.compileSql(
       """
       |CREATE TABLE data (
@@ -217,7 +219,8 @@ class QueriesTypeTest {
     )
   }
 
-  @Test fun `queries file is generated properly with adapter`() {
+  @Test
+  fun `queries file is generated properly with adapter`() {
     val result = FixtureCompiler.compileSql(
       """
       |import foo.S;
@@ -326,7 +329,8 @@ class QueriesTypeTest {
     )
   }
 
-  @Test fun `unused adapters are not passed to the database constructor`() {
+  @Test
+  fun `unused adapters are not passed to the database constructor`() {
     val result = FixtureCompiler.compileSql(
       """
       |import kotlin.Int;
@@ -393,7 +397,8 @@ class QueriesTypeTest {
     )
   }
 
-  @Test fun `queries file is generated properly via compilation1a`() {
+  @Test
+  fun `queries file is generated properly via compilation1a`() {
     val result = FixtureCompiler.compileSql(
       """
       |CREATE VIRTUAL TABLE data USING fts5(
@@ -552,7 +557,8 @@ class QueriesTypeTest {
     )
   }
 
-  @Test fun `queries file is generated properly via compilation with offsets`() {
+  @Test
+  fun `queries file is generated properly via compilation with offsets`() {
     val result = FixtureCompiler.compileSql(
       """
       |CREATE VIRTUAL TABLE search USING fts3(
@@ -708,7 +714,8 @@ class QueriesTypeTest {
     )
   }
 
-  @Test fun `adapter through view resolves correctly`() {
+  @Test
+  fun `adapter through view resolves correctly`() {
     FixtureCompiler.writeSql(
       """
       |import com.chicken.SoupBase.Broth;
@@ -852,7 +859,8 @@ class QueriesTypeTest {
     )
   }
 
-  @Test fun `grouped statement with return and no arguments gets a query type`() {
+  @Test
+  fun `grouped statement with return and no arguments gets a query type`() {
     val result = FixtureCompiler.compileSql(
       """
       |CREATE TABLE data (
@@ -915,4 +923,43 @@ class QueriesTypeTest {
       """.trimMargin(),
     )
   }
+
+  @Test
+  fun `SQL keywords can be used as table names when escaped`() {
+    val result = FixtureCompiler.compileSql(
+      """
+      |CREATE TABLE "order" (
+      |  data_id INTEGER NOT NULL
+      |);
+      |selectForId:
+      |INSERT INTO "order" VALUES ?;
+      """.trimMargin(),
+      temporaryFolder,
+    )
+
+    val dataQueries = File(result.outputDirectory, "com/example/TestQueries.kt")
+    assertThat(result.compilerOutput).containsKey(dataQueries)
+    assertThat(result.compilerOutput[dataQueries].toString()).isEqualTo(
+      """
+        package com.example
+
+        import app.cash.sqldelight.TransacterImpl
+        import app.cash.sqldelight.db.SqlDriver
+
+        public class TestQueries(
+          driver: SqlDriver,
+        ) : TransacterImpl(driver) {
+          public fun selectForId(order: Order) {
+            driver.execute(-304_025_397, ""${'"'}INSERT INTO "order" (data_id) VALUES (?)""${'"'}, 1) {
+                  bindLong(0, order.data_id)
+                }
+            notifyQueries(-304_025_397) { emit ->
+              emit("order")
+            }
+          }
+        }
+
+      """.trimIndent(),
+    )
+  }
 }