add config

austencollins · austencollins · commit ab6d1b07e687 · 2016-11-12T00:23:59.000-08:00
diff --git a/.gitignore b/.gitignore
@@ -13,6 +13,9 @@ node_modules
 **/.idea
 .DS_Store
 .tmp
-admin.env
 .env
+env.js
+env.yml
+env.json
+admin.env
 tmp
diff --git a/serverless-rest-api/.gitignore b/serverless-rest-api/.gitignore
@@ -0,0 +1,21 @@
+.serverless
+*.log
+npm-debug.log
+pids
+*.pid
+*.seed
+dist
+lib-cov
+coverage
+.grunt
+.lock-wscript
+node_modules
+**/.idea
+.DS_Store
+.tmp
+.env
+env.js
+env.yml
+env.json
+admin.env
+tmp
diff --git a/serverless-rest-api/README.md b/serverless-rest-api/README.md
@@ -1,3 +1,5 @@
 # Serverless REST API
 
 Work in progress – This is a ready-to-deploy Serverless REST API.
+
+* Create an `env.js` file
diff --git a/serverless-rest-api/lib/users.js b/serverless-rest-api/lib/users.js
@@ -1,10 +1,13 @@
 'use strict';
 
+const config = require('./utils/config')(); // Load ENV vars and config
+
 const AWS = require('aws-sdk');
 const dynamoDb = new AWS.DynamoDB.DocumentClient();
 const uuid = require('uuid');
+const Res = require('./utils/res');
 
-let Res = require('./utils/res');
+const table = 'users-' + process.env.appStage;
 
 // Users - Create
 module.exports.create = (event, ctx, cb) => {
@@ -20,20 +23,16 @@ module.exports.create = (event, ctx, cb) => {
   data.updatedAt = new Date().getTime();
 
   const params = {
-    TableName: 'todos',
+    TableName: table,
     Item: data
   };
-
-  // Return
-  res.status(200).cors().body({ test: 'data' }).end();
-
-  // return dynamoDb.put(params, (error, data) => {
-  //   console.log(error, data)
-  //
-  //   utils.res(cb, { data: "data!" });
-  //   // if (error) {
-  //   //   callback(error);
-  //   // }
-  //   // callback(error, params.Item);
-  // });
+console.log(params)
+  return dynamoDb.put(params, (error, data) => {
+    if (error) {
+      console.log(error)
+      res.status(400).cors().body({ message: error }).end();
+    } else {
+      res.status(200).cors().body({ test: 'data' }).end();
+    }
+  });
 };
diff --git a/serverless-rest-api/lib/utils/config.js b/serverless-rest-api/lib/utils/config.js
@@ -0,0 +1,13 @@
+module.exports = () => {
+
+  let config;
+  let envPath = '../../env.json';
+
+  try {
+    config = require(envPath) || {};
+  } catch (e) {
+    console.log('Could not load ' + envPath, e);
+  }
+
+  process.env.appStage = config && config.stage ? config.stage : 'dev';
+}
