refactor(core): adjust graphql hook (#5339)

packages/frontend/core/src/components/affine/auth/send-email.tsx

@@ -13,11 +13,11 @@ import {
   sendSetPasswordEmailMutation,
 } from '@affine/graphql';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
-import { useMutation } from '@affine/workspace/affine/gql';
 import { useAsyncCallback } from '@toeverything/hooks/affine-async-hooks';
 import { useSetAtom } from 'jotai/react';
 import { useCallback, useState } from 'react';
 
+import { useMutation } from '../../../hooks/use-mutation';
 import type { AuthPanelProps } from './index';
 
 const useEmailTitle = (emailType: AuthPanelProps['emailType']) => {

packages/frontend/core/src/components/affine/auth/sign-in.tsx

@@ -7,7 +7,6 @@ import { Button } from '@affine/component/ui/button';
 import { type GetUserQuery, getUserQuery } from '@affine/graphql';
 import { Trans } from '@affine/i18n';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
-import { useMutation } from '@affine/workspace/affine/gql';
 import { ArrowDownBigIcon, GoogleDuotoneIcon } from '@blocksuite/icons';
 import { useAsyncCallback } from '@toeverything/hooks/affine-async-hooks';
 import { GraphQLError } from 'graphql';
@@ -15,6 +14,7 @@ import { type FC, useState } from 'react';
 import { useCallback } from 'react';
 
 import { useCurrentLoginStatus } from '../../../hooks/affine/use-current-login-status';
+import { useMutation } from '../../../hooks/use-mutation';
 import { emailRegex } from '../../../utils/email-regex';
 import type { AuthPanelProps } from './index';
 import * as style from './style.css';

packages/frontend/core/src/components/affine/auth/subscription-redirect.tsx

@@ -9,16 +9,17 @@ import {
   subscriptionQuery,
 } from '@affine/graphql';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
-import { useMutation, useQuery } from '@affine/workspace/affine/gql';
 import { useAsyncCallback } from '@toeverything/hooks/affine-async-hooks';
 import { nanoid } from 'nanoid';
 import { Suspense, useCallback, useEffect, useMemo } from 'react';
 
 import { useCurrentUser } from '../../../hooks/affine/use-current-user';
+import { useMutation } from '../../../hooks/use-mutation';
 import {
   RouteLogic,
   useNavigateHelper,
 } from '../../../hooks/use-navigate-helper';
+import { useQuery } from '../../../hooks/use-query';
 import * as styles from './subscription-redirect.css';
 import { useSubscriptionSearch } from './use-subscription';
 

packages/frontend/core/src/components/affine/page-history-modal/data.ts

@@ -9,11 +9,6 @@ import {
   createAffineCloudBlobStorage,
   globalBlockSuiteSchema,
 } from '@affine/workspace';
-import {
-  useMutateQueryResource,
-  useMutation,
-  useQueryInfinite,
-} from '@affine/workspace/affine/gql';
 import { assertEquals } from '@blocksuite/global/utils';
 import { Workspace } from '@blocksuite/store';
 import { usePageMetaHelper } from '@toeverything/hooks/use-block-suite-page-meta';
@@ -23,6 +18,12 @@ import { useMemo } from 'react';
 import useSWRImmutable from 'swr/immutable';
 import { applyUpdate } from 'yjs';
 
+import {
+  useMutateQueryResource,
+  useMutation,
+} from '../../../hooks/use-mutation';
+import { useQueryInfinite } from '../../../hooks/use-query';
+
 const logger = new DebugLogger('page-history');
 
 type DocHistory = ListHistoryQuery['workspace']['histories'][number];

packages/frontend/core/src/components/affine/setting-modal/account-setting/index.tsx

@@ -14,7 +14,6 @@ import {
   uploadAvatarMutation,
 } from '@affine/graphql';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
-import { useMutation, useQuery } from '@affine/workspace/affine/gql';
 import { ArrowRightSmallIcon, CameraIcon } from '@blocksuite/icons';
 import { useAsyncCallback } from '@toeverything/hooks/affine-async-hooks';
 import bytes from 'bytes';
@@ -35,6 +34,8 @@ import {
 } from '../../../../atoms';
 import { useCurrentUser } from '../../../../hooks/affine/use-current-user';
 import { useSelfHosted } from '../../../../hooks/affine/use-server-config';
+import { useMutation } from '../../../../hooks/use-mutation';
+import { useQuery } from '../../../../hooks/use-query';
 import { useUserSubscription } from '../../../../hooks/use-subscription';
 import { validateAndReduceImage } from '../../../../utils/reduce-image';
 import { Upload } from '../../../pure/file-upload';

packages/frontend/core/src/components/affine/setting-modal/general-setting/billing/index.tsx

@@ -20,7 +20,6 @@ import {
 } from '@affine/graphql';
 import { Trans } from '@affine/i18n';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
-import { useMutation, useQuery } from '@affine/workspace/affine/gql';
 import { ArrowRightSmallIcon } from '@blocksuite/icons';
 import { useAsyncCallback } from '@toeverything/hooks/affine-async-hooks';
 import { useSetAtom } from 'jotai';
@@ -28,6 +27,8 @@ import { Suspense, useCallback, useMemo, useState } from 'react';
 
 import { openSettingModalAtom } from '../../../../../atoms';
 import { useCurrentLoginStatus } from '../../../../../hooks/affine/use-current-login-status';
+import { useMutation } from '../../../../../hooks/use-mutation';
+import { useQuery } from '../../../../../hooks/use-query';
 import {
   type SubscriptionMutator,
   useUserSubscription,

packages/frontend/core/src/components/affine/setting-modal/general-setting/plans/actions.tsx

@@ -3,12 +3,12 @@ import {
   cancelSubscriptionMutation,
   resumeSubscriptionMutation,
 } from '@affine/graphql';
-import { useMutation } from '@affine/workspace/affine/gql';
 import { useAsyncCallback } from '@toeverything/hooks/affine-async-hooks';
 import { nanoid } from 'nanoid';
 import type { PropsWithChildren } from 'react';
 import { useState } from 'react';
 
+import { useMutation } from '../../../../../hooks/use-mutation';
 import { ConfirmLoadingModal, DowngradeModal } from './modals';
 
 /**

packages/frontend/core/src/components/affine/setting-modal/general-setting/plans/index.tsx

@@ -7,13 +7,13 @@ import {
 } from '@affine/graphql';
 import { Trans } from '@affine/i18n';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
-import { useQuery } from '@affine/workspace/affine/gql';
 import { useSetAtom } from 'jotai';
 import { Suspense, useEffect, useRef, useState } from 'react';
 import type { FallbackProps } from 'react-error-boundary';
 
 import { SWRErrorBoundary } from '../../../../../components/pure/swr-error-bundary';
 import { useCurrentLoginStatus } from '../../../../../hooks/affine/use-current-login-status';
+import { useQuery } from '../../../../../hooks/use-query';
 import { useUserSubscription } from '../../../../../hooks/use-subscription';
 import { PlanLayout } from './layout';
 import { type FixedPrice, getPlanDetail, PlanCard } from './plan-card';

packages/frontend/core/src/components/affine/setting-modal/general-setting/plans/plan-card.tsx

@@ -13,7 +13,6 @@ import {
 } from '@affine/graphql';
 import { Trans } from '@affine/i18n';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
-import { useMutation } from '@affine/workspace/affine/gql';
 import { DoneIcon } from '@blocksuite/icons';
 import { useAsyncCallback } from '@toeverything/hooks/affine-async-hooks';
 import { useSetAtom } from 'jotai';
@@ -31,6 +30,7 @@ import {
 import { openPaymentDisableAtom } from '../../../../../atoms';
 import { authAtom } from '../../../../../atoms/index';
 import { useCurrentLoginStatus } from '../../../../../hooks/affine/use-current-login-status';
+import { useMutation } from '../../../../../hooks/use-mutation';
 import { CancelAction, ResumeAction } from './actions';
 import { BulledListIcon } from './icons/bulled-list';
 import { ConfirmLoadingModal } from './modals';