org.pageseeder.xlsx.ox.step.Import Maven / Gradle / Ivy

Go to download
package org.pageseeder.xlsx.ox.step;

import org.pageseeder.ox.api.Result;
import org.pageseeder.ox.api.Step;
import org.pageseeder.ox.api.StepInfo;
import org.pageseeder.ox.core.Model;
import org.pageseeder.ox.core.PackageData;
import org.pageseeder.ox.tool.DefaultResult;
import org.pageseeder.ox.util.StepUtils;
import org.pageseeder.ox.util.StringUtils;
import org.pageseeder.xlsx.TransformProcessor;
import org.pageseeder.xlsx.config.Param;
import org.pageseeder.xlsx.config.SplitLevel;
import org.pageseeder.xlsx.config.TransformConfig;
import org.pageseeder.xlsx.config.TransformConfigBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

/**
 *
 * Step Parameters
 * 
 *  input .
 *  output .
 *  richtext .
 *  split-level .
 *  headers .
 *  file-name-column .
 *  working-dir .
 *  xslt .
 * 
 *
 *
 */
public class Import implements Step {

  private static Logger LOGGER = LoggerFactory.getLogger(Import.class);

  @Override
  public Result process(Model model, PackageData data, StepInfo info) {
    DefaultResult result = null;
    File input = StepUtils.getInput(data, info);
    File output = StepUtils.getOutput(data, info, input);

    /* Result content - for step */
    result = new DefaultResult(model, data, info, output);

    try {
      TransformConfig config = getConfig(model,data,info,input,output);
      StringWriter writer = new StringWriter();
      TransformProcessor processor = new TransformProcessor(config, writer);
      processor.process();
    } catch (IOException e) {
      e.printStackTrace();
    }
    return result;
  }

  private TransformConfig getConfig(Model model, PackageData data, StepInfo info, File input, File output) {

    TransformConfigBuilder builder = new TransformConfigBuilder();
    builder.input(input);
    builder.destination(output);
    builder.richtext("true".equalsIgnoreCase(StepUtils.getParameter(data, info, "richtext", "false")));
    builder.level(SplitLevel.valueOf(StepUtils.getParameter(data, info, "split-level", "workbook")));
    builder.headers("true".equalsIgnoreCase(StepUtils.getParameter(data,info, "headers", "false")));
    builder.filenameColumn(StepUtils.getParameterInt(data,info,"file-name-column",0));

    String workingDirectoryParameter = StepUtils.getParameter(data,info,"working-dir", "");
    File working = null;
    if (!StringUtils.isBlank(workingDirectoryParameter)) {
      working = data.getFile(workingDirectoryParameter);
    }
    builder.working(working);

    //## Handle the transformation file
    File xslt = getXSLFile(input.getParentFile(), model, data, info);
    builder.xslt(xslt);

    // Add the parameters from step definition in model.xml
    // these parameters should use the prefix _xslt-
    Param param = null;
    for (Map.Entry p :info.parameters().entrySet()) {
      if (p.getKey().startsWith("_xslt-")) {
        String name = p.getKey().replaceAll("_xslt-", "");
        String value = StepUtils.applyDynamicParameterLogic(data, info, p.getValue());
        builder.parameter(new Param(name, value));
      }
    }

    // Add the parameters from from fields in model.xml
    // these parameters should use the prefix _xslt-
    for (Map.Entry p :data.getParameters().entrySet()) {
      if (p.getKey().startsWith("_xslt-")) {
        String name = p.getKey().replaceAll("_xslt-", "");
        String value = StepUtils.applyDynamicParameterLogic(data, info, p.getValue());
        builder.parameter(new Param(name, value));
      }
    }

    return builder.build();
  }

  /**
   * Gets the XSL file.
   *
   * @param unzipedFolder the unziped folder
   * @param model the model
   * @param data  the data
   * @param info  the info
   * @return the XSL file
   */
  private File getXSLFile(File unzipedFolder, Model model, PackageData data, StepInfo info) {
    String xslParameter = StepUtils.getParameter(data, info, "xslt", "");
    File xsl = null;
    if (!StringUtils.isBlank(xslParameter)) {
      xsl = model.getFile(xslParameter);
      if (xsl == null || !xsl.exists()) {
        xsl = data.getFile(xslParameter);
      }
    }
    return xsl;
  }
}