org.openlca.ipc.handlers.ExportHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of olca-ipc Show documentation
There is a newer version: 2.2.1
package org.openlca.ipc.handlers;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import org.openlca.core.database.Daos;
import org.openlca.core.model.ModelType;
import org.openlca.ipc.Responses;
import org.openlca.ipc.Rpc;
import org.openlca.ipc.RpcRequest;
import org.openlca.ipc.RpcResponse;
import org.openlca.jsonld.Json;
import org.openlca.jsonld.LibraryLink;
import org.openlca.jsonld.ZipStore;
import org.openlca.jsonld.output.JsonExport;

import com.google.gson.JsonObject;

public class ExportHandler {

	private final HandlerContext context;

	public ExportHandler(HandlerContext context) {
		this.context = context;
	}

	@Rpc("export/json-ld")
	public RpcResponse jsonLd(RpcRequest req) {
		if (req == null || req.params == null || !req.params.isJsonObject())
			return Responses.badRequest("No @id given", req);
		var obj = req.params.getAsJsonObject();
		var path = Json.getString(obj, "path");
		if (path == null)
			return Responses.badRequest("No `path` given", req);
		var models = getModels(obj);
		if (models.isEmpty())
			return Responses.badRequest("No `models` given", req);
		try {
			var store = ZipStore.open(new File(path));
			store.putLibraryLinks(LibraryLink.of(
					context.db().getLibraries()));
			var export = new JsonExport(context.db(), store);
			for (var type : models.keySet()) {
				for (var refId : models.get(type)) {
					export.write(Daos.root(context.db(), type).getForRefId(refId));
				}
			}
			store.close();
			return Responses.ok("Exported to " + path, req);
		} catch (IOException e) {
			return Responses.serverError(e, req);
		}
	}

	private Map> getModels(JsonObject obj) {
		var map = new HashMap>();
		Json.forEachObject(obj, "models", model -> {
			String id = Json.getString(model, "@id");
			String type = Json.getString(model, "@type");
			if (id == null || type == null)
				return;
			for (ModelType t : ModelType.values()) {
				var clazz = t.getModelClass();
				if (clazz != null && clazz.getSimpleName().equals(type)) {
					var ids = map.computeIfAbsent(t, k -> new HashSet<>());
					ids.add(id);
				}
			}
		});
		return map;
	}

}