source: default/v2/trunk/de.ugoe.cs.swe.bnftools.ebnf.ui/src/de/ugoe/cs/swe/bnftools/ui/handler/BT_GenerationHandler.java @ 72

package de.ugoe.cs.swe.bnftools.ui.handler;

import java.io.File;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.IHandler;
import org.eclipse.core.commands.common.NotDefinedException;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IWorkspace;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EcorePackage;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.xml.type.SimpleAnyType;
import org.eclipse.emf.ecore.xml.type.XMLTypeFactory;
import org.eclipse.ui.IEditorInput;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.part.FileEditorInput;
import org.eclipse.xtext.builder.EclipseResourceFileSystemAccess2;
import org.eclipse.xtext.generator.IGenerator;
import org.eclipse.xtext.generator.OutputConfiguration;
import org.eclipse.xtext.resource.IResourceDescriptions;
import org.eclipse.xtext.ui.resource.IResourceSetProvider;

import com.google.inject.Inject;
import com.google.inject.Provider;

public class BT_GenerationHandler  extends AbstractHandler implements IHandler {

        @Inject
    private IGenerator generator;
 
    @Inject
    private Provider<EclipseResourceFileSystemAccess2> fileAccessProvider;
     
    @Inject
    IResourceDescriptions resourceDescriptions;
     
    @Inject
    IResourceSetProvider resourceSetProvider;
     
    @Override
    public Object execute(ExecutionEvent event) throws ExecutionException {
         
          //used to save the command name (defined in extensions)
          String mode = "";
         
          try {
             mode = event.getCommand().getName();
          } catch (NotDefinedException e1) {
          // TODO Auto-generated catch block
            e1.printStackTrace();
          }
         
          // stuff to get the workbench and current file
          IWorkbench wb = PlatformUI.getWorkbench();
          IWorkbenchWindow window = wb.getActiveWorkbenchWindow();
          IWorkbenchPage page = window.getActivePage();
          IEditorPart editor = page.getActiveEditor();
          IEditorInput input = editor.getEditorInput();
          IPath path = ((FileEditorInput) input).getPath();
         
          String name = path.toString();
          File myfile = new File(name);
         
          IWorkspace workspace= ResourcesPlugin.getWorkspace();
          IPath location= Path.fromOSString(myfile.getAbsolutePath());
          IFile file= workspace.getRoot().getFileForLocation(location);
         
          IProject project = file.getProject();
          IFolder srcGenFolder = project.getFolder("src-gen");
          if (!srcGenFolder.exists()) {
            try {
              srcGenFolder.create(true, true, new NullProgressMonitor());
            } catch (CoreException e) {
            return null;
            }
          }
         
          final EclipseResourceFileSystemAccess2 fsa = fileAccessProvider.get();
         
          fsa.setProject(project);
         
          //same stuff
          //fsa.setOutputPath("src-gen");
          fsa.setOutputPath(srcGenFolder.getName().toString());
         
          fsa.setMonitor(new NullProgressMonitor());
          Map<String, OutputConfiguration> teste = fsa.getOutputConfigurations();
         
          Iterator<Entry<String, OutputConfiguration>> it = teste.entrySet().iterator();
         
          //make a new Outputconfiguration <- needed
          while(it.hasNext()){
         
            Entry<String, OutputConfiguration> next = it.next();
            OutputConfiguration out = next.getValue();
            out.isOverrideExistingResources();
            out.setCreateOutputDirectory(true); // <--- do not touch this
         
          }
          // ----->
         
          URI uri = URI.createPlatformResourceURI(file.getFullPath().toString(), true);
          ResourceSet rs = resourceSetProvider.get(project);
          Resource r = rs.getResource(uri, true);
         
          // to pass a String inside a resource i have to wrap it in a EOBject
          SimpleAnyType wrapper = XMLTypeFactory.eINSTANCE.createSimpleAnyType();
          wrapper.setInstanceType(EcorePackage.eINSTANCE.getEString());
          wrapper.setValue(mode);
          //
         
          // add string to resource
          r.getContents().add(wrapper);
         
          generator.doGenerate(r, fsa);
         
          return null;
         
        }
    
    @Override
    public boolean isEnabled() {
        return true;
    }
 
}