mirror of
https://github.com/array-in-a-matrix/xinny.git
synced 2024-06-12 17:37:13 -04:00
121 lines
3.6 KiB
TypeScript
121 lines
3.6 KiB
TypeScript
|
import produce from 'immer';
|
||
|
import { atom, useSetAtom, WritableAtom } from 'jotai';
|
||
|
import {
|
||
|
ClientEvent,
|
||
|
MatrixClient,
|
||
|
MatrixEvent,
|
||
|
Room,
|
||
|
RoomEvent,
|
||
|
RoomStateEvent,
|
||
|
} from 'matrix-js-sdk';
|
||
|
import { useEffect } from 'react';
|
||
|
import { Membership, RoomToParents, StateEvent } from '../../types/matrix/room';
|
||
|
import {
|
||
|
getRoomToParents,
|
||
|
getSpaceChildren,
|
||
|
isSpace,
|
||
|
isValidChild,
|
||
|
mapParentWithChildren,
|
||
|
} from '../utils/room';
|
||
|
|
||
|
export type RoomToParentsAction =
|
||
|
| {
|
||
|
type: 'INITIALIZE';
|
||
|
roomToParents: RoomToParents;
|
||
|
}
|
||
|
| {
|
||
|
type: 'PUT';
|
||
|
parent: string;
|
||
|
children: string[];
|
||
|
}
|
||
|
| {
|
||
|
type: 'DELETE';
|
||
|
roomId: string;
|
||
|
};
|
||
|
|
||
|
const baseRoomToParents = atom<RoomToParents>(new Map());
|
||
|
export const roomToParentsAtom = atom<RoomToParents, RoomToParentsAction>(
|
||
|
(get) => get(baseRoomToParents),
|
||
|
(get, set, action) => {
|
||
|
if (action.type === 'INITIALIZE') {
|
||
|
set(baseRoomToParents, action.roomToParents);
|
||
|
return;
|
||
|
}
|
||
|
if (action.type === 'PUT') {
|
||
|
set(
|
||
|
baseRoomToParents,
|
||
|
produce(get(baseRoomToParents), (draftRoomToParents) => {
|
||
|
mapParentWithChildren(draftRoomToParents, action.parent, action.children);
|
||
|
})
|
||
|
);
|
||
|
return;
|
||
|
}
|
||
|
if (action.type === 'DELETE') {
|
||
|
set(
|
||
|
baseRoomToParents,
|
||
|
produce(get(baseRoomToParents), (draftRoomToParents) => {
|
||
|
const noParentRooms: string[] = [];
|
||
|
draftRoomToParents.delete(action.roomId);
|
||
|
draftRoomToParents.forEach((parents, child) => {
|
||
|
parents.delete(action.roomId);
|
||
|
if (parents.size === 0) noParentRooms.push(child);
|
||
|
});
|
||
|
noParentRooms.forEach((room) => draftRoomToParents.delete(room));
|
||
|
})
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
);
|
||
|
|
||
|
export const useBindRoomToParentsAtom = (
|
||
|
mx: MatrixClient,
|
||
|
roomToParents: WritableAtom<RoomToParents, RoomToParentsAction>
|
||
|
) => {
|
||
|
const setRoomToParents = useSetAtom(roomToParents);
|
||
|
|
||
|
useEffect(() => {
|
||
|
setRoomToParents({ type: 'INITIALIZE', roomToParents: getRoomToParents(mx) });
|
||
|
|
||
|
const handleAddRoom = (room: Room) => {
|
||
|
if (isSpace(room) && room.getMyMembership() !== Membership.Invite) {
|
||
|
setRoomToParents({ type: 'PUT', parent: room.roomId, children: getSpaceChildren(room) });
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const handleMembershipChange = (room: Room, membership: string) => {
|
||
|
if (isSpace(room) && membership === Membership.Join) {
|
||
|
setRoomToParents({ type: 'PUT', parent: room.roomId, children: getSpaceChildren(room) });
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const handleStateChange = (mEvent: MatrixEvent) => {
|
||
|
if (mEvent.getType() === StateEvent.SpaceChild) {
|
||
|
const childId = mEvent.getStateKey();
|
||
|
const roomId = mEvent.getRoomId();
|
||
|
if (childId && roomId) {
|
||
|
if (isValidChild(mEvent)) {
|
||
|
setRoomToParents({ type: 'PUT', parent: roomId, children: [childId] });
|
||
|
} else {
|
||
|
setRoomToParents({ type: 'DELETE', roomId: childId });
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const handleDeleteRoom = (roomId: string) => {
|
||
|
setRoomToParents({ type: 'DELETE', roomId });
|
||
|
};
|
||
|
|
||
|
mx.on(ClientEvent.Room, handleAddRoom);
|
||
|
mx.on(RoomEvent.MyMembership, handleMembershipChange);
|
||
|
mx.on(RoomStateEvent.Events, handleStateChange);
|
||
|
mx.on(ClientEvent.DeleteRoom, handleDeleteRoom);
|
||
|
return () => {
|
||
|
mx.removeListener(ClientEvent.Room, handleAddRoom);
|
||
|
mx.removeListener(RoomEvent.MyMembership, handleMembershipChange);
|
||
|
mx.removeListener(RoomStateEvent.Events, handleStateChange);
|
||
|
mx.removeListener(ClientEvent.DeleteRoom, handleDeleteRoom);
|
||
|
};
|
||
|
}, [mx, setRoomToParents]);
|
||
|
};
|