diff options
Diffstat (limited to 'src/main/scala/com/getbootstrap/savage/github/util/package.scala')
-rw-r--r-- | src/main/scala/com/getbootstrap/savage/github/util/package.scala | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/main/scala/com/getbootstrap/savage/github/util/package.scala b/src/main/scala/com/getbootstrap/savage/github/util/package.scala index 520378c..7a5bd39 100644 --- a/src/main/scala/com/getbootstrap/savage/github/util/package.scala +++ b/src/main/scala/com/getbootstrap/savage/github/util/package.scala @@ -1,6 +1,8 @@ package com.getbootstrap.savage.github import org.eclipse.egit.github.core._ +import org.eclipse.egit.github.core.event.PullRequestPayload +import com.getbootstrap.savage.github.pr_action.PullRequestAction package object util { private val SafeRepoRegex = "^[0-9a-zA-Z_-]+/[0-9a-zA-Z_-]+$".r @@ -21,10 +23,15 @@ package object util { } implicit class RichPullRequest(pr: PullRequest) { + import org.eclipse.egit.github.core.service.IssueService def number: PullRequestNumber = PullRequestNumber(pr.getNumber).get + def isOpen: Boolean = (pr.getState == IssueService.STATE_OPEN) } implicit class RichRepositoryId(repoId: RepositoryId) { def asPushRemote: String = s"git@github.com:${repoId.generateId}.git" def asPullRemote: String = s"https://github.com/${repoId.generateId}.git" } + implicit class RichPullRequestPayload(payload: PullRequestPayload) { + def action: PullRequestAction = PullRequestAction(payload.getAction).get + } } |