mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-13 22:14:59 -05:00
dd7d4da5e3
Previously, canvas actions specific to an entity type only needed the id of that entity type. This allowed you to pass in the id of an entity of the wrong type. All actions for a specific entity now take a full entity identifier, and the entity identifier type can be narrowed. `selectEntity` and `selectEntityOrThrow` now need a full entity identifier, and narrow their return values to a specific entity type _if_ the entity identifier is narrowed. The types for canvas entities are updated with optional type parameters for this purpose. All reducers, actions and components have been updated.
17 lines
847 B
TypeScript
17 lines
847 B
TypeScript
import type { CanvasEntityIdentifier, CanvasEntityType } from 'features/controlLayers/store/types';
|
|
import { createContext, useContext } from 'react';
|
|
import { assert } from 'tsafe';
|
|
|
|
export const EntityIdentifierContext = createContext<CanvasEntityIdentifier | null>(null);
|
|
|
|
export const useEntityIdentifierContext = <T extends CanvasEntityType | undefined = CanvasEntityType>(
|
|
type?: T
|
|
): CanvasEntityIdentifier<T extends undefined ? CanvasEntityType : T> => {
|
|
const entityIdentifier = useContext(EntityIdentifierContext);
|
|
assert(entityIdentifier, 'useEntityIdentifier must be used within a EntityIdentifierProvider');
|
|
if (type) {
|
|
assert(entityIdentifier.type === type, 'useEntityIdentifier must be used with the correct type');
|
|
}
|
|
return entityIdentifier as CanvasEntityIdentifier<T extends undefined ? CanvasEntityType : T>;
|
|
};
|