From 95f029c3e3b211571b7e9e7d5319ced33b803a3a Mon Sep 17 00:00:00 2001
From: Adrian Obelmejias <obeladrian@gmail.com>
Date: Mon, 15 Feb 2016 18:19:39 -0500
Subject: [PATCH] Fix error handling and return user within the `user` object
 or response

---
 src/controllers/users.js | 28 +++++++++++++++++++---------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/src/controllers/users.js b/src/controllers/users.js
index 078ad75..7f76af8 100644
--- a/src/controllers/users.js
+++ b/src/controllers/users.js
@@ -60,17 +60,27 @@ router.post('/',
 router.put('/:id',
   ensureUser,
   async (ctx) => {
-    const user = await User.findById(ctx.params.id)
+    try {
+      const user = await User.findById(ctx.params.id, '-password -salt')
+      if (!user) {
+        ctx.throw(404)
+      }
 
-    if (!user) {
-      ctx.throw(404)
+      Object.assign(user, ctx.request.body.user)
+
+      await user.save()
+      ctx.body = {
+        user
+      }
+
+    } catch (err) {
+
+      if (err === 404 || err.name === 'CastError') {
+        ctx.throw(404)
+      }
+
+      ctx.throw(500)
     }
-
-    Object.assign(user, ctx.request.body.user)
-
-    await user.save()
-
-    ctx.body = 200
   }
 )