dev

src/Promise/components/Schema/TableSchemaConfig/table-schema.tsx

@@ -1,10 +1,13 @@
-import React, { useContext } from 'react';
+import React, { useContext, useRef } from 'react';
 import ProProvider from '@ant-design/pro-provider';
 import ProTable from '@ant-design/pro-table';
 import { SchemeColumns, handleSchemeColumns } from '../scheme';
 import { ProTableProps } from '@ant-design/pro-table/lib/typing';
+import { ActionType } from '@ant-design/pro-components';
 
-interface TableSchemaProps extends ProTableProps<any, any, any> {}
+interface TableSchemaProps extends ProTableProps<any, any, any> {
+  [key: string]: any;
+}
 
 // @formatter: off
 const ArchiveSchema: React.FC<TableSchemaProps> = ({
@@ -13,6 +16,7 @@ const ArchiveSchema: React.FC<TableSchemaProps> = ({
   children,
   ...rest
 }) => {
+  const actionRef = useRef<ActionType>();
   // @formatter: on
   const values = useContext(ProProvider);
   let value = {
@@ -26,11 +30,16 @@ const ArchiveSchema: React.FC<TableSchemaProps> = ({
   return (
     <ProProvider.Provider value={value}>
       <ProTable
+        actionRef={actionRef}
         rowKey={rowKey || 'id'}
         search={search}
         tableStyle={{ overflowX: 'auto' }}
         columns={handleSchemeColumns(columns)}
         {...rest}
+        tableAlertOptionRender={(...args) =>
+          // @ts-ignore
+          rest?.tableAlertOptionRender?.(...args, actionRef)
+        }
       >
         {children}
       </ProTable>