feat(mobile): use cached random height for all docs masonry

toeverything · Nov 7, 2024 · 9a6d473 · 9a6d473
1 parent bff5508
commit 9a6d473
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 19 deletions.
diff --git a/packages/frontend/core/src/mobile/components/doc-card/index.tsx b/packages/frontend/core/src/mobile/components/doc-card/index.tsx
@@ -21,10 +21,18 @@ export interface DocCardProps extends Omit<WorkbenchLinkProps, 'to'> {
     title?: ReactNode;
   } & { [key: string]: any };
   showTags?: boolean;
+
+  /**
+   * If provided, content's height will be calculated: `previewRows * line-height`
+   */
+  previewRows?: number;
 }
 
 export const DocCard = forwardRef<HTMLAnchorElement, DocCardProps>(
-  function DocCard({ showTags = true, meta, className, ...attrs }, ref) {
+  function DocCard(
+    { showTags = true, meta, className, previewRows, ...attrs },
+    ref
+  ) {
     const favAdapter = useService(CompatibleFavoriteItemsAdapter);
     const workspace = useService(WorkspaceService).workspace;
 
@@ -38,6 +46,8 @@ export const DocCard = forwardRef<HTMLAnchorElement, DocCardProps>(
       [favAdapter, meta.id]
     );
 
+    const contentHeight = previewRows ? `${previewRows * 18}px` : 'unset';
+
     return (
       <WorkbenchLink
         to={`/${meta.id}`}
@@ -57,7 +67,10 @@ export const DocCard = forwardRef<HTMLAnchorElement, DocCardProps>(
             }
           />
         </header>
-        <main className={styles.content}>
+        <main
+          className={styles.content}
+          style={{ height: contentHeight, flex: contentHeight ? 'unset' : 1 }}
+        >
           <PagePreview
             docCollection={workspace.docCollection}
             pageId={meta.id}

diff --git a/packages/frontend/core/src/mobile/components/doc-card/styles.css.tsx b/packages/frontend/core/src/mobile/components/doc-card/styles.css.tsx
@@ -43,7 +43,6 @@ export const untitled = style({
 export const content = style([
   footnoteRegular,
   {
-    flex: 1,
     overflow: 'hidden',
   },
 ]);

diff --git a/packages/frontend/core/src/mobile/views/all-docs/doc/masonry.css.ts b/packages/frontend/core/src/mobile/views/all-docs/doc/masonry.css.ts
@@ -8,9 +8,6 @@ export const masonry = style({
   columnGap: columnGap,
 });
 export const masonryItem = style({
-  display: 'grid',
-  gridTemplateRows: '1fr auto',
   breakInside: 'avoid',
   marginBottom: 10,
 });
-export const space = style({ width: '100%', height: 16 });
diff --git a/packages/frontend/core/src/mobile/views/all-docs/doc/masonry.tsx b/packages/frontend/core/src/mobile/views/all-docs/doc/masonry.tsx
@@ -1,41 +1,72 @@
 import { useGlobalEvent } from '@affine/core/mobile/hooks/use-global-events';
 import type { DocMeta } from '@blocksuite/affine/store';
-import { useCallback, useEffect, useState } from 'react';
+import type { FrameworkProvider } from '@toeverything/infra';
+import { GlobalCacheService, useFramework } from '@toeverything/infra';
+import { useCallback, useMemo, useState } from 'react';
 
 import { DocCard } from '../../../components';
 import * as styles from './masonry.css';
 
+const calcColumnCount = () => {
+  const maxCardWidth = 220;
+  const windowWidth = window.innerWidth;
+  const newColumnCount = Math.floor(
+    (windowWidth - styles.paddingX * 2 - styles.columnGap) / maxCardWidth
+  );
+  return Math.max(newColumnCount, 2);
+};
+
+/**
+ * Randomly assign preview rows to each item and cache it
+ */
+const getRowsCache = (framework: FrameworkProvider, items: DocMeta[]) => {
+  const ROWS = [3, 4, 5, 6, 7];
+  const CACHE_KEY = 'masonryPreviewRows';
+
+  const globalCache = framework.get(GlobalCacheService).globalCache;
+  const cache = globalCache.get<Record<string, number>>(CACHE_KEY) || {};
+  const newCache = items.reduce(
+    (acc, item) => {
+      if (cache[item.id]) return acc;
+      const row = ROWS[Math.floor(Math.random() * ROWS.length)];
+      acc[item.id] = row;
+      return acc;
+    },
+    { ...cache }
+  );
+  globalCache.set(CACHE_KEY, newCache);
+  return newCache;
+};
+
 export const MasonryDocs = ({
   items,
   showTags,
 }: {
   items: DocMeta[];
   showTags?: boolean;
 }) => {
-  const [columnCount, setColumnCount] = useState(2);
+  const [columnCount, setColumnCount] = useState(calcColumnCount());
+  const framework = useFramework();
 
   const updateColumnCount = useCallback(() => {
-    const maxCardWidth = 220;
-    const windowWidth = window.innerWidth;
-    const newColumnCount = Math.floor(
-      (windowWidth - styles.paddingX * 2 - styles.columnGap) / maxCardWidth
-    );
-    setColumnCount(Math.max(newColumnCount, 2));
+    setColumnCount(calcColumnCount());
   }, []);
-
   useGlobalEvent('resize', updateColumnCount);
-  useEffect(() => {
-    updateColumnCount();
-  }, [updateColumnCount]);
+
+  const previewRowsMap = useMemo(
+    () => getRowsCache(framework, items),
+    [framework, items]
+  );
 
   return (
     <div className={styles.masonry} style={{ columnCount }}>
       {items.map(item => (
         <DocCard
           key={item.id}
+          className={styles.masonryItem}
+          previewRows={previewRowsMap[item.id]}
           showTags={showTags}
           meta={item}
-          className={styles.masonryItem}
         />
       ))}
     </div>