small-program/cmd/api_generate/utils/docUtillTs.ts

import fs from 'fs';
import { IApiDocFuncInterface, IParamsType } from "../base/interface";
import ApiGenerateImpl from "../interface/api_generate_impl";
import { execSync } from 'child_process';

export default class DocUtilTs {

    static runner(app: string) {
        execSync("cd ../" + app + " && flutter pub run build_runner build ");
    }


    static firstUpper(name: string) {
        let nameSpilit = name.split("");
        nameSpilit[0] = name[0].toUpperCase();
        return nameSpilit.join("");
    }

    static lowerCamelCase(name: string) {
        let nameSpilit = name.split("_");
        if (nameSpilit.length > 1) {
            nameSpilit.forEach((p, i) => {
                if (i > 0) {
                    nameSpilit[i] = this.firstUpper(nameSpilit[i]);
                }
            })
        }
        return nameSpilit.join("");
        // nameSpilit[0] = name[0].toUpperCase();
        // return nameSpilit.join("");
    }

    public info: ApiGenerateImpl;

    public entity: string = "";
    public funcs: string = "";




    constructor(_info: ApiGenerateImpl) {
        this.info = _info;
        let api = this.info.doc;
        for (const key in api) {
            const apis = api[key];
            for (const objKey in apis) {
                const element = (apis as any)[objKey] as IApiDocFuncInterface;
                if (['post', 'put', 'get', 'delete'].includes(objKey)) {
                    let name = `${apis.funcName}Using${DocUtilTs.firstUpper(objKey)}`;
                    let entityName = DocUtilTs.firstUpper(name);
                    let body = !!element.body && this.paramsTemple(`${entityName}Body`, element.body) && `${entityName}Body`;
                    let params = !!element.params && this.paramsTemple(`${entityName}Params`, element.params) && `${entityName}Params`;
                    let response: string | false = false;
                    let responseList = false;
                    /// 是否复杂的返回值-需要反序列化
                    let responsecomplex = false;
                    if (element.response) {
                        if (typeof element.response === "string") {
                            response = element.response;
                        }
                        else if (Array.isArray(element.response)) {
                            responseList = true;
                            // response = `List<${element.}>`;
                            let _type = element.response[0];
                            if (typeof _type === "string") {
                                response = _type;
                            }
                            else {
                                responsecomplex = true;
                                response = this.paramsTemple(`${entityName}Response`, _type) && `${entityName}Response`;
                            }
                        }
                        else {
                            responsecomplex = true;
                            response = this.paramsTemple(`${entityName}Response`, element.response) && `${entityName}Response`;
                        }
                    }
                    // let response = !!element.response && ((typeof element.response === "string") ? element.response : this.paramsTemple(`${entityName}Response`, element.response) && `${entityName}Response`);
                    this.funcs += this.funcTemple({
                        desc: apis.desc,
                        name,
                        method: objKey,
                        url: key,
                        body,
                        params,
                        response,
                        responseList,
                        responsecomplex,
                        element,
                        contentType: apis.contentType
                    });

                }
            }
        }
        fs.writeFileSync(this.info.filePath + "/" + this.info.fileName + ".ts", this.buildContent());
    }

    // enum StatusCode {
    //     @JsonValue(200)
    //     success,
    //     @JsonValue('500')
    //     weird,
    //   }

    public buildContent() {
        return `
        import RequestUtil, { AxiosRequestConfigFunc } from "@/utils/request";
        
        ${this.entity}
        
        export default class ${this.info.name} {
            ${this.funcs}
        }
        `
    };

    public mapType(type: any): string {
        let obj: any = {
            String: "string",
            int: "number",
            double: "number"
        }
        return obj[type];
    }


