bitbot-3.11-fork/modules/github.py

233 lines
9.2 KiB
Python

import itertools, json
from src import ModuleManager, utils
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")})
@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):
data = json.loads(event["data"])
github_event = event["headers"]["X-GitHub-Event"]
if github_event == "ping":
return True
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 True 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)
trigger = self._make_trigger(channel, server, output)
self.bot.trigger(trigger)
return True
def _make_trigger(self, channel, server, line):
return lambda: self.events.on("send.stdout").call(target=channel,
module_name="Github", server=server, message=line,
hide_prefix=channel.get_setting("github-hide-prefix", False))
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.bold(utils.irc.color(branch, utils.consts.BLUE))
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
if action == "closed":
if data["pull_request"]["merged"]:
action_desc = utils.irc.color("merged", utils.consts.GREEN)
else:
action_desc = utils.irc.color("closed without merging",
utils.consts.RED)
action_desc = utils.irc.bold(action_desc)
elif action == "synchronize":
action_desc = "committed to"
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["review"]["user"]["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"]
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"]
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, 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)]