#-*- python -*-
import logging
import re
import os
import sys
import shutil
import email
from subprocess import Popen
from datetime import datetime
from itertools import islice, chain
from urllib import urlencode
# Non-stdlib imports
import pkg_resources
from tg import expose, validate, redirect, response
from tg.decorators import with_trailing_slash, without_trailing_slash
import pylons
from pylons import g, c, request
from formencode import validators
from pymongo import bson
from webob import exc
from mercurial import ui, hg
from ming.orm.ormsession import ThreadLocalORMSession
from ming.orm.base import mapper
from pymongo.bson import ObjectId
# Pyforge-specific imports
from pyforge.app import Application, ConfigOption, SitemapEntry, DefaultAdminController
from pyforge.lib.helpers import push_config, DateTimeConverter, mixin_reactors
from pyforge.lib.search import search
from pyforge.lib.decorators import audit, react
from pyforge.lib.security import require, has_artifact_access
from pyforge.model import Project, ProjectRole, User, ArtifactReference, Feed
# Local imports
from forgehg import model
from forgehg import version
from .widgets import HgRevisionWidget
from .reactors import reactors
log = logging.getLogger(__name__)
class W(object):
revision_widget = HgRevisionWidget()
class ForgeHgApp(Application):
'''This is the Hg app for PyForge'''
__version__ = version.__version__
permissions = [ 'read', 'write', 'create', 'admin', 'configure' ]
config_options = Application.config_options + [
ConfigOption('cloned_from', ObjectId, None)
]
def __init__(self, project, config):
Application.__init__(self, project, config)
self.root = RootController()
self.admin = HgAdminController(self)
@property
def sitemap(self):
menu_id = self.config.options.mount_point.title()
with push_config(c, app=self):
return [
SitemapEntry(menu_id, '.')[self.sidebar_menu()] ]
def admin_menu(self):
return super(ForgeHgApp, self).admin_menu()
def sidebar_menu(self):
links = [ SitemapEntry('Home',c.app.url, ui_icon='home') ]
if has_artifact_access('admin', app=c.app)():
links.append(SitemapEntry('Admin', c.project.url()+'admin/'+self.config.options.mount_point, ui_icon='wrench'))
repo = c.app.repo
if repo and repo.status == 'ready':
branches= repo.branchmap().keys()
tags = repo.repo_tags().keys()
if branches:
links.append(SitemapEntry('Branches'))
for b in branches:
links.append(SitemapEntry(
b, c.app.url + '?' + urlencode(dict(branch=b)),
className='nav_child'))
if tags:
links.append(SitemapEntry('Tags'))
for b in tags:
links.append(SitemapEntry(
b, c.app.url + '?' + urlencode(dict(tag=b)),
className='nav_child'))
return links
@property
def repo(self):
return model.HgRepository.query.get(app_config_id=self.config._id)
@property
def templates(self):
return pkg_resources.resource_filename('forgehg', 'templates')
def install(self, project):
'Set up any default permissions and roles here'
self.config.options['project_name'] = project.name
super(ForgeHgApp, self).install(project)
# Setup permissions
role_developer = ProjectRole.query.get(name='Developer')._id
role_auth = ProjectRole.query.get(name='*authenticated')._id
self.config.acl.update(
configure=c.project.acl['tool'],
read=c.project.acl['read'],
write=[role_developer],
create=[role_developer],
admin=c.project.acl['tool'])
repo = model.HgRepository()
repo.name = self.config.options.mount_point
repo.fs_path = '/hg/' + c.project.shortname + '/'
repo.url_path = '/' + c.project.shortname + '/'
repo.tool = 'hg'
repo.status = 'initing'
ThreadLocalORMSession.flush_all()
cloned_from_id = self.config.options.get('cloned_from')
if cloned_from_id is not None:
cloned_from = model.HgRepository.query.get(_id=cloned_from_id)
g.publish('audit', 'scm.hg.clone',
dict(repo_name=repo.name, repo_path=repo.fs_path, cloned_from=cloned_from.full_fs_path))
else:
g.publish('audit', 'scm.hg.init',
dict(repo_name=repo.name, repo_path=repo.fs_path))
def uninstall(self, project):
g.publish('audit', 'scm.hg.uninstall', dict(project_id=project._id))
@audit('scm.hg.uninstall')
def _uninstall(self, routing_key, data):
"Remove all the tool's artifacts and the physical repository"
repo = self.repo
if repo is not None:
shutil.rmtree(repo.full_fs_path, ignore_errors=True)
model.HgRepository.query.remove(dict(app_config_id=self.config._id))
super(ForgeHgApp, self).uninstall(project_id=data['project_id'])
class HgAdminController(DefaultAdminController):
@with_trailing_slash
def index(self):
redirect('permissions')
class RootController(object):
@expose('forgehg.templates.index')
def index(self, offset=0, branch=None, tag=None):
offset=int(offset)
repo = c.app.repo
if repo and repo.status == 'ready':
revisions = repo.log(branch=branch, tag=tag)
revisions = islice(revisions, offset, offset+10)
args = dict(offset=offset+10)
if branch: args['branch'] = branch
if tag: args['tag'] = tag
next_link = '?' + urlencode(args)
else:
revisions = []
next_link=None
c.revision_widget=W.revision_widget
return dict(repo=c.app.repo,
branch=branch,
tag=tag,
host=request.host,
revisions=revisions,
next_link=next_link,
offset=offset,
allow_fork=True)
@with_trailing_slash
@expose('forgehg.templates.fork')
def fork(self, to_name=None):
from_repo = c.app.repo
to_project_name = 'u/' + c.user.username
ThreadLocalORMSession.flush_all()
ThreadLocalORMSession.close_all()
to_project = Project.query.get(shortname=to_project_name)
with push_config(c, project=to_project):
if request.method!='POST' or to_name is None:
prefix_len = len(to_project_name+'/')
in_use = [sp.shortname[prefix_len:] for sp in to_project.direct_subprojects]
in_use += [ac.options['mount_point'] for ac in to_project.app_configs]
return dict(from_repo=from_repo,
to_project_name=to_project_name,
in_use=in_use,
to_name=to_name or '')
else:
to_project.install_app('Hg', to_name, cloned_from=from_repo._id)
redirect('/'+to_project_name+'/'+to_name+'/')
@expose()
def _lookup(self, hash, *remainder):
return CommitController(hash), remainder
class CommitController(object):
def __init__(self, hash):
self._hash = hash
@expose('forgehg.templates.commit')
def index(self):
commit = c.app.repo[self._hash]
c.revision_widget=W.revision_widget
return dict(commit=commit)
mixin_reactors(ForgeHgApp, reactors)