137 lines
3.9 KiB
Python
137 lines
3.9 KiB
Python
#!/usr/bin/env python3
|
|
# encoding: utf-8
|
|
# (c) 2012 Red Hat, Inc.
|
|
# Authored by Ricky Elrod
|
|
# But when it breaks, don't yell at him because that's mean.
|
|
# update hook for FI repos -> zodbot.
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import shlex
|
|
import socket
|
|
|
|
ZODBOT_SERVER = "value01.vpn.fedoraproject.org"
|
|
ZODBOT_PORT = 5050
|
|
ZODBOT_ANNOUNCE = ["ansible", "dns"]
|
|
|
|
hook = sys.argv[0]
|
|
repodir = sys.argv[1]
|
|
channel = sys.argv[2]
|
|
old = sys.argv[3]
|
|
new = sys.argv[4]
|
|
branch = sys.argv[5]
|
|
|
|
# Split on /, nuke empties from the result, use the last nonempty
|
|
# element. This lets us not care if there's a trailing slash.
|
|
repodir = [d for d in repodir.split('/') if d][-1]
|
|
|
|
|
|
def run_command(command):
|
|
""" Run a command and return a hash with the resulting stdout/stderr."""
|
|
escaped = shlex.split(command)
|
|
cmd = subprocess.Popen(escaped,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
universal_newlines=True)
|
|
stdout, stderr = cmd.communicate()
|
|
return {"stdout": stdout, "stderr": stderr}
|
|
|
|
|
|
def construct_url(slug):
|
|
""" Return a space-padded url to the commit.
|
|
|
|
If and only if it is handled by pagure. Otherwise, return an empty string.
|
|
"""
|
|
|
|
# Our long url template.
|
|
tmpl = "https://pagure.io/fedora-infra/{repo}/c/{slug}"
|
|
|
|
if repodir in ZODBOT_ANNOUNCE and slug:
|
|
return " " + tmpl.format(repo=repodir, slug=slug)
|
|
else:
|
|
return ""
|
|
|
|
|
|
def parse_commit(commit):
|
|
"""
|
|
So we're given a commit in the form of:
|
|
|
|
---
|
|
Ricky Elrod - test-repo:a045150 ---- add some more test files...
|
|
|
|
A foobar/asdf/testfile.1
|
|
A foobar/testfile.2
|
|
---
|
|
|
|
Essentially, we rip out the first line and set it aside.
|
|
Then all the other lines will begin with M/C/R/A/D/U.
|
|
Replace those letters with fancy little symbols (like + for A).
|
|
Combine them together in a list/array.
|
|
Show the first 4 and if more exist, append '...' to the list.
|
|
Lastly, replace the "----" in the original line above with these.
|
|
"""
|
|
|
|
lines = commit.split("\n")
|
|
message = lines.pop(0)
|
|
files = []
|
|
|
|
# extract the commit hash from the first line.
|
|
slug = None
|
|
try:
|
|
slug = message.split(' -')[1].strip().split(':')[1]
|
|
except IndexError:
|
|
print("** Couldn't parse slug from git-rev.", message)
|
|
|
|
# The remaining lines are files changed.
|
|
for changed_file in filter(None, lines):
|
|
status, filename = changed_file.split()
|
|
if status == "M" or status == "R":
|
|
symbol = "*"
|
|
elif status == "C" or status == "A":
|
|
symbol = "+"
|
|
elif status == "D":
|
|
symbol = "-"
|
|
else:
|
|
symbol = "?"
|
|
|
|
files.append(symbol + filename)
|
|
|
|
# Show the first 4 files changed, and if there are more, add a '...'
|
|
# If no files were changed don't show empty [] because it looks tacky.
|
|
fileslist = ' '.join(files[0:4])
|
|
if len(files):
|
|
fileslist = '[' + fileslist
|
|
|
|
if len(files) > 4:
|
|
fileslist += ' ...'
|
|
|
|
fileslist += ']'
|
|
else:
|
|
fileslist = '-'
|
|
|
|
padded_url = construct_url(slug)
|
|
|
|
# Replace the ---- with the files list...
|
|
return message.replace('----', fileslist, 1) + padded_url
|
|
|
|
|
|
# Get a list of commits to report.
|
|
cwd = os.getcwd()
|
|
if (cwd.endswith('dns') and branch == 'master') or branch == 'main':
|
|
revs = run_command(f"git rev-list ^{old} {new}")["stdout"].split("\n")
|
|
revs = [r for r in revs if r]
|
|
revs.reverse()
|
|
|
|
for commit_hash in revs:
|
|
# Get the commit in a format that we can deal with
|
|
commit = run_command(
|
|
f"git show --name-status {commit_hash} --oneline "
|
|
f"--format='%an - {repodir}:%h ---- %s'")
|
|
parsed_commit = parse_commit(commit["stdout"])
|
|
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.connect((ZODBOT_SERVER, ZODBOT_PORT))
|
|
msg = f"{channel} {parsed_commit}"
|
|
s.sendall(msg.encode('utf-8'))
|
|
s.close()
|