fix: force update

Signed-off-by: Innei <[email protected]>

src/components/modules/dashboard/writing/BaseWritingProvider.tsx

@@ -1,7 +1,6 @@
 'use client'
 
 import { createContext, useContext, useEffect, useMemo, useState } from 'react'
-import { useForceUpdate } from 'framer-motion'
 import { produce } from 'immer'
 import { atom, useAtom } from 'jotai'
 import type { WriteEditEvent } from '~/events'
@@ -12,6 +11,7 @@ import { StyledButton } from '~/components/ui/button'
 import { useModalStack } from '~/components/ui/modal'
 import { EmitKeyMap } from '~/constants/keys'
 import { useBeforeUnload } from '~/hooks/common/use-before-unload'
+import { useForceUpdate } from '~/hooks/common/use-force-update'
 import { throttle } from '~/lib/lodash'
 import { buildNSKey } from '~/lib/ns'
 

src/components/modules/note/NoteActionAside.tsx

@@ -1,11 +1,12 @@
 'use client'
 
-import { m, useAnimationControls, useForceUpdate } from 'framer-motion'
+import { m, useAnimationControls } from 'framer-motion'
 
 import { useIsMobile } from '~/atoms/hooks'
 import { MotionButtonBase } from '~/components/ui/button'
 import { useModalStack } from '~/components/ui/modal'
 import { NumberSmoothTransition } from '~/components/ui/number-transition/NumberSmoothTransition'
+import { useForceUpdate } from '~/hooks/common/use-force-update'
 import { useIsClient } from '~/hooks/common/use-is-client'
 import { isLikedBefore, setLikeId } from '~/lib/cookie'
 import { clsxm } from '~/lib/helper'

src/components/modules/post/PostActionAside.tsx

@@ -1,12 +1,13 @@
 'use client'
 
-import { m, useAnimationControls, useForceUpdate } from 'framer-motion'
+import { m, useAnimationControls } from 'framer-motion'
 
 import { useIsMobile } from '~/atoms/hooks'
 import { ThumbsupIcon } from '~/components/icons/thumbs-up'
 import { MotionButtonBase } from '~/components/ui/button'
 import { useModalStack } from '~/components/ui/modal'
 import { NumberSmoothTransition } from '~/components/ui/number-transition/NumberSmoothTransition'
+import { useForceUpdate } from '~/hooks/common/use-force-update'
 import { useIsClient } from '~/hooks/common/use-is-client'
 import { isLikedBefore, setLikeId } from '~/lib/cookie'
 import { clsxm } from '~/lib/helper'

src/components/modules/toc/TocAside.tsx

@@ -8,9 +8,9 @@ import React, {
   useMemo,
   useRef,
 } from 'react'
-import { useForceUpdate } from 'framer-motion'
 
 import { DOMCustomEvents } from '~/constants/event'
+import { useForceUpdate } from '~/hooks/common/use-force-update'
 import { clsxm } from '~/lib/helper'
 import { throttle } from '~/lib/lodash'
 import { useWrappedElement } from '~/providers/shared/WrappedElementProvider'

src/components/ui/editor/Milkdown/plugins/Excalidraw.tsx

@@ -1,6 +1,5 @@
 import { useNodeViewContext } from '@prosemirror-adapter/react'
 import { lazy, Suspense, useEffect, useMemo, useRef, useState } from 'react'
-import { useForceUpdate } from 'framer-motion'
 import { useAtom } from 'jotai'
 import { atomWithStorage } from 'jotai/utils'
 import { diff } from 'jsondiffpatch'
@@ -26,6 +25,7 @@ import { BlockLoading } from '~/components/modules/shared/BlockLoading'
 import { StyledButton } from '~/components/ui/button'
 import { CheckBoxLabel } from '~/components/ui/checkbox'
 import { useModalStack } from '~/components/ui/modal'
+import { useForceUpdate } from '~/hooks/common/use-force-update'
 import { safeJsonParse } from '~/lib/helper'
 import { buildNSKey } from '~/lib/ns'
 import { toast } from '~/lib/toast'

src/hooks/biz/use-refetch-data.ts

@@ -1,7 +1,7 @@
 import { useEffect } from 'react'
-import { useForceUpdate } from 'framer-motion'
 
 import { EmitKeyMap } from '~/constants/keys'
+import { useForceUpdate } from '~/hooks/common/use-force-update'
 
 export const useRefetchData = (refetchFn: () => Promise<any>) => {
   const [forceUpdate, key] = useForceUpdate()

src/hooks/common/use-force-update.ts

@@ -0,0 +1,7 @@
+import { useCallback, useState } from 'react'
+
+export const useForceUpdate = () => {
+  const [updated, forceUpdate] = useState(0)
+
+  return [useCallback(() => forceUpdate((v) => ++v), []), updated] as const
+}