48 lines
1.3 KiB
TypeScript
48 lines
1.3 KiB
TypeScript
|
import { atom, useSetAtom, WritableAtom } from 'jotai';
|
||
|
|
import { ClientEvent, MatrixClient, MatrixEvent } from 'matrix-js-sdk';
|
||
|
|
import { useEffect } from 'react';
|
||
|
|
import { AccountDataEvent } from '../../types/matrix/accountData';
|
||
|
|
import { getAccountData, getMDirects } from '../utils/room';
|
||
|
|
|
||
|
|
export type MDirectAction = {
|
||
|
|
type: 'INITIALIZE' | 'UPDATE';
|
||
|
|
rooms: Set<string>;
|
||
|
|
};
|
||
|
|
|
||
|
|
const baseMDirectAtom = atom(new Set<string>());
|
||
|
|
export const mDirectAtom = atom<Set<string>, MDirectAction>(
|
||
|
|
(get) => get(baseMDirectAtom),
|
||
|
|
(get, set, action) => {
|
||
|
|
set(baseMDirectAtom, action.rooms);
|
||
|
|
}
|
||
|
|
);
|
||
|
|
|
||
|
|
export const useBindMDirectAtom = (
|
||
|
|
mx: MatrixClient,
|
||
|
|
mDirect: WritableAtom<Set<string>, MDirectAction>
|
||
|
|
) => {
|
||
|
|
const setMDirect = useSetAtom(mDirect);
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
const mDirectEvent = getAccountData(mx, AccountDataEvent.Direct);
|
||
|
|
if (mDirectEvent) {
|
||
|
|
setMDirect({
|
||
|
|
type: 'INITIALIZE',
|
||
|
|
rooms: getMDirects(mDirectEvent),
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
const handleAccountData = (event: MatrixEvent) => {
|
||
|
|
setMDirect({
|
||
|
|
type: 'UPDATE',
|
||
|
|
rooms: getMDirects(event),
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
mx.on(ClientEvent.AccountData, handleAccountData);
|
||
|
|
return () => {
|
||
|
|
mx.removeListener(ClientEvent.AccountData, handleAccountData);
|
||
|
|
};
|
||
|
|
}, [mx, setMDirect]);
|
||
|
|
};
|