Merge pull request #156 from liam-hq/add_column_comment_parsing_for_s…

…chemarb_parser

Extract column comments for schemarb parser

frontend/packages/db-structure/src/parser/schemarb/index.test.ts

@@ -189,5 +189,25 @@ describe(processor, () => {
 
       expect(result).toEqual(expected)
     })
+
+    it('column commnet', async () => {
+      const result = await processor(/* Ruby */ `
+        create_table "users" do |t|
+          t.string "name", comment: 'this is name'
+        end
+      `)
+
+      const expected = userTable({
+        columns: {
+          name: aColumn({
+            name: 'name',
+            type: 'string',
+            comment: 'this is name',
+          }),
+        },
+      })
+
+      expect(result).toEqual(expected)
+    })
   })
 })

frontend/packages/db-structure/src/parser/schemarb/parser.ts

@@ -157,6 +157,10 @@ function extractColumnOptions(hashNode: KeywordHashNode, column: Column): void {
       case 'unique':
         column.unique = value instanceof TrueNode
         break
+      case 'comment':
+        // @ts-expect-error: unescaped is defined as string but it is actually object
+        column.comment = value.unescaped.value
+        break
     }
   }
 }