"""
Margebot: A Errbot Plugin for Gitlab MR reminders
"""
+import re
from datetime import datetime, timezone
from time import sleep
from dateutil import parser
from dateutil.tz import tzutc
from dateutil.relativedelta import relativedelta
-from errbot import BotPlugin, botcmd, webhook
+from errbot import BotPlugin, botcmd, arg_botcmd, re_botcmd, webhook
+from errbot.templating import tenv
from errcron.bot import CrontabMixin
import gitlab
+import requests
+
+
+def addprojecthook_extra(self, project_id, url, push=False, issues=False, merge_requests=False, tag_push=False, extra_data=None):
+ """
+ A copy parent addprojecthook with an extra_data field
+ """
+ data = {"id": project_id, "url": url}
+ if extra_data:
+ for ed_key, ed_value in extra_data.items():
+ data[ed_key] = ed_value
+ data['push_events'] = int(bool(push))
+ data['issues_events'] = int(bool(issues))
+ data['merge_requests_events'] = int(bool(merge_requests))
+ data['tag_push_events'] = int(bool(tag_push))
+ request = requests.post("{0}/{1}/hooks".format(self.projects_url, project_id),
+ headers=self.headers, data=data, verify=self.verify_ssl)
+ if request.status_code == 201:
+ return request.json()
+ return False
+
+gitlab.Gitlab.addprojecthook_extra = addprojecthook_extra
+
+
+def editprojecthook_extra(self, project_id, hook_id, url, push=False, issues=False, merge_requests=False, tag_push=False, extra_data=None):
+ """
+ A copy of the parent editprojecthook with an extra_data field
+ """
+ data = {"id": project_id, "hook_id": hook_id, "url": url}
+ if extra_data:
+ for ed_key, ed_value in extra_data.items():
+ data[ed_key] = ed_value
+ data['push_events'] = int(bool(push))
+ data['issues_events'] = int(bool(issues))
+ data['merge_requests_events'] = int(bool(merge_requests))
+ data['tag_push_events'] = int(bool(tag_push))
+ request = requests.put("{0}/{1}/hooks/{2}".format(self.projects_url, project_id, hook_id),
+ headers=self.headers, data=data, verify=self.verify_ssl)
+ return request.status_code == 200
+
+gitlab.Gitlab.editprojecthook_extra = editprojecthook_extra
def deltastr(any_delta):
self.chatroom_host = None
self.gitlab = None
self.soak_delta = None
+ self.webhook_url = None
super().__init__(*args, **kwargs)
def get_configuration_template(self):
CRONTAB: Schedule of automated merge request checks in '%M %H %d %m %w' format
VERIFY_SSL : True, False, or path to CA cert to verify cert
CRONTAB_SOAK_HOURS : Don't send out reminders about MRs opened less than this many hours
+ WEBHOOK_URL : URL to use for defining MR integration in gitlab
"""
return {'GITLAB_HOST': 'gitlab.example.com',
'GITLAB_ADMIN_TOKEN': 'gitlab-admin-user-private-token',
'CHATROOM_HOST': 'conference.jabber.example.com',
'CRONTAB': '0 11,17 * * *',
'VERIFY_SSL': True,
- 'CRONTAB_SOAK_HOURS': 1}
+ 'CRONTAB_SOAK_HOURS': 1,
+ 'WEBHOOK_URL': 'https://webhooks.example.com:3142/margebot/'}
def check_configuration(self, configuration):
"""
self.activate_crontab()
self.soak_delta = relativedelta(hours=self.config['CRONTAB_SOAK_HOURS'])
+ self.webhook_url = self.config['WEBHOOK_URL']
+ if self.webhook_url[-1] != '/':
+ self.webhook_url += '/'
super().activate()
def deactivate(self):
"""
self.log.info("webhook: request: {}, rooms: {}".format(request, rooms))
- self.log.info("state: {}".format(request['object_attributes']['state']))
+ # self.log.info("state: {}".format(request['object_attributes']['state']))
# verify it's a merge request
if request['object_kind'] != 'merge_request':
target_project_id = request['object_attributes']['target_project_id']
iid = request['object_attributes']['iid']
- mr_id = request['object_attributes']['id']
+
+ # If the MR is tagged 'never-close' ignore it
+ if 'labels' in request:
+ for a_label in request['labels']:
+ if a_label['title'] == 'never-close':
+ self.log.info("Skipping never-close notice for {} MR".format(url))
+ return "OK"
msg_template = "Hi there ! {} has opened a new {}MR: \"{}\"\n{}/merge_requests/{}"
msg = msg_template.format(author_name, wip, title, url, iid)
open_mrs = self['OPEN_MRS']
- if (target_project_id, mr_id, rooms) not in open_mrs:
+ if (target_project_id, iid, rooms) not in open_mrs:
for a_room in rooms.split(','):
if self.config:
self.send(self.build_identifier(a_room + '@' + self.chatroom_host), msg)
- self.log.info("webhook: Saving ({}, {}, {})".format(target_project_id, mr_id, rooms))
- open_mrs[(target_project_id, mr_id, rooms)] = True
+ self.log.info("webhook: Saving ({}, {}, {})".format(target_project_id, iid, rooms))
+ open_mrs[(target_project_id, iid, rooms)] = True
self['OPEN_MRS'] = open_mrs
+
+ # TODO: Add check if an MR has toggled the WIP indicator
+ # (trigger on updates (what's that look like in request['object_attributes']['state'])
+ # Then check in request['changes']['title']['previous'] starts with 'WIP:'
+ # but not request['changes']['title']['current'], and vice versa
+ # See https://gitlab.com/gitlab-org/gitlab-ce/issues/53529
+
return "OK"
def mr_status_msg(self, a_mr, author=None):
# getapprovals is only available in GitLab 8.9 EE or greater
# (not the open source CE version)
# approvals = self.gitlab.getapprovals(a_mr['id'])
- # also_approved = ""
+ # approved = ""
# for approved in approvals['approved_by']:
- # also_approved += "," + approved['user']['name']
+ # approved += "," + approvals['user']['name']
+
+ # See https://gitlab.com/gitlab-org/gitlab-ce/issues/35498
+ # awards = GET /projects/:id/merge_requests/:merge_request_iid/award_emoji
+ # for an award in awards:
+ # if name==??? and award_type==???:
+ # approved += "," + award["user"]["username"]
upvotes = a_mr['upvotes']
msg = "{} (opened {})".format(a_mr['web_url'], str_open_since)
else:
msg += "."
- return (creation_time, msg)
+ return {'creation_time': creation_time, 'msg': msg}
def crontab_hook(self, polled_time):
"""
# initialize the reminders
rooms = self.rooms()
for a_room in rooms:
+ self.log.info("poller: a_room.node: {}".format(a_room.node))
reminder_msg[a_room.node] = []
- msgs = ""
still_open_mrs = {}
# Let's walk through the MRs we've seen already:
# Lookup the MR from the project/id
a_mr = self.gitlab.getmergerequest(project_id, mr_id)
+ if not a_mr:
+ self.log.debug("Couldn't find project: {}, id: {}".format(project_id, mr_id))
+ continue
+
+ # If the MR is tagged 'never-close' ignore it
+ if 'labels' in a_mr and 'never-close' in a_mr['labels']:
+ continue
self.log.info("a_mr: {} {} {} {}".format(project_id, mr_id, notify_rooms, a_mr['state']))
else:
still_open_mrs[(project_id, mr_id, notify_rooms)] = True
- msg_tuple = self.mr_status_msg(a_mr)
- if msg_tuple is None:
+ msg_dict = self.mr_status_msg(a_mr)
+ if msg_dict is None:
continue
for a_room in notify_rooms.split(','):
- reminder_msg[a_room].append(msg_tuple)
+ if a_room in reminder_msg:
+ reminder_msg[a_room].append(msg_dict)
+ else:
+ self.log.error("{} not in reminder_msg (project_id={}, mr_id={})".format(a_room, project_id, mr_id))
self['OPEN_MRS'] = open_mrs
# Remind each of the rooms about open MRs
for a_room, room_msg_list in reminder_msg.items():
if room_msg_list != []:
-
- # sort by the creation time
- sorted_room_msg_list = sorted(room_msg_list, key=lambda x: x[0])
-
- # extract the msgs from the tuple list
- msgs = [x[1] for x in sorted_room_msg_list]
-
- # join those msgs together.
- room_msg = "\n".join(msgs)
-
if self.config:
- msg_template = "These MRs need some attention:\n{}\n"
- msg_template += "You can get an updated list with the !reviews command."
to_room = a_room + '@' + self.config['CHATROOM_HOST']
- msg = msg_template.format(room_msg)
+ msg = tenv().get_template('reviews.md').render(msg_list=room_msg_list)
self.send(self.build_identifier(to_room), msg)
self['OPEN_MRS'] = still_open_mrs
- @botcmd()
+ @botcmd(template="reviews")
def reviews(self, msg, args):
"""
Returns a list of MRs that are waiting for some luv.
msg = ""
still_open_mrs = {}
open_mrs = self['OPEN_MRS']
+ self.log.info('open_mrs: {}'.format(open_mrs))
for (project, mr_id, notify_rooms) in open_mrs:
# Lookup the MR from the project/id
a_mr = self.gitlab.getmergerequest(project, mr_id)
+ if not a_mr:
+ self.log.debug("Couldn't find project: {}, id: {}".format(project, id))
+ continue
+
+ self.log.info('project: {}, id: {}, a_mr: {}'.format(project, id, a_mr))
+
+ # If the MR is tagged 'never-close' ignore it
+ if 'labels' in a_mr and 'never-close' in a_mr['labels']:
+ continue
# If the MR is no longer open, skip to the next MR,
# and don't include this MR in the next check
if 'opened' not in a_mr['state']:
+ self.log.info('state not opened: {}'.format(a_mr['state']))
continue
else:
still_open_mrs[(project, mr_id, notify_rooms)] = True
- msg_tuple = self.mr_status_msg(a_mr, author=sender_gitlab_id)
- if msg_tuple is None:
+ msg_dict = self.mr_status_msg(a_mr, author=sender_gitlab_id)
+ if msg_dict is None:
continue
- msg_list.append(msg_tuple)
+ msg_list.append(msg_dict)
+
+ self['OPEN_MRS'] = still_open_mrs
+
+ return {'sender': sender, 'msg_list': msg_list}
- if msg_list == []:
- response = 'Hi {}: {}'.format(sender, 'I found no open MRs for you.')
+ @arg_botcmd('rooms', type=str, help="Comma-separated room list without @conference-room suffix")
+ @arg_botcmd('repo', type=str, help="repo to start watching for MRs in NAMESPACE/PROJECT_NAME format")
+ def watchrepo(self, msg, repo, rooms):
+ """
+ Add the margebot webhook to a repo, and prepopulate any open MRs in the repo with margebot
+ """
+ self.log.info("msg={}".format(msg))
+ self.log.info("repo={}".format(repo))
+ self.log.info("rooms={}".format(rooms))
+
+ # get the group/repo repo, error out if it doesn't exist
+ project = self.gitlab.getproject(repo)
+ if not project:
+ msg = "Couldn't find repo {}".format(repo)
+ self.log.info("watchrepo: {}".format(msg))
+ return msg
+
+ self.log.info('project: {}'.format(project))
+
+ target_project_id = project['id']
+
+ # Check is the project already includes the margebot hook
+ # If no hooks, will it return False or [] ?
+ marge_hook = None
+ hooks = self.gitlab.getprojecthooks(target_project_id)
+ self.log.error("hooks = {}".format(hooks))
+ if hooks is False:
+ msg = "Couldn't find {} hooks".format(repo)
+ self.log.error("watchrepo: {}".format(msg))
+ return msg
else:
- # sort by the creation time
- sorted_msg_list = sorted(msg_list, key=lambda x: x[0])
+ for a_hook in hooks:
+ self.log.info('a_hook: {}'.format(a_hook))
+ if a_hook['merge_requests_events'] and a_hook['url'].startswith(self.webhook_url):
+ marge_hook = a_hook
+ break
+
+ # If so replace it (or error out ?)
+ url = "{}{}".format(self.webhook_url, rooms) # webhooks_url will end in '/'
+ if marge_hook:
+
+ old_rooms = marge_hook['url'].split(self.webhook_url, 1)[1]
+ if old_rooms == rooms:
+ msg = "Already reporting {} MRs to the {} room(s)".format(repo, rooms)
+ self.log.info('watchrepo: {}'.format(msg))
+ return msg
+ else:
+ hook_updated = self.gitlab.editprojecthook_extra(target_project_id, marge_hook['id'], url, merge_requests=True, extra_data={'enable_ssl_verification': True})
+ s_watch_msg = "Updating room list for {} MRs from {} to {}".format(repo, old_rooms, rooms)
+ s_action = "update"
+ else:
+ hook_updated = self.gitlab.addprojecthook_extra(target_project_id, url, merge_requests=True, extra_data={'enable_ssl_verification': True})
+ s_watch_msg = "Now watching for new MRs in the {} repo to the {} room(s)".format(repo, rooms)
+ s_action = "add"
- # extract the msgs from the tuple list
- msgs = [x[1] for x in sorted_msg_list]
+ if not hook_updated:
+ msg = "Couldn't {} hook: {}".format(s_action, repo)
+ self.log.error("watchrepo: {}".format(msg))
+ return msg
- # join those msgs together.
- msg = "\n".join(msgs)
- response = 'Hi {}: These MRs need some attention:\n{}'.format(sender, msg)
+ open_mrs = self['OPEN_MRS']
+ mr_count = 0
+ # get the open MRs in the repo
+
+ # If updating the room list, walk through the existing MR list
+ if s_action == "update":
+ new_open_mrs = {}
+ for (project_id, mr_id, old_rooms) in open_mrs:
+ # pragma pylint: disable=simplifiable-if-statement
+ if project_id == target_project_id:
+ new_open_mrs[(project_id, mr_id, rooms)] = True
+ else:
+ new_open_mrs[(project_id, mr_id, old_rooms)] = True
+ # pragma pylint: enable=simplifiable-if-statement
+ open_mrs = new_open_mrs
+
+ # If adding a new repo, check for existing opened/reopened MRs in the repo.
+ else:
+ # For debugging the 'watchrepo didn't find my MR' issue.
+ # mr_list = self.gitlab.getmergerequests(target_project_id, page=1, per_page=100)
+ # self.log.error('juden: mr_list state: {}'.format(mr_list[0]['state']))
+ for state in ['opened', 'reopened']:
+ page = 1
+ mr_list = self.gitlab.getmergerequests(target_project_id, page=page, per_page=100, state=state)
+ while (mr_list is not False) and (mr_list != []):
+ for an_mr in mr_list:
+ mr_count += 1
+ self.log.info('watchrepo: an_mr WATS THE IID\n{}'.format(an_mr))
+ mr_id = an_mr['iid']
+ open_mrs[(target_project_id, mr_id, rooms)] = True
+ # Get the next page of MRs
+ page += 1
+ mr_list = self.gitlab.getmergerequests(target_project_id, page=page, per_page=100)
- self['OPEN_MRS'] = still_open_mrs
+ self['OPEN_MRS'] = open_mrs
- return response
+ if mr_count == 0:
+ mr_msg = "No open MRs were found in the repo."
+ elif mr_count == 1:
+ mr_msg = "1 open MR was found in the repo. Run !reviews to see the updated MR list."
+ else:
+ mr_msg = "{} open MRs were found in the repo. Run !reviews to see the updated MR list."
+ return "{}\n{}".format(s_watch_msg, mr_msg)
# pragma pylint: disable=unused-argument
- @botcmd()
- def hello(self, msg, args):
- """
- A simple command to check if the bot is responding
- """
- return "Hi there"
@botcmd()
def xyzzy(self, msg, args):
sleep(5)
yield "(just kidding)"
+ @re_botcmd(pattern=r"I blame marge(bot)?", prefixed=False, flags=re.IGNORECASE)
+ def dont_blame_margebot(self, msg, match):
+ """
+ margebot is innocent.
+ """
+ yield u"(\u300D\uFF9F\uFF9B\uFF9F)\uFF63NOOOooooo say it ain't so."
+
+ @re_botcmd(pattern=r"\u0028\u256F\u00B0\u25A1\u00B0\uFF09\u256F\uFE35\u0020\u253B(\u2501+)\u253B", prefixed=False)
+ def deflipped(self, msg, match):
+ """
+ Unflip a properly sized table
+ """
+ table_len = len(match.group(1))
+ deflip_table = u"\u252c" + (u"\u2500" * table_len) + u"\u252c \u30ce( \u309c-\u309c\u30ce)"
+ # yield u"\u252c\u2500\u2500\u252c \u30ce( \u309c-\u309c\u30ce)"
+ yield deflip_table
+
+ @re_botcmd(pattern=r"good bot", prefixed=False, flags=re.IGNORECASE)
+ def best_bot(self, msg, match):
+ """
+ margebot is the best.
+ """
+ yield "Best bot"
+
+ @re_botcmd(pattern=r"magfest", prefixed=False, flags=re.IGNORECASE)
+ def margefest(self, msg, args):
+ """
+ margefest4ever
+ """
+ return "More like MargeFest, amirite ?"
+
+# ha the dev-infra room sez koji sooooooooooooooooooooooooooooooooooooooooooo much
+# @re_botcmd(pattern=r"k+o+j+i+", prefixed=False, flags=re.IGNORECASE)
+# def koji(self, msg, args):
+# """
+# More like daikaiju, amirite ?
+# """
+# return "More like kaiju, amirite ?"
+
+ @re_botcmd(pattern=r"peruvian chicken", prefixed=False, flags=re.IGNORECASE)
+ def booruvian(self, msg, args):
+ """
+ They put hard boiled eggs in their tamales too.
+ """
+ return "More like Booruvian chicken, amirite ?"
+
+ @re_botcmd(pattern=r"booruvian chicken", prefixed=False, flags=re.IGNORECASE)
+ def booihoohooruvian(self, msg, args):
+ """
+ That chicken I do not like.
+ """
+ return "More like Boohoohooruvian chicken, amirite ?"
+
+ @re_botcmd(pattern=r"margebot sucks", prefixed=False, flags=re.IGNORECASE)
+ def new_agenda_item(self, msg, args):
+ """
+ Bring it up with the committee
+ """
+ return "Bring it up with the Margebot steering committee."
+
+ @re_botcmd(pattern=r"jackie", prefixed=False, flags=re.IGNORECASE)
+ def jackie(self, msg, args):
+ """
+ Who dat ?
+ """
+ return "I don't know any Jackies. I'm calling security."
+
+ @re_botcmd(pattern=r".*", prefixed=True)
+ def catchall(self, msg, args):
+ """
+ Don't have the bot complain about unknown commands if the first word in a msg is its name
+ """
+ return
+
# pragma pylint: enable=unused-argument