Support create table/view optional arguments

Author: ahmad-khan-educative

src/languages/bigquery.formatter.ts

@@ -718,11 +718,21 @@ const reservedCommands = [
   'SET SCHEMA', // added
   'CREATE SCHEMA',
   'CREATE TABLE',
+  'CREATE TABLE IF NOT EXISTS',
+  'CREATE TEMP TABLE',
+  'CREATE TEMP TABLE IF NOT EXISTS',
+  'CREATE TEMPORARY TABLE',
+  'CREATE TEMPORARY TABLE IF NOT EXISTS',
+  'CREATE OR REPLACE TABLE',
+  'CREATE OR REPLACE TEMP TABLE',
+  'CREATE OR REPLACE TEMPORARY TABLE',
   'CREATE TABLE LIKE',
   'CREATE TABLE COPY',
   'CREATE SNAPSHOT TABLE',
   'CREATE TABLE CLONE',
   'CREATE VIEW',
+  'CREATE VIEW IF NOT EXISTS',
+  'CREATE OR REPLACE VIEW',
   'CREATE MATERIALIZED VIEW',
   'CREATE EXTERNAL TABLE',
   'CREATE FUNCTION',

test/bigquery.test.ts

@@ -166,4 +166,128 @@ describe('BigQueryFormatter', () => {
         Items;
     `);
   });
+
+  it('supports create view optional arguments', () => {
+    expect(
+      format(`
+      CREATE OR REPLACE VIEW my_dataset.my_view AS (
+        SELECT t1.col1, t1.col2 FROM my_dataset.my_table)`)
+    ).toBe(dedent`
+      CREATE OR REPLACE VIEW
+        my_dataset.my_view AS (
+          SELECT
+            t1.col1,
+            t1.col2
+          FROM
+            my_dataset.my_table
+        )
+    `);
+
+    expect(
+      format(`
+      CREATE VIEW IF NOT EXISTS my_dataset.my_view AS (
+        SELECT t1.col1, t1.col2 FROM my_dataset.my_table)`)
+    ).toBe(dedent`
+      CREATE VIEW IF NOT EXISTS
+        my_dataset.my_view AS (
+          SELECT
+            t1.col1,
+            t1.col2
+          FROM
+            my_dataset.my_table
+        )
+    `);
+  });
+
+  it('supports create table optional arguments', () => {
+    expect(
+      format(`
+      CREATE TABLE mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE TABLE
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE TABLE IF NOT EXISTS mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE TABLE IF NOT EXISTS
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE TEMP TABLE mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE TEMP TABLE
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE TEMP TABLE IF NOT EXISTS mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE TEMP TABLE IF NOT EXISTS
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE TEMPORARY TABLE mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE TEMPORARY TABLE
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE TEMPORARY TABLE IF NOT EXISTS mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE TEMPORARY TABLE IF NOT EXISTS
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE OR REPLACE TABLE mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE OR REPLACE TABLE
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE OR REPLACE TEMP TABLE mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE OR REPLACE TEMP TABLE
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+
+    expect(
+      format(`
+      CREATE OR REPLACE TEMPORARY TABLE mydataset.newtable (
+        a INT64 NOT NULL
+      )`)
+    ).toBe(dedent`
+      CREATE OR REPLACE TEMPORARY TABLE
+        mydataset.newtable (a INT64 NOT NULL)
+    `);
+  });
 });