update display names for layout components

morenyang · Jan 28, 2020 · 286e0e9 · 286e0e9
1 parent 0f1629b
commit 286e0e9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/components/layout/layout.tsx b/components/layout/layout.tsx
@@ -6,6 +6,7 @@ import { ConfigConsumer, ConfigConsumerProps } from '../config-provider';
 export interface GeneratorProps {
   suffixCls: string;
   tagName: 'header' | 'footer' | 'main' | 'section';
+  displayName: string;
 }
 export interface BasicProps extends React.HTMLAttributes<HTMLDivElement> {
   prefixCls?: string;
@@ -29,9 +30,11 @@ interface BasicPropsWithTagName extends BasicProps {
   tagName: 'header' | 'footer' | 'main' | 'section';
 }
 
-function generator({ suffixCls, tagName }: GeneratorProps) {
+function generator({ suffixCls, tagName, displayName }: GeneratorProps) {
   return (BasicComponent: any) => {
     return class Adapter extends React.Component<BasicProps, any> {
+      static displayName: string = displayName;
+
       static Header: any;
 
       static Footer: any;
@@ -116,21 +119,25 @@ const Layout: React.ComponentClass<BasicProps> & {
 } = generator({
   suffixCls: 'layout',
   tagName: 'section',
+  displayName: 'Layout',
 })(BasicLayout);
 
 const Header = generator({
   suffixCls: 'layout-header',
   tagName: 'header',
+  displayName: 'Header',
 })(Basic);
 
 const Footer = generator({
   suffixCls: 'layout-footer',
   tagName: 'footer',
+  displayName: 'Footer',
 })(Basic);
 
 const Content = generator({
   suffixCls: 'layout-content',
   tagName: 'main',
+  displayName: 'Content',
 })(Basic);
 
 Layout.Header = Header;