trunk/ruta-ep-ide/src/main/java/org/apache/uima/ruta/ide/core/parser/RutaSourceElementRequestVisitor.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.uima.ruta.ide.core.parser;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.apache.uima.ruta.ide.parser.ast.RutaBasicAnnotationType;
import org.apache.uima.ruta.ide.parser.ast.RutaImportStatement;
import org.apache.uima.ruta.ide.parser.ast.RutaPackageDeclaration;
import org.apache.uima.ruta.ide.parser.ast.RutaTypeConstants;
import org.apache.uima.ruta.ide.parser.ast.RutaTypeDeclaration;
import org.apache.uima.ruta.ide.parser.ast.RutaVariableDeclaration;
import org.eclipse.dltk.ast.ASTNode;
import org.eclipse.dltk.ast.Modifiers;
import org.eclipse.dltk.ast.PositionInformation;
import org.eclipse.dltk.ast.declarations.Argument;
import org.eclipse.dltk.ast.declarations.FieldDeclaration;
import org.eclipse.dltk.ast.declarations.MethodDeclaration;
import org.eclipse.dltk.ast.declarations.ModuleDeclaration;
import org.eclipse.dltk.ast.expressions.Expression;
import org.eclipse.dltk.ast.references.SimpleReference;
import org.eclipse.dltk.ast.references.VariableReference;
import org.eclipse.dltk.ast.statements.Statement;
import org.eclipse.dltk.compiler.ISourceElementRequestor;
import org.eclipse.dltk.compiler.SourceElementRequestVisitor;

public class RutaSourceElementRequestVisitor extends SourceElementRequestVisitor {

  private static class TypeField {
    private String fName;

    private String fInitValue;

    private PositionInformation fPos;

    private Expression fExpression;

    private ASTNode fToNode;

    private ASTNode declaredIn; // The node where the declaration was found

    // (should be either class or method node)
    TypeField(String name, String initValue, PositionInformation pos, Expression expression,
            ASTNode toNode, ASTNode declaredIn) {

      this.fName = name;
      this.fInitValue = initValue;
      this.fPos = pos;
      this.fExpression = expression;
      this.fToNode = toNode;
      this.declaredIn = declaredIn;
    }

    String getName() {

      return this.fName;
    }

    String getInitValue() {

      return this.fInitValue;
    }

    PositionInformation getPos() {

      return this.fPos;
    }

    Expression getExpression() {

      return this.fExpression;
    }

    ASTNode getToNode() {

      return this.fToNode;
    }

    @Override
    public boolean equals(Object obj) {

      if (obj instanceof TypeField) {
        TypeField second = (TypeField) obj;
        return second.fName.equals(this.fName) && second.fToNode.equals(this.fToNode);
      }
      return super.equals(obj);
    }

    @Override
    public String toString() {

      return this.fName;
    }

    public ASTNode getDeclaredIn() {
      return declaredIn;
    }

  }

  private static String ANONYMOUS_LAMBDA_FORM_MARKER = "<anonymous>";

  // Used to prehold fields if adding in methods.
  private List fNotAddedFields = new ArrayList();

  private String lastLambdaFormName = ANONYMOUS_LAMBDA_FORM_MARKER;

  /**
   * Used to determine duplicate names.
   */
  private Map fTypeVariables = new HashMap();

  //
  public RutaSourceElementRequestVisitor(ISourceElementRequestor requestor) {
    super(requestor);
  }

  protected String makeLanguageDependentValue(Expression value) {
    String outValue = "";
    return outValue;
  }

  @Override
  protected void onEndVisitMethod(MethodDeclaration method) {
    if (fNotAddedFields.size() >= 1) {
      TypeField typeField = (TypeField) fNotAddedFields.get(0);
      if (null != typeField && typeField.getDeclaredIn().equals(method)) {
        Iterator i = this.fNotAddedFields.iterator();
        while (i.hasNext()) {
          TypeField field = (TypeField) i.next();
          if (canAddVariables(field.getToNode(), field.getName())) {

            PositionInformation pos = field.getPos();

            ISourceElementRequestor.FieldInfo info = new ISourceElementRequestor.FieldInfo();
            info.modifiers = Modifiers.AccStatic;
            info.name = field.getName();
            info.nameSourceEnd = pos.nameEnd - 2;
            info.nameSourceStart = pos.nameStart;
            info.declarationStart = pos.sourceStart;
            this.fRequestor.enterField(info);
            this.fRequestor.exitField(pos.sourceEnd);

          }
        }
        this.fNotAddedFields.clear();
      }
    }
  }

