fix: type of note field that is string doesn show

packages/dbml-core/package.json

@@ -26,8 +26,8 @@
   "scripts": {
     "test": "jest",
     "benchmark": "jest --projects jest-bench.config.json",
-    "build:lib": "babel src --out-dir lib --copy-files",
-    "build:parser": "npx babel-node src/parse/buildParser.js",
+    "build:lib": "babel src --out-dir lib --copy-files --watch",
+    "build:parser": "npx babel-node src/parse/buildParser.js --watch",
     "build": "npm run build:parser && npm run build:lib",
     "prepublish": "npm run build"
   },
@@ -60,4 +60,4 @@
     }
   },
   "gitHead": "5cb80e1aa38fb9a4dbe3079e39c9ef93cd4dc556"
-}
+}

packages/dbml-core/src/model_structure/database.js

@@ -1,4 +1,4 @@
-import _ from 'lodash';
+import _, { get } from 'lodash';
 import Schema from './schema';
 import Ref from './ref';
 import Enum from './enum';
@@ -25,7 +25,7 @@ class Database extends Element {
     this.generateId();
     this.hasDefaultSchema = false;
     this.schemas = [];
-    this.note = project.note ? project.note.value : null;
+    this.note = project.note ? get(project, 'note.value', project.note) : null;
     this.noteToken = project.note ? project.note.token : null;
     this.databaseType = project.database_type;
     this.name = project.name;

packages/dbml-core/src/model_structure/enum.js

@@ -1,3 +1,4 @@
+import { get } from 'lodash';
 import Element from './element';
 import EnumValue from './enumValue';
 import { shouldPrintSchema } from './utils';
@@ -9,7 +10,7 @@ class Enum extends Element {
     super(token);
     if (!name) { this.error('Enum must have a name'); }
     this.name = name;
-    this.note = note ? note.value : null;
+    this.note = note ? get(note, 'value', note) : null;
     this.noteToken = note ? note.token : null;
     this.values = [];
     this.fields = [];

packages/dbml-core/src/model_structure/enumValue.js

@@ -1,3 +1,4 @@
+import { get } from 'lodash';
 import Element from './element';
 
 class EnumValue extends Element {
@@ -7,7 +8,7 @@ class EnumValue extends Element {
     super(token);
     if (!name) { this.error('Enum value must have a name'); }
     this.name = name;
-    this.note = note ? note.value : null;
+    this.note = note ? get(note, 'value', note) : null;
     this.noteToken = note ? note.token : null;
     this._enum = _enum;
     this.dbState = this._enum.dbState;

packages/dbml-core/src/model_structure/field.js

@@ -1,3 +1,4 @@
+import { get } from 'lodash';
 import Element from './element';
 import { DEFAULT_SCHEMA_NAME } from './config';
 
@@ -15,7 +16,7 @@ class Field extends Element {
     this.unique = unique;
     this.pk = pk;
     this.not_null = notNull;
-    this.note = note ? note.value : null;
+    this.note = note ? get(note, 'value', note) : null;
     this.noteToken = note ? note.token : null;
     this.dbdefault = dbdefault;
     this.increment = increment;

packages/dbml-core/src/model_structure/schema.js

@@ -1,3 +1,4 @@
+import { get } from 'lodash';
 import Table from './table';
 import Element from './element';
 import Enum from './enum';
@@ -15,7 +16,7 @@ class Schema extends Element {
     this.tableGroups = [];
     this.refs = [];
     this.name = name;
-    this.note = note ? note.value : null;
+    this.note = note ? get(note, 'value', note) : null;
     this.noteToken = note ? note.token : null;
     this.alias = alias;
     this.database = database;

packages/dbml-core/src/model_structure/table.js

@@ -1,3 +1,4 @@
+import { get } from 'lodash';
 import Element from './element';
 import Field from './field';
 import Index from './indexes';
@@ -11,7 +12,7 @@ class Table extends Element {
     super(token);
     this.name = name;
     this.alias = alias;
-    this.note = note ? note.value : null;
+    this.note = note ? get(note, 'value', note) : null;
     this.noteToken = note ? note.token : null;
     this.headerColor = headerColor;
     this.fields = [];