From 5b68b860acdcd078d7a9bb6acd18cb01d51a4552 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?LeoYuan=20=E8=A2=81=E5=8A=9B=E7=9A=93?=
 <yuanlihao1988@gmail.com>
Date: Thu, 4 May 2023 17:03:37 +0800
Subject: [PATCH] refactor: moduleName of Component should not be converted

---
 modules/code-generator/src/parser/SchemaParser.ts           | 5 +++--
 .../src/plugins/component/react/containerClass.ts           | 2 +-
 modules/code-generator/src/utils/schema.ts                  | 6 ++++++
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/modules/code-generator/src/parser/SchemaParser.ts b/modules/code-generator/src/parser/SchemaParser.ts
index 3108fee47..c6263877c 100644
--- a/modules/code-generator/src/parser/SchemaParser.ts
+++ b/modules/code-generator/src/parser/SchemaParser.ts
@@ -32,7 +32,7 @@ import {
 import { SUPPORT_SCHEMA_VERSION_LIST } from '../const';
 
 import { getErrorMessage } from '../utils/errors';
-import { handleSubNodes, isValidContainerType } from '../utils/schema';
+import { handleSubNodes, isValidContainerType, ContainerType } from '../utils/schema';
 import { uniqueArray } from '../utils/common';
 import { componentAnalyzer } from '../analyzer/componentAnalyzer';
 import { ensureValidClassName } from '../utils/validate';
@@ -161,7 +161,8 @@ export class SchemaParser implements ISchemaParser {
             ...subRoot,
             componentName: getRootComponentName(subRoot.componentName, compDeps),
             containerType: subRoot.componentName,
-            moduleName: ensureValidClassName(changeCase.pascalCase(subRoot.fileName)),
+            moduleName: ensureValidClassName(subRoot.componentName === ContainerType.Component ?
+              subRoot.fileName : changeCase.pascalCase(subRoot.fileName)),
           };
           return container;
         });
diff --git a/modules/code-generator/src/plugins/component/react/containerClass.ts b/modules/code-generator/src/plugins/component/react/containerClass.ts
index b4c474b38..eab6cbebe 100644
--- a/modules/code-generator/src/plugins/component/react/containerClass.ts
+++ b/modules/code-generator/src/plugins/component/react/containerClass.ts
@@ -48,7 +48,7 @@ const pluginFactory: BuilderComponentPluginFactory<unknown> = () => {
         type: ChunkType.STRING,
         fileType: FileType.JSX,
         name: CLASS_DEFINE_CHUNK_NAME.InsVar,
-        content: `static displayName = '${changeCase.pascalCase(ir.moduleName)}';`,
+        content: `static displayName = '${ir.moduleName}';`,
         linkAfter: [
           CLASS_DEFINE_CHUNK_NAME.Start,
         ],
diff --git a/modules/code-generator/src/utils/schema.ts b/modules/code-generator/src/utils/schema.ts
index 831b38965..f9529945e 100644
--- a/modules/code-generator/src/utils/schema.ts
+++ b/modules/code-generator/src/utils/schema.ts
@@ -147,4 +147,10 @@ export function isValidContainerType(schema: IPublicTypeNodeSchema) {
     'Component',
     'Block',
   ].includes(schema.componentName);
+}
+
+export const enum ContainerType {
+  Page = 'Page',
+  Component = 'Component',
+  Block = 'Block',
 }
\ No newline at end of file