Added getters for the objects notifications refer to

hub4j · Apr 8, 2015 · b058c39 · b058c39
1 parent 277ccb5
commit b058c39
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/main/java/org/kohsuke/github/GHThread.java b/src/main/java/org/kohsuke/github/GHThread.java
@@ -59,6 +59,26 @@ public String getType() {
         return subject.type;
     }
 
+    public GHIssue getBoundIssue() throws IOException {
+        if (!"Issue".equals(subject.type))
+            throw new IllegalStateException("Notification doesn't point to Issue");
+        return repository.getIssue(
+                Integer.parseInt(subject.url.substring(subject.url.lastIndexOf('/') + 1)));
+    }
+
+    public GHPullRequest getBoundPullRequest() throws IOException {
+        if (!"PullRequest".equals(subject.type))
+            throw new IllegalStateException("Notification doesn't point to PullRequest");
+        return repository.getPullRequest(
+                Integer.parseInt(subject.url.substring(subject.url.lastIndexOf('/') + 1)));
+    }
+
+    public GHCommit getBoundCommit() throws IOException {
+        if (!"Commit".equals(subject.type))
+            throw new IllegalStateException("Notification doesn't point to Commit");
+        return repository.getCommit(subject.url.substring(subject.url.lastIndexOf('/') + 1));
+    }
+
     /*package*/ GHThread wrap(GitHub root) {
         this.root = root;
         if (this.repository!=null)