Implement custom prefix mapper to ensure use of longest namespace match. (#169)

Author: balhoff

cli/src/main/scala/org/renci/relationgraph/LongestFirstPrefixMapping.scala

@@ -0,0 +1,24 @@
+package org.renci.relationgraph
+
+case class LongestFirstPrefixMapping(prefixToNS: Map[String, String], oboPrefixes: Boolean) {
+
+  private val nsToPrefix = prefixToNS.toList
+    .map { case (prefix, ns) => (ns, prefix) }
+    .sortBy { case (ns, _) => -ns.length }
+  private val oboNS = "http://purl.obolibrary.org/obo/"
+
+  def compactOpt(uri: String): Option[String] = {
+    nsToPrefix.collectFirst { case (ns, prefix) if uri.startsWith(ns) =>
+      val local = uri.substring(ns.length, uri.length)
+      s"$prefix:$local"
+    }
+  }.orElse {
+    if (oboPrefixes && uri.startsWith(oboNS)) {
+      val id = uri.substring(oboNS.length, uri.length)
+      Some(id.split("_", 2).mkString(":"))
+    } else None
+  }
+
+  def compact(uri: String): String = compactOpt(uri).getOrElse(uri)
+
+}

cli/src/main/scala/org/renci/relationgraph/TSVStreamRDF.scala

@@ -3,26 +3,14 @@ package org.renci.relationgraph
 import org.apache.jena.graph.Triple
 import org.apache.jena.riot.system.StreamRDF
 import org.apache.jena.shared.PrefixMapping
-import org.apache.jena.shared.impl.PrefixMappingImpl
 import org.apache.jena.sparql.core.Quad
 
 import java.io.{File, PrintWriter}
 import scala.jdk.CollectionConverters._
 
 class TSVStreamRDF(file: File, prefixes: Map[String, String], oboPrefixes: Boolean) extends StreamRDF {
 
-  private val prefixMapping: PrefixMapping = {
-    val pm = new PrefixMappingImpl() {
-      override def shortForm(uri: String): String = {
-        val shortForm = super.shortForm(uri)
-        if (oboPrefixes && (shortForm == uri) && (uri.startsWith("http://purl.obolibrary.org/obo/"))) {
-          val tail = uri.replace("http://purl.obolibrary.org/obo/", "")
-          tail.split("_", 2).mkString(":")
-        } else shortForm
-      }
-    }
-    pm.setNsPrefixes(prefixes.asJava).withDefaultMappings(PrefixMapping.Standard)
-  }
+  private val prefixMapping = LongestFirstPrefixMapping(PrefixMapping.Standard.getNsPrefixMap.asScala.toMap ++ prefixes, oboPrefixes)
 
   private var writer: PrintWriter = _
 
@@ -31,17 +19,17 @@ class TSVStreamRDF(file: File, prefixes: Map[String, String], oboPrefixes: Boole
   }
 
   override def triple(triple: Triple): Unit = {
-    val s = triple.getSubject.toString(prefixMapping, true)
-    val p = triple.getPredicate.toString(prefixMapping, true)
-    val o = triple.getObject.toString(prefixMapping, true)
+    val s = prefixMapping.compact(triple.getSubject.toString)
+    val p = prefixMapping.compact(triple.getPredicate.toString)
+    val o = prefixMapping.compact(triple.getObject.toString)
     writer.println(s"$s\t$p\t$o")
   }
 
   override def quad(quad: Quad): Unit = {
-    val s = quad.getSubject.toString(prefixMapping, true)
-    val p = quad.getPredicate.toString(prefixMapping, true)
-    val o = quad.getObject.toString(prefixMapping, true)
-    val g = quad.getGraph.toString(prefixMapping, true)
+    val s = prefixMapping.compact(quad.getSubject.toString)
+    val p = prefixMapping.compact(quad.getPredicate.toString)
+    val o = prefixMapping.compact(quad.getObject.toString)
+    val g = prefixMapping.compact(quad.getGraph.toString)
     writer.println(s"$s\t$p\t$o\t$g")
   }