Added reusable feature set from hapifhir/hapi-fhir-jpaserver-starter#784

Author: jkiddo

hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/packages/AdditionalResourcesParser.java

@@ -0,0 +1,67 @@
+package ca.uhn.fhir.jpa.packages;
+
+import ca.uhn.fhir.context.FhirContext;
+import ca.uhn.fhir.util.BundleBuilder;
+import org.hl7.fhir.instance.model.api.IBaseBundle;
+import org.hl7.fhir.instance.model.api.IBaseResource;
+import org.hl7.fhir.utilities.npm.NpmPackage;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.stereotype.Service;
+
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.util.stream.Collectors;
+import java.util.*;
+
+@Service
+public class AdditionalResourcesParser {
+
+	public IBaseBundle collectAdditionalResources(
+			Set<String> additionalResources, PackageInstallationSpec packageInstallationSpec, FhirContext fhirContext) {
+		NpmPackage npmPackage;
+		try {
+			npmPackage = NpmPackage.fromPackage(new ByteArrayInputStream(packageInstallationSpec.getPackageContents()));
+		} catch (IOException e) {
+			throw new RuntimeException(e);
+		}
+		var resources = getAdditionalResources(additionalResources, npmPackage, fhirContext);
+
+		var bundleBuilder = new BundleBuilder(fhirContext);
+		resources.forEach(bundleBuilder::addTransactionUpdateEntry);
+		return bundleBuilder.getBundle();
+	}
+
+	@NotNull
+	public List<IBaseResource> getAdditionalResources(
+			Set<String> folderNames, NpmPackage npmPackage, FhirContext fhirContext) {
+
+		var npmFolders = folderNames.stream()
+				.map(name -> npmPackage.getFolders().get(name))
+				.filter(Objects::nonNull)
+				.collect(Collectors.toList());
+
+		var resources = new LinkedList<IBaseResource>();
+		for (var folder : npmFolders) {
+			List<String> fileNames;
+			try {
+				fileNames = folder.getTypes().values().stream()
+						.flatMap(Collection::stream)
+						.collect(Collectors.toList());
+			} catch (IOException e) {
+				throw new RuntimeException(e.getMessage(), e);
+			}
+
+			resources.addAll(fileNames.stream()
+					.map(fileName -> {
+						try {
+							return new String(folder.fetchFile(fileName));
+						} catch (IOException e) {
+							throw new RuntimeException(e.getMessage(), e);
+						}
+					})
+					.map(fhirContext.newJsonParser().setSuppressNarratives(true)::parseResource)
+					.collect(Collectors.toList()));
+		}
+		return resources;
+	}
+}

hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/packages/PackageInstallationSpec.java

@@ -29,6 +29,7 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Set;
 import java.util.function.Supplier;
 
 @Schema(name = "PackageInstallationSpec", description = "Defines a set of instructions for package installation")
@@ -39,7 +40,8 @@
 	"installMode",
 	"installResourceTypes",
 	"validationMode",
-	"reloadExisting"
+	"reloadExisting",
+	"additionalResourceFolders"
 })
 @ExampleSupplier({PackageInstallationSpec.ExampleSupplier.class, PackageInstallationSpec.ExampleSupplier2.class})
 @JsonInclude(JsonInclude.Include.NON_NULL)
@@ -90,6 +92,12 @@ public class PackageInstallationSpec {
 					"Any values provided here will be interpreted as a regex. Dependencies with an ID matching any regex will be skipped.")
 	private List<String> myDependencyExcludes;
 
+	@Schema(
+			description =
+					"List of folders in the package to extract additional resources from. Each folder listed will be scanned for resources and installed.")
+	@JsonProperty("additionalResourceFolders")
+	private Set<String> additionalResourceFolders;
+
 	@JsonIgnore
 	private byte[] myPackageContents;
 
@@ -177,6 +185,14 @@ public void setReloadExisting(boolean reloadExisting) {
 		this.myReloadExisting = reloadExisting;
 	}
 
+	public Set<String> getAdditionalResourceFolders() {
+		return additionalResourceFolders;
+	}
+
+	public void setAdditionalResourceFolders(Set<String> additionalResourceFolders) {
+		this.additionalResourceFolders = additionalResourceFolders;
+	}
+
 	public PackageInstallationSpec addDependencyExclude(String theExclude) {
 		getDependencyExcludes().add(theExclude);
 		return this;