X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;ds=sidebyside;f=org.argeo.osgi.boot%2Fsrc%2Forg%2Fargeo%2Fosgi%2Fa2%2FProvisioningManager.java;fp=org.argeo.osgi.boot%2Fsrc%2Forg%2Fargeo%2Fosgi%2Fa2%2FProvisioningManager.java;h=dd7fa3b0eed89e7c30c2c3f8675fc18edf59718c;hb=79e0a2a5d751c7c077e52f9ee54469656dc96a44;hp=0000000000000000000000000000000000000000;hpb=c155192cfcd5ca355eb933fa3f55dbad6d01b958;p=lgpl%2Fargeo-commons.git

diff --git a/org.argeo.osgi.boot/src/org/argeo/osgi/a2/ProvisioningManager.java b/org.argeo.osgi.boot/src/org/argeo/osgi/a2/ProvisioningManager.java
new file mode 100644
index 000000000..dd7fa3b0e
--- /dev/null
+++ b/org.argeo.osgi.boot/src/org/argeo/osgi/a2/ProvisioningManager.java
@@ -0,0 +1,197 @@
+package org.argeo.osgi.a2;
+
+import java.io.File;
+import java.net.URI;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
+import org.argeo.osgi.boot.OsgiBootUtils;
+import org.eclipse.osgi.launch.EquinoxFactory;
+import org.osgi.framework.Bundle;
+import org.osgi.framework.BundleContext;
+import org.osgi.framework.Constants;
+import org.osgi.framework.Version;
+import org.osgi.framework.launch.Framework;
+import org.osgi.framework.wiring.FrameworkWiring;
+
+/** Loads provisioning sources into an OSGi context. */
+public class ProvisioningManager {
+	BundleContext bc;
+	OsgiContext osgiContext;
+	List<ProvisioningSource> sources = Collections.synchronizedList(new ArrayList<>());
+
+	public ProvisioningManager(BundleContext bc) {
+		this.bc = bc;
+		osgiContext = new OsgiContext(bc);
+		osgiContext.load();
+	}
+
+	protected void addSource(ProvisioningSource source) {
+		sources.add(source);
+	}
+
+	void installWholeSource(ProvisioningSource source) {
+		Set<Bundle> updatedBundles = new HashSet<>();
+		for (A2Contribution contribution : source.listContributions(null)) {
+			for (A2Component component : contribution.components.values()) {
+				A2Module module = component.last().last();
+				Bundle bundle = installOrUpdate(module);
+				if (bundle != null)
+					updatedBundles.add(bundle);
+			}
+		}
+		FrameworkWiring frameworkWiring = bc.getBundle(0).adapt(FrameworkWiring.class);
+		frameworkWiring.refreshBundles(updatedBundles);
+	}
+
+	public void registerSource(String uri) {
+		try {
+			URI u = new URI(uri);
+			if (A2Source.SCHEME_A2.equals(u.getScheme())) {
+				if (u.getHost() == null || "".equals(u.getHost())) {
+					String baseStr = u.getPath();
+					if (File.separatorChar == '\\') {// MS Windows
+						baseStr = baseStr.substring(1).replace('/', File.separatorChar);
+					}
+					Path base = Paths.get(baseStr);
+					FsA2Source source = new FsA2Source(base);
+					source.load();
+					addSource(source);
+				}
+			}
+		} catch (Exception e) {
+			throw new A2Exception("Cannot add source " + uri, e);
+		}
+	}
+
+	public boolean registerDefaultSource() {
+		String frameworkLocation = bc.getProperty("osgi.framework");
+		try {
+			URI frameworkLocationUri = new URI(frameworkLocation);
+			if ("file".equals(frameworkLocationUri.getScheme())) {
+				Path frameworkPath = Paths.get(frameworkLocationUri);
+				if (frameworkPath.getParent().getFileName().toString().equals(A2Contribution.BOOT)) {
+					Path base = frameworkPath.getParent().getParent();
+					String baseStr = base.toString();
+					if (File.separatorChar == '\\')// MS Windows
+						baseStr = '/' + baseStr.replace(File.separatorChar, '/');
+					URI baseUri = new URI(A2Source.SCHEME_A2, null, null, 0, baseStr, null, null);
+					registerSource(baseUri.toString());
+					OsgiBootUtils.info("Registered " + baseUri + " as default source");
+					return true;
+				}
+			}
+		} catch (Exception e) {
+			OsgiBootUtils.error("Cannot register default source based on framework location " + frameworkLocation, e);
+		}
+		return false;
+	}
+
+	public void install(String spec) {
+		if (spec == null) {
+			for (ProvisioningSource source : sources) {
+				installWholeSource(source);
+			}
+		}
+	}
+
+	/** @return the new/updated bundle, or null if nothing was done. */
+	protected Bundle installOrUpdate(A2Module module) {
+		try {
+			ProvisioningSource moduleSource = module.getBranch().getComponent().getContribution().getSource();
+			Version moduleVersion = module.getVersion();
+			A2Branch osgiBranch = osgiContext.findBranch(module.getBranch().getComponent().getId(), moduleVersion);
+			if (osgiBranch == null) {
+//				Bundle bundle = bc.installBundle(module.getBranch().getCoordinates(),
+//						moduleSource.newInputStream(module.getLocator()));
+				Bundle bundle = moduleSource.install(bc, module);
+				if (OsgiBootUtils.isDebug())
+					OsgiBootUtils.debug("Installed bundle " + bundle.getLocation() + " with version " + moduleVersion);
+				return bundle;
+			} else {
+				A2Module lastOsgiModule = osgiBranch.last();
+				int compare = moduleVersion.compareTo(lastOsgiModule.getVersion());
+				if (compare > 0) {// update
+					Bundle bundle = (Bundle) lastOsgiModule.getLocator();
+//					bundle.update(moduleSource.newInputStream(module.getLocator()));
+					moduleSource.update(bundle, module);
+					OsgiBootUtils.info("Updated bundle " + bundle.getLocation() + " to version " + moduleVersion);
+					return bundle;
+				}
+			}
+		} catch (Exception e) {
+			OsgiBootUtils.error("Could not install module " + module + ": " + e.getMessage(), e);
+		}
+		return null;
+	}
+
+	public Collection<Bundle> update() {
+		boolean fragmentsUpdated = false;
+		Set<Bundle> updatedBundles = new HashSet<>();
+		bundles: for (Bundle bundle : bc.getBundles()) {
+			for (ProvisioningSource source : sources) {
+				String componentId = bundle.getSymbolicName();
+				Version version = bundle.getVersion();
+				A2Branch branch = source.findBranch(componentId, version);
+				if (branch == null)
+					continue bundles;
+				A2Module module = branch.last();
+				Version moduleVersion = module.getVersion();
+				int compare = moduleVersion.compareTo(version);
+				if (compare > 0) {// update
+					try {
+						source.update(bundle, module);
+//						bundle.update(in);
+						String fragmentHost = bundle.getHeaders().get(Constants.FRAGMENT_HOST);
+						if (fragmentHost != null)
+							fragmentsUpdated = true;
+						OsgiBootUtils.info("Updated bundle " + bundle.getLocation() + " to version " + moduleVersion);
+						updatedBundles.add(bundle);
+					} catch (Exception e) {
+						OsgiBootUtils.error("Cannot update with module " + module, e);
+					}
+				}
+			}
+		}
+		FrameworkWiring frameworkWiring = bc.getBundle(0).adapt(FrameworkWiring.class);
+		if (fragmentsUpdated)// refresh all
+			frameworkWiring.refreshBundles(null);
+		else
+			frameworkWiring.refreshBundles(updatedBundles);
+		return updatedBundles;
+	}
+
+	public static void main(String[] args) {
+		Map<String, String> configuration = new HashMap<>();
+		configuration.put("osgi.console", "2323");
+		Framework framework = OsgiBootUtils.launch(new EquinoxFactory(), configuration);
+		try {
+			ProvisioningManager pm = new ProvisioningManager(framework.getBundleContext());
+			FsA2Source context = new FsA2Source(Paths.get(
+					"/home/mbaudier/dev/git/apache2/argeo-commons/dist/argeo-node/target/argeo-node-2.1.74-SNAPSHOT/argeo-node/share/osgi"));
+			context.load();
+			if (framework.getBundleContext().getBundles().length == 1) {// initial
+				pm.install(null);
+			} else {
+				pm.update();
+			}
+		} catch (Exception e) {
+			e.printStackTrace();
+		} finally {
+			try {
+				// framework.stop();
+			} catch (Exception e) {
+				e.printStackTrace();
+			}
+		}
+	}
+
+}