137 lines
3.6 KiB
Python
137 lines
3.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
import getpass
|
|
import os
|
|
import sys
|
|
|
|
from collections import defaultdict
|
|
|
|
import pygit2
|
|
|
|
import fedmsg
|
|
import fedmsg.config
|
|
|
|
# Use $GIT_DIR to determine where this repo is.
|
|
abspath = os.path.abspath(os.environ['GIT_DIR'])
|
|
# This assumes git root dir is named "repo_name.git"
|
|
repo_name = '.'.join(abspath.split(os.path.sep)[-1].split('.')[:-1])
|
|
|
|
username = getpass.getuser()
|
|
|
|
repo = pygit2.Repository(abspath)
|
|
|
|
print "Emitting a message to the fedmsg bus."
|
|
config = fedmsg.config.load_config([], None)
|
|
config['active'] = True
|
|
config['endpoints']['relay_inbound'] = config['relay_inbound']
|
|
fedmsg.init(name='relay_inbound', cert_prefix='scm', **config)
|
|
|
|
def revs_between(head, base, seen=None):
|
|
""" Yield revisions between HEAD and BASE.
|
|
|
|
Detect if a revision appears multiple times and stop recursion.
|
|
"""
|
|
|
|
seen = seen or []
|
|
rev = unicode(head.id)
|
|
|
|
bail = False
|
|
if not rev in seen:
|
|
seen.append(rev)
|
|
yield rev
|
|
|
|
for parent in head.parents:
|
|
if parent.id == base.id:
|
|
bail = True
|
|
else:
|
|
bail = True
|
|
|
|
if not bail:
|
|
for parent in head.parents:
|
|
for rev in revs_between(parent, base, seen=seen):
|
|
yield rev
|
|
|
|
|
|
def build_stats(commit):
|
|
files = defaultdict(lambda: defaultdict(int))
|
|
|
|
# Calculate diffs against all parent commits
|
|
diffs = [repo.diff(parent, commit) for parent in commit.parents]
|
|
# Unless this is the first commit, with no parents.
|
|
diffs = diffs or [commit.tree.diff_to_tree(swap=True)]
|
|
|
|
for diff in diffs:
|
|
for patch in diff:
|
|
path = patch.new_file_path
|
|
files[path]['additions'] += patch.additions
|
|
files[path]['deletions'] += patch.deletions
|
|
files[path]['lines'] += patch.additions + patch.deletions
|
|
|
|
total = defaultdict(int)
|
|
for name, stats in files.items():
|
|
total['additions'] += stats['additions']
|
|
total['deletions'] += stats['deletions']
|
|
total['lines'] += stats['lines']
|
|
total['files'] += 1
|
|
|
|
return files, total
|
|
|
|
|
|
# Read in all the rev information git-receive-pack hands us.
|
|
lines = [line.split() for line in sys.stdin.readlines()]
|
|
for line in lines:
|
|
base, head, branch = line
|
|
branch = '/'.join(branch.split('/')[2:])
|
|
|
|
try:
|
|
head = repo.revparse_single(head)
|
|
except KeyError:
|
|
# This means they are deleting this branch.. and we don't have a fedmsg
|
|
# for that (yet?). It is disallowed by dist-git in Fedora anyways.
|
|
continue
|
|
|
|
try:
|
|
base = repo.revparse_single(base)
|
|
revs = revs_between(head, base)
|
|
except KeyError:
|
|
revs = [unicode(head.id)]
|
|
|
|
def _build_commit(rev):
|
|
commit = repo.revparse_single(rev)
|
|
|
|
# Tags are a little funny, and vary between versions of pygit2, so we'll
|
|
# just ignore them as far as fedmsg is concerned.
|
|
if isinstance(commit, pygit2.Tag):
|
|
return None
|
|
|
|
files, total = build_stats(commit)
|
|
|
|
return dict(
|
|
name=commit.author.name,
|
|
email=commit.author.email,
|
|
username=username,
|
|
summary=commit.message.split('\n')[0],
|
|
message=commit.message,
|
|
stats=dict(
|
|
files=files,
|
|
total=total,
|
|
),
|
|
rev=rev,
|
|
path=abspath,
|
|
repo=repo_name,
|
|
branch=branch,
|
|
agent=os.getlogin(),
|
|
)
|
|
|
|
commits = map(_build_commit, revs)
|
|
|
|
for commit in commits:
|
|
|
|
if commit is None:
|
|
continue
|
|
|
|
fedmsg.publish(
|
|
topic="receive",
|
|
msg=dict(commit=commit),
|
|
modname="git",
|
|
)
|