org.pageseeder.docx.ox.step.DOCXToPSML Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pso-docx-ox Show documentation
OX tasks and definition for DOCX API
The newest version!
/*
 *  Copyright (c) 2014 Allette Systems pty. ltd.
 */
package org.pageseeder.docx.ox.step;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.HashMap;
import java.util.Map;

import org.pageseeder.docx.PSMLProcessor;
import org.pageseeder.ox.OXErrors;
import org.pageseeder.ox.api.Downloadable;
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.ResultBase;
import org.pageseeder.ox.util.FileUtils;
import org.pageseeder.xmlwriter.XMLWriter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * A docx processing step
 * Step Parameters
 * 
 *  input the xml file needs to be transformed, where is a relative path of package data.
 *  (if not specified, use upper step output as input.)
 *  output the output file, where is a relative path of package data (optional)
 *  config the config file for processing docx.
 *  dotx the doct template.

 *  
media the media folder, where is a relative path of packaged data.

 * 
 *
 * @author Ciber Cai
 * @version 05 November 2014
 */
public class DOCXToPSML implements Step {

  /** The logger. */
  private static Logger LOGGER = LoggerFactory.getLogger(DOCXToPSML.class);

  /**
   * Process.
   *
   * @param model the model
   * @param data the data
   * @param info the info
   * @return the result
   */
  @Override
  public Result process(Model model, PackageData data, StepInfo info) {
    if (data == null) throw new NullPointerException("PackageDate is null");

    // output file
    String output = info.getParameter("output", info.output());
    // the config
    String config = info.getParameter("config");
    // the media folder
    String media = info.getParameter("media", "media");
    
    // input file
    String input = info.getParameter("input", info.input());
    

    
    //Verify the extension .psml
    input = FileUtils.getFileByExtension(data, input,".docx");

    //Verify the extension .xml
    config = FileUtils.getFileByExtension(data, config,".xml");

    
    DocxToPsmlXResult result = new DocxToPsmlXResult(data, model, input, output, config);

    if (input == null) throw new NullPointerException("source must be defined.");
    if (output == null) throw new NullPointerException("destination must be defined.");
    if (config == null) throw new NullPointerException("config must be defined.");

    try {
      //The output file or directory is created here because if it is file and isn't created, the conversion process
      //will think it is a directory and create a directory with the file name.
      createOutput(data, output);
    } catch (IOException ex) {
      LOGGER.error("Exception while creating the output file: " + ex.getMessage());
      result.setError(new IllegalArgumentException("Exception while creating the output file: " + ex.getMessage()));
      return result;
    }



    // the parameters
    Map params = new HashMap();
    params.putAll(info.parameters());

    LOGGER.debug("input {} output {}", input, output);
    try {
      LOGGER.debug("Process the docx");
      PSMLProcessor processor = new PSMLProcessor.Builder()
          .source(data.getFile(input))
          .destination(data.getFile(output))
          .config(getFile(config, model, data))
          .media(media)
          .params(params)
          .working(data.directory())
          .log(new PrintWriter(System.out))
          .build();
      processor.process();
      result.done();
    } catch (Exception ex) {
      LOGGER.warn("Cannot process the docx to psml ", ex);
      result.setError(ex);
    }
    return result;
  }

  /**
   * Creates the input file or directory.
   *
   * @param data the data
   * @param output the output
   * @throws IOException Signals that an I/O exception has occurred.
   */
  private static void createOutput(PackageData data, String output) throws IOException {
    File temp = data.getFile(output);
    if (!temp.getParentFile().exists()) {
      temp.getParentFile().mkdirs();
    }

    if (output.indexOf(".") > 0 ) {
      temp.createNewFile();
    } else {
      temp.mkdir();
    }
  }

//  /**
//   * @param model the name of the model
//   * @param name the name
//   * @return the name of output
//   */
//  private static String getName(String model, String name) {
//    if (name.indexOf(".") > 0) {
//      return name.substring(0, name.indexOf(".")) + ".psml";
//    } else {
//      return model + "-" + System.nanoTime() + ".psml";
//    }
//  }

  /**
 * Gets the file.
 *
 * @param path the path
 * @param model the model
 * @param data the data
 * @return the file
 */
private File getFile(String path, Model model, PackageData data) {
    File file = null;
    if (path != null && !path.isEmpty()) {
      file = model.getFile(path);
      if (file == null || !file.exists()) {
        file = data.getFile(path);
      }
    }
    return file;
  }

  /**
   * The Class DocxToPsmlXResult.
   */
  private final class DocxToPsmlXResult extends ResultBase implements Result, Downloadable {
    
    /** The input. */
    private final String _input;
    
    /** The output. */
    private final String _output;
    
    /** The config. */
    private final String _config;

    /**
     * Instantiates a new docx to psml X result.
     *
     * @param data the data
     * @param model the model
     * @param input the input
     * @param output the output
     * @param config the config
     */
    private DocxToPsmlXResult(PackageData data, Model model, String input, String output, String config) {
      super(model, data);
      this._input = input;
      this._output = output;
      this._config = config;
    }

    /**
     * To XML.
     *
     * @param xml the xml
     * @throws IOException Signals that an I/O exception has occurred.
     */
    @Override
    public void toXML(XMLWriter xml) throws IOException {

      xml.openElement("result");
      xml.attribute("type", "unpackage-docx");
      xml.attribute("id", data().id());
      xml.attribute("model", model().name());
      xml.attribute("status", status().toString().toLowerCase());
      xml.attribute("time", Long.toString(time()));
      xml.attribute("downloadable", String.valueOf(isDownloadable()));
      xml.attribute("path", data().getPath(downloadPath()));

      xml.openElement("source");
      xml.attribute("path", this._input);
      xml.closeElement();

      xml.openElement("destination");
      if (this._output == null) {
        xml.attribute("path", this._input.substring(this._input.lastIndexOf(".")) + ".psml");
      } else {
        xml.attribute("path", "default");
      }
      xml.closeElement();

      xml.openElement("config");
      xml.attribute("path", this._config);
      xml.closeElement();

      // Or the details of any error
      if (this.error() != null) {
        OXErrors.toXML(error(), xml, true);
      }
      xml.closeElement();
    }

    /**
     * Download path.
     *
     * @return the file
     */
    @Override
    public File downloadPath() {
      File outputFile = null;
      if (this._output == null) {
        outputFile = data().getFile(this._input.substring(this._input.lastIndexOf(".")) + ".psml");
      } else {
        outputFile = data().getFile(this._output);
      }
      return outputFile;
    }

    /**
     * Checks if is downloadable.
     *
     * @return true, if is downloadable
     */
    @Override
    public boolean isDownloadable() {
      return true;
    }
  }
}