1 files changed, 75 insertions, 0 deletions

diff --git a/scripts/builder/repository.py b/scripts/builder/repository.py
new file mode 100644
index 0000000..89db9a5
--- a/dev/null
+++ b/scripts/builder/repository.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+
+
+def repositoryWithPath (path):
+        try:
+                from dulwich.repo import Repo
+
+                repo = Repo(path)
+                result = GitRepository(repo, path)
+        except:
+                from mercurial import ui, hg
+
+                repo = hg.repository(ui.ui(), path)
+                result = HgRepository(repo, path)
+
+        return result
+
+
+#===================================================================
+
+
+class Repository(object):
+
+        def __init__ (self, repository, path):
+                self.repository = repository
+                self.path = path
+
+
+        def revision (self):
+                raise NotImplementedError()
+        
+
+        def areTherePendingChanges (self):
+                raise NotImplementedError()
+
+
+        def version (self):
+                result = self.revision()
+                if self.areTherePendingChanges():
+                        result = '>>> ' + result + ' <<<'
+
+                # print "VERSION: " + result
+                return result
+
+
+#===================================================================
+
+
+class GitRepository(Repository):
+
+        def revision (self):
+                return repository.refs['HEAD']
+        
+
+        def areTherePendingChanges (self):
+                return repository.is_dirty()
+
+
+#===================================================================
+
+
+class HgRepository(Repository):
+                #http://mercurial.selenic.com/wiki/MercurialApi
+
+        def revision (self):
+                return 'hg:' + str(self.repository['tip'])
+        
+
+        def areTherePendingChanges (self):
+                # TODO: FIXME: repository.status() does not report 'unknown(?)' files. :(
+                return not all(map(lambda fileList: len(fileList) == 0, self.repository.status()))
+
+
+#===================================================================