/*******************************************************************************
* Copyright (c) 2011, 2011 IBM Corporation and others.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Bruno Medeiros - initial API and implementation
*******************************************************************************/
package mmrnmhrm.core.engine_client;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.Collection;
import melnorme.utilbox.misc.ArrayUtil;
import mmrnmhrm.core.model_elements.DeeModelEngine;
import mmrnmhrm.core.search.SourceModuleFinder;
import org.eclipse.dltk.codeassist.ScriptSelectionEngine;
import org.eclipse.dltk.compiler.env.IModuleSource;
import org.eclipse.dltk.core.IMember;
import org.eclipse.dltk.core.IModelElement;
import org.eclipse.dltk.core.ISourceModule;
import org.eclipse.dltk.core.ModelException;
import dtool.ast.ASTNode;
import dtool.ast.ASTNodeFinder;
import dtool.ast.definitions.DefSymbol;
import dtool.ast.definitions.DefUnit;
import dtool.ast.definitions.INamedElement;
import dtool.ast.definitions.Module;
import dtool.ast.references.Reference;
import dtool.engine.modules.IModuleResolver;
/**
* XXX: what is the exact contract of this class. Do returned model elements have to exist?
* Current known uses:
* - Initializing text for ScripSearchPage
* - Picking element for type hierarchy action
*/
public class DeeSelectionEngine extends ScriptSelectionEngine {
public IModelElement[] select(IModuleSource sourceUnit, int offset) {
return select(sourceUnit, offset, 0);
}
// BM: param i is something like endOffset-1, due to weird API
@Override
public IModelElement[] select(IModuleSource sourceUnit, int offset, int i) {
ISourceModule sourceModule = (ISourceModule) sourceUnit.getModelElement();
Path filePath = DToolClient_Bad.getFilePathOrNull(sourceModule);
if(filePath == null) {
return null;
}
Module deeModule = DToolClient.getDefaultModuleCache().getExistingParsedModuleNode(filePath);
ASTNode node = ASTNodeFinder.findElement(deeModule, offset);
if(node instanceof DefSymbol) {
DefUnit defUnit = ((DefSymbol) node).getDefUnit();
try {
IMember modelElement = DeeModelEngine.findCorrespondingModelElement(defUnit, sourceModule);
return modelElement == null ? null : new IModelElement[] { modelElement };
} catch (ModelException e) {
return null;
}
}
if(!(node instanceof Reference)) {
return new IModelElement[0];
}
Reference ref = (Reference) node;
IModuleResolver moduleResolver = DToolClient_Bad.getResolverFor(filePath);
Collection<INamedElement> defElements = ref.findTargetDefElements(moduleResolver, false);
// We assume namespace Parent is the same
if(defElements == null) {
return new IModelElement[0];
}
ArrayList<IModelElement> list = new ArrayList<IModelElement>();
for (INamedElement defElement : defElements) {
DefUnit defUnit = defElement.resolveDefUnit();
IMember modelElement = getModelElement(defUnit, sourceModule);
if(modelElement != null) {
list.add(modelElement);
}
}
return ArrayUtil.createFrom(list, IModelElement.class);
}
protected IMember getModelElement(DefUnit defUnit, ISourceModule sourceModule) {
if(defUnit == null) {
return null;
}
Module module = defUnit.getModuleNode();
if(module == null) {
return null;
}
try {
ISourceModule moduleUnit = SourceModuleFinder.findModuleUnit(module, sourceModule.getScriptProject());
return DeeModelEngine.findCorrespondingModelElement(defUnit, moduleUnit);
} catch (ModelException e) {
return null;
}
}
}