fix: reset auto save data when saved

Signed-off-by: Innei <i@innei.in>

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

@@ -1,3 +1,5 @@
+'use client'
+
 import { createContext, useContext, useEffect, useMemo, useState } from 'react'
 import { useForceUpdate } from 'framer-motion'
 import { produce } from 'immer'

src/queries/definition/note.ts

@@ -4,6 +4,7 @@ import { revalidateTag } from 'next/cache'
 import type { NoteModel, NoteWrappedPayload } from '@mx-space/api-client'
 import type { NoteDto } from '~/models/writing'
 
+import { useResetAutoSaverData } from '~/components/modules/dashboard/writing/BaseWritingProvider'
 import { cloneDeep } from '~/lib/_'
 import { apiClient } from '~/lib/request'
 import { routeBuilder, Routes } from '~/lib/route-builder'
@@ -73,8 +74,9 @@ export const noteAdmin = {
     }),
 }
 
-export const useCreateNote = () =>
-  useMutation({
+export const useCreateNote = () => {
+  const resetAutoSaver = useResetAutoSaverData()
+  return useMutation({
     mutationFn: (data: NoteDto) => {
       const readonlyKeys = [
         'id',
@@ -95,11 +97,14 @@ export const useCreateNote = () =>
     onSuccess: () => {
       revalidateTag('note')
       toast.success('创建成功')
+      resetAutoSaver('note')
     },
   })
+}
 
-export const useUpdateNote = () =>
-  useMutation({
+export const useUpdateNote = () => {
+  const resetAutoSaver = useResetAutoSaverData()
+  return useMutation({
     mutationFn: (data: NoteDto) => {
       const { id } = data
       const readonlyKeys = [
@@ -118,7 +123,9 @@ export const useUpdateNote = () =>
         data: nextData,
       })
     },
-    onSuccess: () => {
+    onSuccess: ({ id }) => {
       toast.success('更新成功')
+      resetAutoSaver('note', id)
     },
   })
+}

src/queries/definition/post.ts

@@ -2,6 +2,7 @@ import { useMutation } from '@tanstack/react-query'
 import type { PaginateResult, PostModel, TagModel } from '@mx-space/api-client'
 import type { PostDto } from '~/models/writing'
 
+import { useResetAutoSaverData } from '~/components/modules/dashboard/writing/BaseWritingProvider'
 import { cloneDeep } from '~/lib/_'
 import { apiClient } from '~/lib/request'
 import { routeBuilder, Routes } from '~/lib/route-builder'
@@ -102,8 +103,9 @@ export const postAdmin = {
     }),
 }
 
-export const useCreatePost = () =>
-  useMutation({
+export const useCreatePost = () => {
+  const resetAutoSaver = useResetAutoSaverData()
+  return useMutation({
     mutationFn: (data: PostDto) => {
       const readonlyKeys = [
         'id',
@@ -123,11 +125,14 @@ export const useCreatePost = () =>
     },
     onSuccess: () => {
       toast.success('创建成功')
+      resetAutoSaver('post')
     },
   })
+}
 
-export const useUpdatePost = () =>
-  useMutation({
+export const useUpdatePost = () => {
+  const resetAutoSaver = useResetAutoSaverData()
+  return useMutation({
     mutationFn: (data: PostDto) => {
       const { id } = data
       const readonlyKeys = [
@@ -146,7 +151,9 @@ export const useUpdatePost = () =>
         data: nextData,
       })
     },
-    onSuccess: () => {
+    onSuccess: ({ id }) => {
       toast.success('更新成功')
+      resetAutoSaver('post', id)
     },
   })
+}