cloud.unionj.generator.backend.springboot.ProtoPomGenerator Maven / Gradle / Ivy

Go to download
package cloud.unionj.generator.backend.springboot;

import java.io.File;
import java.util.Map;

import static cloud.unionj.generator.backend.springboot.Constants.OUTPUT_DIR;

/**
 * @author created by tqccc
 * @version v0.0.1
 * description: cloud.unionj.generator.backend.springboot
 * date:2021/1/17
 */
public class ProtoPomGenerator extends BasePomGenerator {

  protected String voGroupId;

  protected String voArtifactId;

  protected String voVersion;

  public ProtoPomGenerator() {
    super();

    this.outputDir = OUTPUT_DIR + File.separator + "proto";
    this.artifactId = this.parentArtifactId + "-proto";

    this.hasParent = true;

    this.voGroupId = this.parentGroupId;
    this.voArtifactId = this.parentArtifactId + "-vo";
    this.voVersion = this.parentVersion;
  }

  public ProtoPomGenerator(String protoDir) {
    super();

    this.outputDir = protoDir;
    this.artifactId = this.parentArtifactId + "-proto";

    this.hasParent = true;

    this.voGroupId = this.parentGroupId;
    this.voArtifactId = this.parentArtifactId + "-vo";
    this.voVersion = this.parentVersion;
  }

  public ProtoPomGenerator voGroupId(String voGroupId) {
    this.voGroupId = voGroupId;
    return this;
  }

  public ProtoPomGenerator voGroupIdAsParent() {
    this.voGroupId = this.parentGroupId;
    return this;
  }

  public ProtoPomGenerator voArtifactId(String voArtifactId) {
    this.voArtifactId = voArtifactId;
    return this;
  }

  public ProtoPomGenerator voVersion(String voVersion) {
    this.voVersion = voVersion;
    return this;
  }

  public ProtoPomGenerator voVersionAsParent() {
    this.voVersion = this.parentVersion;
    return this;
  }

  @Override
  public Map getInput() {
    Map input = super.getInput();

    input.put("voGroupId", voGroupId);
    input.put("voArtifactId", voArtifactId);
    input.put("voVersion", voVersion);

    return input;
  }

  @Override
  public String getTemplate() {
    return OUTPUT_DIR + "/protopom.xml.ftl";
  }

}