Actually disable all REST

README.md

@@ -42,11 +42,11 @@ The usage is based on the node-db-migrate project.
 
 - `dataSource`
 
-  [String] : Datasource to connect the Migration and MigrationMap models to. *(default: db)*
+  [String] : Datasource to connect the Migration model to. *(default: db)*
 
 - `acls`
 
-  [Array] : ACLs to apply to Migration and MigrationMap models. *(default: [])*
+  [Array] : ACLs to apply to Migration model. *(default: [])*
 
 
 ## Running Migrations

lib/index.js

@@ -1,7 +1,6 @@
 var debug = require('debug')('loopback-component-migrate');
 var loopback = require('loopback');
 var migrationDef = require('./models/migration.json');
-var migrationMapDef = require('./models/migration-map.json');
 
 // Remove proerties that will confuse LB
 function getSettings(def) {
@@ -30,12 +29,11 @@ module.exports = function(app, options) {
   }
 
   var migrationModelSettings = getSettings(migrationDef);
-  var migrationMapModelSettings = getSettings(migrationMapDef);
 
   if (typeof options.acls !== 'object') {
-    migrationModelSettings.acls = migrationMapModelSettings.acls = [];
+    migrationModelSettings.acls = [];
   } else {
-    migrationModelSettings.acls = migrationMapModelSettings.acls = options.acls;
+    migrationModelSettings.acls = options.acls;
   }
 
   // Support for loopback 2.x.
@@ -51,27 +49,10 @@ module.exports = function(app, options) {
     migrationDef.properties,
     migrationModelSettings);
 
-  debug('Creating MigrationMap model using settings: %o', migrationModelSettings);
-  var MigrationMapModel = dataSource.createModel(
-    migrationMapDef.name,
-    migrationMapDef.properties,
-    migrationMapModelSettings);
-
   var Migration = require('./models/migration')(MigrationModel, options);
-  var MigrationMap = require('./models/migration-map')(MigrationMapModel, options);
-
-  app.model(Migration);
-  app.model(MigrationMap);
 
-  if (!options.enableRest) {
-    if (Migration.disableRemoteMethodByName) {
-      // Loopback 3.x+
-      Migration.disableRemoteMethodByName('migrateTo');
-      Migration.disableRemoteMethodByName('rollbackTo');
-    } else {
-      // Loopback 2.x
-      Migration.disableRemoteMethod('migrateTo', true);
-      Migration.disableRemoteMethod('rollbackTo', true);
-    }
-  }
+  app.model(Migration, {
+    public: !!options.enableRest,
+    dataSource: dataSource
+  });
 };

lib/models/migration.json

@@ -17,7 +17,14 @@
   },
   "indexes": {
     "name_index": {
-      "name": 1
+      "keys": {
+        "name": 1
+      },
+      "options": {
+        "unique": true
+      },
+      "columns": "name",
+      "kind": "unique"
     }
   },
   "methods": {

test/test.js

@@ -26,9 +26,6 @@ describe('loopback db migrate', function() {
       expect(app.models.Migration).to.exist;
       expect(app.models.Migration).itself.to.respondTo('migrate');
     });
-    it('should attach a MigrationMap model to the app', function() {
-      expect(app.models.Migration).to.exist;
-    });
     it('should provide a Migration.migrate() method', function() {
       expect(app.models.Migration).itself.to.respondTo('migrate');
     });