package com.witwicky.plugins;

import org.mybatis.generator.api.GeneratedJavaFile;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.DefaultJavaFormatter;
import org.mybatis.generator.api.dom.java.JavaVisibility;
import org.mybatis.generator.api.dom.java.TopLevelClass;

import java.util.Collections;
import java.util.List;

public class GenQueryVoPlugin extends PluginAdapter {

    public GenQueryVoPlugin() {
        super();
    }

    public boolean validate(List<String> warnings) {
        return true;
    }

    @Override
    public List<GeneratedJavaFile> contextGenerateAdditionalJavaFiles(IntrospectedTable introspectedTable) {
        String queryVoName = introspectedTable.getFullyQualifiedTable().getDomainObjectName();

        String model = context.getJavaModelGeneratorConfiguration().getTargetPackage();
        String querVoPkgName = properties.getProperty("QUERVO_PKG_NAME", "queryvo");
        String queryVo = join(model.substring(0, model.lastIndexOf(".")), querVoPkgName);

        String querVoSuffix = properties.getProperty("QUERVO_VO_SUFFIX", "Vo");
        TopLevelClass root = new TopLevelClass(join(queryVo, queryVoName + querVoSuffix));
        root.addImportedType(model + "." + queryVoName);
        root.setSuperClass(queryVoName);
        root.setVisibility(JavaVisibility.PUBLIC);

        String targetProject = context.getJavaClientGeneratorConfiguration().getTargetProject();
        GeneratedJavaFile gjf = new GeneratedJavaFile(root, targetProject, "UTF-8", new DefaultJavaFormatter());

        return Collections.singletonList(gjf);
    }

    private String join(String... strs) {
        StringBuilder sb = new StringBuilder();
        for (String str : strs) {
            sb.append(str).append(".");
        }
        return sb.deleteCharAt(sb.length() - 1).toString();
    }
}