--- /dev/null
+package org.argeo.init.osgi;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.net.URL;
+import java.nio.file.DirectoryStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.PathMatcher;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.SortedMap;
+import java.util.StringTokenizer;
+import java.util.TreeMap;
+import java.util.jar.JarEntry;
+import java.util.jar.JarInputStream;
+import java.util.jar.Manifest;
+
+import org.osgi.framework.Constants;
+import org.osgi.framework.Version;
+
+/**
+ * A distribution bundle is a bundle within a maven-like distribution
+ * groupId:Bundle-SymbolicName:Bundle-Version which references others OSGi
+ * bundle. It is not required to be OSGi complete also it will generally be
+ * expected that it is. The root of the repository is computed based on the file
+ * name of the URL and of the content of the index.
+ */
+public class DistributionBundle {
+ private final static String INDEX_FILE_NAME = "modularDistribution.csv";
+
+ private final String url;
+
+ private Manifest manifest;
+ private String symbolicName;
+ private String version;
+
+ /** can be null */
+ private String baseUrl;
+ /** can be null */
+ private String relativeUrl;
+ private String localCache;
+
+ private List<OsgiArtifact> artifacts;
+
+ private String separator = ",";
+
+ public DistributionBundle(String url) {
+ this.url = url;
+ }
+
+ public DistributionBundle(String baseUrl, String relativeUrl, String localCache) {
+ if (baseUrl == null || !baseUrl.endsWith("/"))
+ throw new IllegalArgumentException("Base url " + baseUrl + " badly formatted");
+ if (relativeUrl.startsWith("http") || relativeUrl.startsWith("file:"))
+ throw new IllegalArgumentException("Relative URL " + relativeUrl + " badly formatted");
+ this.url = constructUrl(baseUrl, relativeUrl);
+ this.baseUrl = baseUrl;
+ this.relativeUrl = relativeUrl;
+ this.localCache = localCache;
+ }
+
+ protected String constructUrl(String baseUrl, String relativeUrl) {
+ try {
+ if (relativeUrl.indexOf('*') >= 0) {
+ if (!baseUrl.startsWith("file:"))
+ throw new IllegalArgumentException(
+ "Wildcard support only for file:, badly formatted " + baseUrl + " and " + relativeUrl);
+ Path basePath = Paths.get(new URI(baseUrl));
+ // Path basePath = Paths.get(new URI(baseUrl));
+ // int li = relativeUrl.lastIndexOf('/');
+ // String relativeDir = relativeUrl.substring(0, li);
+ // String relativeFile = relativeUrl.substring(li,
+ // relativeUrl.length());
+ String pattern = "glob:" + basePath + '/' + relativeUrl;
+ PathMatcher pm = basePath.getFileSystem().getPathMatcher(pattern);
+ SortedMap<Version, Path> res = new TreeMap<>();
+ checkDir(basePath, pm, res);
+ if (res.size() == 0)
+ throw new IllegalArgumentException("No file matching " + relativeUrl + " found in " + baseUrl);
+ return res.get(res.firstKey()).toUri().toString();
+ } else {
+ return baseUrl + relativeUrl;
+ }
+ } catch (Exception e) {
+ throw new RuntimeException("Cannot build URL from " + baseUrl + " and " + relativeUrl, e);
+ }
+ }
+
+ private void checkDir(Path dir, PathMatcher pm, SortedMap<Version, Path> res) throws IOException {
+ try (DirectoryStream<Path> ds = Files.newDirectoryStream(dir)) {
+ for (Path path : ds) {
+ if (Files.isDirectory(path))
+ checkDir(path, pm, res);
+ else if (pm.matches(path)) {
+ String fileName = path.getFileName().toString();
+ fileName = fileName.substring(0, fileName.lastIndexOf('.'));
+ if (fileName.endsWith("-SNAPSHOT"))
+ fileName = fileName.substring(0, fileName.lastIndexOf('-')) + ".SNAPSHOT";
+ fileName = fileName.substring(fileName.lastIndexOf('-') + 1);
+ Version version = new Version(fileName);
+ res.put(version, path);
+ }
+ }
+ }
+ }
+
+ public void processUrl() {
+ JarInputStream jarIn = null;
+ try {
+ URL u = new URL(url);
+
+ // local cache
+ URI localUri = new URI(localCache + relativeUrl);
+ Path localPath = Paths.get(localUri);
+ if (Files.exists(localPath))
+ u = localUri.toURL();
+ jarIn = new JarInputStream(u.openStream());
+
+ // meta data
+ manifest = jarIn.getManifest();
+ symbolicName = manifest.getMainAttributes().getValue(Constants.BUNDLE_SYMBOLICNAME);
+ version = manifest.getMainAttributes().getValue(Constants.BUNDLE_VERSION);
+
+ JarEntry indexEntry;
+ while ((indexEntry = jarIn.getNextJarEntry()) != null) {
+ String entryName = indexEntry.getName();
+ if (entryName.equals(INDEX_FILE_NAME)) {
+ break;
+ }
+ jarIn.closeEntry();
+ }
+
+ // list artifacts
+ if (indexEntry == null)
+ throw new IllegalArgumentException("No index " + INDEX_FILE_NAME + " in " + url);
+ artifacts = listArtifacts(jarIn);
+ jarIn.closeEntry();
+
+ // find base URL
+ // won't work if distribution artifact is not listed
+ for (int i = 0; i < artifacts.size(); i++) {
+ OsgiArtifact osgiArtifact = (OsgiArtifact) artifacts.get(i);
+ if (osgiArtifact.getSymbolicName().equals(symbolicName) && osgiArtifact.getVersion().equals(version)) {
+ String relativeUrl = osgiArtifact.getRelativeUrl();
+ if (url.endsWith(relativeUrl)) {
+ baseUrl = url.substring(0, url.length() - osgiArtifact.getRelativeUrl().length());
+ break;
+ }
+ }
+ }
+ } catch (Exception e) {
+ throw new RuntimeException("Cannot list URLs from " + url, e);
+ } finally {
+ if (jarIn != null)
+ try {
+ jarIn.close();
+ } catch (IOException e) {
+ // silent
+ }
+ }
+ }
+
+ protected List<OsgiArtifact> listArtifacts(InputStream in) {
+ List<OsgiArtifact> osgiArtifacts = new ArrayList<OsgiArtifact>();
+ BufferedReader reader = null;
+ try {
+ reader = new BufferedReader(new InputStreamReader(in));
+ String line = null;
+ lines: while ((line = reader.readLine()) != null) {
+ StringTokenizer st = new StringTokenizer(line, separator);
+ String moduleName = st.nextToken();
+ String moduleVersion = st.nextToken();
+ String relativeUrl = st.nextToken();
+ if (relativeUrl.endsWith(".pom"))
+ continue lines;
+ osgiArtifacts.add(new OsgiArtifact(moduleName, moduleVersion, relativeUrl));
+ }
+ } catch (Exception e) {
+ throw new RuntimeException("Cannot list artifacts", e);
+ }
+ return osgiArtifacts;
+ }
+
+ /** Convenience method */
+ public static DistributionBundle processUrl(String baseUrl, String relativeUrl, String localCache) {
+ DistributionBundle distributionBundle = new DistributionBundle(baseUrl, relativeUrl, localCache);
+ distributionBundle.processUrl();
+ return distributionBundle;
+ }
+
+ /**
+ * List full URLs of the bundles, based on base URL, usable directly for
+ * download.
+ */
+ public List<String> listUrls() {
+ if (baseUrl == null)
+ throw new IllegalArgumentException("Base URL is not set");
+
+ if (artifacts == null)
+ throw new IllegalStateException("Artifact list not initialized");
+
+ List<String> urls = new ArrayList<String>();
+ for (int i = 0; i < artifacts.size(); i++) {
+ OsgiArtifact osgiArtifact = (OsgiArtifact) artifacts.get(i);
+ // local cache
+ URI localUri;
+ try {
+ localUri = new URI(localCache + relativeUrl);
+ } catch (URISyntaxException e) {
+ OsgiBootUtils.warn(e.getMessage());
+ localUri = null;
+ }
+ Version version = new Version(osgiArtifact.getVersion());
+ if (localUri != null && Files.exists(Paths.get(localUri)) && version.getQualifier() != null
+ && version.getQualifier().startsWith("SNAPSHOT")) {
+ urls.add(localCache + osgiArtifact.getRelativeUrl());
+ } else {
+ urls.add(baseUrl + osgiArtifact.getRelativeUrl());
+ }
+ }
+ return urls;
+ }
+
+ public void setBaseUrl(String baseUrl) {
+ this.baseUrl = baseUrl;
+ }
+
+ /** Separator used to parse the tabular file */
+ public void setSeparator(String modulesUrlSeparator) {
+ this.separator = modulesUrlSeparator;
+ }
+
+ public String getRelativeUrl() {
+ return relativeUrl;
+ }
+
+ /** One of the listed artifact */
+ protected static class OsgiArtifact {
+ private final String symbolicName;
+ private final String version;
+ private final String relativeUrl;
+
+ public OsgiArtifact(String symbolicName, String version, String relativeUrl) {
+ super();
+ this.symbolicName = symbolicName;
+ this.version = version;
+ this.relativeUrl = relativeUrl;
+ }
+
+ public String getSymbolicName() {
+ return symbolicName;
+ }
+
+ public String getVersion() {
+ return version;
+ }
+
+ public String getRelativeUrl() {
+ return relativeUrl;
+ }
+
+ }
+}