/**
 * Copyright (c) 2015-2022, Michael Yang 杨福海 (fuhai999@gmail.com).
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.jboot.codegen.service;

import com.jfinal.core.JFinal;
import com.jfinal.kit.Kv;
import com.jfinal.kit.StrKit;
import com.jfinal.plugin.activerecord.generator.BaseModelGenerator;
import com.jfinal.plugin.activerecord.generator.MetaBuilder;
import com.jfinal.plugin.activerecord.generator.TableMeta;
import com.jfinal.template.Engine;
import com.jfinal.template.source.ClassPathSourceFactory;
import io.jboot.codegen.CodeGenHelpler;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.List;

public class JbootServiceInterfaceGenerator extends BaseModelGenerator {

    private String modelPacket;
    private String basePackage;
    private String classSuffix = "Service";
    private String classPrefix = "";

    private MetaBuilder metaBuilder;


    public JbootServiceInterfaceGenerator(String basePackage, String modelPacket) {
        super(basePackage, CodeGenHelpler.getUserDir() + "/src/main/java/" + basePackage.replace(".", "/"));


        this.modelPacket = modelPacket;
        this.basePackage = basePackage;
        this.template = "io/jboot/codegen/service/service_template.tp";
        this.metaBuilder = CodeGenHelpler.createMetaBuilder();
    }

    public JbootServiceInterfaceGenerator(String basePackage, String outputDir, String modelPacket) {
        super(basePackage, outputDir);
        this.modelPacket = modelPacket;
        this.basePackage = basePackage;
        this.template = "io/jboot/codegen/service/service_template.tp";
        this.metaBuilder = CodeGenHelpler.createMetaBuilder();
    }

    public String getClassSuffix() {
        return classSuffix;
    }

    public JbootServiceInterfaceGenerator setClassSuffix(String classSuffix) {
        this.classSuffix = classSuffix;
        return this;
    }

    public String getClassPrefix() {
        return classPrefix;
    }

    public JbootServiceInterfaceGenerator setClassPrefix(String classPrefix) {
        this.classPrefix = classPrefix;
        return this;
    }

    public void generate() {
        generate(metaBuilder.build());
    }

    /**
     * 设置需要被移除的表名前缀
     * 例如表名  "tb_account"，移除前缀 "tb_" 后变为 "account"
     */
    public JbootServiceInterfaceGenerator setRemovedTableNamePrefixes(String... prefixes) {
        metaBuilder.setRemovedTableNamePrefixes(prefixes);
        return this;
    }

    public JbootServiceInterfaceGenerator addExcludedTable(String... excludedTables) {
        metaBuilder.addExcludedTable(excludedTables);
        return this;
    }

    @Override
    public void generate(List<TableMeta> tableMetas) {
        System.out.println("Generate base model ...");
        System.out.println("Base Model Output Dir: " + baseModelOutputDir);

        Engine engine = Engine.create("forService");
        engine.setSourceFactory(new ClassPathSourceFactory());
        engine.addSharedMethod(new StrKit());
        engine.addSharedObject("getterTypeMap", getterTypeMap);
        engine.addSharedObject("javaKeyword", javaKeyword);

        for (TableMeta tableMeta : tableMetas) {
            genBaseModelContent(tableMeta);
        }
        writeToFile(tableMetas);
    }

    @Override
    protected void genBaseModelContent(TableMeta tableMeta) {
        Kv data = Kv.by("baseModelPackageName", baseModelPackageName);
        data.set("generateChainSetter", generateChainSetter);
        data.set("tableMeta", tableMeta);
        data.set("modelPacket", modelPacket);
        data.set("basePackage", basePackage);
        data.set("classSuffix", this.classSuffix);
        data.set("classPrefix", this.classPrefix);

        Engine engine = Engine.use("forService");
        tableMeta.baseModelContent = engine.getTemplate(template).renderToString(data);
    }


    /**
     * base model 覆盖写入
     */
    @Override
    protected void writeToFile(TableMeta tableMeta) throws IOException {
        File dir = new File(baseModelOutputDir);
        if (!dir.exists()) {
            dir.mkdirs();
        }

        String target = baseModelOutputDir + File.separator + getClassPrefix() + tableMeta.modelName + classSuffix + ".java";

        File targetFile = new File(target);
        if (targetFile.exists()) {
            return;
        }

        try (OutputStreamWriter osw = new OutputStreamWriter(new FileOutputStream(target), JFinal.me().getConstants().getEncoding())) {
            osw.write(tableMeta.baseModelContent);
        }
    }


}

@@ -5110,12 +5110,216 @@
 %7D%0A    %7D%0A
+    public JbootServiceInterfaceGenerator addWhitelist(String... tableNames) %7B%0A        if (tableNames != null) %7B%0A            this.metaBuilder.addWhitelist(tableNames);%0A        %7D%0A        return this;%0A    %7D
 %0A%0A%7D%0A

update src/main/java/io/jboot/codegen/service/JbootServiceInterfaceGenerator.java. 增加白名单功能 public JbootServiceInterfaceGenerator addWhitelist(String... tableNames) { if (tableNames != null) { this.metaBuilder.addWhitelist(tableNames); } return this; }