--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/hg_hooks/bugzilla/versiontobugzilla.py	Thu Jul 23 12:50:28 2009 +0100
@@ -0,0 +1,136 @@
+'''Bugzilla integration for adding versions from tags
+
+The hook updates the Bugzilla database directly. Only Bugzilla installations
+using MySQL are supported.
+
+This hook uses the same .hgrc parameters as the default Bugzilla hook. There
+is no need for configuring the same stuff twice. (connection, etc.)
+
+Configuring the extension: (same as Bugzilla -hook)
+
+    [bugzilla]
+    host       Hostname of the MySQL server holding the Bugzilla database.
+    db         Name of the Bugzilla database in MySQL. Default 'bugs'.
+    user       Username to use to access MySQL server. Default 'bugs'.
+    password   Password to use to access MySQL server.
+    timeout    Database connection timeout (seconds). Default 5.
+
+Additional elements under Bugzilla -section: (new items)
+    [bugzilla]
+    product    The name on the Bugzilla product that is used for adding 
+               the new versions.
+
+Activating the extension:
+
+    [extensions]
+    hgext.versiontobugzilla =
+
+    [hooks]
+    incoming.versiontobugzilla = python:hgext.versiontobugzilla.hook
+
+Example configuration in hgrc:
+    [bugzilla]
+    host = localhost
+    user = bugs
+    password = password
+    product = my_product
+
+    [extensions]
+    hgext.versiontobugzilla =
+
+    [hooks]
+    incoming.versiontobugzilla = python:hgext.versiontobugzilla.hook
+'''
+
+from mercurial import util
+import re
+
+MySQLdb = None
+
+class BugzillaClient:
+    
+    def __init__(self, ui, repo, node):
+        self.tag = None
+        self.ui = ui
+        self.repo = repo
+        self.node = node
+        self.product = ui.config('bugzilla', 'product')
+        self.host = ui.config('bugzilla', 'host', 'localhost')
+        self.user = ui.config('bugzilla', 'user', 'bugs')
+        self.passwd = ui.config('bugzilla', 'password')
+        self.db = ui.config('bugzilla', 'db', 'bugs')
+        self.timeout = int(ui.config('bugzilla', 'timeout', 10))
+        self.connection = MySQLdb.connect(host=self.host, user=self.user, passwd=self.passwd,
+                                    db=self.db, connect_timeout=self.timeout)
+        self.cursor = self.connection.cursor()
+
+    def printMessageInVerboseMode(self, message):
+        '''Prints a message to console if hg has been executed with -v option.'''
+        self.ui.note(message)
+
+    def executeDatabaseQuery(self, *args, **kwargs):
+        self.printMessageInVerboseMode('Bugzilla: query: %s %s\n' % (args, kwargs))
+        try:
+            self.cursor.execute(*args, **kwargs)
+        except MySQLdb.MySQLError:
+            self.printMessageInVerboseMode('Bugzilla: failed query: %s %s\n' % (args, kwargs))
+            raise
+
+    def commitContainsTag(self):
+        self.parseTagFromCommitMessage()
+        if self.tag:
+            return True
+        else:
+            return False
+
+    def parseTagFromCommitMessage(self):
+        ctx = self.repo[self.node]
+        version_re = re.compile(('Added tag (.+) for changeset [0-9a-h]+'), re.IGNORECASE)
+        m = version_re.search(ctx.description())
+        if m:
+            self.tag = m.group(1)
+
+    def insertTagIntoDatabase(self):
+        self.makeSureThatProductExists()
+        if not self.doesVersionAlreadyExist():
+            self.printMessageInVerboseMode("Bugzilla: adding version '%s' to product '%s' in database.\n" % (self.tag, self.product))
+            self.insertNewVersionIntoDatabase()
+        else:
+            self.printMessageInVerboseMode("Bugzilla: product '%s' already has a version '%s' in database. Not trying to add it again." % (self.product, self.tag))
+
+    def makeSureThatProductExists(self):
+        self.executeDatabaseQuery('select id from products where name = %s', (self.product,))
+        ids = self.cursor.fetchall()
+        if len(ids) != 1:
+            raise util.Abort("Product '%s' does not exist in database, please check the [bugzilla] -section in hgrc." % self.product)
+
+    def doesVersionAlreadyExist(self):
+        self.executeDatabaseQuery('select * from versions where value = %s and product_id in (select id from products where name=%s )', (self.tag, self.product))
+        ids = self.cursor.fetchall()
+        if len(ids) == 1:
+            return True
+        else:
+            return False
+
+    def insertNewVersionIntoDatabase(self):
+        self.executeDatabaseQuery('insert into versions (value, product_id) values (%s, (select id from products where name=%s ))', (self.tag, self.product))
+        self.connection.commit()
+
+def hook(ui, repo, hooktype, node=None, **kwargs):
+
+    try:
+        import MySQLdb as mysql
+        global MySQLdb
+        MySQLdb = mysql
+    except ImportError, err:
+        raise util.Abort('MySQL driver not installed: %s' % err)
+
+    if node is None:
+        raise util.Abort('Only hooks that have changesetid''s can be used.')
+
+    try: 
+        bzClient = BugzillaClient(ui, repo, node)
+        if bzClient.commitContainsTag():
+            bzClient.insertTagIntoDatabase()
+    except MySQLdb.MySQLError, err:
+        raise util.Abort('Database error: %s' % err[1])

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/hg_hooks/filecheck/filecheck.py	Thu Jul 23 12:50:28 2009 +0100
@@ -0,0 +1,56 @@
+# 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)