feat(component): improve fallback skeleton (#2323)

apps/web/src/components/pure/loading/page-loading.tsx

@@ -1,6 +1,5 @@
 import { styled } from '@affine/component';
 import { AffineLoading } from '@affine/component/affine-loading';
-import { useAFFiNEI18N } from '@affine/i18n/hooks';
 import { memo, Suspense } from 'react';
 
 export const Loading = memo(function Loading() {
@@ -35,12 +34,15 @@ const StyledLoadingContainer = styled('div')(() => {
   };
 });
 
-export const PageLoading = ({ text }: { text?: string }) => {
-  const t = useAFFiNEI18N();
+export const PageLoading = () => {
+  // We disable the loading on desktop, because want it looks faster.
+  //  This is a design requirement.
+  if (environment.isDesktop) {
+    return null;
+  }
   return (
     <StyledLoadingContainer>
       <Loading />
-      <h1>{text ? text : t['Loading']()}</h1>
     </StyledLoadingContainer>
   );
 };

apps/web/src/layouts/workspace-layout.tsx

@@ -2,12 +2,12 @@ import {
   AppContainer,
   MainContainer,
   ToolContainer,
+  WorkspaceFallback,
 } from '@affine/component/workspace';
 import { DebugLogger } from '@affine/debug';
 import { DEFAULT_HELLO_WORLD_PAGE_ID } from '@affine/env';
 import { initPage } from '@affine/env/blocksuite';
 import { setUpLanguage, useI18N } from '@affine/i18n';
-import { useAFFiNEI18N } from '@affine/i18n/hooks';
 import { createAffineGlobalChannel } from '@affine/workspace/affine/sync';
 import {
   rootCurrentPageIdAtom,
@@ -31,7 +31,6 @@ import {
   publicWorkspaceIdAtom,
 } from '../atoms/public-workspace';
 import { HelpIsland } from '../components/pure/help-island';
-import { PageLoading } from '../components/pure/loading';
 import { RootAppSidebar } from '../components/root-app-sidebar';
 import { useCurrentWorkspace } from '../hooks/current/use-current-workspace';
 import { useRouterHelper } from '../hooks/use-router-helper';
@@ -141,15 +140,14 @@ export const CurrentWorkspaceContext = ({
   useRouterWithWorkspaceIdDefense(router);
   const metadata = useAtomValue(rootWorkspacesMetadataAtom);
   const exist = metadata.find(m => m.id === workspaceId);
-  const t = useAFFiNEI18N();
   if (!router.isReady) {
-    return <PageLoading text={t['Router is Loading']()} />;
+    return <WorkspaceFallback key="router-is-loading" />;
   }
   if (!workspaceId) {
-    return <PageLoading text={t['Finding Workspace ID']()} />;
+    return <WorkspaceFallback key="finding-workspace-id" />;
   }
   if (!exist) {
-    return <PageLoading text={t['Workspace Not Found']()} />;
+    return <WorkspaceFallback key="workspace-not-found" />;
   }
   return <>{children}</>;
 };
@@ -157,7 +155,6 @@ export const CurrentWorkspaceContext = ({
 export const WorkspaceLayout: FC<PropsWithChildren> =
   function WorkspacesSuspense({ children }) {
     const i18n = useI18N();
-    const t = useAFFiNEI18N();
     useEffect(() => {
       document.documentElement.lang = i18n.language;
       // todo(himself65): this is a hack, we should use a better way to set the language
@@ -233,9 +230,7 @@ export const WorkspaceLayout: FC<PropsWithChildren> =
           </CurrentWorkspaceContext>
         </AllWorkspaceContext>
         <CurrentWorkspaceContext>
-          <Suspense
-            fallback={<PageLoading text={t['Finding Current Workspace']()} />}
-          >
+          <Suspense fallback={<WorkspaceFallback />}>
             <Provider>
               <WorkspaceLayoutInner>{children}</WorkspaceLayoutInner>
             </Provider>
@@ -251,7 +246,6 @@ export const WorkspaceLayoutInner: FC<PropsWithChildren> = ({ children }) => {
   const currentPageId = useAtomValue(rootCurrentPageIdAtom);
   const router = useRouter();
   const { jumpToPage } = useRouterHelper(router);
-  const t = useAFFiNEI18N();
 
   //#region init workspace
   if (currentWorkspace.blockSuiteWorkspace.isEmpty) {
@@ -360,9 +354,7 @@ export const WorkspaceLayoutInner: FC<PropsWithChildren> = ({ children }) => {
           paths={isPublicWorkspace ? publicPathGenerator : pathGenerator}
         />
         <MainContainer>
-          <Suspense fallback={<PageLoading text={t['Page is Loading']()} />}>
-            {children}
-          </Suspense>
+          <Suspense fallback={<WorkspaceFallback />}>{children}</Suspense>
           <ToolContainer>
             {/* fixme(himself65): remove this */}
             <div id="toolWrapper" style={{ marginBottom: '12px' }}>

apps/web/src/pages/_app.tsx

@@ -1,6 +1,7 @@
 import '@affine/component/theme/global.css';
 import '@affine/component/theme/theme.css';
 
+import { WorkspaceFallback } from '@affine/component/workspace';
 import { config, setupGlobal } from '@affine/env';
 import { createI18n, I18nextProvider } from '@affine/i18n';
 import { rootStore } from '@affine/workspace/atom';
@@ -15,7 +16,6 @@ import React, { lazy, Suspense, useEffect, useMemo } from 'react';
 
 import { AffineErrorBoundary } from '../components/affine/affine-error-eoundary';
 import { ProviderComposer } from '../components/provider-composer';
-import { PageLoading } from '../components/pure/loading';
 import { MessageCenter } from '../components/pure/message-center';
 import { ThemeProvider } from '../providers/theme-provider';
 import type { NextPageWithLayout } from '../shared';
@@ -66,7 +66,7 @@ const App = function App({
       <I18nextProvider i18n={i18n}>
         <MessageCenter />
         <AffineErrorBoundary router={useRouter()}>
-          <Suspense fallback={<PageLoading key="RootPageLoading" />}>
+          <Suspense fallback={<WorkspaceFallback key="RootPageLoading" />}>
             <ProviderComposer
               contexts={useMemo(
                 () =>

apps/web/src/pages/index.tsx

@@ -1,5 +1,4 @@
 import { DebugLogger } from '@affine/debug';
-import { useAFFiNEI18N } from '@affine/i18n/hooks';
 import type { NextPage } from 'next';
 import { useRouter } from 'next/router';
 import { Suspense, useEffect } from 'react';
@@ -63,9 +62,8 @@ const IndexPageInner = () => {
 };
 
 const IndexPage: NextPage = () => {
-  const t = useAFFiNEI18N();
   return (
-    <Suspense fallback={<PageLoading text={t['Loading All Workspaces']()} />}>
+    <Suspense fallback={<PageLoading />}>
       <IndexPageInner />
     </Suspense>
   );

apps/web/src/plugins/affine/index.tsx

@@ -84,9 +84,7 @@ function AuthContext({ children }: PropsWithChildren): ReactElement {
     }
   }, [login]);
   if (!login) {
-    return (
-      <PageLoading text="No login, redirecting to local workspace page..." />
-    );
+    return <PageLoading />;
   }
   return <>{children}</>;
 }
@@ -278,7 +276,7 @@ export const AffinePlugin: WorkspacePlugin<WorkspaceFlavour.AFFINE> = {
   UI: {
     Provider: ({ children }) => {
       return (
-        <Suspense fallback={<PageLoading text="Checking login status..." />}>
+        <Suspense fallback={<PageLoading />}>
           <AuthContext>{children}</AuthContext>
         </Suspense>
       );