implement a version checker

[mass-build.git] / vcs.py

diff --git a/vcs.py b/vcs.py
index 564b0c14818069dd552e97759689411a6992bc67..02a919223157157f3bf655c0bf76da6d9128532e 100644 (file)
--- a/vcs.py
+++ b/vcs.py
@@ -15,12 +15,21 @@
 # along with this program; if not, write to the Free Software
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 
-import os, git, subprocess
+import os, git, subprocess, re
 
-'''A VCS must have an "update" method with an optional "forceVersion" parameter, and optionally a "version" method.'''
+'''A VCS must have an "update" method with an optional "mode" parameter taking one of the three values below,
+   a "version" method returning a version name (or None),
+   and a "newVersions" method which checks for new versions and prints the result to standard output.'''
+MODE_FETCH = 0
+MODE_REBASE = 1
+MODE_RESET = 2
+
+def natural_sort_key(val):
+       return [ (int(c) if c.isdigit() else c) for c in re.split('([0-9]+)', val) ]
 
 # Fetch updates from git
 class Git:
+       
        def __init__(self, folder, url, commit):
                self.folder = os.path.abspath(folder)
                self.url = url
@@ -30,7 +39,7 @@ class Git:
                def update(self, op_code, cur_count, max_count=None, message=''):
                        print self._cur_line+(" "*30)+"\r",
 
-       def update(self, forceVersion=False):
+       def update(self, mode = MODE_REBASE):
                isBranch = (self.commit.startswith('origin/'))
                if isBranch:
                        branchname = self.commit[len('origin/'):]
@@ -49,6 +58,8 @@ class Git:
                        origin = repo.create_remote('origin', self.url)
                origin.fetch(progress=Git._ProgressPrinter()) # download new data
                print " "*80+"\r", # clean the line we are in
+               if mode == MODE_FETCH:
+                       return
                # create/find correct branch
                if branchname in repo.heads:
                        branch = repo.heads[branchname]
@@ -58,7 +69,7 @@ class Git:
                                branch.set_tracking_branch(origin.refs[branchname])
                # update it to the latest remote commit
                branch.checkout()
-               if forceVersion:
+               if mode == MODE_RESET:
                        repo.head.reset(self.commit, working_tree=True)
                else:
                        repo.git.rebase(self.commit)
@@ -76,6 +87,23 @@ class Git:
                if v.startswith('v'): v = v[1:]
                return v
 
+       def checkVersions(self):
+               repo = git.Repo(self.folder)
+               # get tag for current commit, if any
+               commit = repo.commit(self.commit)
+               commitTag = filter(lambda t: t.commit == commit, repo.tags)
+               if not commitTag:
+                       print "Version is not a tag"
+                       return
+               currentVersion = str(commitTag[0])
+               # get sorted list of tag names
+               tags = map(str, repo.tags)
+               tags = filter(lambda t: natural_sort_key(t) > natural_sort_key(currentVersion), tags)
+               if not tags: return
+               tags.sort(key = natural_sort_key)
+               print "Versions newer than "+self.commit+" available:"
+               print tags
+
 # Fetch updates via SVN
 class SVN:
        def __init__(self, folder, url):
@@ -90,3 +118,9 @@ class SVN:
                else:
                        os.makedirs(self.folder) # if even the parent folder does not exist, svn fails
                        subprocess.check_call(['svn', 'co', self.url, self.folder]) # just download it
+       
+       def version(self):
+               return None
+       
+       def checkVersions(self):
+               print "Version checking not supporting with SVN"