feat: searching operation set by short code

Author: guansss

src/apis/operation-set.ts

@@ -1,4 +1,5 @@
 import { useAtomValue } from 'jotai'
+import { noop } from 'lodash-es'
 import {
   CopilotSetPageRes,
   CopilotSetQuery,
@@ -11,6 +12,8 @@ import { authAtom } from 'store/auth'
 import { OperationSetApi } from 'utils/maa-copilot-client'
 import { useSWRRefresh } from 'utils/swr'
 
+import { parseShortCode } from '../models/shortCode'
+
 export type OrderBy = 'views' | 'hot' | 'id'
 
 export interface UseOperationSetsParams {
@@ -81,7 +84,65 @@ export function useOperationSets({
 
 export function useRefreshOperationSets() {
   const refresh = useSWRRefresh()
-  return () => refresh((key) => key.includes('operationSets'))
+  return () =>
+    refresh(
+      (key) =>
+        key.includes('operationSets') ||
+        (key.includes('operationSet') && key.includes('fromList')),
+    )
+}
+
+export function useOperationSetSearch({
+  keyword,
+  suspense,
+  disabled,
+  ...params
+}: UseOperationSetsParams) {
+  if (!suspense) {
+    throw new Error('useOperationSetSearch must be used with suspense')
+  }
+  if (disabled) {
+    throw new Error('useOperationSetSearch cannot be disabled')
+  }
+
+  let id: number | undefined
+
+  if (keyword) {
+    const shortCodeContent = parseShortCode(keyword)
+
+    if (shortCodeContent) {
+      if (shortCodeContent.type === 'operation') {
+        throw new Error('该神秘代码属于作业，无法在此使用⊙﹏⊙∥')
+      }
+
+      id = shortCodeContent.id
+    }
+  }
+
+  const { data: operationSet } = useOperationSet({ id, suspense })
+
+  const listResponse = useOperationSets({
+    keyword,
+    suspense,
+    ...params,
+
+    // disable the list query if we are fetching a single operation set
+    disabled: !!id,
+  })
+
+  if (id) {
+    return {
+      operationSets: [operationSet],
+      isReachingEnd: true,
+      setSize: noop,
+
+      // these are fixed values in suspense mode
+      error: undefined,
+      isValidating: false,
+    }
+  }
+
+  return listResponse
 }
 
 interface UseOperationSetParams {

src/components/OperationSetList.tsx

@@ -1,6 +1,9 @@
 import { Button, NonIdealState } from '@blueprintjs/core'
 
-import { UseOperationSetsParams, useOperationSets } from 'apis/operation-set'
+import {
+  UseOperationSetsParams,
+  useOperationSetSearch,
+} from 'apis/operation-set'
 import { useAtomValue } from 'jotai'
 import { ComponentType, ReactNode } from 'react'
 
@@ -15,7 +18,7 @@ export const OperationSetList: ComponentType<UseOperationSetsParams> =
       const neoLayout = useAtomValue(neoLayoutAtom)
 
       const { operationSets, setSize, isValidating, isReachingEnd } =
-        useOperationSets({ ...props, suspense: true })
+        useOperationSetSearch({ ...props, suspense: true })
 
       // make TS happy: we got Suspense out there
       if (!operationSets) throw new Error('unreachable')