103 lines
3.1 KiB
TypeScript
103 lines
3.1 KiB
TypeScript
import type { Chat, Message, MutationCreate_MessageArgs } from '../graphql/schema/chat.gen'
|
|
import type { Accessor, JSX } from 'solid-js'
|
|
|
|
import { createContext, createMemo, createSignal, useContext } from 'solid-js'
|
|
|
|
import { inboxClient } from '../graphql/client/chat'
|
|
import { loadMessages } from '../stores/inbox'
|
|
|
|
import { SSEMessage, useConnect } from './connect'
|
|
|
|
type InboxContextType = {
|
|
chats: Accessor<Chat[]>
|
|
messages?: Accessor<Message[]>
|
|
actions: {
|
|
createChat: (members: number[], title: string) => Promise<{ chat: Chat }>
|
|
loadChats: () => Promise<void>
|
|
getMessages?: (chatId: string) => Promise<void>
|
|
sendMessage?: (args: MutationCreate_MessageArgs) => void
|
|
}
|
|
}
|
|
|
|
const InboxContext = createContext<InboxContextType>()
|
|
|
|
export function useInbox() {
|
|
return useContext(InboxContext)
|
|
}
|
|
|
|
export const InboxProvider = (props: { children: JSX.Element }) => {
|
|
const [chats, setChats] = createSignal<Chat[]>([])
|
|
const [messages, setMessages] = createSignal<Message[]>([])
|
|
const handleMessage = (sseMessage: SSEMessage) => {
|
|
console.log('[context.inbox]:', sseMessage)
|
|
// TODO: handle all action types: create update delete join left
|
|
if (sseMessage.entity === 'message') {
|
|
const relivedMessage = sseMessage.payload
|
|
setMessages((prev) => [...prev, relivedMessage])
|
|
} else if (sseMessage.entity === 'chat') {
|
|
const relivedChat = sseMessage.payload
|
|
setChats((prev) => [...prev, relivedChat])
|
|
}
|
|
}
|
|
const apiClient = createMemo(() => {
|
|
if (!inboxClient.private) inboxClient.connect()
|
|
return inboxClient
|
|
})
|
|
const { addHandler } = useConnect()
|
|
addHandler(handleMessage)
|
|
|
|
const loadChats = async () => {
|
|
try {
|
|
const newChats = await apiClient().loadChats({ limit: 50, offset: 0 })
|
|
setChats(newChats)
|
|
} catch (error) {
|
|
console.log('[loadChats]', error)
|
|
}
|
|
}
|
|
|
|
const getMessages = async (chatId: string) => {
|
|
if (!chatId) return
|
|
try {
|
|
const response = await loadMessages({ chat: chatId })
|
|
setMessages(response as unknown as Message[])
|
|
} catch (error) {
|
|
console.error('Error loading messages:', error)
|
|
}
|
|
}
|
|
|
|
const sendMessage = async (args: MutationCreate_MessageArgs) => {
|
|
try {
|
|
const message = await apiClient().createMessage(args)
|
|
setMessages((prev) => [...prev, message])
|
|
const currentChat = chats().find((chat) => chat.id === args.chat_id)
|
|
setChats((prev) => [
|
|
...prev.filter((c) => c.id !== currentChat.id),
|
|
{ ...currentChat, updated_at: message.created_at },
|
|
])
|
|
} catch (error) {
|
|
console.error('Error sending message:', error)
|
|
}
|
|
}
|
|
|
|
const createChat = async (members: number[], title: string) => {
|
|
try {
|
|
const chat = await inboxClient.createChat({ members, title })
|
|
setChats((prevChats) => [chat, ...prevChats])
|
|
return chat
|
|
} catch (error) {
|
|
console.error('Error creating chat:', error)
|
|
}
|
|
}
|
|
|
|
const actions = {
|
|
createChat,
|
|
loadChats,
|
|
getMessages,
|
|
sendMessage,
|
|
}
|
|
|
|
const value: InboxContextType = { chats, messages, actions }
|
|
|
|
return <InboxContext.Provider value={value}>{props.children}</InboxContext.Provider>
|
|
}
|