1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
package com.getbootstrap.rorschach.server
import com.getbootstrap.rorschach.auditing.{BaseAndHeadBranchesAuditor, ModifiedFilesAuditor}
import scala.collection.JavaConverters._
import scala.util.{Try,Success,Failure}
import akka.actor.ActorRef
import org.eclipse.egit.github.core._
import org.eclipse.egit.github.core.service.{CommitService, OrganizationService}
import com.getbootstrap.rorschach.github._
import com.getbootstrap.rorschach.github.util._
class PullRequestEventHandler(commenter: ActorRef) extends GitHubActorWithLogging {
private def modifiedFilesFor(repoId: RepositoryId, base: CommitSha, head: CommitSha) = {
val commitService = new CommitService(gitHubClient)
Try { commitService.compare(repoId, base.sha, head.sha) }.map { comparison =>
val affectedFiles = comparison.getFiles.asScala
affectedFiles.filter{ _.status == Modified }.map{ _.getFilename }.toSet[String]
}
}
def isTrusted(user: User): Boolean = {
val orgService = new OrganizationService(gitHubClient)
settings.TrustedOrganizations.exists{ org => Try{ orgService.isPublicMember(org, user.getLogin) }.toOption.getOrElse(false) }
}
override def receive = {
case pr: PullRequest => {
val bsBase = pr.getBase
val prHead = pr.getHead
val destinationRepo = bsBase.getRepo.repositoryId
if (settings.repoIds contains destinationRepo) {
if (isTrusted(pr.getUser)) {
log.info(s"Skipping audit because user ${pr.getUser.getLogin} is member of trusted org.")
}
else {
log.info(s"Auditing ${destinationRepo} ${pr.number} ...")
val base = bsBase.commitSha
val head = prHead.commitSha
val foreignRepoId = prHead.getRepo.repositoryId
val fileMessages = modifiedFilesFor(foreignRepoId, base, head) match {
case Failure(exc) => {
log.error(exc, s"Could not get modified files for commits ${base}...${head} for ${foreignRepoId}")
Nil
}
case Success(modifiedFiles) => {
ModifiedFilesAuditor.audit(modifiedFiles)
}
}
val branchMessages = BaseAndHeadBranchesAuditor.audit(baseBranch = bsBase.getRef, headBranch = prHead.getRef)
val allMessages = fileMessages ++ branchMessages
if (allMessages.nonEmpty) {
commenter ! PullRequestFeedback(destinationRepo, pr.number, pr.getUser, allMessages)
}
else {
log.info(s"Repo ${destinationRepo} ${pr.number} successfully passed all audits.")
}
}
}
else {
log.error(s"Received event from GitHub about irrelevant repository: ${destinationRepo}")
}
}
}
}
|
