package org.argeo.osgi.a2;
   
   import java.io.File;
   import java.net.URI;
   import java.nio.file.Files;
   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);
  					if (Files.exists(base)) {
  						FsA2Source source = new FsA2Source(base);
  						source.load();
  						addSource(source);
  						OsgiBootUtils.info("Registered " + uri + " as 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.debug("Default source from framework location " + frameworkLocation);
  					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();
 			}
 		}
 	}
 
 }