From 5ac69a54d1d7451daeddf2a72325523985246b83 Mon Sep 17 00:00:00 2001
From: Muhsin Keloth <muhsinkeramam@gmail.com>
Date: Mon, 5 Sep 2022 12:45:38 +0530
Subject: [PATCH] Set last account id on account change

---
 app/javascript/dashboard/api/auth.js          |  9 ++++++++
 app/javascript/dashboard/api/endPoints.js     |  4 ++++
 .../sidebarComponents/AccountSelector.vue     | 11 ++++++++-
 .../dashboard/store/modules/auth.js           | 10 ++++++++
 .../api/v1/profiles_controller_spec.rb        | 23 +++++++++++++++++++
 5 files changed, 56 insertions(+), 1 deletion(-)

diff --git a/app/javascript/dashboard/api/auth.js b/app/javascript/dashboard/api/auth.js
index 76042103f..ef1762f46 100644
--- a/app/javascript/dashboard/api/auth.js
+++ b/app/javascript/dashboard/api/auth.js
@@ -147,4 +147,13 @@ export default {
   deleteAvatar() {
     return axios.delete(endPoints('deleteAvatar').url);
   },
+
+  setActiveAccount({ accountId }) {
+    const urlData = endPoints('setActiveAccount');
+    return axios.put(urlData.url, {
+      profile: {
+        account_id: accountId,
+      },
+    });
+  },
 };
diff --git a/app/javascript/dashboard/api/endPoints.js b/app/javascript/dashboard/api/endPoints.js
index c9d0955ef..8deb8d56a 100644
--- a/app/javascript/dashboard/api/endPoints.js
+++ b/app/javascript/dashboard/api/endPoints.js
@@ -40,6 +40,10 @@ const endPoints = {
   deleteAvatar: {
     url: '/api/v1/profile/avatar',
   },
+
+  setActiveAccount: {
+    url: '/api/v1/profile/set_active_account',
+  },
 };
 
 export default page => {
diff --git a/app/javascript/dashboard/components/layout/sidebarComponents/AccountSelector.vue b/app/javascript/dashboard/components/layout/sidebarComponents/AccountSelector.vue
index e6e7df48e..482b36d3d 100644
--- a/app/javascript/dashboard/components/layout/sidebarComponents/AccountSelector.vue
+++ b/app/javascript/dashboard/components/layout/sidebarComponents/AccountSelector.vue
@@ -13,7 +13,7 @@
       :key="account.id"
       class="account-selector"
     >
-      <a :href="`/app/accounts/${account.id}/dashboard`">
+      <a @click="onChangeAccount(account.id)">
         <fluent-icon
           v-if="account.id === accountId"
           class="selected--account"
@@ -58,5 +58,14 @@ export default {
       globalConfig: 'globalConfig/get',
     }),
   },
+  methods: {
+    async onChangeAccount(accountId) {
+      await this.$store.dispatch('setActiveAccount', {
+        accountId: accountId,
+      });
+      const accountUrl = `/app/accounts/${accountId}/dashboard`;
+      window.location.href = accountUrl;
+    },
+  },
 };
 </script>
diff --git a/app/javascript/dashboard/store/modules/auth.js b/app/javascript/dashboard/store/modules/auth.js
index 4d33dd0cc..6fa214e69 100644
--- a/app/javascript/dashboard/store/modules/auth.js
+++ b/app/javascript/dashboard/store/modules/auth.js
@@ -143,6 +143,16 @@ export const actions = {
     }
   },
 
+  // eslint-disable-next-line no-empty-pattern
+  setActiveAccount: async ({}, { accountId }) => {
+    // eslint-disable-next-line no-useless-catch
+    try {
+      await authAPI.setActiveAccount({ accountId });
+    } catch (error) {
+      throw error;
+    }
+  },
+
   updateUISettings: async ({ commit }, params) => {
     try {
       commit(types.SET_CURRENT_USER_UI_SETTINGS, params);
diff --git a/spec/controllers/api/v1/profiles_controller_spec.rb b/spec/controllers/api/v1/profiles_controller_spec.rb
index 0e386752f..ed06fcaa6 100644
--- a/spec/controllers/api/v1/profiles_controller_spec.rb
+++ b/spec/controllers/api/v1/profiles_controller_spec.rb
@@ -195,4 +195,27 @@ RSpec.describe 'Profile API', type: :request do
       end
     end
   end
+
+  describe 'PUT /api/v1/profile/set_active_account' do
+    context 'when it is an unauthenticated user' do
+      it 'returns unauthorized' do
+        post '/api/v1/profile/set_active_account'
+
+        expect(response).to have_http_status(:unauthorized)
+      end
+    end
+
+    context 'when it is an authenticated user' do
+      let(:agent) { create(:user, password: 'Test123!', account: account, role: :agent) }
+
+      it 'updates the last active account id' do
+        put '/api/v1/profile/set_active_account',
+            params: { profile: { account_id: account.id } },
+            headers: agent.create_new_auth_token,
+            as: :json
+
+        expect(response).to have_http_status(:success)
+      end
+    end
+  end
 end