2 files changed, 9 insertions, 40 deletions
diff --git a/src/modules/pushNotifications.js b/src/modules/pushNotifications.js
deleted file mode 100644
index ea92f811..00000000
--- a/src/modules/pushNotifications.js
+++ /dev/null
@@ -1,36 +0,0 @@
-import registerPushNotifications from '../services/push/push.js'
-
-const subscribe = {
-  state: {
-    token: null,
-    vapidPublicKey: null
-  },
-  mutations: {
-    setApiToken (state, user) {
-      state.token = user.credentials
-    },
-    setVapidPublicKey (state, vapidPublicKey) {
-      state.vapidPublicKey = vapidPublicKey
-    }
-  },
-  actions: {
-    setInstanceOption (store, { name, value }) {
-      if (name === 'vapidPublicKey') {
-        store.commit('setVapidPublicKey', value)
-
-        if (store.state.token) {
-          registerPushNotifications(store.rootState.config.webPushNotifications, value, store.state.token)
-        }
-      }
-    },
-    setCurrentUser (store, user) {
-      store.commit('setApiToken', user.credentials)
-
-      if (store.state.vapidPublicKey) {
-        registerPushNotifications(store.rootState.config.webPushNotifications, store.state.vapidPublicKey, user.credentials)
-      }
-    }
-  }
-}
-
-export default subscribe
diff --git a/src/modules/users.js b/src/modules/users.js
index 1256e9df..5e0c087d 100644
--- a/src/modules/users.js
+++ b/src/modules/users.js
@@ -1,6 +1,7 @@
 import backendInteractorService from '../services/backend_interactor_service/backend_interactor_service.js'
 import { compact, map, each, merge } from 'lodash'
 import { set } from 'vue'
+import registerPushNotifications from '../services/push/push.js'
 
 // TODO: Unify with mergeOrAdd in statuses.js
 export const mergeOrAdd = (arr, obj, item) => {
@@ -65,6 +66,13 @@ const users = {
       store.rootState.api.backendInteractor.fetchUser({id})
         .then((user) => store.commit('addNewUsers', user))
     },
+    registerPushNotifications (store) {
+      const token = store.state.currentUser.credentials
+      const vapidPublicKey = store.rootState.instance.vapidPublicKey
+      const isEnabled = store.rootState.config.webPushNotifications
+
+      registerPushNotifications(isEnabled, vapidPublicKey, token)
+    },
     addNewStatuses (store, { statuses }) {
       const users = map(statuses, 'user')
       const retweetedUsers = compact(map(statuses, 'retweeted_status.user'))
@@ -86,9 +94,6 @@ const users = {
       store.dispatch('stopFetching', 'friends')
       store.commit('setBackendInteractor', backendInteractorService())
     },
-    setCurrentUser (store, user) {
-      store.commit('setCurrentUser', user)
-    },
     loginUser (store, accessToken) {
       return new Promise((resolve, reject) => {
         const commit = store.commit
@@ -100,7 +105,7 @@ const users = {
                 .then((user) => {
                   // user.credentials = userCredentials
                   user.credentials = accessToken
-                  store.dispatch('setCurrentUser', user)
+                  commit('setCurrentUser', user)
                   commit('addNewUsers', [user])
 
                   // Set our new backend interactor