37 lines
1.1 KiB
JavaScript
37 lines
1.1 KiB
JavaScript
|
import {
|
||
|
|
CHAT_MESSAGES_FETCH_SUCCESS,
|
||
|
|
CHAT_MESSAGE_SEND_SUCCESS,
|
||
|
|
} from 'soapbox/actions/chats';
|
||
|
|
import { CHAT_MESSAGES_IMPORT } from 'soapbox/actions/importer';
|
||
|
|
import { Map as ImmutableMap, OrderedSet as ImmutableOrderedSet } from 'immutable';
|
||
|
|
|
||
|
|
const initialState = ImmutableMap();
|
||
|
|
|
||
|
|
const updateList = (state, chatId, messageIds) => {
|
||
|
|
const newIds = state.get(chatId, ImmutableOrderedSet()).union(messageIds);
|
||
|
|
return state.set(chatId, newIds);
|
||
|
|
};
|
||
|
|
|
||
|
|
const importMessage = (state, chatMessage) => {
|
||
|
|
return updateList(state, chatMessage.chat_id, [chatMessage.id]);
|
||
|
|
};
|
||
|
|
|
||
|
|
const importMessages = (state, chatMessages) => (
|
||
|
|
state.withMutations(map =>
|
||
|
|
chatMessages.forEach(chatMessage =>
|
||
|
|
importMessage(map, chatMessage)))
|
||
|
|
);
|
||
|
|
|
||
|
|
export default function chatMessageLists(state = initialState, action) {
|
||
|
|
switch(action.type) {
|
||
|
|
case CHAT_MESSAGES_IMPORT:
|
||
|
|
return importMessages(state, action.chatMessages);
|
||
|
|
case CHAT_MESSAGES_FETCH_SUCCESS:
|
||
|
|
return updateList(state, action.chatId, action.data.map(chat => chat.id));
|
||
|
|
case CHAT_MESSAGE_SEND_SUCCESS:
|
||
|
|
return updateList(state, action.chatId, action.data.id);
|
||
|
|
default:
|
||
|
|
return state;
|
||
|
|
}
|
||
|
|
};
|