From aa62346c401054808d5fe482f438d7a84e8b5f5a Mon Sep 17 00:00:00 2001
From: Muhammad Aaqil <aaqilcs102@gmail.com>
Date: Sun, 14 Apr 2024 23:23:42 +0500
Subject: [PATCH] fix: enable migration for enum property

Signed-off-by: Muhammad Aaqil <aaqilcs102@gmail.com>
---
 lib/migration.js       | 11 +++++++++++
 test/migration.test.js | 10 +++++++++-
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/lib/migration.js b/lib/migration.js
index 46f75c5..c55b68a 100644
--- a/lib/migration.js
+++ b/lib/migration.js
@@ -770,10 +770,21 @@ function mixinMigration(MySQL, mysql) {
     const colLength = columnMetadata && columnMetadata.dataLength || prop.length || prop.limit;
     const colPrecision = columnMetadata && columnMetadata.dataPrecision;
     const colScale = columnMetadata && columnMetadata.dataScale;
+    let enumList = '';
+    if (colType && colType === 'ENUM') {
+      if (prop.jsonSchema && prop.jsonSchema.enum) {
+        prop.jsonSchema.enum.forEach(item => {
+          enumList += `'${item}',`;
+        });
+        // remove trailing comma
+        enumList = enumList.substring(0, enumList.length - 1);
+      }
+    }
     // info on setting column specific properties
     // i.e dataLength, dataPrecision, dataScale
     // https://loopback.io/doc/en/lb3/Model-definition-JSON-file.html
     if (colType) {
+      if (colType === 'ENUM') return colType + '(' + enumList + ')';
       if (colLength) return colType + '(' + colLength + ')';
       if (colPrecision && colScale) return colType + '(' + colPrecision + ',' + colScale + ')';
       if (colPrecision) return colType + '(' + colPrecision + ')';
diff --git a/test/migration.test.js b/test/migration.test.js
index 622f1e9..b7345ce 100644
--- a/test/migration.test.js
+++ b/test/migration.test.js
@@ -85,6 +85,13 @@ describe('migrations', function() {
           Key: '',
           Default: null,
           Extra: ''},
+        typings: {
+          Field: 'typings',
+          Type: "enum('A','B')",
+          Null: 'YES',
+          Key: '',
+          Default: null,
+          Extra: ''},
       });
       done();
     });
@@ -548,9 +555,10 @@ function setup(done) {
   require('./init.js');
 
   db = global.getSchema();
-
+  const customType = db.EnumFactory('A', 'B');
   UserData = db.define('UserData', {
     email: {type: String, null: false, index: true},
+    typings: {type: customType},
     name: String,
     bio: Schema.Text,
     birthDate: Date,