/**

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * Copyright 2013-2014 Oracle and/or its affiliates. All rights reserved.

 *

 * Oracle and Java are registered trademarks of Oracle and/or its affiliates.

 * Other names may be trademarks of their respective owners.

 *

 * The contents of this file are subject to the terms of either the GNU

 * General Public License Version 2 only ("GPL") or the Common

 * Development and Distribution License("CDDL") (collectively, the

 * "License"). You may not use this file except in compliance with the

 * License. You can obtain a copy of the License at

 * http://www.netbeans.org/cddl-gplv2.html

 * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the

 * specific language governing permissions and limitations under the

 * License.  When distributing the software, include this License Header

 * Notice in each file and include the License file at

 * nbbuild/licenses/CDDL-GPL-2-CP.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the GPL Version 2 section of the License file that

 * accompanied this code. If applicable, add the following below the

 * License Header, with the fields enclosed by brackets [] replaced by

 * your own identifying information:

 * "Portions Copyrighted [year] [name of copyright owner]"

 *

 * Contributor(s):

 *

 * The Original Software is NetBeans. The Initial Developer of the Original

 * Software is Oracle. Portions Copyright 2013-2014 Oracle. All Rights Reserved.

 *

 * If you wish your version of this file to be governed by only the CDDL

 * or only the GPL Version 2, indicate your decision by adding

 * "[Contributor] elects to include this software in this distribution

 * under the [CDDL or GPL Version 2] license." If you do not indicate a

 * single choice of license, a recipient has the option to distribute

 * your version of this file under either the CDDL, the GPL Version 2 or

 * to extend the choice of license to its licensees as provided above.

 * However, if you add GPL Version 2 code and therefore, elected the GPL

 * Version 2 license, then the option applies only if the new code is

 * made subject to such option by the copyright holder.

 */

package org.netbeans.html.equinox.agentclass;

import java.lang.instrument.IllegalClassFormatException;

import java.security.ProtectionDomain;

import java.util.ArrayList;

import java.util.logging.Logger;

import org.eclipse.osgi.baseadaptor.BaseData;

import org.eclipse.osgi.baseadaptor.HookConfigurator;

import org.eclipse.osgi.baseadaptor.HookRegistry;

import org.eclipse.osgi.baseadaptor.bundlefile.BundleEntry;

import org.eclipse.osgi.baseadaptor.hooks.ClassLoadingHook;

import org.eclipse.osgi.baseadaptor.loader.BaseClassLoader;

import org.eclipse.osgi.baseadaptor.loader.ClasspathEntry;

import org.eclipse.osgi.baseadaptor.loader.ClasspathManager;

import org.eclipse.osgi.framework.adaptor.BundleProtectionDomain;

import org.eclipse.osgi.framework.adaptor.BundleWatcher;

import org.eclipse.osgi.framework.adaptor.ClassLoaderDelegate;

import org.osgi.framework.Bundle;

import org.osgi.framework.BundleContext;

import org.osgi.framework.wiring.BundleWiring;

public class AgentHook implements HookConfigurator, BundleWatcher, ClassLoadingHook {

    private static final Logger LOG = Logger.getLogger(AgentHook.class.getName());

	private boolean all;

	@Override

	public void addHooks(HookRegistry hookRegistry) {

		LOG.info("Agent hook for Equinox initialized!");

		hookRegistry.addWatcher(this);

		hookRegistry.addClassLoadingHook(this);

	}

	@Override

	public void watchBundle(Bundle bundle, int type) {

		if (!all) {

			BundleContext c = bundle.getBundleContext();

			if (c != null) {

				Bundle[] arr = bundle.getBundleContext().getBundles();

				for (Bundle b : arr) {

					agentBundle(b);

				}

				all = true;

			}

		}

		if (type == BundleWatcher.END_ACTIVATION) {

			agentBundle(bundle);

		}

	}

	private void agentBundle(Bundle bundle) {

		String agentClass = (String)bundle.getHeaders().get("Agent-Class");

		if (agentClass != null) {

			Class<?> agent;

			try {

				agent = bundle.loadClass(agentClass);

				NbInstrumentation.registerAgent(agent.getClassLoader(), agent.getName());

			} catch (ClassNotFoundException e) {

				throw new IllegalStateException(e);

			}

		}

	}

	@Override

	public byte[] processClass(String name, byte[] bytes,

			ClasspathEntry ce, BundleEntry entry,

			ClasspathManager manager) {

        final BaseData bd = ce.getBaseData();

        if (bd == null) {

            return bytes;

        }

        final Bundle b = bd.getBundle();

        if (b == null) {

            return bytes;

        }

        BundleWiring w = (BundleWiring)b.adapt(BundleWiring.class);

        if (w == null) {

            return bytes;

        }

        ClassLoader loader = w.getClassLoader();

		try {

			return NbInstrumentation.patchByteCode(loader, name, ce.getDomain(), bytes);

		} catch (IllegalClassFormatException e) {

			return bytes;

		}

	}

	@Override

	public boolean addClassPathEntry(ArrayList cpEntries,

			String cp, ClasspathManager hostmanager, BaseData sourcedata,

			ProtectionDomain sourcedomain) {

		// TODO Auto-generated method stub

		return false;

	}

	@Override

	public String findLibrary(BaseData data, String libName) {

		// TODO Auto-generated method stub

		return null;

	}

	@Override

	public ClassLoader getBundleClassLoaderParent() {

		// TODO Auto-generated method stub

		return null;

	}

	@Override

	public BaseClassLoader createClassLoader(ClassLoader parent,

			ClassLoaderDelegate delegate, BundleProtectionDomain domain,

			BaseData data, String[] bundleclasspath) {

		// TODO Auto-generated method stub

		return null;

	}

	@Override

	public void initializedClassLoader(BaseClassLoader baseClassLoader,

			BaseData data) {

		// TODO Auto-generated method stub

	}

}