24 lines
877 B
TypeScript
24 lines
877 B
TypeScript
|
import { importEntities } from 'soapbox/entity-store/actions';
|
||
|
|
import { Entities } from 'soapbox/entity-store/entities';
|
||
|
|
import { type Entity } from 'soapbox/entity-store/types';
|
||
|
|
import { useAppDispatch, useGetState } from 'soapbox/hooks';
|
||
|
|
|
||
|
|
type ChangeEntityFn<TEntity extends Entity> = (entity: TEntity) => TEntity
|
||
|
|
|
||
|
|
function useChangeEntity<TEntity extends Entity = Entity>(entityType: Entities) {
|
||
|
|
const getState = useGetState();
|
||
|
|
const dispatch = useAppDispatch();
|
||
|
|
|
||
|
|
function changeEntity(entityId: string, change: ChangeEntityFn<TEntity>): void {
|
||
|
|
if (!entityId) return;
|
||
|
|
const entity = getState().entities[entityType]?.store[entityId] as TEntity | undefined;
|
||
|
|
if (entity) {
|
||
|
|
const newEntity = change(entity);
|
||
|
|
dispatch(importEntities([newEntity], entityType));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return { changeEntity };
|
||
|
|
}
|
||
|
|
|
||
|
|
export { useChangeEntity, type ChangeEntityFn };
|