Drop table restrict

package.json

@@ -1,6 +1,6 @@
 {
   "name": "pgsql-ast-parser",
-  "version": "10.0.4",
+  "version": "10.0.5",
   "description": "Yet another simple Postgres SQL parser/modifier",
   "main": "index.js",
   "repository": "https://github.com/oguimbal/pgsql-ast-parser",

src/syntax/ast.ts

@@ -195,7 +195,7 @@ export interface DropTableStatement extends PGNode {
     type: 'drop table';
     name: QName;
     ifExists?: boolean;
-    cascade?: boolean;
+    cascade?: 'cascade' | 'restrict';
 }
 
 export interface DropSequenceStatement extends PGNode {

src/syntax/drop.ne

@@ -3,13 +3,13 @@
 
 # https://www.postgresql.org/docs/9.3/sql-dropindex.html
 
-drop_statement -> kw_drop drop_what kw_ifexists:? qualified_name kw_cascade:? {% (x: any, rej: any) => {
+drop_statement -> kw_drop drop_what kw_ifexists:? qualified_name (kw_cascade | kw_restrict):? {% (x: any, rej: any) => {
     const v = unwrap(x[1]);
     return track(x, {
         ...v,
         ... x[2] && {ifExists: true},
         name: unwrap(x[3]),
-        ... x[4] && {cascade: true},
+        ... x[4] && {cascade: toStr(x[4]) },
     });
 }%}
 

src/syntax/drop.spec.ts

@@ -19,7 +19,14 @@ describe('Drop', () => {
         type: 'drop table',
         name: { name: 'Users' },
         ifExists: true,
-        cascade: true,
+        cascade: 'cascade',
+    });
+
+    checkStatement([`DROP TABLE IF EXISTS "Users" RESTRICT`], {
+        type: 'drop table',
+        name: { name: 'Users' },
+        ifExists: true,
+        cascade: 'restrict',
     });
 
 

src/to-sql.ts

@@ -626,7 +626,7 @@ const visitor = astVisitor<IAstFullVisitor>(m => ({
         }
         m.tableRef(val.name);
         if (val.cascade) {
-            ret.push(' CASCADE ');
+            ret.push(' ', val.cascade, ' ');
         }
     },
     dropIndex: val => {