1 files changed, 80 insertions, 0 deletions
diff --git a/bundle-plugin/src/main/java/com/yahoo/container/plugin/mojo/GenerateTestBundleOsgiManifestMojo.java b/bundle-plugin/src/main/java/com/yahoo/container/plugin/mojo/GenerateTestBundleOsgiManifestMojo.java
new file mode 100644
index 00000000000..811aff87b7e
--- /dev/null
+++ b/bundle-plugin/src/main/java/com/yahoo/container/plugin/mojo/GenerateTestBundleOsgiManifestMojo.java
@@ -0,0 +1,80 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.plugin.mojo;
+
+import com.yahoo.container.plugin.classanalysis.Analyze;
+import com.yahoo.container.plugin.classanalysis.ClassFileMetaData;
+import com.yahoo.container.plugin.classanalysis.PackageTally;
+import com.yahoo.container.plugin.osgi.ExportPackages.Export;
+import com.yahoo.container.plugin.osgi.ImportPackages;
+import org.apache.maven.artifact.Artifact;
+import org.apache.maven.plugin.MojoExecutionException;
+import org.apache.maven.plugins.annotations.Mojo;
+import org.apache.maven.plugins.annotations.Parameter;
+import org.apache.maven.plugins.annotations.ResolutionScope;
+
+import java.io.File;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Stream;
+
+import static com.yahoo.container.plugin.bundle.AnalyzeBundle.exportedPackagesAggregated;
+import static com.yahoo.container.plugin.mojo.TestBundleUtils.outputDirectory;
+import static com.yahoo.container.plugin.osgi.ExportPackages.exportsByPackageName;
+import static com.yahoo.container.plugin.osgi.ImportPackages.calculateImports;
+import static com.yahoo.container.plugin.util.Files.allDescendantFiles;
+import static java.util.stream.Collectors.toList;
+
+/**
+ * @author bjorncs
+ */
+@Mojo(name = "generate-test-bundle-osgi-manifest", requiresDependencyResolution = ResolutionScope.TEST, threadSafe = true)
+public class GenerateTestBundleOsgiManifestMojo extends AbstractGenerateOsgiManifestMojo {
+
+    @Parameter
+    private String testProvidedArtifacts;
+
+    public void execute() throws MojoExecutionException {
+        try {
+            Artifacts.ArtifactSet artifactSet = Artifacts.getArtifacts(project, true, testProvidedArtifacts);
+
+            List<File> providedJars = artifactSet.getJarArtifactsProvided().stream()
+                    .map(Artifact::getFile)
+                    .collect(toList());
+
+            List<Export> exportedPackagesFromProvidedJars = exportedPackagesAggregated(providedJars);
+
+            PackageTally projectPackages = getProjectMainAndTestClassesTally();
+
+            PackageTally jarArtifactsToInclude = definedPackages(artifactSet.getJarArtifactsToInclude());
+
+            PackageTally includedPackages = projectPackages.combine(jarArtifactsToInclude);
+
+            Map<String, ImportPackages.Import> calculatedImports = calculateImports(includedPackages.referencedPackages(),
+                    includedPackages.definedPackages(),
+                    exportsByPackageName(exportedPackagesFromProvidedJars));
+
+            Map<String, String> manifestContent = generateManifestContent(artifactSet.getJarArtifactsToInclude(), calculatedImports, includedPackages);
+            addAdditionalManifestProperties(manifestContent);
+            createManifestFile(outputDirectory(project), manifestContent);
+
+        } catch (Exception e) {
+            throw new MojoExecutionException("Failed generating osgi manifest", e);
+        }
+    }
+
+    private void addAdditionalManifestProperties(Map<String, String> manifestContent) {
+        manifestContent.put("X-JDisc-Test-Bundle-Version", "1.0");
+    }
+
+    private PackageTally getProjectMainAndTestClassesTally() {
+        List<ClassFileMetaData> analyzedClasses =
+                Stream.concat(
+                        allDescendantFiles(new File(project.getBuild().getOutputDirectory())),
+                        allDescendantFiles(new File(project.getBuild().getTestOutputDirectory())))
+                        .filter(file -> file.getName().endsWith(".class"))
+                        .map(classFile -> Analyze.analyzeClass(classFile, null))
+                        .collect(toList());
+        return PackageTally.fromAnalyzedClassFiles(analyzedClasses);
+    }
+
+}