reformat sql parser

Author: lastlink

src/generate-sql-ddl.ts

@@ -1,7 +1,12 @@
+import {
+  DbDefinition,
+  DbEntityDefinition,
+  DbRelationshipDefinition,
+} from "./types";
 
 interface Table {
-    name: string;
-    columns: string[];
+  name: string;
+  columns: string[];
 }
 
 class Node<T> {
@@ -10,57 +15,72 @@ class Node<T> {
   constructor(public data: T) {}
 }
 
-export class Parser {
+export class DbParser {
+  private symbols = ["INTEGER", "NVARCHAR", "DATETIME", "NUMERIC"];
+  private db: DbDefinition;
+  private dbType: string;
+  private dataDefinitions!: Array<string>;
+  private entities!: Record<string, DbEntityDefinition>;
+  private relationships!: DbRelationshipDefinition[];
 
-    private symbols = ['INTEGER', 'NVARCHAR', 'DATETIME', 'NUMERIC'];
-    private ast: string;
-    private dataDefinitions! : Array<string>;
+  constructor(dbType: string, db: DbDefinition) {
+    this.db = db;
+    this.dbType = dbType;
+  }
 
-    constructor(abstractSyntaxTree: string){
-        this.ast = abstractSyntaxTree;
-    }
-
-    public getSQLDataDefinition(){
-        const tokens = this.ast.trim().split(" ");
-        console.log(tokens);
-        if(tokens[0] !== 'erDiagram'){
-            throw 'Expecting erDiagram keyword';
-        }
-        return this.lexer(tokens);
-    }
+  public getSQLDataDefinition() {
+    this.entities = this.db.getEntities();
+    this.relationships = this.db.getRelationships();
+    // const tokens = this.ast.trim().split(" ");
+    // console.log(tokens);
+    // if(tokens[0] !== 'erDiagram'){
+    //     throw 'Expecting erDiagram keyword';
+    // }
+    return this.lexer();
+  }
 
-    private lexer(tokens : Array<string>): string{
-        let statementGeneration : Array<string> = [];
-        for(let i = 1; i < tokens.length; i++){
-            if(this.symbols.includes(tokens[i])){
-                throw `Keyword reserved '${tokens[i]}'`;
-            }
-            const tableName = tokens[i];
-            i++;
-            if(tokens[i] !== "{"){
-                throw `Expecting '{' got : '${tokens[i]}'`;
-            }
-            i++;
-            while(tokens[i] !== '}' && i < tokens.length){
-                if(!this.symbols.includes(tokens[i])){
-                    throw `Expecting keyword got : '${tokens[i]}'`;
-                }
-                const dataType = tokens[i];
-                i++;
-                statementGeneration.push(this.createColumn(tableName, tokens[i], dataType));
-                i++;
-            }
-            statementGeneration.unshift(this.createTable(tableName));
+  private lexer(): string {
+    let statementGeneration: Array<string> = [];
+    if (this.entities) {
+      for (const key in this.entities) {
+        if (Object.prototype.hasOwnProperty.call(this.entities, key)) {
+          const element = this.entities[key];
         }
-        return statementGeneration.join("");
+      }
     }
+    // for(let i = 1; i < tokens.length; i++){
+    //     if(this.symbols.includes(tokens[i])){
+    //         throw `Keyword reserved '${tokens[i]}'`;
+    //     }
+    //     const tableName = tokens[i];
+    //     i++;
+    //     if(tokens[i] !== "{"){
+    //         throw `Expecting '{' got : '${tokens[i]}'`;
+    //     }
+    //     i++;
+    //     while(tokens[i] !== '}' && i < tokens.length){
+    //         if(!this.symbols.includes(tokens[i])){
+    //             throw `Expecting keyword got : '${tokens[i]}'`;
+    //         }
+    //         const dataType = tokens[i];
+    //         i++;
+    //         statementGeneration.push(this.createColumn(tableName, tokens[i], dataType));
+    //         i++;
+    //     }
+    //     statementGeneration.unshift(this.createTable(tableName));
+    // }
+    return statementGeneration.join("");
+  }
 
-    private createTable(tableName: string,){
-        return `CREATE TABLE ${tableName}`;
-    }
-
-    private createColumn(tableName: string, columnName: string, dataType: string){
-        return `ALTER TABLE ${tableName } ADD ${columnName} ${dataType}`;
-    }
+  private createTable(tableName: string) {
+    return `CREATE TABLE ${tableName}`;
+  }
 
-}
+  private createColumn(
+    tableName: string,
+    columnName: string,
+    dataType: string
+  ) {
+    return `ALTER TABLE ${tableName} ADD ${columnName} ${dataType}`;
+  }
+}

tests/generate-sql-ddl.spec.ts

@@ -1,9 +1,10 @@
-import { Parser } from "../src/generate-sql-ddl";
-
+import { DbParser } from "../src/generate-sql-ddl";
+import erDb from "../src/mermaid/src/diagrams/er/erDb";
 describe("Generate DDL examples", () => {
   it("Should generate create table SQL syntax", () => {
-    const abt = `erDiagram artists { }`;
-    const ddlSyntax = new Parser(abt).getSQLDataDefinition();
-    expect(ddlSyntax).toBe('CREATE TABLE artists');
+    const db = erDb;
+    // `erDiagram artists { }`;
+    const ddlSyntax = new DbParser('sqlite', db).getSQLDataDefinition();
+    // expect(ddlSyntax).toBe('CREATE TABLE artists');
   });
 });