Scala code in container-di converted to Java

1 files changed, 92 insertions, 0 deletions

diff --git a/container-di/src/main/java/com/yahoo/container/di/componentgraph/core/JerseyNode.java b/container-di/src/main/java/com/yahoo/container/di/componentgraph/core/JerseyNode.java
new file mode 100644
index 00000000000..79b849bff8f
--- /dev/null
+++ b/container-di/src/main/java/com/yahoo/container/di/componentgraph/core/JerseyNode.java
@@ -0,0 +1,92 @@
+// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.di.componentgraph.core;
+
+import com.yahoo.component.ComponentId;
+import com.yahoo.component.ComponentSpecification;
+import com.yahoo.container.di.Osgi;
+import com.yahoo.container.di.config.JerseyBundlesConfig;
+import com.yahoo.container.di.config.RestApiContext;
+import com.yahoo.container.di.config.RestApiContext.BundleInfo;
+import com.yahoo.container.di.osgi.BundleClasses;
+import org.osgi.framework.Bundle;
+import org.osgi.framework.wiring.BundleWiring;
+
+import java.net.URL;
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.List;
+
+/**
+ * Represents an instance of RestApiContext
+ *
+ * @author gjoranv
+ * @author Tony Vaagenes
+ * @author ollivir
+ */
+public class JerseyNode extends ComponentNode {
+    private static final String WEB_INF_URL = "WebInfUrl";
+
+    private final Osgi osgi;
+
+    public JerseyNode(ComponentId componentId, String configId, Class<?> clazz, Osgi osgi) {
+        super(componentId, configId, clazz, null);
+        this.osgi = osgi;
+    }
+
+    @Override
+    protected RestApiContext newInstance() {
+        Object instance = super.newInstance();
+        RestApiContext restApiContext = (RestApiContext) instance;
+
+        List<JerseyBundlesConfig.Bundles> bundles = restApiContext.bundlesConfig.bundles();
+        for (JerseyBundlesConfig.Bundles bundleConfig : bundles) {
+            BundleClasses bundleClasses = osgi.getBundleClasses(ComponentSpecification.fromString(bundleConfig.spec()),
+                    new HashSet<>(bundleConfig.packages()));
+
+            restApiContext.addBundle(createBundleInfo(bundleClasses.bundle(), bundleClasses.classEntries()));
+        }
+
+        componentsToInject.forEach(component -> restApiContext.addInjectableComponent(component.instanceKey(), component.componentId(),
+                component.newOrCachedInstance()));
+
+        return restApiContext;
+    }
+
+    @Override
+    public int hashCode() {
+        return super.hashCode();
+    }
+
+    @Override
+    public boolean equals(Object other) {
+        return super.equals(other)
+                && (other instanceof JerseyNode && this.componentsToInject.equals(((JerseyNode) other).componentsToInject));
+    }
+
+    public static BundleInfo createBundleInfo(Bundle bundle, Collection<String> classEntries) {
+        BundleInfo bundleInfo = new BundleInfo(bundle.getSymbolicName(), bundle.getVersion(), bundle.getLocation(), webInfUrl(bundle),
+                bundle.adapt(BundleWiring.class).getClassLoader());
+
+        bundleInfo.setClassEntries(classEntries);
+        return bundleInfo;
+    }
+
+    public static Bundle getBundle(Osgi osgi, String bundleSpec) {
+        Bundle bundle = osgi.getBundle(ComponentSpecification.fromString(bundleSpec));
+        if (bundle == null) {
+            throw new IllegalArgumentException("Bundle not found: " + bundleSpec);
+        }
+        return bundle;
+    }
+
+    private static URL webInfUrl(Bundle bundle) {
+        String webInfUrlHeader = bundle.getHeaders().get(WEB_INF_URL);
+
+        if (webInfUrlHeader == null) {
+            return null;
+        } else {
+            return bundle.getEntry(webInfUrlHeader);
+        }
+    }
+
+}