feat(core): add display group to collection page and tag page

toeverything · May 10, 2024 · d64baeb · d64baeb
1 parent 09a78ef
commit d64baeb
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 11 deletions.
diff --git a/packages/frontend/core/src/pages/workspace/collection/header.tsx b/packages/frontend/core/src/pages/workspace/collection/header.tsx
@@ -1,4 +1,5 @@
 import { IconButton } from '@affine/component';
+import { PageDisplayMenu } from '@affine/core/components/page-list';
 import { Header } from '@affine/core/components/pure/header';
 import { WorkspaceModeFilterTab } from '@affine/core/components/pure/workspace-mode-filter-tab';
 import { PlusIcon } from '@blocksuite/icons';
@@ -16,16 +17,19 @@ export const CollectionDetailHeader = ({
  return (
  <Header
  right={
- <IconButton
- type="default"
- icon={<PlusIcon fontSize={16} />}
- onClick={onCreate}
- className={clsx(
- styles.headerCreateNewButton,
- styles.headerCreateNewCollectionIconButton,
- !showCreateNew && styles.headerCreateNewButtonHidden
- )}
- />
+ <>
+ <IconButton
+ type="default"
+ icon={<PlusIcon fontSize={16} />}
+ onClick={onCreate}
+ className={clsx(
+ styles.headerCreateNewButton,
+ styles.headerCreateNewCollectionIconButton,
+ !showCreateNew && styles.headerCreateNewButtonHidden
+ )}
+ />
+ <PageDisplayMenu />
+ </>
  }
  center={<WorkspaceModeFilterTab activeFilter={'collections'} />}
  />

diff --git a/packages/frontend/core/src/pages/workspace/tag/header.tsx b/packages/frontend/core/src/pages/workspace/tag/header.tsx
@@ -1,6 +1,12 @@
+import { PageDisplayMenu } from '@affine/core/components/page-list';
 import { Header } from '@affine/core/components/pure/header';
 import { WorkspaceModeFilterTab } from '@affine/core/components/pure/workspace-mode-filter-tab';
 
 export const TagDetailHeader = () => {
- return <Header center={<WorkspaceModeFilterTab activeFilter={'tags'} />} />;
+ return (
+ <Header
+ center={<WorkspaceModeFilterTab activeFilter={'tags'} />}
+ right={<PageDisplayMenu />}
+ />
+ );
 };