org.molgenis.semanticmapper.service.impl.AlgorithmTemplate Maven / Gradle / Ivy

Go to download
package org.molgenis.semanticmapper.service.impl;

import static java.util.Objects.requireNonNull;

import java.util.Map;
import org.molgenis.script.core.Script;

/**
 * Mapping algorithm template for JavaScript Magma scripts.
 *
 * Example: $('weight').div($('height').pow(2)).value()

 * Example rendered:

 * Model: {weight: 'attr1', height: 'attr2'}

 * Example: $('attr1').div($('attr2').pow(2)).value()
 */
public class AlgorithmTemplate {
  private final Script script;
  private final Map model;

  public AlgorithmTemplate(Script script, Map model) {
    this.script = requireNonNull(script);
    this.model = requireNonNull(model);
  }

  public String render() {
    String content = script.getContent();
    for (Map.Entry entry : model.entrySet()) {
      content =
          content.replaceAll(
              String.format("\\$\\('%s'\\)", entry.getKey()),
              String.format("\\$\\('%s'\\)", entry.getValue()));
    }
    return content;
  }

  @Override
  public int hashCode() {
    final int prime = 31;
    int result = 1;
    result = prime * result + ((model == null) ? 0 : model.hashCode());
    result = prime * result + ((script == null) ? 0 : script.hashCode());
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (this == obj) return true;
    if (obj == null) return false;
    if (getClass() != obj.getClass()) return false;
    AlgorithmTemplate other = (AlgorithmTemplate) obj;
    if (model == null) {
      if (other.model != null) return false;
    } else if (!model.equals(other.model)) return false;
    if (script == null) {
      if (other.script != null) return false;
    } else if (!script.equals(other.script)) return false;
    return true;
  }

  @Override
  public String toString() {
    return "AlgorithmTemplate [script=" + script + ", model=" + model + "]";
  }
}