io.ebean.codegen.DoAddManifest 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

There is a newer version: 4.2

package io.ebean.codegen;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.List;

class DoAddManifest {

  private final Detection detection;
  private final InteractionHelp help;

  DoAddManifest(Detection detection, InteractionHelp help) {
    this.detection = detection;
    this.help = help;
  }

  void run() {
    help.questionEntityBeanPackage();
    if (help.isContinue()) {
      help.questionTransactionalPackage();
    }
    if (help.isContinue()) {
      help.questionQueryBeanPackage();
    }
    if(help.isContinue()) {
      // write the ebean.mf
      writeManifest();
    }
  }

  private void writeManifest() {

    List resourceDirs = detection.getMeta().resourceDirs;
    if (resourceDirs.isEmpty()) {
      help.acknowledge("  Unsuccessful - could not determine the resources directory?");

    } else {
      File res = new File(resourceDirs.get(0));
      if (!res.exists() && res.isDirectory()) {
        throw new IllegalStateException("Expected resource directory at " + res.getAbsolutePath());
      }
      try {
        File file = new File(res, "ebean.mf");
        FileWriter writer = new FileWriter(file);

        Actions actions = help.actions();
        String entityPkg = actions.getManifestEntityPackage();
        String transPkg = actions.getManifestTransactionalPackage();
        String queryPkg = actions.getManifestQueryBeanPackage();

        writer.append("entity-packages: ").append(entityPkg).append("\n");
        writer.append("transactional-packages: ").append(transPkg).append("\n");
        writer.append("querybean-packages: ").append(queryPkg).append("\n");
        writer.append("\n");
        writer.flush();
        writer.close();

        help.acknowledge("  ... added " + file.getAbsolutePath());
        detection.addedEbeanManifest();

      } catch (IOException e) {
        throw new RuntimeException("Failed to write ebean.mf", e);
      }
    }
  }
}