refactor(infra): directory structure (#4615)

tests/storybook/src/stories/share-menu.stories.tsx

@@ -0,0 +1,149 @@
+import { toast } from '@affine/component';
+import {
+  PublicLinkDisableModal,
+  StyledDisableButton,
+} from '@affine/component/share-menu';
+import { ShareMenu } from '@affine/component/share-menu';
+import type {
+  AffineCloudWorkspace,
+  LocalWorkspace,
+} from '@affine/env/workspace';
+import { WorkspaceFlavour } from '@affine/env/workspace';
+import { getOrCreateWorkspace } from '@affine/workspace/manager';
+import type { Page } from '@blocksuite/store';
+import { expect } from '@storybook/jest';
+import type { Meta, StoryFn } from '@storybook/react';
+import { use } from 'foxact/use';
+import { useState } from 'react';
+
+export default {
+  title: 'AFFiNE/ShareMenu',
+  component: ShareMenu,
+  parameters: {
+    chromatic: { disableSnapshot: true },
+  },
+} satisfies Meta;
+
+const sharePageMap = new Map<string, boolean>([]);
+// todo: use a real hook
+const useIsSharedPage = (
+  _workspaceId: string,
+  pageId: string
+): [isSharePage: boolean, setIsSharePage: (enable: boolean) => void] => {
+  const [isShared, setIsShared] = useState(sharePageMap.get(pageId) ?? false);
+  const togglePagePublic = (enable: boolean) => {
+    setIsShared(enable);
+    sharePageMap.set(pageId, enable);
+  };
+  return [isShared, togglePagePublic];
+};
+
+async function initPage(page: Page) {
+  await page.waitForLoaded();
+  // Add page block and surface block at root level
+  const pageBlockId = page.addBlock('affine:page', {
+    title: new page.Text('Hello, world!'),
+  });
+  page.addBlock('affine:surface', {}, pageBlockId);
+  const frameId = page.addBlock('affine:note', {}, pageBlockId);
+  page.addBlock(
+    'affine:paragraph',
+    {
+      text: new page.Text('This is a paragraph.'),
+    },
+    frameId
+  );
+  page.resetHistory();
+}
+
+const blockSuiteWorkspace = getOrCreateWorkspace(
+  'test-workspace',
+  WorkspaceFlavour.LOCAL
+);
+
+const promise = Promise.all([
+  initPage(blockSuiteWorkspace.createPage({ id: 'page0' })),
+  initPage(blockSuiteWorkspace.createPage({ id: 'page1' })),
+  initPage(blockSuiteWorkspace.createPage({ id: 'page2' })),
+]);
+
+const localWorkspace: LocalWorkspace = {
+  id: 'test-workspace',
+  flavour: WorkspaceFlavour.LOCAL,
+  blockSuiteWorkspace,
+};
+
+const affineWorkspace: AffineCloudWorkspace = {
+  id: 'test-workspace',
+  flavour: WorkspaceFlavour.AFFINE_CLOUD,
+  blockSuiteWorkspace,
+};
+
+async function unimplemented() {
+  toast('work in progress');
+}
+
+export const Basic: StoryFn = () => {
+  use(promise);
+  return (
+    <ShareMenu
+      currentPage={blockSuiteWorkspace.getPage('page0') as Page}
+      useIsSharedPage={useIsSharedPage}
+      workspace={localWorkspace}
+      onEnableAffineCloud={unimplemented}
+      togglePagePublic={unimplemented}
+      exportHandler={unimplemented}
+    />
+  );
+};
+
+Basic.play = async ({ canvasElement }) => {
+  {
+    const button = canvasElement.querySelector(
+      '[data-testid="share-menu-button"]'
+    ) as HTMLButtonElement;
+    expect(button).not.toBeNull();
+    button.click();
+  }
+  await new Promise(resolve => window.setTimeout(resolve, 100));
+  {
+    const button = canvasElement.querySelector(
+      '[data-testid="share-menu-enable-affine-cloud-button"]'
+    );
+    expect(button).not.toBeNull();
+  }
+};
+
+export const AffineBasic: StoryFn = () => {
+  use(promise);
+  return (
+    <ShareMenu
+      currentPage={blockSuiteWorkspace.getPage('page0') as Page}
+      useIsSharedPage={useIsSharedPage}
+      workspace={affineWorkspace}
+      onEnableAffineCloud={unimplemented}
+      togglePagePublic={unimplemented}
+      exportHandler={unimplemented}
+    />
+  );
+};
+
+export const DisableModal: StoryFn = () => {
+  const [open, setOpen] = useState(false);
+  use(promise);
+  return (
+    <>
+      <StyledDisableButton onClick={() => setOpen(!open)}>
+        Disable Public Link
+      </StyledDisableButton>
+      <PublicLinkDisableModal
+        open={open}
+        onConfirm={() => {
+          toast('Disabled');
+          setOpen(false);
+        }}
+        onOpenChange={setOpen}
+      />
+    </>
+  );
+};