import itertools, json, urllib.parse from src import ModuleManager, utils FORM_ENCODED = "application/x-www-form-urlencoded" 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" DEFAULT_EVENTS = [ "push", "commit_comment", "pull_request", "pull_request_review", "pull_request_review_comment", "issue_comment", "issues", "create", "delete", "release" ] COMMENT_ACTIONS = { "created": "commented", "edited": "edited a comment", "deleted": "deleted a comment" } @utils.export("channelset", {"setting": "github-hook", "help": ("Disable/Enable showing BitBot's github commits in the " "current channel"), "array": True}) @utils.export("channelset", {"setting": "github-hide-prefix", "help": "Hide/show command-like prefix on Github hook outputs", "validate": utils.bool_or_none}) class Module(ModuleManager.BaseModule): @utils.hook("api.post.github") def github(self, event): payload = event["data"].decode("utf8") if event["headers"]["Content-Type"] == FORM_ENCODED: payload = urllib.parse.parse_qs(urllib.parse.unquote(payload) )["payload"][0] data = json.loads(payload) github_event = event["headers"]["X-GitHub-Event"] if github_event == "ping": return "" full_name = data["repository"]["full_name"] hooks = self.bot.database.channel_settings.find_by_setting( "github-hook") targets = [] repo_hooked = False for i, (server_id, channel_name, hooked_repos) in list( enumerate(hooks))[::-1]: if full_name in hooked_repos: repo_hooked = True server = self.bot.get_server(server_id) if server and channel_name in server.channels: channel = server.channels.get(channel_name) github_events = channel.get_setting("github-events", DEFAULT_EVENTS) if github_event in github_events: targets.append([server, channel]) if not targets: return "" if repo_hooked else None outputs = None if github_event == "push": outputs = self.push(event, full_name, data) elif github_event == "commit_comment": outputs = self.commit_comment(event, full_name, data) elif github_event == "pull_request": outputs = self.pull_request(event, full_name, data) elif github_event == "pull_request_review": outputs = self.pull_request_review(event, full_name, data) elif github_event == "pull_request_review_comment": outputs = self.pull_request_review_comment(event, full_name, data) elif github_event == "issue_comment": outputs = self.issue_comment(event, full_name, data) elif github_event == "issues": outputs = self.issues(event, full_name, data) elif github_event == "create": outputs = self.create(event, full_name, data) elif github_event == "delete": outputs = self.delete(event, full_name, data) elif github_event == "release": outputs = self.release(event, full_name, data) if outputs: for server, channel in targets: for output in outputs: output = "(%s) %s" % (full_name, output) self.events.on("send.stdout").call(target=channel, module_name="Github", server=server, message=output, hide_prefix=channel.get_setting( "github-hide-prefix", False)) return "" 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, "+", utils.consts.GREEN) def _removed(self, n): return self._change_count(n, "-", utils.consts.RED) def _modified(self, n): return self._change_count(n, "~", utils.consts.PURPLE) def _short_hash(self, hash): return hash[:8] def _flat_unique(self, commits, key): return set(itertools.chain(*(commit[key] for commit in commits))) def push(self, event, full_name, data): outputs = [] branch = data["ref"].split("/", 2)[2] branch = utils.irc.color(branch, utils.consts.LIGHTBLUE) if len(data["commits"]) <= 3: for commit in data["commits"]: id = self._short_hash(commit["id"]) message = commit["message"].split("\n")[0].strip() author = commit["author"]["name"] or commit["author"]["login"] author = utils.irc.bold(author) url = COMMIT_URL % (full_name, id) added = self._added(len(commit["added"])) removed = self._removed(len(commit["removed"])) modified = self._modified(len(commit["modified"])) outputs.append("[%s/%s/%s files] commit by %s to %s: %s - %s" % (added, removed, modified, author, branch, message, url)) else: first_id = self._short_hash(data["before"]) last_id = self._short_hash(data["commits"][-1]["id"]) pusher = utils.irc.bold(data["pusher"]["name"]) url = COMMIT_RANGE_URL % (full_name, first_id, last_id) commits = data["commits"] added = self._added(len(self._flat_unique(commits, "added"))) removed = self._removed(len(self._flat_unique(commits, "removed"))) modified = self._modified(len(self._flat_unique(commits, "modified"))) outputs.append("[%s/%s/%s files] %s pushed %d commits to %s - %s" % (added, removed, modified, pusher, len(data["commits"]), branch, url)) return outputs def commit_comment(self, event, full_name, data): action = data["action"] commit = data["commit_id"][:8] commenter = utils.irc.bold(data["comment"]["user"]["login"]) url = data["comment"]["html_url"] return ["[commit/%s] %s commented" % (commit, commenter, action)] def pull_request(self, event, full_name, data): action = data["action"] action_desc = action branch = data["pull_request"]["base"]["ref"] colored_branch = utils.irc.color(branch, utils.consts.LIGHTBLUE) if action == "opened": action_desc = "requested merge into %s" % colored_branch elif action == "closed": if data["pull_request"]["merged"]: action_desc = "%s into %s" % ( utils.irc.color("merged", utils.consts.GREEN), colored_branch) else: action_desc = utils.irc.color("closed without merging", utils.consts.RED) elif action == "synchronize": action_desc = "committed to" elif action == "labeled": action_desc = "labeled as '%s'" % data["label"]["name"] elif action == "unlabeled": action_desc = "unlabeled as '%s'" % data["label"]["name"] pr_title = data["pull_request"]["title"] author = utils.irc.bold(data["sender"]["login"]) url = data["pull_request"]["html_url"] return ["[pr] %s %s: %s - %s" % (author, action_desc, pr_title, url)] def pull_request_review(self, event, full_name, data): if data["review"]["state"] == "commented": return [] action = data["action"] pr_title = data["pull_request"]["title"] reviewer = utils.irc.bold(data["sender"]["login"]) url = data["review"]["html_url"] return ["[pr] %s %s a review on: %s - %s" % (reviewer, action, pr_title, url)] def pull_request_review_comment(self, event, full_name, data): action = data["action"] pr_title = data["pull_request"]["title"] commenter = data["comment"]["user"]["login"] url = data["comment"]["html_url"] return ["[pr] %s %s on a review: %s - %s" % (commenter, COMMENT_ACTIONS[action], pr_title, url)] def issues(self, event, full_name, data): action = data["action"] action_desc = action if action == "labeled": action_desc = "labeled as '%s'" % data["label"]["name"] elif action == "unlabeled": action_desc = "unlabeled as '%s'" % data["label"]["name"] issue_title = data["issue"]["title"] author = utils.irc.bold(data["sender"]["login"]) url = data["issue"]["html_url"] return ["[issue] %s %s: %s - %s" % (author, action, issue_title, url)] def issue_comment(self, event, full_name, data): action = data["action"] issue_title = data["issue"]["title"] type = "pr" if "pull_request" in data["issue"] else "issue" commenter = utils.irc.bold(data["comment"]["user"]["login"]) url = data["comment"]["html_url"] return ["[%s] %s %s on: %s - %s" % (type, commenter, COMMENT_ACTIONS[action], issue_title, url)] def create(self, event, full_name, data): ref = data["ref"] ref_color = utils.irc.color(ref, utils.consts.LIGHTBLUE) type = data["ref_type"] sender = utils.irc.bold(data["sender"]["login"]) url = CREATE_URL % (full_name, ref) return ["%s created a %s: %s - %s" % (sender, type, ref_color, url)] def delete(self, event, full_name, data): ref = data["ref"] type = data["ref_type"] sender = utils.irc.bold(data["sender"]["login"]) return ["%s deleted a %s: %s" % (sender, type, ref)] def release(self, event, full_name, data): action = data["action"] tag = data["release"]["tag_name"] name = data["release"]["name"] or "" if name: name = ": %s" author = utils.irc.bold(data["release"]["author"]["login"]) url = data["release"]["html_url"] return ["%s %s a release%s - %s" % (author, action, name, url)]