From 79211b79a7aec6a0429e757be40e79f8a7a0a7c9 Mon Sep 17 00:00:00 2001
From: Adam Soltys <asoltys@gmail.com>
Date: Wed, 29 Apr 2020 14:12:16 -0700
Subject: get user before emitting

---
 routes/users.js | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'routes/users.js')

diff --git a/routes/users.js b/routes/users.js
index 9e2700b0..82f4264a 100644
--- a/routes/users.js
+++ b/routes/users.js
@@ -144,6 +144,7 @@ app.post("/disable2fa", auth, twofa, async (req, res) => {
   let { user } = req;
   user.twofa = false;
   await user.save();
+  user = await getUser(user.username);
   emit(user.username, "user", user);
   emit(user.username, "otpsecret", user.otpsecret);
   l.info("disabled 2fa", user.username);
@@ -163,6 +164,7 @@ app.post("/2fa", auth, async (req, res) => {
     if (isValid) {
       user.twofa = true;
       user.save();
+      user = await getUser(user.username);
       emit(user.username, "user", req.user);
     } else {
       return res.status(500).send("Invalid token");
@@ -364,6 +366,7 @@ app.post("/address", auth, async (req, res) => {
   user.address = await bc.getNewAddress("", req.body.type || "bech32");
   await user.save();
   addresses[user.address] = user.username;
+  user = await getUser(user.username);
   emit(user.username, "user", user);
   res.send(user.address);
 });
-- 
cgit v1.2.3


From c03559bd7285d6608974ab02be78b859a6cac689 Mon Sep 17 00:00:00 2001
From: Adam Soltys <asoltys@gmail.com>
Date: Wed, 29 Apr 2020 14:16:09 -0700
Subject: catch account shift errors

---
 routes/users.js | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

(limited to 'routes/users.js')

diff --git a/routes/users.js b/routes/users.js
index 82f4264a..6b814e81 100644
--- a/routes/users.js
+++ b/routes/users.js
@@ -375,16 +375,21 @@ app.post("/shiftAccount", auth, async (req, res) => {
   let { user } = req;
   let { asset } = req.body;
 
-  const account = await db.Account.findOne({
-    where: { user_id: user.id, asset } 
-  });
+  try {
+    const account = await db.Account.findOne({
+      where: { user_id: user.id, asset } 
+    });
 
-  user.account_id = account.id;
-  await user.save();
-  
-  user = await getUser(user.username);
-  emit(user.username, "user", user);
-  res.send(account);
+    user.account_id = account.id;
+    await user.save();
+    
+    user = await getUser(user.username);
+    emit(user.username, "user", user);
+    res.send(account);
+  } catch(e) {
+    l.error(e.message);
+    return res.status(500).send("There was a problem switching accounts");
+  } 
 });
 
 setInterval(() => (faucet = 2000), DAY);
-- 
cgit v1.2.3