8d8d836284a95360222bb16b786777af7d6bc9f3
[mass-build.git] / vcs.py
1 # mass-build - Easily Build Software Involving a Large Amount of Source Repositories
2 # Copyright (C) 2012-2013 Ralf Jung <post@ralfj.de>
3 #
4 # This program is free software; you can redistribute it and/or modify
5 # it under the terms of the GNU General Public License as published by
6 # the Free Software Foundation; either version 2 of the License, or
7 # (at your option) any later version.
8 #
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 # GNU General Public License for more details.
13 #
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software
16 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17
18 import os, subprocess, re
19
20 '''A VCS must have an "update" method with an optional "mode" parameter taking one of the three values below,
21    a "version" method returning a version name (or None),
22    and a "newVersions" method which checks for new versions and prints the result to standard output.'''
23 MODE_FETCH = 0
24 MODE_REBASE = 1
25 MODE_RESET = 2
26
27 def natural_sort_key(val):
28         return [ (int(c) if c.isdigit() else c) for c in re.split('([0-9]+)', val) ]
29
30 def get_non_digit_prefix(val):
31         return re.match('[^0-9]*', val).group(0)
32
33 class GitCommand:
34         def __getattr__(self, name):
35                 def call(*args):
36                         cmd = ["git", name.replace('_', '-')] + list(args)
37                         p = subprocess.Popen(cmd, stdout=subprocess.PIPE) # TODO use with
38                         (stdout, stderr) = p.communicate()
39                         if p.returncode != 0:
40                                 raise Exception("Running %s returned non-zero exit code %d" % (str(cmd), p.returncode))
41                         return filter(len, stdout.split('\n')) # return list of non-empty lines
42                 return call
43 git = GitCommand()
44
45 # Fetch updates from git
46 class Git:
47         def __init__(self, folder, config):
48                 self.folder = os.path.abspath(folder)
49                 self.url = config['url']
50                 self.commit = config['version']
51
52         def update(self, mode = MODE_REBASE):
53                 isBranch = (self.commit.startswith('origin/'))
54                 if isBranch:
55                         branchname = self.commit[len('origin/'):]
56                 else:
57                         branchname = "tag"
58                 # get us a git repository, and the "origin" remote
59                 if os.path.exists(self.folder):
60                         # load existing repo
61                         os.chdir(self.folder)
62                         git.remote("set-url", "origin", self.url) # make sure we use the current URL
63                 else:
64                         # create a new one
65                         os.makedirs(self.folder)
66                         os.chdir(self.folder)
67                         git.init()
68                         git.remote("add", "origin", self.url)
69                 git.fetch("origin")
70                 if mode == MODE_FETCH:
71                         return
72                 # create/find correct branch
73                 if not git.branch("--list", branchname): # the branch does not yet exit
74                         git.branch(branchname, self.commit)
75                         if isBranch: # make sure we track remote branch
76                                 git.branch("-u", self.commit, branchname)
77                 # update it to the latest remote commit
78                 git.checkout(branchname)
79                 if mode == MODE_RESET:
80                         git.reset("--hard", self.commit)
81                 else:
82                         git.rebase(self.commit)
83                 # update submodules
84                 git.submodule("update", "--init", "--recursive", "--rebase")
85                 # done
86                 print "...done",
87                 if git.rev_parse("HEAD")[0] != git.rev_parse(self.commit)[0]:
88                         print "(keeping local patches around)",
89                 print
90
91         def version(self):
92                 repo = git.Repo(self.folder)
93                 v = repo.git.describe()
94                 return v[len(get_non_digit_prefix(v)):] # remove the non-digit prefix from v (so that it starts with a number)
95
96         def checkVersions(self):
97                 self.update(mode = MODE_FETCH)
98                 currentVersion = git.describe()[0]
99                 # get sorted list of tag names with the same non-digit prefix and higher version number
100                 tags = git.tag()
101                 tags = filter(lambda t: get_non_digit_prefix(t) == get_non_digit_prefix(currentVersion) and natural_sort_key(t) > natural_sort_key(currentVersion), tags)
102                 if not tags: return
103                 tags.sort(key = natural_sort_key)
104                 print "Versions newer than "+currentVersion+" available:"
105                 print tags
106
107 # Fetch updates via SVN
108 class SVN:
109         def __init__(self, folder, url):
110                 self.folder = os.path.abspath(folder)
111                 self.url = url
112
113         def update(self, mode = MODE_REBASE):
114                 if mode == MODE_FETCH: raise Exception("Just fetching is not supported with SVN")
115                 if os.path.exists(self.folder):
116                         os.chdir(self.folder) # go into repository
117                         if mode == MODE_RESET: subprocess.check_call(['svn', 'revert', '-R', '.'])
118                         subprocess.check_call(['svn', 'switch', self.url]) # and update to the URL we got
119                 else:
120                         os.makedirs(self.folder) # if even the parent folder does not exist, svn fails
121                         subprocess.check_call(['svn', 'co', self.url, self.folder]) # just download it
122         
123         def version(self):
124                 return None
125         
126         def checkVersions(self):
127                 print "Version checking not supporting with SVN"