JsMagmaScriptExecutor.java

package org.molgenis.js.magma;

import org.molgenis.data.Entity;
import org.molgenis.data.meta.model.AttributeFactory;
import org.molgenis.data.meta.model.EntityType;
import org.molgenis.data.meta.model.EntityTypeFactory;
import org.molgenis.data.support.DynamicEntity;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Map;

import static java.util.Objects.requireNonNull;

/**
 * Executes a JavaScript using the Magma API
 */
@Service
public class JsMagmaScriptExecutor
{
	private final JsMagmaScriptEvaluator jsMagmaScriptEvaluator;
	private final EntityTypeFactory entityTypeFactory;
	private final AttributeFactory attributeFactory;

	@Autowired
	public JsMagmaScriptExecutor(JsMagmaScriptEvaluator jsMagmaScriptEvaluator, EntityTypeFactory entityTypeFactory,
			AttributeFactory attributeFactory)
	{
		this.jsMagmaScriptEvaluator = jsMagmaScriptEvaluator;
		this.entityTypeFactory = requireNonNull(entityTypeFactory);
		this.attributeFactory = requireNonNull(attributeFactory);
	}

	/**
	 * Execute a JavaScript using the Magma API
	 *
	 * @param jsScript
	 * @param parameters
	 * @return
	 */
	Object executeScript(String jsScript, Map<String, Object> parameters)
	{
		EntityType entityType = entityTypeFactory.create().setSimpleName("entity");
		parameters.keySet().forEach(key -> entityType.addAttribute(attributeFactory.create().setName(key)));
		Entity entity = new DynamicEntity(entityType);
		parameters.entrySet().forEach(parameter -> entity.set(parameter.getKey(), parameter.getValue()));
		return jsMagmaScriptEvaluator.eval(jsScript, entity);
	}
}