/*
GanttProject is an opensource project management tool. License: GPL3
Copyright (C) 2010-2012 Dmitry Barashev, GanttProject Team
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 3
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
package biz.ganttproject.impex.msproject2;
import java.awt.Component;
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import biz.ganttproject.core.option.DefaultEnumerationOption;
import biz.ganttproject.core.option.EnumerationOption;
import biz.ganttproject.core.option.GPOption;
import biz.ganttproject.core.option.GPOptionGroup;
import net.sf.mpxj.MPXJException;
import net.sf.mpxj.ProjectFile;
import net.sf.mpxj.mpx.MPXWriter;
import net.sf.mpxj.mspdi.MSPDIWriter;
import net.sf.mpxj.writer.ProjectWriter;
import net.sourceforge.ganttproject.export.ExporterBase;
/**
* @author dbarashev (Dmitry Barashev)
*/
public class ExporterToMsProjectFile extends ExporterBase {
private static final String[] FILE_FORMAT_IDS = new String[] { "impex.msproject.fileformat.mpx",
"impex.msproject.fileformat.mspdi" };
private static final String[] FILE_EXTENSIONS = new String[] { "mpx", "xml" };
private String myFileFormat = FILE_FORMAT_IDS[0];
private EnumerationOption myFileFormatOption = new DefaultEnumerationOption<Object>("impex.msproject.fileformat",
FILE_FORMAT_IDS) {
@Override
public void commit() {
super.commit();
ExporterToMsProjectFile.this.myFileFormat = getValue();
}
};
private LocaleOption myLanguageOption = new LocaleOption();
private GPOptionGroup myOptions = new GPOptionGroup("exporter.msproject", new GPOption[] { myFileFormatOption });
private GPOptionGroup myMPXOptions = new GPOptionGroup("exporter.msproject.mpx", new GPOption[] { myLanguageOption });
public ExporterToMsProjectFile() {
myOptions.setTitled(false);
myMPXOptions.setTitled(false);
myFileFormatOption.lock();
myFileFormatOption.setValue(FILE_FORMAT_IDS[0]);
myFileFormatOption.commit();
myLanguageOption.setSelectedLocale(language.getLocale());
}
@Override
public String getFileTypeDescription() {
return language.getText("impex.msproject.description");
}
@Override
public GPOptionGroup getOptions() {
return myOptions;
}
@Override
public List<GPOptionGroup> getSecondaryOptions() {
return FILE_FORMAT_IDS[0].equals(myFileFormat) ? Collections.singletonList(myMPXOptions)
: Collections.<GPOptionGroup> emptyList();
}
@Override
public Component getCustomOptionsUI() {
return null;
}
@Override
public String getFileNamePattern() {
return myFileFormat;
}
@Override
protected ExporterJob[] createJobs(final File outputFile, List<File> resultFiles) {
ExporterJob job = createExportJob(outputFile);
resultFiles.add(outputFile);
return new ExporterJob[] { job };
}
private ExporterJob createExportJob(final File outputFile) {
ExporterJob result = new ExporterJob("Export project") {
@Override
protected IStatus run() {
ProjectFile outProject;
try {
outProject = new ProjectFileExporter(getProject()).run();
ProjectWriter writer = createProjectWriter();
writer.write(outProject, outputFile);
} catch (MPXJException e) {
getUIFacade().showErrorDialog(e);
return Status.CANCEL_STATUS;
} catch (IOException e) {
getUIFacade().showErrorDialog(e);
return Status.CANCEL_STATUS;
}
return Status.OK_STATUS;
}
};
return result;
}
private ProjectWriter createProjectWriter() {
if (FILE_FORMAT_IDS[0].equals(myFileFormat)) {
MPXWriter result = new MPXWriter();
if (myLanguageOption.getSelectedLocale() != null) {
result.setLocale(myLanguageOption.getSelectedLocale());
}
return result;
}
if (FILE_FORMAT_IDS[1].equals(myFileFormat)) {
return new MSPDIWriter();
}
assert false : "Should not be here";
return null;
}
@Override
public String proposeFileExtension() {
return getSelectedFormatExtension();
}
private String getSelectedFormatExtension() {
for (int i = 0; i < FILE_FORMAT_IDS.length; i++) {
if (myFileFormat.equals(FILE_FORMAT_IDS[i])) {
return FILE_EXTENSIONS[i];
}
}
throw new IllegalStateException("Selected format=" + myFileFormat + " has not been found in known formats:"
+ Arrays.asList(FILE_FORMAT_IDS));
}
@Override
public String[] getFileExtensions() {
return FILE_EXTENSIONS;
}
}