Fix initialization

jledentu · Dec 20, 2020 · 8e912a4 · 8e912a4
1 parent 3199570
commit 8e912a4
Showing 1 changed file with 9 additions and 13 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -42,7 +42,7 @@ module.exports = function(sails) {
                 let attr = model.attributes[name];
                 if (attr.type === 'slug' && attr.from) {
                     attr.type = 'string';
-
+                    
                     model.beforeCreate = ((previousBeforeCreate, from, blacklist) =>
                         function(values, cb) {
                             const slugName = slug(values[from], {
@@ -80,18 +80,14 @@ module.exports = function(sails) {
             },
         },
 
-        initialize(done) {
-            sails.after('hook:moduleloader:loaded', () => {
-                const originalLoadModels = sails.modules.loadModels;
-                sails.modules.loadModels = function(cb) {
-                    originalLoadModels((error, models) => {
-                        patchModels(models);
-                        console.log(models);
-                        cb(error, models);
-                        done();
-                    });
-                };
-            });
+        configure() {
+            const originalLoadModels = sails.hooks.moduleloader.loadModels;
+            sails.hooks.moduleloader.loadModels = function(cb) {
+                originalLoadModels((error, models) => {
+                    patchModels(models);
+                    cb(error, models);
+                });
+            };
         },
     };
 };