Merge PR #269: Support create table/view optional arguments

Author: nene

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,55 @@ describe('BigQueryFormatter', () => {
         Items;
     `);
   });
+
+  it('supports create view optional arguments', () => {
+    const createViewVariations = [
+      'CREATE VIEW',
+      'CREATE OR REPLACE VIEW',
+      'CREATE VIEW IF NOT EXISTS',
+    ];
+
+    createViewVariations.forEach((createViewVariation: string) => {
+      expect(
+        format(`
+        ${createViewVariation} my_dataset.my_view AS (
+          SELECT t1.col1, t1.col2 FROM my_dataset.my_table)`)
+      ).toBe(dedent`
+        ${createViewVariation}
+          my_dataset.my_view AS (
+            SELECT
+              t1.col1,
+              t1.col2
+            FROM
+              my_dataset.my_table
+          )
+      `);
+    });
+  });
+
+  it('supports create table optional arguments', () => {
+    const createTableVariations = [
+      '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',
+    ];
+
+    createTableVariations.forEach((createTableVariation: string) => {
+      expect(
+        format(`
+        ${createTableVariation} mydataset.newtable (
+          a INT64 NOT NULL
+        )`)
+      ).toBe(dedent`
+        ${createTableVariation}
+          mydataset.newtable (a INT64 NOT NULL)
+      `);
+    });
+  });
 });