feat: add KytheURI#toCorpusPath() (#5592)

kythe · Apr 21, 2023 · eb169e6 · eb169e6
1 parent b69b49b
commit eb169e6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/kythe/java/com/google/devtools/kythe/util/KytheURI.java b/kythe/java/com/google/devtools/kythe/util/KytheURI.java
@@ -118,6 +118,15 @@ public VName toVName() {
     return vName;
   }
 
+  /** Returns the {@link CorpusPath} components of the uri. */
+  public CorpusPath toCorpusPath() {
+    return CorpusPath.newBuilder()
+        .setCorpus(vName.getCorpus())
+        .setRoot(vName.getRoot())
+        .setPath(vName.getPath())
+        .build();
+  }
+
   @Override
   public String toString() {
     StringBuilder b =

diff --git a/kythe/javatests/com/google/devtools/kythe/util/KytheURITest.java b/kythe/javatests/com/google/devtools/kythe/util/KytheURITest.java
@@ -170,6 +170,18 @@ public void testToVName() throws URISyntaxException {
     assertThat(vname.getLanguage()).isEqualTo(lang);
   }
 
+  public void testToCorpusPath() throws URISyntaxException {
+    String signature = "magic school truck";
+    String corpus = "crazyTown";
+    String path = "usa/2.0";
+    String root = null;
+    String lang = "c++";
+    CorpusPath cp = new KytheURI(signature, corpus, root, path, lang).toCorpusPath();
+    assertThat(cp.getCorpus()).isEqualTo(corpus);
+    assertThat(cp.getRoot()).isEmpty(); // Proto fields are never null
+    assertThat(cp.getPath()).isEqualTo(path);
+  }
+
   public void testParseErrors() {
     String[] tests =
         new String[] {