X-Git-Url: https://git.ralfj.de/mass-build.git/blobdiff_plain/0847fb06e91eacd061980ca614957f2134692600..e477d3a808951963a80c6e5918eb075a8c5e086c:/vcs.py

diff --git a/vcs.py b/vcs.py
index b2387c7..0dc9685 100644
--- a/vcs.py
+++ b/vcs.py
@@ -1,6 +1,6 @@
 import os, git, subprocess
 
-'''A VCS must have an "update" method with an optional "force" parameter.'''
+'''A VCS must have an "update" method with an optional "forceVersion" parameter, and optionally a "version" method.'''
 
 # Fetch updates from git
 class Git:
@@ -13,7 +13,7 @@ class Git:
 		def update(self, op_code, cur_count, max_count=None, message=''):
 			print self._cur_line+(" "*30)+"\r",
 
-	def update(self):
+	def update(self, forceVersion=False):
 		isBranch = (self.commit.startswith('origin/'))
 		if isBranch:
 			branchname = self.commit[len('origin/'):]
@@ -24,6 +24,7 @@ class Git:
 			# load existing repo
 			repo = git.Repo(self.folder)
 			origin = repo.remotes.origin
+			origin.config_writer.set_value("url", self.url) # make sure we use the current URL
 		else:
 			# create a new one
 			os.makedirs(self.folder)
@@ -40,7 +41,13 @@ class Git:
 				branch.set_tracking_branch(origin.refs[branchname])
 		# update it to the latest remote commit
 		branch.checkout()
-		repo.git.rebase(self.commit)
+		if forceVersion:
+			repo.head.reset(self.commit, working_tree=True)
+		else:
+			repo.git.rebase(self.commit)
+		# update submodules
+		repo.git.submodule("update", "--init", "--recursive", "--rebase")
+		# done
 		print "...done",
 		if repo.head.reference.commit != repo.commit(self.commit):
 			print "(keeping local patches around)",
@@ -54,17 +61,15 @@ class Git:
 
 # Fetch updates via SVN
 class SVN:
-	def __init__(self, folder, svnPath, versionName):
+	def __init__(self, folder, url):
 		self.folder = os.path.abspath(folder)
-		self.svnPath = svnPath
-		self.versionName = versionName
+		self.url = url
 
-	def update(self):
+	def update(self, forceVersion=False):
 		if os.path.exists(self.folder):
 			os.chdir(self.folder) # go into repository
-			subprocess.check_call(['svn', 'switch', self.svnPath]) # and update to the URL we got
+			if forceVersion: subprocess.check_call(['svn', 'revert', '-R', '.'])
+			subprocess.check_call(['svn', 'switch', self.url]) # and update to the URL we got
 		else:
-			subprocess.check_call(['svn', 'co', self.svnPath, self.folder])# just download it
-
-	def version(self):
-		return self.versionName
+			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