feat: Add support for dropping temporary tables with temporary() modifier (#1615)

Co-authored-by: Igal Klebanov <igalklebanov@gmail.com>

Author: szalonna

src/operation-node/drop-table-node.ts

@@ -11,6 +11,7 @@ export interface DropTableNode extends OperationNode {
   readonly table: TableNode
   readonly ifExists?: boolean
   readonly cascade?: boolean
+  readonly temporary?: boolean
 }
 
 /**

src/operation-node/operation-node-transformer.ts

@@ -533,6 +533,7 @@ export class OperationNodeTransformer {
       table: this.transformNode(node.table, queryId),
       ifExists: node.ifExists,
       cascade: node.cascade,
+      temporary: node.temporary,
     })
   }
 

src/query-compiler/default-query-compiler.ts

@@ -745,7 +745,13 @@ export class DefaultQueryCompiler
   }
 
   protected override visitDropTable(node: DropTableNode): void {
-    this.append('drop table ')
+    this.append('drop ')
+
+    if (node.temporary) {
+      this.append('temporary ')
+    }
+
+    this.append('table ')
 
     if (node.ifExists) {
       this.append('if exists ')

src/schema/drop-table-builder.ts

@@ -13,6 +13,20 @@ export class DropTableBuilder implements OperationNodeSource, Compilable {
     this.#props = freeze(props)
   }
 
+  /**
+   * Adds the "temporary" modifier.
+   *
+   * This is only supported by some dialects like MySQL.
+   */
+  temporary(): DropTableBuilder {
+    return new DropTableBuilder({
+      ...this.#props,
+      node: DropTableNode.cloneWith(this.#props.node, {
+        temporary: true,
+      }),
+    })
+  }
+
   ifExists(): DropTableBuilder {
     return new DropTableBuilder({
       ...this.#props,

test/node/src/schema.test.ts

@@ -1512,6 +1512,22 @@ for (const dialect of DIALECTS) {
             sqlite: NOT_SUPPORTED,
           })
         })
+
+        if (sqlSpec === 'mysql') {
+          it('should drop a temporary table if it exists', async () => {
+            const builder = ctx.db.schema.dropTable('test').temporary().ifExists()
+            
+            testSql(builder, dialect, {
+              postgres: NOT_SUPPORTED,
+              mysql: {
+                sql: 'drop temporary table if exists `test`',
+                parameters: [],
+              },
+              mssql: NOT_SUPPORTED,
+              sqlite: NOT_SUPPORTED,
+            })
+          })
+        }
       }
     })