feat(test-studio): add custom restore document action

dev/test-studio/documentActions/actions/TestCustomRestoreAction.tsx

@@ -0,0 +1,16 @@
+import {RocketIcon} from '@sanity/icons'
+import {type DocumentActionComponent} from 'sanity'
+
+export const TestCustomRestoreAction: (
+  action: DocumentActionComponent,
+) => DocumentActionComponent = (restoreAction) => {
+  const action: DocumentActionComponent = (props) => ({
+    ...restoreAction(props),
+    label: 'Custom restore',
+    tone: 'positive',
+    icon: RocketIcon,
+  })
+
+  action.action = 'restore'
+  return action
+}

dev/test-studio/documentActions/index.ts

@@ -2,6 +2,7 @@ import {type DocumentActionsResolver} from 'sanity'
 
 import {TestConfirmDialogAction} from './actions/TestConfirmDialogAction'
 import {TestCustomComponentAction} from './actions/TestCustomComponentAction'
+import {TestCustomRestoreAction} from './actions/TestCustomRestoreAction'
 import {TestModalDialogAction} from './actions/TestModalDialogAction'
 import {TestPopoverDialogAction} from './actions/TestPopoverDialogAction'
 
@@ -13,7 +14,13 @@ export const resolveDocumentActions: DocumentActionsResolver = (prev, {schemaTyp
       TestPopoverDialogAction,
       TestCustomComponentAction,
       ...prev,
-    ]
+    ].map((action) => {
+      if (action.action === 'restore') {
+        return TestCustomRestoreAction(action)
+      }
+
+      return action
+    })
   }
 
   return prev