AsmReferenceRecorder.java

/*******************************************************************************
 * Copyright (c) 2011 Gerd Wuetherich (gerd@gerd-wuetherich.de).
 * 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:
 *     Gerd Wuetherich (gerd@gerd-wuetherich.de) - initial API and implementation
 ******************************************************************************/
package org.bundlemaker.core.parser.bytecode.asm;

import org.bundlemaker.core.jtype.IModifiableType;
import org.bundlemaker.core.jtype.IParsableTypeResource;
import org.bundlemaker.core.jtype.IReferenceRecorder;
import org.bundlemaker.core.jtype.IType;
import org.bundlemaker.core.jtype.ITypeResource;
import org.bundlemaker.core.jtype.JavaTypeUtils;
import org.bundlemaker.core.jtype.ReferenceAttributes;
import org.bundlemaker.core.jtype.ReferenceType;
import org.bundlemaker.core.jtype.TypeEnum;
import org.bundlemaker.core.spi.parser.IParsableResource;
import org.eclipse.core.runtime.Assert;

/**
 * <p>
 * </p>
 * 
 * @author Gerd Wütherich (gerd@gerd-wuetherich.de)
 */
public class AsmReferenceRecorder implements IReferenceRecorder {

  /** - */
  private IParsableResource _resource;

  /** - */
  private IParsableResource _enclosingClassFileResource;

  /** - */
  private IModifiableType     _bundleMakerType;

  /** - */
  private String              _fullyQualifiedTypeName;

  /** - */
  private String              _fullyQualifiedEnclosingTypeName;

  /**
   * <p>
   * Creates a new instance of type {@link AsmReferenceRecorder}.
   * </p>
   * 
   * @param resource
   * @param enclosingClassFileResource
   */
  public AsmReferenceRecorder(IParsableResource resource, IParsableResource enclosingClassFileResource) {

    Assert.isNotNull(resource);
    Assert.isNotNull(enclosingClassFileResource);

    _resource = resource;
    _enclosingClassFileResource = enclosingClassFileResource;
  }

  /**
   * <p>
   * </p>
   * 
   * @param fullyQualifiedName
   */
  public void recordContainedType(String fullyQualifiedName, TypeEnum typeEnum, boolean abstractType) {

    try {

      //
      Assert.isNotNull(fullyQualifiedName);

      // set the type name
      _fullyQualifiedTypeName = fullyQualifiedName;

      //
      if (JavaTypeUtils.isLocalOrAnonymousTypeName(fullyQualifiedName)) {

        IType type = _resource.adaptAs(IParsableTypeResource.class).getOrCreateType(fullyQualifiedName, typeEnum, abstractType);
        _resource.adaptAs(IParsableTypeResource.class).setPrimaryType(type);

        // we have to check for the existence of contained types:
        // in the rare case of an (erroneous) non-set type we fall back
        // on the resource type
        if (!(_enclosingClassFileResource.adaptAs(ITypeResource.class).getContainedTypes().isEmpty())) {

          //
          _bundleMakerType = ((IModifiableType[]) _enclosingClassFileResource.adaptAs(ITypeResource.class).getContainedTypes().toArray(
              new IModifiableType[0]))[0];

          _fullyQualifiedEnclosingTypeName = _bundleMakerType.getFullyQualifiedName();

        } else {

          // create the fall-back type
          _bundleMakerType = _resource.adaptAs(IParsableTypeResource.class).getOrCreateType(fullyQualifiedName, typeEnum, abstractType);
        }

        // add as sticky
        _enclosingClassFileResource.addStickyResource(_resource);

      } else {

        // create the type
        _bundleMakerType = _resource.adaptAs(IParsableTypeResource.class).getOrCreateType(fullyQualifiedName, typeEnum,abstractType);
      }

    } catch (RuntimeException runtimeException) {

      //
      System.out.println(String.format("Exception while parsing '%s' [enclosing: '%s']", _resource,
          _enclosingClassFileResource));
      runtimeException.printStackTrace();
    }
  }

  @Override
  public void recordReference(String fullyQualifiedName, ReferenceAttributes attributes) {

    try {

      if (fullyQualifiedName != null && !fullyQualifiedName.equals(_fullyQualifiedTypeName)
          && !fullyQualifiedName.equals(_fullyQualifiedEnclosingTypeName)) {

        //
        if (!_resource.equals(_enclosingClassFileResource)) {

          attributes = new ReferenceAttributes(ReferenceType.TYPE_REFERENCE, false, false, false,
              attributes.isCompileTime(), attributes.isRuntimeTime(), attributes.isDirectlyReferenced(),
              attributes.isIndirectlyReferenced());
        }

        //
        org.objectweb.asm.Type objectType = org.objectweb.asm.Type.getObjectType(fullyQualifiedName);

        String name = VisitorUtils.getFullyQualifiedTypeName(objectType);

        _bundleMakerType.recordReference(name, attributes);
      }

    } catch (RuntimeException runtimeException) {

      //
      System.out.println(String.format("Exception while parsing '%s' [enclosing: '%s']", _resource,
          _enclosingClassFileResource));
      runtimeException.printStackTrace();
    }
  }
}