From bce0aeac4837ad9e3a2755c734a9f961674e5dce Mon Sep 17 00:00:00 2001
From: 1ilit <lilitmousheghyan@gmail.com>
Date: Sat, 20 Apr 2024 13:59:05 +0300
Subject: [PATCH] Handle enum and set value parsing

---
 src/utils/astToDiagram.js | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/utils/astToDiagram.js b/src/utils/astToDiagram.js
index 660e3bf..136ffa5 100644
--- a/src/utils/astToDiagram.js
+++ b/src/utils/astToDiagram.js
@@ -21,6 +21,9 @@ export function astToDiagram(ast) {
             const field = {};
             field.name = d.column.column;
             field.type = d.definition.dataType;
+            if (d.definition.expr && d.definition.expr.type === "expr_list") {
+              field.values = d.definition.expr.value.map((v) => v.value);
+            }
             field.comment = "";
             field.unique = false;
             if (d.unique) field.unique = true;