@tldiagram/core-ui 1.94.1 → 1.94.3

package/src/store/useStore.test.ts

@@ -112,7 +112,6 @@ beforeEach(() => {
     incomingLinks: [],
     treeData: [],
     allElements: [],
-    libraryRefresh: 0,
   })
 })
 
@@ -221,7 +220,6 @@ describe('zustand canvas store', () => {
     useStore.getState().setIncomingLinks([{ id: 1, element_id: 10, element_name: 'E', from_view_id: 1, from_view_name: 'Root', to_view_id: 2 }])
     useStore.getState().setTreeData(tree)
     useStore.getState().setAllElements([libraryElement(10)])
-    useStore.getState().setLibraryRefresh((previous) => previous + 1)
 
     const state = useStore.getState()
     expect(state.view?.id).toBe(1)
@@ -234,7 +232,6 @@ describe('zustand canvas store', () => {
     expect(state.incomingLinks).toHaveLength(1)
     expect(state.treeData).toBe(tree)
     expect(state.allElements).toHaveLength(1)
-    expect(state.libraryRefresh).toBe(1)
   })
 
   it('hydrates and resets canvas data', () => {
@@ -269,13 +266,14 @@ describe('zustand canvas store', () => {
     expect(useStore.getState().viewElements.map((item) => item.element_id)).toEqual([20])
 
     useStore.getState().setViewElements([element(10)])
+    useStore.getState().setAllElements([libraryElement(10)])
     useStore.getState().mergeSavedElement(libraryElement(10))
     expect(useStore.getState().viewElements[0].name).toBe('Saved')
-    expect(useStore.getState().libraryRefresh).toBe(1)
+    expect(useStore.getState().allElements[0].name).toBe('Saved')
 
     useStore.getState().removeElementEverywhere(10)
     expect(useStore.getState().viewElements).toEqual([])
-    expect(useStore.getState().libraryRefresh).toBe(2)
+    expect(useStore.getState().allElements).toEqual([])
 
     useStore.getState().upsertConnector(connector(2))
     useStore.getState().replaceConnector({ ...connector(2), label: 'updated' })

package/src/store/useStore.ts

@@ -45,7 +45,6 @@ export type CanvasStoreState = ViewEditorUiState & {
   incomingLinks: IncomingViewConnector[]
   treeData: ViewTreeNode[]
   allElements: LibraryElement[]
-  libraryRefresh: number
 
   setViewEditorUi: (patch: Partial<ViewEditorUiState>) => void
   setSnapToGrid: (snapToGrid: boolean) => void
@@ -61,7 +60,6 @@ export type CanvasStoreState = ViewEditorUiState & {
   setIncomingLinks: (next: StoreSetter<IncomingViewConnector[]>) => void
   setTreeData: (next: StoreSetter<ViewTreeNode[]>) => void
   setAllElements: (next: StoreSetter<LibraryElement[]>) => void
-  setLibraryRefresh: (next: StoreSetter<number>) => void
   resetCanvas: () => void
   hydrateViewContent: (payload: ViewContentPayload) => void
   updateElementPosition: (elementId: number, x: number, y: number) => void
@@ -268,7 +266,6 @@ export const useStore = create<CanvasStoreState>((set) => ({
   incomingLinks: [],
   treeData: [],
   allElements: [],
-  libraryRefresh: 0,
 
   setViewEditorUi: (patch) => set((state) => ({ ...state, ...patch })),
   setSnapToGrid: (snapToGrid) => set({ snapToGrid }),
@@ -284,7 +281,6 @@ export const useStore = create<CanvasStoreState>((set) => ({
   setIncomingLinks: (next) => set((state) => ({ incomingLinks: resolveSetter(next, state.incomingLinks) })),
   setTreeData: (next) => set((state) => ({ treeData: resolveSetter(next, state.treeData) })),
   setAllElements: (next) => set((state) => ({ allElements: resolveSetter(next, state.allElements) })),
-  setLibraryRefresh: (next) => set((state) => ({ libraryRefresh: resolveSetter(next, state.libraryRefresh) })),
   resetCanvas: () => set({ nodes: [], edges: [] }),
   hydrateViewContent: (payload) => set({
     view: payload.view,
@@ -303,11 +299,11 @@ export const useStore = create<CanvasStoreState>((set) => ({
   })),
   removeElementEverywhere: (elementId) => set((state) => ({
     viewElements: removePlacedElement(state.viewElements, elementId),
-    libraryRefresh: state.libraryRefresh + 1,
+    allElements: state.allElements.filter((el) => el.id !== elementId),
   })),
   mergeSavedElement: (saved) => set((state) => ({
     viewElements: mergeSavedElementIntoPlacements(state.viewElements, saved),
-    libraryRefresh: state.libraryRefresh + 1,
+    allElements: state.allElements.map((el) => (el.id === saved.id ? saved : el)),
   })),
   upsertConnector: (connector) => set((state) => ({
     connectors: upsertConnectorInList(state.connectors, connector),