from src import ModuleManager, utils from . import colors COMMIT_URL = "https://github.com/%s/commit/%s" COMMIT_RANGE_URL = "https://github.com/%s/compare/%s...%s" CREATE_URL = "https://github.com/%s/tree/%s" PR_URL = "https://github.com/%s/pull/%s" PR_COMMIT_RANGE_URL = "https://github.com/%s/pull/%s/files/%s..%s" PR_COMMIT_URL = "https://github.com/%s/pull/%s/commits/%s" DEFAULT_EVENT_CATEGORIES = [ "ping", "code", "pr", "issue", "repo" ] EVENT_CATEGORIES = { "ping": [ "ping" # new webhook received ], "code": [ "push", "commit_comment" ], "pr-minimal": [ "pull_request/opened", "pull_request/closed", "pull_request/reopened" ], "pr": [ "pull_request/opened", "pull_request/closed", "pull_request/reopened", "pull_request/edited", "pull_request/assigned", "pull_request/unassigned", "pull_request_review", "pull_request/locked", "pull_request/unlocked", "pull_request_review_comment" ], "pr-all": [ "pull_request", "pull_request_review", "pull_request_review_comment" ], "pr-review-minimal": [ "pull_request_review/submitted", "pull_request_review/dismissed" ], "pr-review-comment-minimal": [ "pull_request_review_comment/created", "pull_request_review_comment/deleted" ], "issue-minimal": [ "issues/opened", "issues/closed", "issues/reopened", "issues/deleted", "issues/transferred" ], "issue": [ "issues/opened", "issues/closed", "issues/reopened", "issues/deleted", "issues/edited", "issues/assigned", "issues/unassigned", "issues/locked", "issues/unlocked", "issues/transferred", "issue_comment", ], "issue-all": [ "issues", "issue_comment" ], "issue-comment-minimal": [ "issue_comment/created", "issue_comment/deleted" ], "repo": [ "create", # a repository, branch or tag has been created "delete", # same as above but deleted "release", "fork" ], "team": [ "membership" ], "star": [ # "watch" is a misleading name for this event so this add "star" as an # alias for "watch" "watch" ] } COMMENT_ACTIONS = { "created": "commented", "edited": "edited a comment", "deleted": "deleted a comment" } COMMENT_MAX = 100 CHECK_SUITE_CONCLUSION = { "success": ("passed", colors.COLOR_POSITIVE), "failure": ("failed", colors.COLOR_NEGATIVE), "neutral": ("finished", colors.COLOR_NEUTRAL), "cancelled": ("was cancelled", colors.COLOR_NEGATIVE), "timed_out": ("timed out", colors.COLOR_NEGATIVE), "action_required": ("requires action", colors.COLOR_NEUTRAL) } class GitHub(object): def __init__(self, log, exports): self.log = log self.exports = exports def is_private(self, data, headers): if "repository" in data: return data["repository"]["private"] return False def names(self, data, headers): full_name = None repo_username = None repo_name = None if "repository" in data: full_name = data["repository"]["full_name"] repo_username, repo_name = full_name.split("/", 1) organisation = None if "organization" in data: organisation = data["organization"]["login"] return full_name, repo_username, repo_name, organisation def branch(self, data, headers): if "ref" in data: return data["ref"].rpartition("/")[2] return None def event(self, data, headers): event = headers["X-GitHub-Event"] action = None event_action = None if "action" in data: action = data["action"] category = None category_action = None if "review" in data and "state" in data["review"]: category = "%s+%s" % (event, data["review"]["state"]) elif "check_suite" in data and "conclusion" in data["check_suite"]: category = "%s+%s" % (event, data["check_suite"]["conclusion"]) if action: if category: category_action = "%s/%s" % (category, action) event_action = "%s/%s" % (event, action) return [event]+list(filter(None, [event_action, category, category_action])) def event_categories(self, event): return EVENT_CATEGORIES.get(event, [event]) def webhook(self, full_name, event, data, headers): out = [] if event == "push": out = self.push(full_name, data) elif event == "commit_comment": out = self.commit_comment(full_name, data) elif event == "pull_request": out = self.pull_request(full_name, data) elif event == "pull_request_review": out = self.pull_request_review(full_name, data) elif event == "pull_request_review_comment": out = self.pull_request_review_comment(full_name, data) elif event == "issue_comment": out = self.issue_comment(full_name, data) elif event == "issues": out = self.issues(full_name, data) elif event == "create": out = self.create(full_name, data) elif event == "delete": out = self.delete(full_name, data) elif event == "release": out = self.release(full_name, data) elif event == "check_suite": out = self.check_suite(full_name, data) elif event == "fork": out = self.fork(full_name, data) elif event == "ping": out = self.ping(data) elif event == "membership": out = self.membership(organisation, data) elif event == "watch": out = self.watch(data) return out def _iso8601(self, s): return utils.datetime.parse.iso8601(s) def ping(self, data): return ["Received new webhook"] def _change_count(self, n, symbol, color): return utils.irc.color("%s%d" % (symbol, n), color)+utils.irc.bold("") def _added(self, n): return self._change_count(n, "+", colors.COLOR_POSITIVE) def _removed(self, n): return self._change_count(n, "-", colors.COLOR_NEGATIVE) def _modified(self, n): return self._change_count(n, "~", utils.consts.PURPLE) def _short_hash(self, hash): return hash[:7] def _flat_unique(self, commits, key): return set(itertools.chain(*(commit[key] for commit in commits))) def push(self, full_name, data): outputs = [] branch = data["ref"].split("/", 2)[2] branch = utils.irc.color(branch, colors.COLOR_BRANCH) author = utils.irc.bold(data["pusher"]["name"]) range_url = None if len(data["commits"]): first_id = data["before"] last_id = data["commits"][-1]["id"] range_url = COMMIT_RANGE_URL % (full_name, first_id, last_id) single_url = COMMIT_URL % (full_name, "%s") return self._format_push(branch, author, data["commits"], data["forced"], single_url, range_url) def _format_push(self, branch, author, commits, forced, single_url, range_url): outputs = [] forced_str = "" if forced: forced_str = "%s " % utils.irc.color("force", utils.consts.RED) if len(commits) == 0 and forced: outputs.append( "%s %spushed to %s" % (author, forced_str, branch), None) elif len(commits) <= 3: for commit in commits: hash = commit["id"] hash_colored = utils.irc.color(self._short_hash(hash), colors.COLOR_ID) message = commit["message"].split("\n")[0].strip() url = single_url % hash outputs.append(( "%s %spushed %s to %s: %s" % (author, forced_str, hash_colored, branch, message), url)) else: outputs.append(("%s %spushed %d commits to %s" % (author, forced_str, len(commits), branch), url)) return outputs def _comment(self, s): s_line = s.split("\n")[0].strip() left, right = s_line[:COMMENT_MAX], s_line[COMMENT_MAX:] if not right: return left else: if " " in left: left = left.rsplit(" ", 1)[0] return "%s[...]" % left def commit_comment(self, full_name, data): action = data["action"] commit = self._short_hash(data["comment"]["commit_id"]) commenter = utils.irc.bold(data["comment"]["user"]["login"]) url = data["comment"]["html_url"] return [("[commit/%s] %s %s a comment" % (commit, commenter, action), url)] def pull_request(self, full_name, data): raw_number = data["pull_request"]["number"] branch = data["pull_request"]["base"]["ref"] colored_branch = utils.irc.color(branch, colors.COLOR_BRANCH) sender = utils.irc.bold(data["sender"]["login"]) author = utils.irc.bold(data["pull_request"]["user"]["login"]) number = utils.irc.color("#%s" % data["pull_request"]["number"], colors.COLOR_ID) identifier = "%s by %s" % (number, author) action = data["action"] action_desc = "%s %s" % (action, identifier) if action == "opened": action_desc = "requested %s merge into %s" % (number, colored_branch) elif action == "closed": if data["pull_request"]["merged"]: action_desc = "%s %s into %s" % ( utils.irc.color("merged", colors.COLOR_POSITIVE), identifier, colored_branch) else: action_desc = "%s %s" % ( utils.irc.color("closed", colors.COLOR_NEGATIVE), identifier) elif action == "ready_for_review": action_desc = "marked %s ready for review" % number elif action == "synchronize": action_desc = "committed to %s" % identifier commits_url = data["pull_request"]["commits_url"] commits = utils.http.request(commits_url).json() if commits: seen_before = False new_commits = [] for commit in commits: if seen_before: new_commits.append({"id": commit["sha"], "message": commit["commit"]["message"]}) elif commit["sha"] == data["before"]: seen_before = True range_url = PR_COMMIT_RANGE_URL % (full_name, raw_number, data["before"], data["after"]) single_url = PR_COMMIT_URL % (full_name, raw_number, "%s") if new_commits: pr_identifier = "%s (%s)" % ( number, data["pull_request"]["title"]) outputs = self._format_push(pr_identifier, author, new_commits, False, single_url, range_url) for i, output in enumerate(outputs): outputs[i] = "[PR] %s" % output return outputs elif action == "labeled": action_desc = "labled %s as '%s'" % ( identifier, data["label"]["name"]) elif action == "edited" and "title" in data["changes"]: action_desc = "renamed %s" % identifier pr_title = data["pull_request"]["title"] url = data["pull_request"]["html_url"] return [("[PR] %s %s: %s" % ( sender, action_desc, pr_title), url)] def pull_request_review(self, full_name, data): if not data["action"] == "submitted": return [] if not "submitted_at" in data["review"]: return [] state = data["review"]["state"] if state == "commented": return [] number = utils.irc.color("#%s" % data["pull_request"]["number"], colors.COLOR_ID) action = data["action"] pr_title = data["pull_request"]["title"] reviewer = utils.irc.bold(data["sender"]["login"]) url = data["review"]["html_url"] state_desc = state if state == "approved": state_desc = "approved changes" elif state == "changes_requested": state_desc = "requested changes" elif state == "dismissed": state_desc = "dismissed a review" return [("[PR] %s %s on %s: %s" % (reviewer, state_desc, number, pr_title), url)] def pull_request_review_comment(self, full_name, data): number = utils.irc.color("#%s" % data["pull_request"]["number"], colors.COLOR_ID) action = data["action"] pr_title = data["pull_request"]["title"] sender = utils.irc.bold(data["sender"]["login"]) url = data["comment"]["html_url"] return [("[PR] %s %s on a review on %s: %s" % (sender, COMMENT_ACTIONS[action], number, pr_title), url)] def issues(self, full_name, data): number = utils.irc.color("#%s" % data["issue"]["number"], colors.COLOR_ID) action = data["action"] action_str = "%s %s" % (action, number) if action == "labeled": action_str = "labeled %s as '%s'" % (number, data["label"]["name"]) elif action == "edited" and "title" in data["changes"]: action_str = "renamed %s" % number issue_title = data["issue"]["title"] author = utils.irc.bold(data["sender"]["login"]) url = data["issue"]["html_url"] return [("[issue] %s %s: %s" % (author, action_str, issue_title), url)] def issue_comment(self, full_name, data): if "changes" in data: # don't show this event when nothing has actually changed if data["changes"]["body"]["from"] == data["comment"]["body"]: return number = utils.irc.color("#%s" % data["issue"]["number"], colors.COLOR_ID) action = data["action"] issue_title = data["issue"]["title"] type = "PR" if "pull_request" in data["issue"] else "issue" title = data["issue"]["title"] commenter = utils.irc.bold(data["sender"]["login"]) url = data["comment"]["html_url"] body = "" if not action == "deleted": body = ": %s" % self._comment(data["comment"]["body"]) return [("[%s] %s %s on %s (%s)%s" % (type, commenter, COMMENT_ACTIONS[action], number, title, body), url)] def create(self, full_name, data): ref = data["ref"] ref_color = utils.irc.color(ref, colors.COLOR_BRANCH) type = data["ref_type"] sender = utils.irc.bold(data["sender"]["login"]) url = CREATE_URL % (full_name, ref) return [("%s created a %s: %s" % (sender, type, ref_color), url)] def delete(self, full_name, data): ref = data["ref"] ref_color = utils.irc.color(ref, colors.COLOR_BRANCH) type = data["ref_type"] sender = utils.irc.bold(data["sender"]["login"]) return [("%s deleted a %s: %s" % (sender, type, ref_color), None)] def release(self, full_name, data): action = data["action"] tag = data["release"]["tag_name"] name = data["release"]["name"] or "" if name: name = ": %s" % name author = utils.irc.bold(data["release"]["author"]["login"]) url = data["release"]["html_url"] return [("%s %s a release%s" % (author, action, name), url)] def check_suite(self, full_name, data): suite = data["check_suite"] commit = self._short_hash(suite["head_sha"]) commit = utils.irc.color(commit, utils.consts.LIGHTBLUE) pr = "" url = "" if suite["pull_requests"]: pr_num = suite["pull_requests"][0]["number"] pr = "/PR%s" % utils.irc.color("#%s" % pr_num, colors.COLOR_ID) url = PR_URL % (full_name, pr_num) name = suite["app"]["name"] conclusion = suite["conclusion"] conclusion, conclusion_color = CHECK_SUITE_CONCLUSION[conclusion] conclusion = utils.irc.color(conclusion, conclusion_color) created_at = self._iso8601(suite["created_at"]) updated_at = self._iso8601(suite["updated_at"]) seconds = (updated_at-created_at).total_seconds() duration = utils.datetime.format.to_pretty_time(seconds) return [("[build @%s%s] %s: %s in %s" % ( commit, pr, name, conclusion, duration), url)] def fork(self, full_name, data): forker = utils.irc.bold(data["sender"]["login"]) fork_full_name = utils.irc.color(data["forkee"]["full_name"], utils.consts.LIGHTBLUE) url = data["forkee"]["html_url"] return [("%s forked into %s" % (forker, fork_full_name), url)] def membership(self, organisation, data): return [("%s %s %s to team %s" % (data["sender"]["login"], data["action"], data["member"]["login"], data["team"]["name"]), None)] def watch(self, data): return [("%s starred the repository" % data["sender"]["login"], None)]