refactor: page layout

src/pages/playground/rerank.tsx

@@ -1,21 +1,22 @@
-import IconFont from '@/components/icon-font';
 import HotKeys from '@/config/hotkeys';
+import { ExtraContent } from '@/layouts/extraRender';
 import { modelCategoriesMap } from '@/pages/llmodels/config';
-import { PageContainer } from '@ant-design/pro-components';
 import { useIntl } from '@umijs/max';
-import { Button, Space } from 'antd';
+import useMemoizedFn from 'ahooks/lib/useMemoizedFn';
+import { Divider } from 'antd';
 import classNames from 'classnames';
 import _ from 'lodash';
-import { useCallback, useEffect, useRef, useState } from 'react';
+import { useEffect, useRef, useState } from 'react';
 import { useHotkeys } from 'react-hotkeys-hook';
+import { PageContainerInner } from '../_components/page-box';
 import { queryModelsList } from './apis';
 import GroundReranker from './components/ground-reranker';
+import ViewCodeButtons from './components/view-code-buttons';
 import useCollapseLayout from './hooks/use-collapse-layout';
 import './style/play-ground.less';
 
 const PlaygroundRerank: React.FC = () => {
   const intl = useIntl();
-  const groundLeftRef = useRef<any>(null);
   const groundRerankerRef = useRef<any>(null);
   const [rerankerModelList, setRerankerModelList] = useState<
     Global.BaseOption<string>[]
@@ -29,13 +30,13 @@ const PlaygroundRerank: React.FC = () => {
     triggeredRef: groundRerankerRef.current
   });
 
-  const handleViewCode = useCallback(() => {
+  const handleViewCode = useMemoizedFn(() => {
     groundRerankerRef.current?.viewCode?.();
-  }, [groundRerankerRef]);
+  });
 
-  const handleToggleCollapse = useCallback(() => {
+  const handleToggleCollapse = useMemoizedFn(() => {
     groundRerankerRef.current?.setCollapse?.();
-  }, [groundRerankerRef]);
+  });
 
   useEffect(() => {
     const getModelListByReranker = async () => {
@@ -71,30 +72,6 @@ const PlaygroundRerank: React.FC = () => {
     fetchData();
   }, []);
 
-  const renderExtra = () => {
-    return (
-      <Space key="buttons">
-        <Button
-          size="middle"
-          onClick={handleViewCode}
-          icon={<IconFont type="icon-code" className="font-size-16"></IconFont>}
-        >
-          {intl.formatMessage({ id: 'playground.viewcode' })}
-        </Button>
-        <Button
-          size="middle"
-          onClick={handleToggleCollapse}
-          icon={
-            <IconFont
-              type="icon-a-layout6-line"
-              className="font-size-16"
-            ></IconFont>
-          }
-        ></Button>
-      </Space>
-    );
-  };
-
   useHotkeys(
     HotKeys.RIGHT.join(','),
     () => {
@@ -106,17 +83,22 @@ const PlaygroundRerank: React.FC = () => {
   );
 
   return (
-    <PageContainer
-      ghost
-      header={{
-        title: intl.formatMessage({ id: 'menu.playground.rerank' }),
-        style: {
-          paddingInline: 'var(--layout-content-header-inlinepadding)'
-        },
-        breadcrumb: {}
-      }}
-      extra={renderExtra()}
+    <PageContainerInner
       className={classNames('playground-container chat')}
+      extra={[
+        <ViewCodeButtons
+          activeKey=""
+          handleViewCode={handleViewCode}
+          handleToggleCollapse={handleToggleCollapse}
+          key="view-code-buttons"
+        ></ViewCodeButtons>,
+        <Divider
+          key="divider"
+          type="vertical"
+          style={{ height: 24, marginInline: 16 }}
+        />,
+        <ExtraContent key="extra-content" />
+      ]}
     >
       <div className="play-ground">
         <div className="chat">
@@ -127,7 +109,7 @@ const PlaygroundRerank: React.FC = () => {
           ></GroundReranker>
         </div>
       </div>
-    </PageContainer>
+    </PageContainerInner>
   );
 };