add brain custom statement

Author: Wangggym

craco.config.js

@@ -7,6 +7,7 @@ module.exports = {
       "@pages": path.resolve(__dirname, "src/pages"),
       "@router": path.resolve(__dirname, "src/router"),
       "@views": path.resolve(__dirname, "src/views"),
+      "@port": path.resolve(__dirname, "src/port"),
     },
     configure: {
       module: {

src/base/Json2Ts.ts

@@ -1,3 +1,7 @@
+import { BrainStatementImpl } from "../port/StatementCustom/BrainStatementImpl";
+import { DefaultStatementImpl } from "../port/StatementCustom/DefaultStatementImpl";
+import { StatementCustom } from "../port/StatementCustom/StatementCustom";
+
 type Partial<T> = {
   [P in keyof T]?: T[P];
 };
@@ -7,12 +11,14 @@ interface IJson2TsConfigPrivate {
   optionalFields: boolean;
   rootObjectName: string;
   addPrefix: boolean;
+  brainCustom: boolean;
 }
 
 export type IJson2TsConfig = Partial<IJson2TsConfigPrivate>;
 
 export class Json2Ts {
   private config: IJson2TsConfigPrivate;
+  private statementCustom: StatementCustom;
 
   private interfaces: {
     [name: string]: {
@@ -26,8 +32,10 @@ export class Json2Ts {
       optionalFields: false,
       rootObjectName: "RootObject",
       addPrefix: false,
+      brainCustom: true,
       ...config,
     };
+    this.statementCustom = config.brainCustom ? new BrainStatementImpl() : new DefaultStatementImpl();
   }
 
   convert(json: {}) {
@@ -136,12 +144,13 @@ export class Json2Ts {
         }
         fields.forEach(field => {
           const type = this.interfaces[name][field];
-          const defaultValue = optionalFields ? "" : " = " + this.getDefaultValue(type);
+          const defaultTypeValue = this.getDefaultValue(type);
+          const defaultValue = optionalFields ? "" : " = " + defaultTypeValue;
 
           const customType = type.replace("[]", "");
           const customTypeCheck = Object.keys(this.interfaces).includes(customType);
           if (customTypeCheck) {
-            interfaceStr.push(`  @Type(() => ${customType})`);
+            interfaceStr.push(this.statementCustom.typeStatementGenerator(customType, defaultTypeValue));
           }
           const rewriteFiled = this.hasSpaceInMiddle(field) ? `"${field}"` : field;
           interfaceStr.push(`  ${rewriteFiled}${optionalFields ? "?" : ""}: ${type}${defaultValue};`);
@@ -151,7 +160,7 @@ export class Json2Ts {
       })
       .join("\n");
 
-    if (outout.includes("@Type")) outout = 'import { Type } from "class-transformer";\n\n' + outout;
+    if (outout.includes("@Type")) outout = this.statementCustom.importTypeStatement + "\n\n" + outout;
     return outout;
   }
 

src/pages/default/DefaultPage.tsx

@@ -52,13 +52,17 @@ export function DefaultPage(props: {}): React.JSX.Element {
                 <Form.Item label="添加前缀">
                   <Switch value={bloc.state.value.options.addPrefix} onChange={e => bloc.handleAddPrefixChange(e)} />
                 </Form.Item>
+                <Form.Item label="Brain 自定义">
+                  <Switch value={bloc.state.value.options.brainCustom} onChange={e => bloc.handleBrainCustomChange(e)} />
+                </Form.Item>
                 <Form.Item label="类名">
                   <Input
                     placeholder=""
                     value={bloc.state.value.options.rootObjectName}
                     onChange={e => bloc.handleRootObjectNameChange(e)}
                   />
                 </Form.Item>
+
                 <Form.Item label="&nbsp;" noStyle={false}>
                   <Button type="primary" onClick={() => bloc.transform()}>
                     生成

src/pages/default/DefaultPageBloc.ts

@@ -13,6 +13,7 @@ export class DefaultPageState {
     optionalFields: false,
     rootObjectName: "RootObject",
     addPrefix: false,
+    brainCustom: true,
   };
 
   copyWith(params: {
@@ -69,4 +70,9 @@ export class DefaultPageBloc {
     this.state.value.options.addPrefix = e;
     this.state.value = this.state.value.copyWith({});
   }
+
+  handleBrainCustomChange(e: boolean) {
+    this.state.value.options.brainCustom = e;
+    this.state.value = this.state.value.copyWith({});
+  }
 }

src/port/StatementCustom/BrainStatementImpl.ts

@@ -0,0 +1,8 @@
+import { StatementCustom } from "./StatementCustom";
+
+export class BrainStatementImpl extends StatementCustom {
+  public importTypeStatement: string = 'import { Type } from "@base/cutil/ClassTransformerDecorators";';
+  public typeStatementGenerator(customType: string, defaultTypeValue?: string): string {
+    return `  @Type(() => ${customType}${defaultTypeValue === "[]" ? ", []" : ""})`;
+  }
+}

src/port/StatementCustom/DefaultStatementImpl.ts

@@ -0,0 +1,8 @@
+import { StatementCustom } from "./StatementCustom";
+
+export class DefaultStatementImpl extends StatementCustom {
+  public importTypeStatement: string = 'import { Type } from "class-transformer";';
+  public typeStatementGenerator(customType: string): string {
+    return `  @Type(() => ${customType})`;
+  }
+}

src/port/StatementCustom/StatementCustom.ts

@@ -0,0 +1,4 @@
+export abstract class StatementCustom {
+  public abstract importTypeStatement: string;
+  public abstract typeStatementGenerator(customType: string, defaultTypeValue?: string): string;
+}

tsconfig.json

@@ -1,16 +1,10 @@
 {
   "compilerOptions": {
-    "types": [
-      "reflect-metadata"
-    ],
+    "types": ["reflect-metadata"],
     "experimentalDecorators": true,
     "emitDecoratorMetadata": true,
     "target": "es5",
-    "lib": [
-      "dom",
-      "dom.iterable",
-      "esnext"
-    ],
+    "lib": ["dom", "dom.iterable", "esnext"],
     "allowJs": true,
     "skipLibCheck": true,
     "esModuleInterop": true,
@@ -31,9 +25,8 @@
       "@pages/*": ["pages/*"],
       "@router/*": ["router/*"],
       "@views/*": ["views/*"],
+      "@port/*": ["port/*"]
     }
   },
-  "include": [
-    "src"
-  ]
+  "include": ["src"]
 }