"""Mercurial access control hook"""

from mercurial.i18n import _
import mercurial.util
import mercurial.node

import os
from mercurialserver import ruleset
from mercurialserver import changes

def allow(ctx):
    branch = ctx.branch()
    if not ruleset.rules.allow("write", branch=branch, file=None):
        return False
    for f in ctx.files():
        if not ruleset.rules.allow("write", branch=branch, file=f):
            return False
    return True

def hook(ui, repo, hooktype, node=None, source=None, **kwargs):
    if hooktype != 'pretxnchangegroup':
        raise mercurial.util.Abort(_('config error - hook type "%s" cannot stop '
                           'incoming changesets') % hooktype)
    for ctx in changes.changes(repo, node):
        if not allow(ctx):
            raise mercurial.util.Abort(_('%s: access denied for changeset %s') %
                (__name__, mercurial.node.short(ctx.node())))

def phasehook(ui, repo, hooktype, **kwargs):
    if hooktype != 'prepushkey':
        raise mercurial.util.Abort(_('config error - hook type "%s" cannot stop '
                                     'incoming phase changes') % hooktype)
    if kwargs.get('namespace') == 'phases':
        if not ruleset.rules.allow("publish"):
            raise mercurial.util.Abort(_('%s: access denied for making %s public') %
                                       (__name__, kwargs['key'][:12]))