ContentGenerator.java

package com.proalpha.pds.generator;
 
import java.io.File;
 
import org.eclipse.core.resources.IFile;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.progress.UIJob;
 
import com.openedge.pdt.core.template.TemplateGenerationException;
import com.proalpha.pds.templates.data.IGeneratorDataContainer;
import com.proalpha.pds.templates.helper.IDEUtils;
import com.proalpha.pds.templates.jet.IpaTemplateGenerator;
import com.proalpha.pds.templates.jet.JetTemplateFactory;
 
public class ContentGenerator extends UIJob {
 
    public static final String JOB_FAMILY = "com.proalpha.java.oea.plugins.generator.AbstractGenerator";
 
    private final IpaTemplateGenerator templateGenerator;
 
    private final IGeneratorDataContainer dataContainer;
 
    private Boolean isOpenFileInEditor = Boolean.TRUE;
 
    public ContentGenerator(String name, IGeneratorDataContainer dataContainer, String templateGeneratorName)
            throws Exception {
 
        this(name, dataContainer,
                JetTemplateFactory.getFactoryInstance(dataContainer.getVersion()).getTemplate(templateGeneratorName));
 
    }
 
    public ContentGenerator(String name, IGeneratorDataContainer dataContainer, IpaTemplateGenerator templateGenerator)
            throws Exception {
        super(name);
 
        this.dataContainer = dataContainer;
        this.templateGenerator = templateGenerator;
        setUser(true);
        setPriority(Job.SHORT);
 
    }
 
    private void generate(IProgressMonitor monitor) throws Exception {
 
        IFile outputFile = dataContainer.getOutputFile();
 
        if (templateGenerator == null) {
            throw new TemplateGenerationException("Template generator is not set.");
        }
 
        String outputContent = templateGenerator.generate(dataContainer.getDataAsMap());
 
        File file = outputFile.getLocation().toFile();
        ContentWriter contentWriter = new ContentWriter(file, outputContent);
 
        contentWriter.writeContentToHarddisk();
 
        if (isOpenFileInEditor) {
            IDEUtils.openFileInEditor(monitor, outputFile);
        }
    }
 
    @Override
    public IStatus runInUIThread(IProgressMonitor monitor) {
        monitor.beginTask("User Exit Include File", 2);
        monitor.worked(1);
 
        try {
 
            if (!dataContainer.isGenerationConfirmed()) {
                return Status.CANCEL_STATUS;
            }
 
            generate(monitor);
 
            monitor.done();
 
            return Status.OK_STATUS;
 
        } catch (Exception e) {
            MessageDialog.openInformation(PlatformUI.getWorkbench().getDisplay().getActiveShell(), "NO USER EXITS",
                    e.getMessage());
            return Status.CANCEL_STATUS;
        }
    }
 
    @Override
    public boolean belongsTo(Object family) {
        if (family instanceof String) {
            return ((String) family).equals(JOB_FAMILY);
        } else {
            return false;
        }
    }
 
    public void disableOpenFileInEditor() {
 
        isOpenFileInEditor = Boolean.FALSE;
 
    }
 
    public void enableOpenFileInEditro() {
 
        isOpenFileInEditor = Boolean.TRUE;
 
    }
 
}