import logging
from pylons import c, g
from ming.orm.base import state
from pyforge.lib.decorators import audit, react
from pyforge.lib.helpers import push_context, set_context, encode_keys
from forgescm.lib import hg, git
from pyforge import model as M
from pymongo import bson
log = logging.getLogger(__name__)
@react('scm.initialized')
def initialized(routing_key, data):
shortname = M.Project.query.get(_id=bson.ObjectId(data['project_id'])).shortname
set_context(shortname, data['mount_point'])
repo = c.app.repo
log.info('Setting repo status for %s', repo)
repo.status = 'Ready'
@react('scm.forked')
def forked_update_source(routing_key, data):
log.info('Updating the source of a fork (%s)', data['forked_from'])
set_context(**encode_keys(data['forked_from']))
repo = c.app.repo
repo.forks.append(data['forked_to'])
@react('scm.forked')
def forked_update_dest(routing_key, data):
log.info('Updating the dest of a fork', data['forked_to'])
set_context(**encode_keys(data['forked_to']))
# Update dest repo
repo = c.app.repo
repo.forked_from.update(data['forked_from'])
repo.clear_commits()
# Copy history from source
commit_extra = dict(
app_config_id = c.app.config._id,
repository_id = repo._id)
# Copy the history
log.info('Begin history copy')
with push_context(**encode_keys(repo.forked_from)):
parent_repo = c.app.repo
for commit in parent_repo.commits:
patches = commit.patches
with push_context(**encode_keys(data['forked_to'])):
commit_dict = state(commit).document.deinstrumented_clone()
commit_dict.update(commit_extra)
del commit_dict['_id']
commit = M.Commit(**commit_dict)
for p in patches:
patch_dict = state(p).document.deinstrumented_clone()
patch_dict.update(commit_extra, commit_id=commit._id)
del patch_dict['_id']
M.Patch(**patch_dict)
log.info('History copy complete')
@react('scm.cloned')
def cloned(routing_key, data):
shortname = M.Project.query.get(_id=bson.ObjectId(data['project_id'])).shortname
set_context(shortname, data['mount_point'])
repo = c.app.repo
# Update the repo status
repo.status = 'Ready'
repo.parent = data['url']
# Load the log & create refresh commit messages
log.info('Begin log %s', data['url'])
type = c.app.config.options['type']
if type == 'hg':
cmd = hg.scm_log('-g', '-p')
parser = hg.LogParser(repo._id)
cmd.run(output_consumer=parser.feed)
elif type == 'git':
cmd = git.scm_log('-p')
parser = git.LogParser(repo._id)
cmd.run(output_consumer=parser.feed)
elif type == 'svn':
cmd = hg.scm_log('-g', '-p', '--debug', 'hg_repo')
parser = hg.LogParser(repo._id)
cmd.run(output_consumer=parser.feed)
else:
log.warning('Cannot index repos of type %s', type)
log.info('Log complete %s', data['url'])