feat: add note on the node

alswl · May 4, 2024 · 2ec5814 · 2ec5814
1 parent f060e37
commit 2ec5814
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 0 deletions.
diff --git a/src/nodes/er.ts b/src/nodes/er.ts
@@ -19,6 +19,20 @@ function registerER() {
     },
     true,
   );
+  Graph.registerPortLayout(
+    'erNotePosition',
+    (portsPositionArgs) => {
+      return portsPositionArgs.map((_, index) => {
+        return {
+          position: {
+            x: 0,
+            y: 0,
+          },
+        };
+      });
+    },
+    true,
+  );
 
   Graph.registerNode(
     'er-rect',
@@ -48,6 +62,21 @@ function registerER() {
       },
       ports: {
         groups: {
+          note: {
+            position: 'erNotePosition',
+            markup: [
+              {
+                tagName: 'text',
+                selector: 'note',
+              },
+            ],
+            attrs: {
+              note: {
+                refX: 0,
+                refY: -LINE_HEIGHT,
+              },
+            },
+          },
           list: {
             position: 'erPortPosition',
             markup: [

diff --git a/src/pages/index.tsx b/src/pages/index.tsx
@@ -20,6 +20,8 @@ export default () => {
   username varchar
   role varchar
   created_at timestamp
+
+  Note: 'User Table'
 }
 
 Table posts {

diff --git a/src/parser/parser.ts b/src/parser/parser.ts
@@ -29,13 +29,33 @@ function parseFieldToPort(
   };
 }
 
+function parseNoteToPort(
+  note: string,
+  schemaName: string,
+  tableName: string,
+): any {
+  return {
+    id: `${schemaName}-${tableName}-note`,
+    group: 'note',
+    attrs: {
+      note: {
+        text: note,
+      },
+    },
+  };
+}
+
 function parseTableToNode(table: Table, schemaName: string): any {
   let fields: any[] = [];
   for (let k = 0; k < table.fields.length; k++) {
     const f = table.fields[k];
     const field = parseFieldToPort(f, schemaName, table.name);
     fields.push(field);
   }
+  if (table.note) {
+    const note = parseNoteToPort(table.note, schemaName, table.name);
+    fields.push(note);
+  }
   return {
     id: `${schemaName}-${table.name}`,
     shape: 'er-rect',
@@ -62,6 +82,14 @@ function parseRef(ref: Ref): any {
   return {
     id: ``,
     shape: 'edge',
+    router: {
+      name: 'er',
+      args: {
+        offset: 16,
+        min: 4,
+        direction: 'H',
+      },
+    },
     source: {
       cell: `${sSchemaName}-${source.tableName}`,
       port: `${sSchemaName}-${source.tableName}-${sourceFieldName}`,