br.com.jarch.apt.generate.RestCodeGenerate Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jarch-apt Show documentation
There is a newer version: 24.9.0
package br.com.jarch.apt.generate;

import br.com.jarch.util.type.CodeType;
import br.com.jarch.util.type.ModuleType;

import javax.lang.model.element.Element;

import static br.com.jarch.util.ProcessorUtils.addCode;
import static br.com.jarch.util.ProcessorUtils.addLineBlank;

public class RestCodeGenerate extends BaseCodeGenerate {

    public static void generate(Element element, String useCase, String nameSubPackage) {
        new RestCodeGenerate(element, useCase, nameSubPackage).generate();
    }

    RestCodeGenerate(Element element, String useCase, String nameSubPackage) {
        super(element, useCase, nameSubPackage);
    }

    @Override
    public ModuleType getModule() {
        return ModuleType.WS;
    }

    @Override
    public CodeType getType() {
        return CodeType.REST;
    }

    @Override
    public String getContent() {
        StringBuilder sbListCode = new StringBuilder();
        addPackage(sbListCode);
        addImports(sbListCode);
        addBody(sbListCode);
        return sbListCode.toString();
    }

    private void addPackage(StringBuilder sbListCode) {
        addCode(sbListCode, "package " + getNamePackage() + ";");
        addLineBlank(sbListCode);
    }

    private void addImports(StringBuilder sbListCode) {
        addCode(sbListCode, "import " + getFullNameEntity() + ";");
        addCode(sbListCode, "import " + getFullNameService() + ";");
        addCode(sbListCode, "import " + getFullNameRepository() + ";");
        addCode(sbListCode, "import " + getFullNameSearch() + ";");
        addCode(sbListCode, "import br.com.jarch.annotation.JArchRest;");
        addCode(sbListCode, "import br.com.jarch.crud.rest.BaseRest;");
        addCode(sbListCode, "import io.swagger.annotations.Api;");
        addCode(sbListCode, "import javax.ws.rs.Path;");
        addLineBlank(sbListCode);
    }

    private void addBody(StringBuilder sbListCode) {
        addCode(sbListCode, "@Api");
        addCode(sbListCode, "@Path(\"" + getVarName() + "\")");
        addCode(sbListCode, "@JArchRest");
        addCode(sbListCode, "public class " + getName() + " extends BaseRest<" + getNameEntity() + ", " + getNameService() + ", " + getNameRepository() + ", " + getNameSearch() + "> {");
        addCode(sbListCode, "}");
    }
}