From 7a2ad9e25cac8a42ae7ce28af1087bb906685117 Mon Sep 17 00:00:00 2001
From: changfuguo <changfuguo@qq.com>
Date: Fri, 3 Jul 2020 16:36:46 +0800
Subject: [PATCH] feat(components): add FormMegaLayout className (#935)

---
 packages/antd/src/components/FormMegaLayout/index.tsx | 5 ++---
 packages/next/src/components/FormMegaLayout/index.tsx | 4 ++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/packages/antd/src/components/FormMegaLayout/index.tsx b/packages/antd/src/components/FormMegaLayout/index.tsx
index 43870a54eca..ba642f542e8 100644
--- a/packages/antd/src/components/FormMegaLayout/index.tsx
+++ b/packages/antd/src/components/FormMegaLayout/index.tsx
@@ -69,7 +69,7 @@ const StyledLayoutNestWrapper = styled(props => {
 })`${props => computeStyle(props, true)}`
 
 const MegaLayout = (props: ILayoutProps) => {
-    const { children, addonBefore, addonAfter, description, ...others } = props
+    const { children, addonBefore, addonAfter, description, className: megaLayoutClassName, ...others } = props
     const layoutProps = props.layoutProps || {}
     const { size } = useDeepFormItem()
 
@@ -119,9 +119,8 @@ const MegaLayout = (props: ILayoutProps) => {
                 if (labelWidth !== -1) itemProps.labelWidth = labelWidth
                 if (wrapperWidth !== -1) itemProps.wrapperWidth = wrapperWidth
             }
-
             let ele = <StyledLayoutWrapper
-                className="mega-layout-container"
+                className={classnames("mega-layout-container", megaLayoutClassName || '')}
                 label={label}
                 required={required}
                 help={description}
diff --git a/packages/next/src/components/FormMegaLayout/index.tsx b/packages/next/src/components/FormMegaLayout/index.tsx
index f036ff2f1ae..ea076194f46 100644
--- a/packages/next/src/components/FormMegaLayout/index.tsx
+++ b/packages/next/src/components/FormMegaLayout/index.tsx
@@ -69,7 +69,7 @@ return <div style={style} className={classnames('mega-layout-nest-container', cl
 })`${props => computeStyle(props)}`
 
 const MegaLayout = (props: ILayoutProps) => {
-    const { children, addonBefore, addonAfter, description, ...others } = props
+    const { children, addonBefore, addonAfter, description, className: megaLayoutClassName, ...others } = props
     const layoutProps = props.layoutProps || {}
     const { size } = useDeepFormItem()
 
@@ -122,7 +122,7 @@ const MegaLayout = (props: ILayoutProps) => {
             }
 
             let ele = <StyledLayoutWrapper
-                className="mega-layout-container"
+                className={classnames("mega-layout-container", megaLayoutClassName || '')}
                 label={label}
                 required={required}
                 help={description}