@lobehub/chat 1.27.3 → 1.28.0

package/src/store/chat/slices/share/action.test.ts

@@ -6,208 +6,17 @@ import { useChatStore } from '@/store/chat';
 import { messageMapKey } from '@/store/chat/utils/messageMapKey';
 import { ChatMessage } from '@/types/message';
 
-describe('shareSlice actions', () => {
-  let shareServiceSpy: any;
-  let windowOpenSpy;
-
-  beforeEach(() => {
-    shareServiceSpy = vi.spyOn(shareService, 'createShareGPTUrl').mockResolvedValue('test-url');
-    windowOpenSpy = vi.spyOn(window, 'open');
-  });
-
-  afterEach(() => {
-    vi.restoreAllMocks();
-  });
-
-  describe('shareToShareGPT', () => {
-    it('should share to ShareGPT and open a new window', async () => {
-      const { result } = renderHook(() => useChatStore());
-      const shareServiceSpy = vi.spyOn(shareService, 'createShareGPTUrl');
-      const windowOpenSpy = vi.spyOn(window, 'open');
-      const avatar = 'avatar-url';
-      const withPluginInfo = true;
-      const withSystemRole = true;
-
-      await act(async () => {
-        await result.current.shareToShareGPT({ avatar, withPluginInfo, withSystemRole });
-      });
-
-      expect(shareServiceSpy).toHaveBeenCalled();
-      expect(windowOpenSpy).toHaveBeenCalled();
-    });
-    it('should handle messages from different roles correctly', async () => {
-      // 注意：此处需要你根据实际情况模拟 chatSelectors.currentChats 和 agentSelectors 返回的数据
-      // 此外，你可能需要调整 useChatStore 的引用路径
-      const { result } = renderHook(() => useChatStore());
-      await act(async () => {
-        await result.current.shareToShareGPT({
-          withPluginInfo: true,
-          withSystemRole: true,
-        });
-      });
-      // 根据你的逻辑添加对应的expect断言
-    });
-
-    it('should not include system role information when withSystemRole is false or systemRole is undefined', async () => {
-      // 模拟不同的配置以测试这个行为
-      // 你可能需要调整 useChatStore 的引用路径
-      const { result } = renderHook(() => useChatStore());
-      await act(async () => {
-        await result.current.shareToShareGPT({
-          withSystemRole: false,
-        });
-      });
-      // 根据你的逻辑添加对应的expect断言
-    });
-
-    it('should use default avatar URL when avatar is not provided', async () => {
-      const { result } = renderHook(() => useChatStore());
-      await act(async () => {
-        await result.current.shareToShareGPT({});
-      });
-
-      expect(shareServiceSpy).toHaveBeenCalledWith(
-        expect.objectContaining({
-          avatarUrl: DEFAULT_USER_AVATAR_URL,
-        }),
-      );
-    });
-
-    it('should set shareLoading to true before sharing and to false after sharing', async () => {
-      const { result } = renderHook(() => useChatStore());
-      expect(result.current.shareLoading).toBe(false);
-      await act(async () => {
-        await result.current.shareToShareGPT({});
-      });
-      expect(result.current.shareLoading).toBe(false);
-      // 注意：这里的验证可能需要你根据实际的状态管理逻辑进行调整
-    });
-
-    it('should include plugin information when withPluginInfo is true', async () => {
-      // 模拟带有插件信息的消息
-      const pluginMessage = {
-        role: 'tool',
-        content: 'plugin content',
-        plugin: {
-          type: 'default',
-          arguments: '{}',
-          apiName: 'test-api',
-          identifier: 'test-identifier',
-        },
-        id: 'abc',
-      } as ChatMessage;
-
-      act(() => {
-        useChatStore.setState({
-          messagesMap: {
-            [messageMapKey('abc')]: [pluginMessage],
-          },
-          activeId: 'abc',
-        });
-      });
-
-      const { result } = renderHook(() => useChatStore());
-      await act(async () => {
-        result.current.shareToShareGPT({ withPluginInfo: true });
-      });
-      expect(shareServiceSpy).toHaveBeenCalledWith(
-        expect.objectContaining({
-          items: expect.arrayContaining([
-            expect.objectContaining({
-              from: 'gpt',
-              value: expect.stringContaining('Function Calling Plugin'),
-            }),
-          ]),
-        }),
-      );
-    });
-
-    it('should not include plugin information when withPluginInfo is false', async () => {
-      const pluginMessage = {
-        role: 'tool',
-        content: 'plugin content',
-        plugin: {
-          type: 'default',
-          arguments: '{}',
-          apiName: 'test-api',
-          identifier: 'test-identifier',
-        },
-        id: 'abc',
-      } as ChatMessage;
-
-      act(() => {
-        useChatStore.setState({
-          messagesMap: {
-            [messageMapKey('abc')]: [pluginMessage],
-          },
-          activeId: 'abc',
-        });
-      });
-
-      const { result } = renderHook(() => useChatStore());
-      await act(async () => {
-        result.current.shareToShareGPT({ withPluginInfo: false });
-      });
-      expect(shareServiceSpy).toHaveBeenCalledWith(
-        expect.objectContaining({
-          items: expect.not.arrayContaining([
-            expect.objectContaining({
-              from: 'gpt',
-              value: expect.stringContaining('Function Calling Plugin'),
-            }),
-          ]),
-        }),
-      );
-    });
-
-    it('should handle messages from different roles correctly', async () => {
-      const messages = [
-        { role: 'user', content: 'user message', id: '1' },
-        { role: 'assistant', content: 'assistant message', id: '2' },
-        {
-          role: 'tool',
-          content: 'plugin content',
-          plugin: {
-            type: 'default',
-            arguments: '{}',
-            apiName: 'test-api',
-            identifier: 'test-identifier',
-          },
-          id: '3',
-        },
-      ] as ChatMessage[];
-
-      act(() => {
-        useChatStore.setState({
-          messagesMap: {
-            [messageMapKey('abc')]: messages,
-          },
-          activeId: 'abc',
-        });
-      });
+afterEach(() => {
+  vi.restoreAllMocks();
+});
 
+describe('shareSlice actions', () => {
+  describe('genShareUrl', () => {
+    it('TODO', async () => {
       const { result } = renderHook(() => useChatStore());
       await act(async () => {
-        await result.current.shareToShareGPT({
-          withPluginInfo: true,
-          withSystemRole: true,
-        });
+        await result.current.genShareUrl();
       });
-
-      expect(shareServiceSpy).toHaveBeenCalledWith(
-        expect.objectContaining({
-          items: [
-            expect.objectContaining({ from: 'gpt' }), // Agent meta info
-            expect.objectContaining({ from: 'human', value: 'user message' }),
-            expect.objectContaining({ from: 'gpt', value: 'assistant message' }),
-            expect.objectContaining({
-              from: 'gpt',
-              value: expect.stringContaining('Function Calling Plugin'),
-            }),
-            expect.objectContaining({ from: 'gpt', value: expect.stringContaining('Share from') }), // Footer
-          ],
-        }),
-      );
     });
   });
 });

package/src/store/chat/slices/share/action.ts

@@ -1,122 +1,18 @@
-import dayjs from 'dayjs';
-import { produce } from 'immer';
 import { StateCreator } from 'zustand/vanilla';
 
-import { DEFAULT_USER_AVATAR_URL } from '@/const/meta';
-import { shareService } from '@/services/share';
-import { useAgentStore } from '@/store/agent';
-import { agentSelectors } from '@/store/agent/selectors';
-import { useSessionStore } from '@/store/session';
-import { sessionMetaSelectors } from '@/store/session/selectors';
-import { ShareGPTConversation } from '@/types/share';
-
-import { chatSelectors } from '../../selectors';
 import { ChatStore } from '../../store';
 
-interface ShareMessage {
-  from: 'human' | 'gpt';
-  value: string;
-}
-
-const Footer: ShareMessage = {
-  from: 'gpt',
-  value: `Share from [**🤯 LobeChat**](https://github.com/lobehub/lobe-chat) - ${dayjs().format(
-    'YYYY-MM-DD',
-  )}`,
-};
-
-const PLUGIN_INFO = (plugin: {
-  apiName: string;
-  content: string;
-  identifier: string;
-}): ShareMessage => ({
-  from: 'gpt',
-  value: [
-    `**🧩 Function Calling Plugin**`,
-    `- Identifier: \`${plugin.identifier}\``,
-    `- API name: \`${plugin.apiName}\``,
-    `- Result:`,
-    ``,
-    '```json',
-    plugin.content,
-    '```',
-  ].join('\n'),
-});
-
-// const t = setNamespace('chat/share');
 export interface ShareAction {
-  shareToShareGPT: (props: {
-    avatar?: string;
-    withPluginInfo?: boolean;
-    withSystemRole?: boolean;
-  }) => Promise<void>;
+  genShareUrl: () => Promise<string>;
 }
 
-export const chatShare: StateCreator<ChatStore, [['zustand/devtools', never]], [], ShareAction> = (
-  set,
-  get,
-) => ({
-  shareToShareGPT: async ({ withSystemRole, withPluginInfo, avatar }) => {
-    const messages = chatSelectors.currentChats(get());
-    const config = agentSelectors.currentAgentConfig(useAgentStore.getState());
-    const meta = sessionMetaSelectors.currentAgentMeta(useSessionStore.getState());
-
-    const defaultMsg: ShareGPTConversation['items'] = [];
-    const showSystemRole = withSystemRole && !!config.systemRole;
-    const shareMsgs = produce(defaultMsg, (draft) => {
-      draft.push({
-        from: 'gpt',
-        value: [
-          `${meta.avatar} **${meta.title}** - ${meta.description}`,
-          showSystemRole && '---',
-          showSystemRole && config.systemRole,
-        ]
-          .filter(Boolean)
-          .join('\n\n'),
-      });
-
-      for (const i of messages) {
-        switch (i.role) {
-          case 'assistant': {
-            draft.push({ from: 'gpt', value: i.content });
-            break;
-          }
-          case 'tool': {
-            if (withPluginInfo)
-              draft.push(
-                PLUGIN_INFO({
-                  apiName: i.plugin?.apiName || 'undefined',
-                  content: i.content,
-                  identifier: i.plugin?.identifier || 'undefined',
-                }),
-              );
-            break;
-          }
-          case 'user': {
-            draft.push({ from: 'human', value: i.content });
-            break;
-          }
-        }
-      }
-
-      draft.push(Footer);
-    });
-
-    set({ shareLoading: true });
-
-    const res = await shareService.createShareGPTUrl({
-      avatarUrl: avatar || DEFAULT_USER_AVATAR_URL,
-      items: shareMsgs,
-    });
-    set({ shareLoading: false });
-
-    window.open(res, '_blank');
+export const chatShare: StateCreator<
+  ChatStore,
+  [['zustand/devtools', never]],
+  [],
+  ShareAction
+> = () => ({
+  genShareUrl: () => {
+    return Promise.resolve('TODO');
   },
-  // genShareUrl: () => {
-  //   const session = sessionSelectors.currentSession(get());
-  //   if (!session) return '';
-  //
-  //   const agent = session.config;
-  //   return genShareMessagesUrl(session.chats, agent.systemRole);
-  // },
 });

package/src/utils/client/exportFile.ts

@@ -0,0 +1,20 @@
+export const exportFile = (content: string, filename?: string) => {
+  // 创建一个 Blob 对象
+  const blob = new Blob([content], { type: 'plain/text' });
+
+  // 创建一个 URL 对象，用于下载
+  const url = URL.createObjectURL(blob);
+
+  // 创建一个 <a> 元素，设置下载链接和文件名
+  const a = document.createElement('a');
+  a.href = url;
+  a.download = filename || 'file.txt';
+
+  // 触发 <a> 元素的点击事件，开始下载
+  document.body.append(a);
+  a.click();
+
+  // 下载完成后，清除 URL 对象
+  URL.revokeObjectURL(url);
+  a.remove();
+};

package/src/app/(main)/chat/(workspace)/features/ShareButton/ShareModal.tsx

@@ -1,164 +0,0 @@
-import { Form, type FormItemProps, Modal, type ModalProps } from '@lobehub/ui';
-import { Button, Segmented, SegmentedProps, Switch } from 'antd';
-import { memo, useMemo, useState } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import { FORM_STYLE } from '@/const/layoutTokens';
-import { useChatStore } from '@/store/chat';
-import { useUserStore } from '@/store/user';
-import { userProfileSelectors } from '@/store/user/selectors';
-
-import Preview from './Preview';
-import { FieldType, ImageType } from './type';
-import { useScreenshot } from './useScreenshot';
-
-enum Tab {
-  Screenshot = 'screenshot',
-  ShareGPT = 'sharegpt',
-}
-
-export const imageTypeOptions: SegmentedProps['options'] = [
-  {
-    label: 'JPG',
-    value: ImageType.JPG,
-  },
-  {
-    label: 'PNG',
-    value: ImageType.PNG,
-  },
-  {
-    label: 'SVG',
-    value: ImageType.SVG,
-  },
-  {
-    label: 'WEBP',
-    value: ImageType.WEBP,
-  },
-];
-
-const DEFAULT_FIELD_VALUE: FieldType = {
-  imageType: ImageType.JPG,
-  withBackground: true,
-  withFooter: false,
-  withPluginInfo: false,
-  withSystemRole: false,
-};
-
-const ShareModal = memo<ModalProps>(({ onCancel, open }) => {
-  const [fieldValue, setFieldValue] = useState<FieldType>(DEFAULT_FIELD_VALUE);
-  const [tab, setTab] = useState<Tab>(Tab.Screenshot);
-  const { t } = useTranslation('chat');
-  const avatar = useUserStore(userProfileSelectors.userAvatar);
-  const [shareLoading, shareToShareGPT] = useChatStore((s) => [s.shareLoading, s.shareToShareGPT]);
-  const { loading, onDownload, title } = useScreenshot(fieldValue.imageType);
-
-  const options: SegmentedProps['options'] = useMemo(
-    () => [
-      {
-        label: t('shareModal.screenshot'),
-        value: Tab.Screenshot,
-      },
-      {
-        label: 'ShareGPT',
-        value: Tab.ShareGPT,
-      },
-    ],
-    [],
-  );
-
-  const settings: FormItemProps[] = useMemo(
-    () => [
-      {
-        children: <Switch />,
-        label: t('shareModal.withSystemRole'),
-        minWidth: undefined,
-        name: 'withSystemRole',
-        valuePropName: 'checked',
-      },
-      {
-        children: <Switch />,
-        hidden: tab !== Tab.Screenshot,
-        label: t('shareModal.withBackground'),
-        minWidth: undefined,
-        name: 'withBackground',
-        valuePropName: 'checked',
-      },
-      {
-        children: <Switch />,
-        hidden: tab !== Tab.Screenshot,
-        label: t('shareModal.withFooter'),
-        minWidth: undefined,
-        name: 'withFooter',
-        valuePropName: 'checked',
-      },
-      {
-        children: <Segmented options={imageTypeOptions} />,
-        hidden: tab !== Tab.Screenshot,
-        label: t('shareModal.imageType'),
-        minWidth: undefined,
-        name: 'imageType',
-      },
-      {
-        children: <Switch />,
-        hidden: tab !== Tab.ShareGPT,
-        label: t('shareModal.withPluginInfo'),
-        minWidth: undefined,
-        name: 'withPluginInfo',
-        valuePropName: 'checked',
-      },
-    ],
-    [tab],
-  );
-
-  return (
-    <Modal
-      allowFullscreen
-      centered={false}
-      footer={
-        <>
-          {tab === Tab.Screenshot && (
-            <Button block loading={loading} onClick={onDownload} size={'large'} type={'primary'}>
-              {t('shareModal.download')}
-            </Button>
-          )}
-          {tab === Tab.ShareGPT && (
-            <Button
-              block
-              loading={shareLoading}
-              onClick={() => shareToShareGPT({ avatar, ...fieldValue })}
-              size={'large'}
-              type={'primary'}
-            >
-              {t('shareModal.shareToShareGPT')}
-            </Button>
-          )}
-        </>
-      }
-      maxHeight={false}
-      onCancel={onCancel}
-      open={open}
-      title={t('share', { ns: 'common' })}
-    >
-      <Flexbox gap={16}>
-        <Segmented
-          block
-          onChange={(value) => setTab(value as Tab)}
-          options={options}
-          style={{ width: '100%' }}
-          value={tab}
-        />
-        {tab === Tab.Screenshot && <Preview title={title} {...fieldValue} />}
-        <Form
-          initialValues={DEFAULT_FIELD_VALUE}
-          items={settings}
-          itemsType={'flat'}
-          onValuesChange={(_, v) => setFieldValue(v)}
-          {...FORM_STYLE}
-        />
-      </Flexbox>
-    </Modal>
-  );
-});
-
-export default ShareModal;