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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
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 = event["data"]
try:
data = json.loads(data)
except:
return
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)]
|
