37 lines
No EOL
1.2 KiB
TypeScript
37 lines
No EOL
1.2 KiB
TypeScript
import { useAppDispatch, useGetState } from 'soapbox/hooks';
|
|
|
|
import { deleteEntities, importEntities } from '../actions';
|
|
|
|
interface DeleteEntityResult<T> {
|
|
result: T
|
|
}
|
|
|
|
type DeleteFn<T> = (entityId: string) => Promise<T> | T;
|
|
|
|
function useDeleteEntity<T = unknown>(entityType: string, deleteFn: DeleteFn<T>) {
|
|
const dispatch = useAppDispatch();
|
|
const getState = useGetState();
|
|
|
|
return async function deleteEntity(entityId: string): Promise<DeleteEntityResult<T>> {
|
|
// Get the entity before deleting, so we can reverse the action if the API request fails.
|
|
const entity = getState().entities[entityType]?.store[entityId];
|
|
|
|
// Optimistically delete the entity from the _store_ but keep the lists in tact.
|
|
dispatch(deleteEntities([entityId], entityType, { preserveLists: true }));
|
|
|
|
try {
|
|
const result = await deleteFn(entityId);
|
|
// Success - finish deleting entity from the state.
|
|
dispatch(deleteEntities([entityId], entityType));
|
|
return { result };
|
|
} catch (e) {
|
|
if (entity) {
|
|
// If the API failed, reimport the entity.
|
|
dispatch(importEntities([entity], entityType));
|
|
}
|
|
throw e;
|
|
}
|
|
};
|
|
}
|
|
|
|
export { useDeleteEntity }; |