bitbot-3.11-fork/modules/github.py

import json
from src import ModuleManager, utils

COMMIT_URL = "https://github.com/%s/commit/%s"

@utils.export("channelset", {"setting": "github-hook",
    "help": ("Disable/Enable showing BitBot's github commits in the "
    "current channel"), "hidden": True})
class Module(ModuleManager.BaseModule):
    @utils.hook("api.post.github")
    def github(self, event):
        data = json.loads(event["data"])

        full_name = data["repository"]["full_name"]
        hooks = self.bot.database.channel_settings.find_by_setting(
            "github-hook")
        for i, (server_id, channel_name, values) in list(
                enumerate(hooks))[::-1]:
            if not full_name in values:
                hooks.pop(i)
        if not hooks:
            return

        github_event = event["headers"]["X-GitHub-Event"]

        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)

        if outputs:
            for server_id, channel_name, _ in hooks:
                for output in outputs:
                    server = self.bot.get_server(server_id)
                    channel = server.get_channel(channel_name)
                    trigger = self._make_trigger(channel, server, output)
                    self.bot.trigger(trigger)

    def _make_trigger(self, channel, server, line):
        return lambda: self.events.on("send.stdout").call(
            target=channel, module_name="Github", server=server, message=line)

    def push(self, event, full_name, data):
        outputs = []
        for commit in data["commits"]:
            id = commit["id"]

            message = commit["message"].split("\n")
            message = [line.strip() for line in message]
            message = " ".join(message)

            author = "%s <%s>" % (commit["author"]["username"],
                commit["author"]["email"])
            modified_count = len(commit["modified"])
            added_count = len(commit["added"])
            removed_count = len(commit["removed"])
            url = COMMIT_URL % (full_name, id[:8])

            outputs.append("(%s) [files: +%d ∆%d -%d] commit by '%s': %s - %s"
                % (full_name, added_count, modified_count, removed_count,
                author, message, url))
        return outputs

    def commit_comment(self, event, full_name, data):
        action = data["action"]
        commit = data["commit_id"][:8]
        commenter = data["comment"]["user"]["login"]
        url = data["comment"]["html_url"]
        return ["(%s) [commit/%s] %s %s a comment" %
            (full_name, commit, commenter, action)]

    def pull_request(self, event, full_name, data):
        action = data["action"]
        pr_number = data["pull_request"]["number"]
        pr_title = data["pull_request"]["title"]
        author = data["sender"]["login"]
        url = data["pull_request"]["html_url"]
        return ["(%s) [pr#%d] %s %s: %s - %s" %
            (full_name, pr_number, author, action, pr_title, url)]

    def pull_request_review(self, event, full_name, data):
        action = data["action"]
        pr_number = data["pull_request"]["number"]
        pr_title = data["pull_request"]["title"]
        reviewer = data["review"]["user"]["login"]
        url = data["review"]["html_url"]
        return ["(%s) [pr#%d] %s %s a review: %s - %s" %
            (full_name, pr_number, reviewer, action, pr_title, url)]

    def pull_request_review_comment(self, event, full_name, data):
        action = data["action"]
        pr_number = data["pull_request"]["number"]
        pr_title = data["pull_request"]["title"]
        commenter = data["comment"]["user"]["login"]
        url = data["comment"]["html_url"]
        return ["(%s) [pr#%d] %s %s a comment: %s - %s" %
            (full_name, pr_number, commenter, action, pr_title, url)]

    def issues(self, event, full_name, data):
        action = data["action"]
        issue_number = data["issue"]["number"]
        issue_title = data["issue"]["title"]
        author = data["sender"]["login"]
        url = data["issue"]["html_url"]
        return ["(%s) [issue#%d] %s %s: %s - %s" %
            (full_name, issue_number, author, action, issue_title, url)]
    def issue_comment(self, event, full_name, data):
        action = data["action"]
        issue_number = data["issue"]["number"]
        issue_title = data["issue"]["title"]
        commenter = data["comment"]["user"]["login"]
        url = data["comment"]["html_url"]
        return ["(%s) [issue#%d] %s %s a comment: %s - %s" %
            (full_name, issue_number, commenter, action, issue_title, url)]