fix: cleanup

Author: isekovanic

package/src/components/Message/hooks/useMessageActionHandlers.ts

@@ -124,10 +124,9 @@ export const useMessageActionHandlers = ({
     }
   });
 
-  const setEditingMessage = useStableCallback((messageToEdit: LocalMessage = message) => {
-    setEditingState(messageToEdit);
+  const handleEditMessage = useWithPortalKeyboardSafety(() => {
+    setEditingState(message);
   });
-  const handleEditMessage = useWithPortalKeyboardSafety(setEditingMessage);
 
   const handleFlagMessage = useStableCallback(() => {
     if (!message.id) {

package/src/hooks/useAfterKeyboardOpenCallback.ts

@@ -21,6 +21,7 @@ export const useAfterKeyboardOpenCallback = <T extends unknown[]>(
   const isKeyboardVisible = useKeyboardVisibility();
   const { inputBoxRef } = useMessageInputContext();
   const keyboardSubscriptionRef = useRef<EventSubscription | undefined>(undefined);
+  // This callback runs from a keyboard event listener, so it must stay fresh across rerenders.
   const stableCallback = useStableCallback(callback);
 
   /** Clears the pending keyboard listener, if any. */

package/src/hooks/usePortalSettledCallback.ts

@@ -61,6 +61,7 @@ const scheduleAfterFrames = (callback: () => void, frames: number, rafIds: numbe
  */
 export const usePortalSettledCallback = <T extends unknown[]>(callback: (...args: T) => void) => {
   const rafIdsRef = useRef<number[]>([]);
+  // This callback runs from deferred RAF work, so it must stay fresh across rerenders.
   const stableCallback = useStableCallback(callback);
 
   const clearScheduledFrames = useStableCallback(() => {