@dbml/core 2.4.0 → 2.4.1

package/lib/parse/postgresql/Commands/Alter_table/Alter_table.pegjs

@@ -1,11 +1,50 @@
-alter_table = alter_table_action:alter_table_action {
+alter_table = alter_sub_syntax:(
+	alter_table_action /
+	alter_table_rename /
+	alter_table_set_schema /
+	alter_table_set_tablespace /
+	alter_table_attach /
+	alter_table_detach 
+) {
 	return {
 		command_name: "alter_table",
-		value: alter_table_action
+		value: alter_sub_syntax
 	}
 }
 
-alter_table_action = _ ALTER __ TABLE (__ IF_EXISTS)? (__ ONLY)? __ name:table_name __
+alter_table_rename = _ ALTER __ TABLE (__ IF_EXISTS)? (__ ONLY)? __ table_name (_ "*")? __ RENAME (cmt / !semicolon .)* _ semicolon _ {
+	return {
+		syntax_name: "alter_table_rename",
+	}
+}
+
+alter_table_set_schema = _ ALTER __ TABLE (__ IF_EXISTS)? __ table_name __ SET __ SCHEMA (cmt / !semicolon .)* _ semicolon _ {
+	return {
+		syntax_name: "alter_table_set_schema",
+	}
+}
+
+alter_table_set_tablespace = _ ALTER __ TABLE __ ALL __ IN __ TABLESPACE __ table_name
+	(__ OWNED __ BY __ identifier (_ comma _ identifier)*)? __ SET __ TABLESPACE
+	(cmt / !semicolon .)* _ semicolon _ {
+	return {
+		syntax_name: "alter_set_tablespace",
+	}
+}
+
+alter_table_attach = _ ALTER __ TABLE (__ IF_EXISTS)? __ table_name __ ATTACH __ PARTITION (cmt / !semicolon .)* _ semicolon _ {
+	return {
+		syntax_name: "alter_table_attach",
+	}
+}
+
+alter_table_detach = _ ALTER __ TABLE (__ IF_EXISTS)? __ table_name __ DETACH __ PARTITION (cmt / !semicolon .)* _ semicolon _ {
+	return {
+		syntax_name: "alter_table_detach",
+	}
+}
+
+alter_table_action = _ ALTER __ TABLE (__ IF_EXISTS)? (__ ONLY)? __ name:table_name (_ "*")? __
 	actions:actions _ semicolon _ {
 		actions.forEach(({ type, t_value}) => {
 			switch(type.toLowerCase()) {
@@ -79,6 +118,11 @@ action = ADD __ table_constraint:table_constraint (__ NOT __ VALID)? { // reuse
 			value: value.toLowerCase()
 		}
 	}
+	/ (cmt / !semicolon .)* {
+		return {
+			type: "unknown",
+		}
+	}
 
 
 set_attribute_options = first:set_attribute_option rest:(_ comma _ set_attribute_option)* {

package/lib/parse/postgresql/Commands/Comment.pegjs

@@ -2,9 +2,7 @@ comment = _ COMMENT __ ON __ comment_option:comment_option __ IS __ text:(string
   _ semicolon _ {
   if (text.toLowerCase() !== "null") {
     comment_option.value.text = text;
-  } else {
-    comment_option.value.syntax_name = "remove_comment";
-  }
+  } else comment_option.value.text = null;
 
   return {
     command_name: "comment",
@@ -13,12 +11,26 @@ comment = _ COMMENT __ ON __ comment_option:comment_option __ IS __ text:(string
 }
 
 comment_option = (
-  COLUMN __ relation_name:identifier "." column_name:column_name {
+  COLUMN __ path:(identifier '.')+ column_name:column_name {
+    let dbName = null, schemaName = null, tableName;
+    if (path.length === 1) {
+      tableName = path[0][0];
+    } else if (path.length === 2) {
+      schemaName = path[0][0];
+      tableName = path[1][0];
+    }
+    else {
+      dbName = path[0][0];
+      schemaName = path[1][0];
+      tableName = path[2][0];
+    }
     return {
       syntax_name: "column",
       value: {
-        relation_name: relation_name,
-        column_name
+        dbName,
+        schemaName,
+        tableName,
+        columnName: column_name
       }
     }
   }

package/lib/parse/postgresql/Commands/Ignore_syntax.pegjs

@@ -8,10 +8,19 @@ ignore_syntax = _ value:(
   / CREATE __ SEQUENCE [^;]* { return { syntax_name: "create_sequence" } }
   / CREATE __ SCHEMA [^;]* { return { syntax_name: "create_schema" } }
   / CREATE __ VIEW [^;]* { return { syntax_name: "create_view" } }
+  / ALTER __ (cmt / !(semicolon) .)* { return { syntax_name: "alter_not_table" } }
   / __ { return { syntax_name: "comment_and_space" } }
 ) semicolon _ {
+  const loc = location();
+  const t = text();
   return {
     command_name: "ignore_syntax",
-    value
+    value,
+    warning: {
+      type: 'ignore',
+      location: loc,
+      text: t,
+      message: `ignoring "${t}" at line: ${loc.start.line}`,
+    },
   }
 }

package/lib/parse/postgresql/InitializerUtils.pegjs

@@ -1,8 +1,8 @@
 {
   // intput:
-  // ` 
+  // `
   //      'created'
-  //                   ,            
+  //                   ,
   //         'pending',          'done'
   //  `
   //  => `'created', 'pending', 'done'`
@@ -24,4 +24,6 @@
     });
     return table;
   };
+
+  const findField = (table, fieldName) => table.fields.find(field => field.name === fieldName);
 }

package/lib/parse/postgresql/Keywords.pegjs

@@ -110,4 +110,8 @@ SELECT = "SELECT"i
 USE = "USE"i
 SEQUENCE = "SEQUENCE"i
 SCHEMA = "SCHEMA"i
-VIEW = "VIEW"i
+VIEW = "VIEW"i
+RENAME = "RENAME"i
+OWNED = "OWNED"i
+ATTACH = "ATTACH"i
+DETACH = "DETACH"i

package/lib/parse/postgresql/parser.pegjs

@@ -2,10 +2,12 @@
 	const tables = [];
 	const refs = [];
 	const enums = [];
+	const warnings = [];
 }
 
 parser = commands:command* {
-	commands.forEach(({ command_name, value: { syntax_name, value } }) => {
+	commands.forEach((cmd) => {
+		const { command_name, value: { syntax_name, value }, warning } = cmd;
 		switch(command_name.toLowerCase()){
 			case "create_table":
 				const table = value;
@@ -104,18 +106,29 @@ parser = commands:command* {
 				break;
 			case "comment":
 				switch(syntax_name.toLowerCase()) {
-					case "column":
-						const table_comment = tables.find(table => table.name === value.relation_name);
-						const field_comment = table_comment.fields.find(field => field.name === value.column_name);
-						field_comment.note = value.text;
+					case "column": {
+						const { schemaName, tableName, columnName } = value;
+						const foundTable = findTable(schemaName, tableName);
+						if (foundTable) {
+							const foundField = findField(foundTable, columnName);
+							if (foundField) foundField.note = value.text;
+						}
+						break;
+					}
+					case "table":	{
+						const { schemaName, name: tableName } = value.table_name;
+						const foundTable = findTable(schemaName, tableName);
+						if (foundTable) foundTable.note = value.text;
 						break;
+					}
 				}
 				break;
-			case "ignore_commands":
+			case "ignore_syntax":
+				// warnings.push(warning);
 				break;
 		}
 	})
-	// console.log({tables, refs, enums, indexes});
+
 	return {tables, refs, enums};
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dbml/core",
-  "version": "2.4.0",
+  "version": "2.4.1",
   "description": "> TODO: description",
   "author": "Holistics <dev@holistics.io>",
   "license": "Apache-2.0",
@@ -56,5 +56,5 @@
       "\\.(?!json$)[^.]*$": "jest-raw-loader"
     }
   },
-  "gitHead": "bbc1d589c6dcefce089bd6007a24cad2808a3818"
+  "gitHead": "4a4ec93db569e677bf15eb53964bd6b26762a421"
 }