# filecheck.py - changeset filename check for mercurial

#

# should be configured as pretxnchangegroup to vet incoming changesets

# and as pretxncommit to vet local commit

#

# will receive a group of changesets from provided node to tip

from mercurial import util

from mercurial.i18n import _

import re

badpatterns = ('.*\.ttt\s*$','c.ttf','.*\.ttf\s*$','.*\.bbb\s*$',

               '.*\.bbb\s*$','.*\.ccc\s*$','.*\.ddd\s*$','.*\.eee\s*$','.*\.fff\s*$','.*\.ggg\s*$')

badexpr=[]

runonce=0

def init():

    global badexpr

    for p in badpatterns:

        badexpr.append((re.compile((p),re.IGNORECASE)))

def deny(f):

    global runonce

    if (not runonce):

        init()

        runonce =1

    for pat in badexpr:

        if(pat.match(f)):

            return(1)

    return(0)

def push_hook(ui, repo, hooktype, node=None, source=None, **kwargs):

    if hooktype != 'pretxnchangegroup':

        raise util.Abort(_('config error - hook type "%s" cannot stop '

                           'incoming changesets') % hooktype)

    # iterate over all the added changesets between node and tip

    for rev in xrange(repo[node], len(repo)):

        ctx = repo[rev]

        for f in ctx.files():

            if deny(f):

                ui.debug(_('filecheck: file %s not allowed \n') % (f))

                raise util.Abort(_('filecheck: access denied for changeset %s file %s blocked') % (ctx,f))

        ui.debug(_('filecheck: allowing changeset %s\n') % ctx)

def commit_hook(ui, repo, hooktype, node=None, source=None, **kwargs):

    # iterate over all the files in added changeset

    ctx = repo[node]

    for f in ctx.files():

        if deny(f):

            ui.debug(_('filecheck: file %s not allowed \n') % (f))

            raise util.Abort(_('filecheck: access denied for changeset %s file %s blocked') % (ctx,f))

    ui.debug(_('filecheck: allowing changeset %s\n') % ctx)