forked from AkkomaGang/akkoma-fe
93 lines
2.6 KiB
JavaScript
93 lines
2.6 KiB
JavaScript
import merge from 'lodash.merge'
|
|
import objectPath from 'object-path'
|
|
import localforage from 'localforage'
|
|
import { each } from 'lodash'
|
|
|
|
let loaded = false
|
|
|
|
const defaultReducer = (state, paths) => (
|
|
paths.length === 0 ? state : paths.reduce((substate, path) => {
|
|
objectPath.set(substate, path, objectPath.get(state, path))
|
|
return substate
|
|
}, {})
|
|
)
|
|
|
|
const saveImmedeatelyActions = [
|
|
'markNotificationsAsSeen',
|
|
'clearCurrentUser',
|
|
'setCurrentUser',
|
|
'setHighlight',
|
|
'setOption',
|
|
'setClientData',
|
|
'setToken',
|
|
'clearToken'
|
|
]
|
|
|
|
const defaultStorage = (() => {
|
|
return localforage
|
|
})()
|
|
|
|
export default function createPersistedState ({
|
|
key = 'vuex-lz',
|
|
paths = [],
|
|
getState = (key, storage) => {
|
|
let value = storage.getItem(key)
|
|
return value
|
|
},
|
|
setState = (key, state, storage) => {
|
|
if (!loaded) {
|
|
console.log('waiting for old state to be loaded...')
|
|
return Promise.resolve()
|
|
} else {
|
|
return storage.setItem(key, state)
|
|
}
|
|
},
|
|
reducer = defaultReducer,
|
|
storage = defaultStorage,
|
|
subscriber = store => handler => store.subscribe(handler)
|
|
} = {}) {
|
|
return getState(key, storage).then((savedState) => {
|
|
return store => {
|
|
try {
|
|
if (savedState !== null && typeof savedState === 'object') {
|
|
// build user cache
|
|
const usersState = savedState.users || {}
|
|
usersState.usersObject = {}
|
|
const users = usersState.users || []
|
|
each(users, (user) => { usersState.usersObject[user.id] = user })
|
|
savedState.users = usersState
|
|
|
|
store.replaceState(
|
|
merge({}, store.state, savedState)
|
|
)
|
|
}
|
|
loaded = true
|
|
} catch (e) {
|
|
console.log("Couldn't load state")
|
|
console.error(e)
|
|
loaded = true
|
|
}
|
|
subscriber(store)((mutation, state) => {
|
|
try {
|
|
if (saveImmedeatelyActions.includes(mutation.type)) {
|
|
setState(key, reducer(state, paths), storage)
|
|
.then(success => {
|
|
if (typeof success !== 'undefined') {
|
|
if (mutation.type === 'setOption' || mutation.type === 'setCurrentUser') {
|
|
store.dispatch('settingsSaved', { success })
|
|
}
|
|
}
|
|
}, error => {
|
|
if (mutation.type === 'setOption' || mutation.type === 'setCurrentUser') {
|
|
store.dispatch('settingsSaved', { error })
|
|
}
|
|
})
|
|
}
|
|
} catch (e) {
|
|
console.log("Couldn't persist state:")
|
|
console.log(e)
|
|
}
|
|
})
|
|
}
|
|
})
|
|
}
|