org.avaje.ebean.codegen.InitMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of codegen-maven-plugin Show documentation
The newest version!
package org.avaje.ebean.codegen;

import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;
import org.codehaus.plexus.util.IOUtil;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

@Mojo(name = "init", defaultPhase = LifecyclePhase.NONE)
public class InitMojo extends AbstractMojo {

  public void execute() {

    getLog().debug("executing init ...");
    copyTestResources();
    copyMainResources();
  }

  private String getSourceMain() {
    return System.getProperty("src.main", "src/main");
  }

  private String getSourceTest() {
    return System.getProperty("src.test", "src/test");
  }

  private void copyMainResources() {

    File mainSource = new File(getSourceMain());
    if (!mainSource.exists()) {
      getLog().error("No " + getSourceMain() + " directory to add resources to?");
    } else {
      copyMainProperties(mainSource);
    }
  }

  private void copyTestResources() {

    File testSource = new File(getSourceTest());
    if (!testSource.exists()) {
      getLog().error("No " + getSourceTest() + " directory to add test resources to?");
    } else {
      copyDbMigration(testSource);
      copyTestProperties(testSource);
      copyTestLogging(testSource);
    }
  }

  private void copyMainProperties(File mainSource) {

    File resources = new File(mainSource, "resources");
    if (!resources.exists() && !resources.mkdirs()) {
      getLog().error("Could not create src/main/resources ?");

    } else {
      try {
        File props = new File(resources, "ebean.properties");
        if (props.exists()) {
          getLog().info("ebean.properties already exists, leaving as is.");
        } else {
          copyResource(props, "/tp-ebean.properties");
          getLog().info("... added ebean.properties");
        }
      } catch (IOException e) {
        getLog().error("Failed to copy ebean.properties", e);
      }
    }
  }

  private void copyTestProperties(File testSource) {

    File testResources = new File(testSource, "resources");
    if (!testResources.exists() && !testResources.mkdirs()) {
      getLog().error("Could not create src/test/resources ?");

    } else {
      try {
        File testProps = new File(testResources, "test-ebean.properties");
        if (testProps.exists()) {
          getLog().info("test-ebean.properties already exists, leaving as is.");
        } else {
          copyResource(testProps, "/tp-test-ebean.properties");
          getLog().info("... added test-ebean.properties");
        }
      } catch (IOException e) {
        getLog().error("Failed to copy test-ebean.properties", e);
      }
    }
  }


  private void copyTestLogging(File testSource) {

    File testResources = new File(testSource, "resources");
    if (!testResources.exists() && !testResources.mkdirs()) {
      getLog().error("Could not create " + getSourceTest() + "/resources ?");

    } else {
      try {
        File testProps = new File(testResources, "logback-test.xml");
        if (testProps.exists()) {
          getLog().info("logback-test.xml already exists, leaving as is.");
        } else {
          copyResource(testProps, "/tp-logback-test.xml");
          getLog().info("... added logback-test.xml");
        }
      } catch (IOException e) {
        getLog().error("Failed to copy logback-test.xml", e);
      }
    }
  }

  private void copyDbMigration(File testSource) {

    File testJavaMain = new File(testSource, "java/main");
    if (!testJavaMain.exists() && !testJavaMain.mkdirs()) {
      getLog().error("Failed to make " + getSourceTest() + "/java/main ?");
    } else {
      try {
        File dbMigJava = new File(testJavaMain, "MainDbMigration.java");
        if (dbMigJava.exists()) {
          getLog().info("MainDbMigration.java already exists, leaving as is.");
        } else {
          copyResource(dbMigJava, "/tp-MainDbMigration.java");
          getLog().info("... added MainDbMigration.java");
        }

      } catch (IOException e) {
        getLog().error("Failed to copy MainDbMigration.java", e);
      }
    }
  }

  private void copyResource(File testProps, String name) throws IOException {

    InputStream in = this.getClass().getResourceAsStream(name);
    FileOutputStream fos = new FileOutputStream(testProps);
    IOUtil.copy(in, fos);

    fos.flush();
    fos.close();
  }
}