Add a method for getting the root of a rule workspace to the Label method

This method is exposed to Skylark and will enable correct handling of protobuf skylark files.
See #784

--
MOS_MIGRATED_REVID=112235357

1 files changed, 14 insertions, 0 deletions

diff --git a/src/main/java/com/google/devtools/build/lib/cmdline/Label.java b/src/main/java/com/google/devtools/build/lib/cmdline/Label.java
index 8fe14e780c..97955dcd84 100644
--- a/src/main/java/com/google/devtools/build/lib/cmdline/Label.java
+++ b/src/main/java/com/google/devtools/build/lib/cmdline/Label.java
@@ -263,6 +263,20 @@ public final class Label implements Comparable<Label>, Serializable, SkylarkPrin
   public String getPackageName() {
     return packageIdentifier.getPackageFragment().getPathString();
   }
+  
+  /**
+   * Returns the execution root for the workspace, relative to the execroot (e.g., for label
+   * {@code @repo//pkg:b}, it will returns {@code external/repo/pkg} and for label {@code //pkg:a},
+   * it will returns an empty string.
+   */
+  @SkylarkCallable(name = "workspace_root", structField = true,
+      doc = "Returns the execution root for the workspace of this label, relative to the execroot. "
+      + "For instance:<br>"
+      + "<pre class=language-python>Label(\"@repo//pkg/foo:abc\").workspace_root =="
+      + " \"external/repo\"</pre>")
+  public String getWorkspaceRoot() {
+    return packageIdentifier.getRepository().getPathFragment().toString();
+  }
 
   /**
    * Returns the path fragment of the package in which this rule was declared (e.g. {@code