plugins/org.eclipse.objectteams.otequinox.hook/src/org/eclipse/objectteams/otequinox/hook/ClassScanner.java

/**********************************************************************
 * This file is part of "Object Teams Development Tooling"-Software
 * 
 * Copyright 2008, 2010 Technical University Berlin, Germany.
 * 
 * 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
 * $Id: ClassScanner.java 23461 2010-02-04 22:10:39Z stephan $
 * 
 * Please visit http://www.eclipse.org/objectteams for updates and contact.
 * 
 * Contributors:
 * Technical University Berlin - Initial API and implementation
 **********************************************************************/
package org.eclipse.objectteams.otequinox.hook;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;

import org.eclipse.objectteams.otequinox.internal.hook.Util;
import org.osgi.framework.Bundle;

/**
 * Bridge for the TransformerPlugin by which it can access the ObjectTeamsTransformer
 * without accessing that OTRE class.
 * 
 * @author stephan
 * @since 1.2.0
 */
@SuppressWarnings("nls")
public class ClassScanner 
{
	// default is "on", leave this switch for trouble shooting and profiling:
	public static final boolean REPOSITORY_USE_RESOURCE_LOADER = !"off".equals(System.getProperty("otequinox.repository.hook"));

	// access to the OTRE
	IOTTransformer transformerService;
	
	// collect class names recorded by readOTAttributes:
	
	//   * this version used by the MasterTeamLoader.loadTeams:
	HashMap<String,ArrayList<String>> baseClassNamesByTeam = new HashMap<String, ArrayList<String>>();
	//   * these fields used by TransformerHook.processClass (collected over multiple readOTAttributes):
	ArrayList<String> allBaseClassNames = new ArrayList<String>();
	ArrayList<String> roleClassNames = new ArrayList<String>();


	public ClassScanner(IOTTransformer transformerService) {
		this.transformerService = transformerService;
	}

	/** 
	 * Read all OT byte code attributes for the specified class.
	 * While doing so the names of roles and adapted base classes are collected.
	 * 
	 * @param bundle    where to look
	 * @param className the class to investigate (team or role)
	 * @param loader    the loader (could be null) to use for further classFile lookup
	 * @throws ClassFormatError
	 * @throws IOException
	 * @throws ClassNotFoundException the team or role class was not found
	 */
	public void readOTAttributes(Bundle bundle, String className, ClassLoader loader)
			throws ClassFormatError, IOException, ClassNotFoundException 
	{
		if (!REPOSITORY_USE_RESOURCE_LOADER)
			loader = null;
		URL classFile = bundle.getResource(className.replace('.', '/')+".class");
		if (classFile == null) 
			throw new ClassNotFoundException(className);
		InputStream inputStream = classFile.openStream();
		Object token;
		try {
			token = this.transformerService.readOTAttributes(inputStream, classFile.getFile(), loader);
		} finally {
			inputStream.close();
		}
		Collection<String> currentBaseNames = this.transformerService.fetchAdaptedBases(token); // destructive read
		if (currentBaseNames != null) {
			// store per team:
			ArrayList<String> basesPerTeam = this.baseClassNamesByTeam.get(className);
			if (basesPerTeam == null) {
				basesPerTeam = new ArrayList<String>();
				this.baseClassNamesByTeam.put(className, basesPerTeam);
			}
			basesPerTeam.addAll(currentBaseNames);
			// accumulated store:
			allBaseClassNames.addAll(currentBaseNames);
		}
		readMemberTypeAttributes(bundle, className, loader);
	}
	
	/** 
	 * Get the names of the base classes adapted by the given team and 
	 * encountered while reading the byte code attributes.
	 * (Destructive read). 
	 */
	public Collection<String> getCollectedBaseClassNames(String teamName) {
		return this.baseClassNamesByTeam.remove(teamName);
	}

	/** 
	 * Get the names of all adapted base classes encountered while reading the byte code attributes.
	 * (Destructive read). 
	 */
	public Collection<String> getCollectedBaseClassNames() {
		try {
			return this.allBaseClassNames;
		} finally {
			this.allBaseClassNames = new ArrayList<String>();
		}
	}
	
	/** 
	 * Get the names of all member roles encountered while reading the byte code attributes. 
	 * (Destructive read).
	 */
	public Collection<String> getCollectedRoleClassNames() {
		return this.roleClassNames;
	}
	
	/*
	 * Recurse into member types scanning OT attributes.
	 */
	private void readMemberTypeAttributes(Bundle                 bundle,
										  String                 className, 
										  ClassLoader			 resourceLoader)
	{
		List<String> roles = this.transformerService.getRolesPerTeam(className);
		if (roles != null) {
			ILogger logger = HookConfigurator.getLogger();
			for (String roleName: roles) {
				logger.log(Util.OK, "scanning role "+roleName);
				try {
					this.roleClassNames.add(roleName);
					readOTAttributes(bundle, roleName, resourceLoader);					
				} catch (Throwable t) {
					logger.log(t, "Failed to read OT-Attributes of role "+roleName);
				}
				readMemberTypeAttributes(bundle, roleName, resourceLoader);
			}
		}
	}
};