    public funcTemple(obj: {
        name: string,
        desc?: string,
        method: string,
        url: string,
        body: string | false,
        params: string | false,
        response: string | false,
        responseList: boolean,
        responsecomplex: boolean,
        element: IApiDocFuncInterface,
        contentType?: "json" | "urlencoded" | "formData"
    }) {
        let serialize = "";
        let _response = obj.response || "any";
        if (obj.response) {
            if (obj.responseList) {
                _response = `${obj.response}[]`;
            }
            if (obj.responsecomplex) {
                if (obj.responseList) {
                    serialize = `, serialize: (data)=> [for (var i in data) ${obj.response}.fromJson(i)]`;
                } else {
                    serialize = `, serialize: (data)=> ${obj.response}.fromJson(data)`;
                }
            }
            else {
                serialize = `,serialize:(dynamic data)=>data`
            }
        }
        let url = obj.url;
        if (obj.element.params) {
            for (const key in obj.element.params) {
                const element = obj.element.params[key];
                if (element.url === true) {
                    url = url.replace(`{${key}}`, `\${options.params!.${key}}`);
                }
            }
        }

        return (
            `   
        /// ${obj.desc}
        static  ${obj.name}(options:AxiosRequestConfigFunc<${obj.body || "any"},${obj.params || "any"}> ):RequestUtil<${_response || "any"}> {
            return new RequestUtil<${_response || "any"}>(\`${url}\`,'${obj.method.toUpperCase()}' ,options);
        }
    `
        )
    };

    public enumTemple(name: string, params: Record<string, number | string>) {
        let _list: string[] = []
        for (const key in params) {
            const element = params[key];
            let _element = typeof element === "string" ? `"${element}"` : element;
            _list.push(`@JsonValue(${_element})`);
            _list.push(DocUtilTs.lowerCamelCase(element.toString()) + ",");
        }
        this.entity +=
            `
       export enum ${DocUtilTs.firstUpper(name)} {
            ${_list.join("\r\n")}
          }
        `
    }

    public paramsTemple(name: string, params: Record<string, IParamsType>) {
        let str = "";
        let iniStr: string[] = [];

        for (const key in params) {
            const element = params[key];
            let jsonKey = element.jsonKey ? `@JsonKey(name: "${element.jsonKey}")` : "";
            let _final = element.final != false ? "" : "";
            if (element.type === "object" && element.child && typeof element.child !== "string") {
                this.paramsTemple(name + DocUtilTs.firstUpper(key), element.child);
                str += `
                /// ${element.desc}
                ${_final} ${key}${element.required != false ? "!" : "?"}:  ${name + DocUtilTs.firstUpper(key)} ;
                `
                iniStr.push(` ${element.required != false ? "required" : ""} this.${key}`);
            }
            else if (element.type === "enum") {
                if (element.enum) {
                    this.enumTemple(name + DocUtilTs.firstUpper(key), element.enum);
                    str += `
                    /// ${element.desc}
                    ${_final}  ${key}${element.required != false ? "!" : "?"}: ${name + DocUtilTs.firstUpper(key)};
                    `
                    iniStr.push(` ${element.required != false ? "required" : ""} this.${key}`);
                }
            }
            else if (element.type === "list" && element.child) {
                let _type = "";
                if (typeof element.child !== "string") {
                    this.paramsTemple(name + DocUtilTs.firstUpper(key), element.child);
                    _type = name + DocUtilTs.firstUpper(key);
                }
                else {
                    _type = element.child;
                }
                str += `
                /// ${element.desc}
                ${_final}  ${key}${element.required != false ? "!" : "?"}: ${_type}[] ;
                `
                iniStr.push(` ${element.required != false ? "required" : ""} this.${key}`);
            }
            else {
                str += `
                /// ${element.desc}
                ${_final} ${key}${element.required != false ? "!" : "?"}: ${this.mapType(element.type)} ;
                `
                iniStr.push(` ${element.required != false ? "required" : ""} this.${key}`);
            }
        }

        this.entity += `
export  class ${name} {
        ${str}
}
    `
        return true;
    }
}