Replace dynamodb migration handling functionality with dynamodb-migrations library #27

Author: mjzone

index.js

@@ -3,6 +3,8 @@
 const _ = require('lodash'),
     BbPromise = require('bluebird'),
     tables = require('./tables/core'),
+    dm = require('dynamodb-migrations'),
+    AWS = require('aws-sdk'),
     dynamodbLocal = require('dynamodb-localhost');
 
 module.exports = function(S) { // Always pass in the ServerlessPlugin Class
@@ -125,16 +127,17 @@ module.exports = function(S) { // Always pass in the ServerlessPlugin Class
                     table = config && config.table || {},
                     rootPath = S.getProject().getRootPath(),
                     tablesPath = rootPath + '/' + (table.dir || 'dynamodb'),
+                    dbPort = config && config.start && config.start.port || 8000,
+                    dbOptions = { region: 'localhost', endpoint: 'http://localhost:' + dbPort },
+                    dynamodb = { raw: new AWS.DynamoDB(dbOptions), doc: new AWS.DynamoDB.DocumentClient(dbOptions) },
                     suffix = table.suffix || '',
                     prefix = table.prefix || '';
 
+                dm.init(dynamodb, tablesPath);
                 if (options.new) {
-                    tables.newTemplate(options.new, tablesPath).then(resolve, reject);
+                    dm.create(options.new).then(resolve, reject);
                 } else if (options.create) {
-                    tables.create(tablesPath, {
-                        prefix: prefix,
-                        suffix: suffix
-                    }).then(resolve, reject);
+                    dm.executeAll({ tablePrefix: prefix, tableSuffix: suffix }).then(resolve, reject);
                 }
             });
         }

package.json

@@ -39,6 +39,7 @@
     "aws-sdk": "^2.3.19",
     "bluebird": "^3.0.6",
     "dynamodb-localhost": "0.0.2",
+    "dynamodb-migrations": "0.0.3",
     "lodash": "^4.13.1",
     "mkdirp": "^0.5.0",
     "tar": "^2.0.0"