import { remove, slice, each, findIndex, find, maxBy, minBy, merge, first, last, isArray, omitBy } from 'lodash' import { set } from 'vue' import { isStatusNotification, isValidNotification, maybeShowNotification } from '../services/notification_utils/notification_utils.js' import apiService from '../services/api/api.service.js' const emptyTl = (userId = 0) => ({ statuses: [], statusesObject: {}, faves: [], visibleStatuses: [], visibleStatusesObject: {}, newStatusCount: 0, maxId: 0, minId: 0, minVisibleId: 0, loading: false, followers: [], friends: [], userId, flushMarker: 0 }) const emptyNotifications = () => ({ desktopNotificationSilence: true, maxId: 0, minId: Number.POSITIVE_INFINITY, data: [], idStore: {}, loading: false }) export const defaultState = () => ({ allStatuses: [], allStatusesObject: {}, conversationsObject: {}, maxId: 0, notifications: emptyNotifications(), favorites: new Set(), timelines: { mentions: emptyTl(), public: emptyTl(), user: emptyTl(), favorites: emptyTl(), media: emptyTl(), publicAndExternal: emptyTl(), friends: emptyTl(), tag: emptyTl(), dms: emptyTl(), bookmarks: emptyTl() } }) export const prepareStatus = (status) => { // Set deleted flag status.deleted = false // To make the array reactive status.attachments = status.attachments || [] return status } const mergeOrAdd = (arr, obj, item) => { const oldItem = obj[item.id] if (oldItem) { // We already have this, so only merge the new info. // We ignore null values to avoid overwriting existing properties with missing data // we also skip 'user' because that is handled by users module merge(oldItem, omitBy(item, (v, k) => v === null || k === 'user')) // Reactivity fix. oldItem.attachments.splice(oldItem.attachments.length) return { item: oldItem, new: false } } else { // This is a new item, prepare it prepareStatus(item) arr.push(item) set(obj, item.id, item) return { item, new: true } } } const sortById = (a, b) => { const seqA = Number(a.id) const seqB = Number(b.id) const isSeqA = !Number.isNaN(seqA) const isSeqB = !Number.isNaN(seqB) if (isSeqA && isSeqB) { return seqA > seqB ? -1 : 1 } else if (isSeqA && !isSeqB) { return 1 } else if (!isSeqA && isSeqB) { return -1 } else { return a.id > b.id ? -1 : 1 } } const sortTimeline = (timeline) => { timeline.visibleStatuses = timeline.visibleStatuses.sort(sortById) timeline.statuses = timeline.statuses.sort(sortById) timeline.minVisibleId = (last(timeline.visibleStatuses) || {}).id return timeline } // Add status to the global storages (arrays and objects maintaining statuses) except timelines const addStatusToGlobalStorage = (state, data) => { const result = mergeOrAdd(state.allStatuses, state.allStatusesObject, data) if (result.new) { // Add to conversation const status = result.item const conversationsObject = state.conversationsObject const conversationId = status.statusnet_conversation_id if (conversationsObject[conversationId]) { conversationsObject[conversationId].push(status) } else { set(conversationsObject, conversationId, [status]) } } return result } // Remove status from the global storages (arrays and objects maintaining statuses) except timelines const removeStatusFromGlobalStorage = (state, status) => { remove(state.allStatuses, { id: status.id }) // TODO: Need to remove from allStatusesObject? // Remove possible notification remove(state.notifications.data, ({ action: { id } }) => id === status.id) // Remove from conversation const conversationId = status.statusnet_conversation_id if (state.conversationsObject[conversationId]) { remove(state.conversationsObject[conversationId], { id: status.id }) } } const addNewStatuses = (state, { statuses, showImmediately = false, timeline, user = {}, noIdUpdate = false, userId, pagination = {} }) => { // Sanity check if (!isArray(statuses)) { return false } const allStatuses = state.allStatuses const timelineObject = state.timelines[timeline] // Mismatch between API pagination and our internal minId/maxId tracking systems: // pagination.maxId is the oldest of the returned statuses when fetching older, // and pagination.minId is the newest when fetching newer. The names come directly // from the arguments they're supposed to be passed as for the next fetch. const minNew = pagination.maxId || (statuses.length > 0 ? minBy(statuses, 'id').id : 0) const maxNew = pagination.minId || (statuses.length > 0 ? maxBy(statuses, 'id').id : 0) const newer = timeline && (maxNew > timelineObject.maxId || timelineObject.maxId === 0) && statuses.length > 0 const older = timeline && (minNew < timelineObject.minId || timelineObject.minId === 0) && statuses.length > 0 if (!noIdUpdate && newer) { timelineObject.maxId = maxNew } if (!noIdUpdate && older) { timelineObject.minId = minNew } // This makes sure that user timeline won't get data meant for other // user. I.e. opening different user profiles makes request which could // return data late after user already viewing different user profile if ((timeline === 'user' || timeline === 'media') && timelineObject.userId !== userId) { return } const addStatus = (data, showImmediately, addToTimeline = true) => { const result = addStatusToGlobalStorage(state, data) const status = result.item if (result.new) { // We are mentioned in a post if (status.type === 'status' && find(status.attentions, { id: user.id })) { const mentions = state.timelines.mentions // Add the mention to the mentions timeline if (timelineObject !== mentions) { mergeOrAdd(mentions.statuses, mentions.statusesObject, status) mentions.newStatusCount += 1 sortTimeline(mentions) } } if (status.visibility === 'direct') { const dms = state.timelines.dms mergeOrAdd(dms.statuses, dms.statusesObject, status) dms.newStatusCount += 1 sortTimeline(dms) } } // Decide if we should treat the status as new for this timeline. let resultForCurrentTimeline // Some statuses should only be added to the global status repository. if (timeline && addToTimeline) { resultForCurrentTimeline = mergeOrAdd(timelineObject.statuses, timelineObject.statusesObject, status) } if (timeline && showImmediately) { // Add it directly to the visibleStatuses, don't change // newStatusCount mergeOrAdd(timelineObject.visibleStatuses, timelineObject.visibleStatusesObject, status) } else if (timeline && addToTimeline && resultForCurrentTimeline.new) { // Just change newStatuscount timelineObject.newStatusCount += 1 } return status } const favoriteStatus = (favorite, counter) => { const status = find(allStatuses, { id: favorite.in_reply_to_status_id }) if (status) { // This is our favorite, so the relevant bit. if (favorite.user.id === user.id) { status.favorited = true } else { status.fave_num += 1 } } return status } const processors = { 'status': (status) => { addStatus(status, showImmediately) }, 'retweet': (status) => { // RetweetedStatuses are never shown immediately const retweetedStatus = addStatus(status.retweeted_status, false, false) let retweet // If the retweeted status is already there, don't add the retweet // to the timeline. if (timeline && find(timelineObject.statuses, (s) => { if (s.retweeted_status) { return s.id === retweetedStatus.id || s.retweeted_status.id === retweetedStatus.id } else { return s.id === retweetedStatus.id } })) { // Already have it visible (either as the original or another RT), don't add to timeline, don't show. retweet = addStatus(status, false, false) } else { retweet = addStatus(status, showImmediately) } retweet.retweeted_status = retweetedStatus }, 'favorite': (favorite) => { // Only update if this is a new favorite. // Ignore our own favorites because we get info about likes as response to like request if (!state.favorites.has(favorite.id)) { state.favorites.add(favorite.id) favoriteStatus(favorite) } }, 'deletion': (deletion) => { const uri = deletion.uri const status = find(allStatuses, { uri }) if (!status) { return } removeStatusFromGlobalStorage(state, status) if (timeline) { remove(timelineObject.statuses, { uri }) remove(timelineObject.visibleStatuses, { uri }) } }, 'follow': (follow) => { // NOOP, it is known status but we don't do anything about it for now }, 'default': (unknown) => { console.log('unknown status type') console.log(unknown) } } each(statuses, (status) => { const type = status.type const processor = processors[type] || processors['default'] processor(status) }) // Keep the visible statuses sorted if (timeline && !(timeline === 'bookmarks')) { sortTimeline(timelineObject) } } const updateNotificationsMinMaxId = (state, notification) => { state.notifications.maxId = notification.id > state.notifications.maxId ? notification.id : state.notifications.maxId state.notifications.minId = notification.id < state.notifications.minId ? notification.id : state.notifications.minId } const addNewNotifications = (state, { dispatch, notifications, older, visibleNotificationTypes, rootGetters, newNotificationSideEffects }) => { each(notifications, (notification) => { // If invalid notification, update ids but don't add it to store if (!isValidNotification(notification)) { console.error('Invalid notification:', notification) updateNotificationsMinMaxId(state, notification) return } if (isStatusNotification(notification.type)) { notification.action = addStatusToGlobalStorage(state, notification.action).item notification.status = notification.status && addStatusToGlobalStorage(state, notification.status).item } if (notification.type === 'pleroma:report') { dispatch('addReport', notification.report) } if (notification.type === 'pleroma:emoji_reaction') { dispatch('fetchEmojiReactionsBy', notification.status.id) } // Only add a new notification if we don't have one for the same action if (!state.notifications.idStore.hasOwnProperty(notification.id)) { updateNotificationsMinMaxId(state, notification) state.notifications.data.push(notification) state.notifications.idStore[notification.id] = notification newNotificationSideEffects(notification) } else if (notification.seen) { state.notifications.idStore[notification.id].seen = true } }) } const removeStatus = (state, { timeline, userId }) => { const timelineObject = state.timelines[timeline] if (userId) { remove(timelineObject.statuses, { user: { id: userId } }) remove(timelineObject.visibleStatuses, { user: { id: userId } }) timelineObject.minVisibleId = timelineObject.visibleStatuses.length > 0 ? last(timelineObject.visibleStatuses).id : 0 timelineObject.maxId = timelineObject.statuses.length > 0 ? first(timelineObject.statuses).id : 0 } } export const mutations = { addNewStatuses, addNewNotifications, removeStatus, showNewStatuses (state, { timeline }) { const oldTimeline = (state.timelines[timeline]) oldTimeline.newStatusCount = 0 oldTimeline.visibleStatuses = slice(oldTimeline.statuses, 0, 50) oldTimeline.minVisibleId = last(oldTimeline.visibleStatuses).id oldTimeline.minId = oldTimeline.minVisibleId oldTimeline.visibleStatusesObject = {} each(oldTimeline.visibleStatuses, (status) => { oldTimeline.visibleStatusesObject[status.id] = status }) }, resetStatuses (state) { const emptyState = defaultState() Object.entries(emptyState).forEach(([key, value]) => { state[key] = value }) }, clearTimeline (state, { timeline, excludeUserId = false }) { const userId = excludeUserId ? state.timelines[timeline].userId : undefined state.timelines[timeline] = emptyTl(userId) }, clearNotifications (state) { state.notifications = emptyNotifications() }, setFavorited (state, { status, value }) { const newStatus = state.allStatusesObject[status.id] if (newStatus.favorited !== value) { if (value) { newStatus.fave_num++ } else { newStatus.fave_num-- } } newStatus.favorited = value }, setFavoritedConfirm (state, { status, user }) { const newStatus = state.allStatusesObject[status.id] newStatus.favorited = status.favorited newStatus.fave_num = status.fave_num const index = findIndex(newStatus.favoritedBy, { id: user.id }) if (index !== -1 && !newStatus.favorited) { newStatus.favoritedBy.splice(index, 1) } else if (index === -1 && newStatus.favorited) { newStatus.favoritedBy.push(user) } }, setMutedStatus (state, status) { const newStatus = state.allStatusesObject[status.id] newStatus.thread_muted = status.thread_muted if (newStatus.thread_muted !== undefined) { state.conversationsObject[newStatus.statusnet_conversation_id].forEach(status => { status.thread_muted = newStatus.thread_muted }) } }, setRetweeted (state, { status, value }) { const newStatus = state.allStatusesObject[status.id] if (newStatus.repeated !== value) { if (value) { newStatus.repeat_num++ } else { newStatus.repeat_num-- } } newStatus.repeated = value }, setRetweetedConfirm (state, { status, user }) { const newStatus = state.allStatusesObject[status.id] newStatus.repeated = status.repeated newStatus.repeat_num = status.repeat_num const index = findIndex(newStatus.rebloggedBy, { id: user.id }) if (index !== -1 && !newStatus.repeated) { newStatus.rebloggedBy.splice(index, 1) } else if (index === -1 && newStatus.repeated) { newStatus.rebloggedBy.push(user) } }, setBookmarked (state, { status, value }) { const newStatus = state.allStatusesObject[status.id] newStatus.bookmarked = value }, setBookmarkedConfirm (state, { status }) { const newStatus = state.allStatusesObject[status.id] newStatus.bookmarked = status.bookmarked }, setDeleted (state, { status }) { const newStatus = state.allStatusesObject[status.id] if (newStatus) newStatus.deleted = true }, setManyDeleted (state, condition) { Object.values(state.allStatusesObject).forEach(status => { if (condition(status)) { status.deleted = true } }) }, setLoading (state, { timeline, value }) { state.timelines[timeline].loading = value }, setNsfw (state, { id, nsfw }) { const newStatus = state.allStatusesObject[id] newStatus.nsfw = nsfw }, setNotificationsLoading (state, { value }) { state.notifications.loading = value }, setNotificationsSilence (state, { value }) { state.notifications.desktopNotificationSilence = value }, markNotificationsAsSeen (state) { each(state.notifications.data, (notification) => { notification.seen = true }) }, markSingleNotificationAsSeen (state, { id }) { const notification = find(state.notifications.data, n => n.id === id) if (notification) notification.seen = true }, dismissNotification (state, { id }) { state.notifications.data = state.notifications.data.filter(n => n.id !== id) }, dismissNotifications (state, { finder }) { state.notifications.data = state.notifications.data.filter(n => finder) }, updateNotification (state, { id, updater }) { const notification = find(state.notifications.data, n => n.id === id) notification && updater(notification) }, queueFlush (state, { timeline, id }) { state.timelines[timeline].flushMarker = id }, queueFlushAll (state) { Object.keys(state.timelines).forEach((timeline) => { state.timelines[timeline].flushMarker = state.timelines[timeline].maxId }) }, addRepeats (state, { id, rebloggedByUsers, currentUser }) { const newStatus = state.allStatusesObject[id] newStatus.rebloggedBy = rebloggedByUsers.filter(_ => _) // repeats stats can be incorrect based on polling condition, let's update them using the most recent data newStatus.repeat_num = newStatus.rebloggedBy.length newStatus.repeated = !!newStatus.rebloggedBy.find(({ id }) => currentUser.id === id) }, addFavs (state, { id, favoritedByUsers, currentUser }) { const newStatus = state.allStatusesObject[id] newStatus.favoritedBy = favoritedByUsers.filter(_ => _) // favorites stats can be incorrect based on polling condition, let's update them using the most recent data newStatus.fave_num = newStatus.favoritedBy.length newStatus.favorited = !!newStatus.favoritedBy.find(({ id }) => currentUser.id === id) }, addEmojiReactionsBy (state, { id, emojiReactions, currentUser }) { const status = state.allStatusesObject[id] set(status, 'emoji_reactions', emojiReactions) }, addOwnReaction (state, { id, emoji, currentUser }) { const status = state.allStatusesObject[id] const reactionIndex = findIndex(status.emoji_reactions, { name: emoji }) const reaction = status.emoji_reactions[reactionIndex] || { name: emoji, count: 0, accounts: [] } const newReaction = { ...reaction, count: reaction.count + 1, me: true, accounts: [ ...reaction.accounts, currentUser ] } // Update count of existing reaction if it exists, otherwise append at the end if (reactionIndex >= 0) { set(status.emoji_reactions, reactionIndex, newReaction) } else { set(status, 'emoji_reactions', [...status.emoji_reactions, newReaction]) } }, removeOwnReaction (state, { id, emoji, currentUser }) { const status = state.allStatusesObject[id] const reactionIndex = findIndex(status.emoji_reactions, { name: emoji }) if (reactionIndex < 0) return const reaction = status.emoji_reactions[reactionIndex] const accounts = reaction.accounts || [] const newReaction = { ...reaction, count: reaction.count - 1, me: false, accounts: accounts.filter(acc => acc.id !== currentUser.id) } if (newReaction.count > 0) { set(status.emoji_reactions, reactionIndex, newReaction) } else { set(status, 'emoji_reactions', status.emoji_reactions.filter(r => r.name !== emoji)) } }, updateStatusWithPoll (state, { id, poll }) { const status = state.allStatusesObject[id] status.poll = poll }, setVirtualHeight (state, { statusId, height }) { state.allStatusesObject[statusId].virtualHeight = height } } const statuses = { state: defaultState(), actions: { addNewStatuses ({ rootState, commit }, { statuses, showImmediately = false, timeline = false, noIdUpdate = false, userId, pagination }) { commit('addNewStatuses', { statuses, showImmediately, timeline, noIdUpdate, user: rootState.users.currentUser, userId, pagination }) }, addNewNotifications (store, { notifications, older }) { const { commit, dispatch, rootGetters } = store const newNotificationSideEffects = (notification) => { maybeShowNotification(store, notification) } commit('addNewNotifications', { dispatch, notifications, older, rootGetters, newNotificationSideEffects }) }, setNotificationsLoading ({ rootState, commit }, { value }) { commit('setNotificationsLoading', { value }) }, setNotificationsSilence ({ rootState, commit }, { value }) { commit('setNotificationsSilence', { value }) }, fetchStatus ({ rootState, dispatch }, id) { return rootState.api.backendInteractor.fetchStatus({ id }) .then((status) => dispatch('addNewStatuses', { statuses: [status] })) }, deleteStatus ({ rootState, commit }, status) { commit('setDeleted', { status }) apiService.deleteStatus({ id: status.id, credentials: rootState.users.currentUser.credentials }) }, deleteStatusById ({ rootState, commit }, id) { const status = rootState.statuses.allStatusesObject[id] commit('setDeleted', { status }) }, markStatusesAsDeleted ({ commit }, condition) { commit('setManyDeleted', condition) }, favorite ({ rootState, commit }, status) { // Optimistic favoriting... commit('setFavorited', { status, value: true }) rootState.api.backendInteractor.favorite({ id: status.id }) .then(status => commit('setFavoritedConfirm', { status, user: rootState.users.currentUser })) }, unfavorite ({ rootState, commit }, status) { // Optimistic unfavoriting... commit('setFavorited', { status, value: false }) rootState.api.backendInteractor.unfavorite({ id: status.id }) .then(status => commit('setFavoritedConfirm', { status, user: rootState.users.currentUser })) }, fetchPinnedStatuses ({ rootState, dispatch }, userId) { rootState.api.backendInteractor.fetchPinnedStatuses({ id: userId }) .then(statuses => dispatch('addNewStatuses', { statuses, timeline: 'user', userId, showImmediately: true, noIdUpdate: true })) }, pinStatus ({ rootState, dispatch }, statusId) { return rootState.api.backendInteractor.pinOwnStatus({ id: statusId }) .then((status) => dispatch('addNewStatuses', { statuses: [status] })) }, unpinStatus ({ rootState, dispatch }, statusId) { rootState.api.backendInteractor.unpinOwnStatus({ id: statusId }) .then((status) => dispatch('addNewStatuses', { statuses: [status] })) }, muteConversation ({ rootState, commit }, statusId) { return rootState.api.backendInteractor.muteConversation({ id: statusId }) .then((status) => commit('setMutedStatus', status)) }, unmuteConversation ({ rootState, commit }, statusId) { return rootState.api.backendInteractor.unmuteConversation({ id: statusId }) .then((status) => commit('setMutedStatus', status)) }, retweet ({ rootState, commit }, status) { // Optimistic retweeting... commit('setRetweeted', { status, value: true }) rootState.api.backendInteractor.retweet({ id: status.id }) .then(status => commit('setRetweetedConfirm', { status: status.retweeted_status, user: rootState.users.currentUser })) }, unretweet ({ rootState, commit }, status) { // Optimistic unretweeting... commit('setRetweeted', { status, value: false }) rootState.api.backendInteractor.unretweet({ id: status.id }) .then(status => commit('setRetweetedConfirm', { status, user: rootState.users.currentUser })) }, bookmark ({ rootState, commit }, status) { commit('setBookmarked', { status, value: true }) rootState.api.backendInteractor.bookmarkStatus({ id: status.id }) .then(status => { commit('setBookmarkedConfirm', { status }) }) }, unbookmark ({ rootState, commit }, status) { commit('setBookmarked', { status, value: false }) rootState.api.backendInteractor.unbookmarkStatus({ id: status.id }) .then(status => { commit('setBookmarkedConfirm', { status }) }) }, queueFlush ({ rootState, commit }, { timeline, id }) { commit('queueFlush', { timeline, id }) }, queueFlushAll ({ rootState, commit }) { commit('queueFlushAll') }, markNotificationsAsSeen ({ rootState, commit }) { commit('markNotificationsAsSeen') apiService.markNotificationsAsSeen({ id: rootState.statuses.notifications.maxId, credentials: rootState.users.currentUser.credentials }) }, markSingleNotificationAsSeen ({ rootState, commit }, { id }) { commit('markSingleNotificationAsSeen', { id }) apiService.markNotificationsAsSeen({ single: true, id, credentials: rootState.users.currentUser.credentials }) }, dismissNotificationLocal ({ rootState, commit }, { id }) { commit('dismissNotification', { id }) }, dismissNotification ({ rootState, commit }, { id }) { commit('dismissNotification', { id }) rootState.api.backendInteractor.dismissNotification({ id }) }, updateNotification ({ rootState, commit }, { id, updater }) { commit('updateNotification', { id, updater }) }, fetchFavsAndRepeats ({ rootState, commit }, id) { Promise.all([ rootState.api.backendInteractor.fetchFavoritedByUsers({ id }), rootState.api.backendInteractor.fetchRebloggedByUsers({ id }) ]).then(([favoritedByUsers, rebloggedByUsers]) => { commit('addFavs', { id, favoritedByUsers, currentUser: rootState.users.currentUser }) commit('addRepeats', { id, rebloggedByUsers, currentUser: rootState.users.currentUser }) }) }, reactWithEmoji ({ rootState, dispatch, commit }, { id, emoji }) { const currentUser = rootState.users.currentUser if (!currentUser) return commit('addOwnReaction', { id, emoji, currentUser }) rootState.api.backendInteractor.reactWithEmoji({ id, emoji }).then( ok => { dispatch('fetchEmojiReactionsBy', id) } ) }, unreactWithEmoji ({ rootState, dispatch, commit }, { id, emoji }) { const currentUser = rootState.users.currentUser if (!currentUser) return commit('removeOwnReaction', { id, emoji, currentUser }) rootState.api.backendInteractor.unreactWithEmoji({ id, emoji }).then( ok => { dispatch('fetchEmojiReactionsBy', id) } ) }, fetchEmojiReactionsBy ({ rootState, commit }, id) { rootState.api.backendInteractor.fetchEmojiReactions({ id }).then( emojiReactions => { commit('addEmojiReactionsBy', { id, emojiReactions, currentUser: rootState.users.currentUser }) } ) }, fetchFavs ({ rootState, commit }, id) { rootState.api.backendInteractor.fetchFavoritedByUsers({ id }) .then(favoritedByUsers => commit('addFavs', { id, favoritedByUsers, currentUser: rootState.users.currentUser })) }, fetchRepeats ({ rootState, commit }, id) { rootState.api.backendInteractor.fetchRebloggedByUsers({ id }) .then(rebloggedByUsers => commit('addRepeats', { id, rebloggedByUsers, currentUser: rootState.users.currentUser })) }, search (store, { q, resolve, limit, offset, following }) { return store.rootState.api.backendInteractor.search2({ q, resolve, limit, offset, following }) .then((data) => { store.commit('addNewUsers', data.accounts) store.commit('addNewStatuses', { statuses: data.statuses }) return data }) }, setVirtualHeight ({ commit }, { statusId, height }) { commit('setVirtualHeight', { statusId, height }) } }, mutations } export default statuses