Activator

/**
 * Copyright By Grandsoft Company Limited.
 * 2012-1-16 下午02:18:02
 */
package gboat2.base.core;

import gboat2.base.core.service.ModuleService;
import gboat2.base.core.util.SpringContextUtil;
import gboat2.base.core.validate.PrivilegeCheckServiceFactory;
import gboat2.base.core.web.SystemButtonProviderTracker;

import java.io.File;
import java.io.FileFilter;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.jar.JarFile;
import java.util.zip.ZipEntry;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;

import org.apache.commons.lang3.StringUtils;
import org.osgi.framework.Bundle;
import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;
import org.osgi.framework.BundleEvent;
import org.osgi.framework.BundleListener;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

/**
 *
 * gboat2.base.core的Activator
 * @author lysming
 * @since jdk1.6
 * @date 2012-1-16
 *
 */

public class Activator implements BundleActivator, BundleListener {

	private static Logger logger = LoggerFactory.getLogger(Activator.class);

	public static Bundle LOCAL_BUNDLE;

	private static final String HEADER_STRUTS_ENABLED = "Struts2-Enabled";

	// true 为项目bundle, false 为产品bundle
	public static final String HEADER_PROJECT_ENABLED = "Project-Enabled";

	// 扩展的bundles，多个用逗号隔开
	public static final String HEADER_EXTENDER_BUNDLES = "Extend-Bundles";

	public static Map<String, Map<String, String>> pomCache = new HashMap<String, Map<String, String>>();

	/**
	  * @see org.osgi.framework.BundleActivator#start(org.osgi.framework.BundleContext)
	  * @param context BundleContext对象
	  * @throws Exception felix异常
	  */
	public void start(BundleContext context) throws Exception {
		LOCAL_BUNDLE = context.getBundle();

		context.addBundleListener(this);
		context.addBundleListener(SpringContextUtil.getInstance());
		//处理在该Bundle启动之前已经启动的Bundle
		Bundle[] allBundles = context.getBundles();
		GBoatClassLoader loader = GBoatClassLoader.getInstance();
		for (Bundle tempBnd : allBundles) {
			String seTag = (String) tempBnd.getHeaders().get(HEADER_STRUTS_ENABLED);
			if (seTag != null && seTag.equals("true") && Bundle.ACTIVE == tempBnd.getState()) {
				registBundle(tempBnd, loader);
			}

			// 读取pom.xml项目扩展参数
			Map<String, String> value = new HashMap<String, String>();
			String project = tempBnd.getHeaders().get(HEADER_PROJECT_ENABLED);
			if (StringUtils.isEmpty(project)) continue;
			String bundles = tempBnd.getHeaders().get(HEADER_EXTENDER_BUNDLES);
			value.put(HEADER_PROJECT_ENABLED, project);
			value.put(HEADER_EXTENDER_BUNDLES, bundles);

			pomCache.put(tempBnd.getSymbolicName(), value);
		}
	}

	public void stop(BundleContext context) throws Exception {
		context.removeBundleListener(this);
		PrivilegeCheckServiceFactory.close();
		SystemButtonProviderTracker.close();
	}

	public void bundleChanged(BundleEvent event) {
		GBoatClassLoader loader = GBoatClassLoader.getInstance();
		if (event.getType() == BundleEvent.STARTED) {
			Bundle bundle = event.getBundle();
			String seTag = (String) bundle.getHeaders().get(HEADER_STRUTS_ENABLED);
			if (seTag != null && seTag.equals("true")) {
				registBundle(bundle, loader);
			}
		}
		//清空SpringContextUtil中的ApplicationContext，否则重启后无法设置
		if ((event.getType() & BundleEvent.STOPPED) != 0) {
			ModuleService.unRegistModule(event.getBundle());
		}
	}

	//注册Struts类型的Bundle,解析所有该Bundle下Action的Module
	private void registBundle(Bundle bundle, GBoatClassLoader loader) {
		List<String> pkgForBundle = getPkgForBundle(bundle);
		for (String tempPkg : pkgForBundle) {
			loader.add(tempPkg, bundle);
			ModuleService.registModule(tempPkg);
		}
	}

	/**
	 * 取出某个Bundle下所有Action的package
	 * @param bundle bundle对象
	 * @return 该Bundle下Action的package
	 */
	private List<String> getPkgForBundle(Bundle bundle) {
		List<String> pkgList = new ArrayList<String>();
		try {
			logger.debug("register actions for " + bundle.getSymbolicName() + " ...");
			if (bundle.getLocation().startsWith("file:")) {
				JarFile file = new JarFile(bundle.getLocation().replaceFirst("file:[/|\\\\]", ""));

				ZipEntry entry = file.getEntry("struts.xml");
				if (entry != null) {
					parseActionPackages(file.getInputStream(entry), pkgList);
				} else {
					logger.debug("no struts.xml found.");
				}
			} else if (bundle.getLocation().startsWith("jardir:")) {//自动部署工具安装的bundle
				File dir = new File(bundle.getLocation().replaceFirst("jardir:", ""));
				if (dir.isDirectory()) {
					File[] configFiles = dir.listFiles(new FileFilter() {

						@Override
						public boolean accept(File pathname) {
							if (pathname.getName().equals("struts.xml")) {
								return true;
							}
							return false;
						}
					});
					if (configFiles.length == 0) {
						logger.debug("no struts.xml found.");
					} else {
						File strutsConfig = configFiles[0];
						FileInputStream is = new FileInputStream(strutsConfig);
						parseActionPackages(is, pkgList);
					}
				} else {
					logger.debug(bundle.getLocation() + " is not a direction");
				}
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
		return pkgList;
	}

	private void parseActionPackages(InputStream io, List<String> pkgList) {
		try {
			DocumentBuilderFactory domFactory = DocumentBuilderFactory.newInstance();
			domFactory.setValidating(false);
			domFactory.setNamespaceAware(true); // never forget this!
			DocumentBuilder builder = domFactory.newDocumentBuilder();

			Document doc = builder.parse(io);
			NodeList ns = doc.getElementsByTagName("constant");
			Element element;
			for (int i = 0; i < ns.getLength(); i++) {
				element = (Element) ns.item(i);
				if (element.getAttribute("name").equals("struts.convention.package.locators.basePackage")) {
					String pkg = element.getAttribute("value");
					logger.debug("packages defined in struts.xml is " + pkg);
					/**
					 * 修改Struts 注册 constant 的action package 方法：
					 * 用","分割，可以注册多个action 包
					 * modify by wanghb
					 */
					if (StringUtils.isNotEmpty(pkg)) {
						String[] pkgs = pkg.split(",");
						for(String pkgStr : pkgs) {
							pkgStr = pkgStr.trim();
							if (StringUtils.isNotEmpty(pkgStr)) {
								pkgList.add(pkgStr);
							}
						}
					}
					break;
				}
			}
		} catch (ParserConfigurationException e) {
			logger.error(e.getMessage(), e);
		} catch (SAXException e) {
			logger.error(e.getMessage(), e);
		} catch (IOException e) {
			logger.error(e.getMessage(), e);
		}
	}

}