Can I add custom data to EditorStateHistory?

[yangheecho]

I want to extend the EditorStateHistory class to store custom data for my background removal feature. Specifically, I need to save both the original image and the background-removed image in the state history.

[hm21]

It's not possible to directly extend EditorStateHistory unless you fork the entire project.

But you can handle it directly in your code. Simply add an empty history editorKey.currentState!.addHistory(). However, you'll also need to specify in your code at which history index the background changed. To get the active history index, use editorKey.currentState!.stateManager.historyPointer. The final step is to listen for undo/redo actions and apply the correct background image based on your state.

 callbacks: ProImageEditorCallbacks(
  mainEditorCallbacks: MainEditorCallbacks(
    onUndo: () {
      editorKey.currentState!.updateBackgroundImage(image);
    },
    onRedo: () {
      editorKey.currentState!.updateBackgroundImage(image);
    },
  ),
),