feat(codegen): add new option exclude for plugin containerLifeCycle

alibaba · Jul 13, 2023 · e19931b · e19931b
1 parent aa1bb1c
commit e19931b
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/modules/code-generator/src/plugins/component/react/containerLifeCycle.ts b/modules/code-generator/src/plugins/component/react/containerLifeCycle.ts
@@ -16,13 +16,14 @@ import { isJSFunction, isJSExpression } from '@alilc/lowcode-types';
 import { isJSExpressionFn } from '../../../utils/common';
 
 export interface PluginConfig {
-  fileType: string;
-  exportNameMapping: Record<string, string>;
-  normalizeNameMapping: Record<string, string>;
+  fileType?: string;
+  exportNameMapping?: Record<string, string>;
+  normalizeNameMapping?: Record<string, string>;
+  exclude?: string[];
 }
 
 const pluginFactory: BuilderComponentPluginFactory<PluginConfig> = (config?) => {
-  const cfg: PluginConfig = {
+  const cfg = {
     fileType: FileType.JSX,
     exportNameMapping: {},
     normalizeNameMapping: {},
@@ -56,6 +57,10 @@ const pluginFactory: BuilderComponentPluginFactory<PluginConfig> = (config?) =>
           normalizeName = cfg.normalizeNameMapping[lifeCycleName] || lifeCycleName;
         }
 
+        if (cfg?.exclude?.includes(normalizeName)) {
+          return null;
+        }
+
         const exportName = cfg.exportNameMapping[lifeCycleName] || lifeCycleName;
         if (normalizeName === 'constructor') {
           return {
@@ -97,7 +102,7 @@ const pluginFactory: BuilderComponentPluginFactory<PluginConfig> = (config?) =>
           }),
           linkAfter: [...DEFAULT_LINK_AFTER[CLASS_DEFINE_CHUNK_NAME.InsMethod]],
         };
-      });
+      }).filter((i) => !!i);
 
       next.chunks.push(...chunks.filter((x): x is ICodeChunk => x !== null));
     }