[#12] put method updated (added reset)

hyun-park · Feb 23, 2018 · f6e02f8 · f6e02f8
1 parent e68d6da
commit f6e02f8
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 4 deletions.
diff --git a/controllers/userController.js b/controllers/userController.js
@@ -77,5 +77,13 @@ module.exports.signInOrUpUser = function(req, res) {
 }
 module.exports.updateUser = function(req, res){
     logger.debug(req.params, req.body);
-    User.updateUser(req.params.uuid, req.body, okWithBodyResponseCb(res), iseWithBodyResponseCb(res));
-}
+    if(Object.keys(req.body).length === 0 && req.body.constructor === Object){
+        User.resetUser(req.params.uuid, okWithBodyResponseCb(res), iseWithBodyResponseCb(res));
+    } else {
+        User.updateUser(req.params.uuid, req.body, okWithBodyResponseCb(res), iseWithBodyResponseCb(res));
+    }
+}
+// module.exports.resetUser = function(req, res){
+//     logger.debug(req.params);
+//     User.resetUser(req.)
+// }
diff --git a/models/user.js b/models/user.js
@@ -54,7 +54,7 @@ var signInOrUpUser = function(_user, successCb, errorCb){
                 if(user.oauthKey === _user.oauthKey) {
                     var resultData = {
                         result: true,
-                        user: user,
+                        user: user
                     }
                 } else {
                     var resultData = {
@@ -138,6 +138,25 @@ var updateUsersFloor = function(currentFloor, updatedFloor) {
         })
 };
 
+var resetUser = function(uuid, successCb, errorCb) {
+    var userResetCb = function(user) {
+        user.buyFloor = 0;
+        user.currentFloor = 0;
+        user.goalFloor = 0;
+        user.startedAt = ff.getCurrentDate();
+
+        usersRef.child(uuid).set(user)
+            .then(function(){
+                successCb(user)
+            })
+            .catch(function(err){
+                errorCb({ "message": "error occurred: " + err.code});
+                throw new Error("error occurred: " + err.code);
+            });
+    };
+    getUserByUuid(uuid, userResetCb, errorCb);
+};
+
 var getUsersPopulation = function(cb){
     var usersPopulation = new Array(60+1).join('0').split('').map(parseFloat);
     usersRef.orderByChild("currentFloor")
@@ -175,4 +194,8 @@ module.exports.updateUsersFloor = function(currentFloor, updatedFloor) {
 
 module.exports.getUsersPopulation = function(cb) {
     return getUsersPopulation(cb);
-}
+};
+
+module.exports.resetUser = function(uuid, successCb, errorCb){
+    return resetUser(uuid, successCb, errorCb);
+};