  @Override
  public boolean visit(Statement statement) throws Exception {
    super.visit(statement);
    if (statement instanceof RutaPackageDeclaration) {
      this.processPackage(statement);
      super.fNodes.pop();
      return false;
    }
    if (statement instanceof FieldDeclaration) {
      FieldDeclaration fieldDecl = (FieldDeclaration) statement;
      processFieldDeclaration(fieldDecl.getRef(), fieldDecl);
      super.fNodes.pop();
      return false;
    }
    // TODO handle tm-import statement
    if (statement instanceof RutaImportStatement) {
      RutaImportStatement tmImport = (RutaImportStatement) statement;
      super.fNodes.pop();
      return false;
    }
    return true;
  }

  @Override
  public boolean visit(Expression expression) throws Exception {
    if (expression instanceof VariableReference) {
      VariableReference varRef = (VariableReference) expression;
      this.fRequestor.acceptFieldReference(varRef.getName(), varRef.sourceStart());
    }
    return super.visit(expression);
  }

  private void processPackage(Statement statement) {
    RutaPackageDeclaration pack = (RutaPackageDeclaration) statement;
    this.fRequestor.acceptPackage(pack.getNameStart(), pack.getNameEnd(), pack.getName());
  }

  private void processFieldDeclaration(SimpleReference variableIDRef, Statement fullDeclaration) {
    int modifier = Modifiers.AccDefault;
    if (fullDeclaration instanceof RutaVariableDeclaration) {
      modifier = Modifiers.AccPrivate;
      modifier |= ((RutaVariableDeclaration) fullDeclaration).getKind();
    } else if (fullDeclaration instanceof RutaBasicAnnotationType) {
      modifier = Modifiers.AccConstant;
      modifier |= RutaTypeConstants.RUTA_TYPE_AT;
    } else if (fullDeclaration instanceof RutaTypeDeclaration) {
      modifier = Modifiers.AccPublic;
      modifier |= RutaTypeConstants.RUTA_TYPE_AT;
    }
    if (canAddVariables((ASTNode) this.fNodes.peek(), variableIDRef.getName())) {
      ISourceElementRequestor.FieldInfo info = new ISourceElementRequestor.FieldInfo();
      info.modifiers = modifier;
      info.name = variableIDRef.getName();
      info.nameSourceEnd = variableIDRef.sourceEnd() - 1;
      info.nameSourceStart = variableIDRef.sourceStart();
      info.declarationStart = variableIDRef.sourceStart();
      this.fRequestor.enterField(info);
      if (fullDeclaration != null) {
        this.fRequestor.exitField(fullDeclaration.sourceEnd());
      } else {
        this.fRequestor.exitField(variableIDRef.sourceEnd());
      }
    }
  }

  private boolean canAddVariables(ASTNode type, String name) {
    if (this.fTypeVariables.containsKey(type)) {
      List variables = (List) this.fTypeVariables.get(type);
      if (variables.contains(name)) {
        return false;
      }
      variables.add(name);
      return true;
    } else {
      List variables = new ArrayList();
      variables.add(name);
      this.fTypeVariables.put(type, variables);
      return true;
    }
  }

  @Override
  public boolean endvisit(Statement s) throws Exception {
    return true;
  }

  @Override
  public boolean visit(MethodDeclaration method) throws Exception {
    this.fNodes.push(method);
    List<Argument> args = method.getArguments();
    String[] parameter = new String[args.size()];
    for (int a = 0; a < args.size(); a++) {
      Argument arg = args.get(a);
      parameter[a] = arg.getName();
    }

    ISourceElementRequestor.MethodInfo mi = new ISourceElementRequestor.MethodInfo();
    mi.parameterNames = parameter;
    mi.name = method.getName();
    mi.modifiers = method.getModifiers();
    mi.nameSourceStart = method.getNameStart();
    mi.nameSourceEnd = method.getNameEnd() - 1;
    mi.declarationStart = method.sourceStart();

    this.fInMethod = true;
    this.fCurrentMethod = method;
    this.fRequestor.enterMethod(mi);
    return true;
  }

  @Override
  public boolean visit(ModuleDeclaration declaration) throws Exception {
    return super.visit(declaration);
  }
}