| Index: ensure_dependencies.py |
| diff --git a/ensure_dependencies.py b/ensure_dependencies.py |
| index d8af9e718b27d802acd7cde70301341220c6f7eb..b6557d051f6661645dea95edfde280ea288f7533 100755 |
| --- a/ensure_dependencies.py |
| +++ b/ensure_dependencies.py |
| @@ -79,6 +79,9 @@ class Mercurial(): |
| module = os.path.relpath(target, repo) |
| _ensure_line_exists(ignore_path, module) |
| + def postprocess_url(self, url): |
| + return url |
| + |
| class Git(): |
| def istype(self, repodir): |
| return os.path.exists(os.path.join(repodir, ".git")) |
| @@ -104,6 +107,12 @@ class Git(): |
| exclude_file = os.path.join(repo, ".git", "info", "exclude") |
| _ensure_line_exists(exclude_file, module) |
| + def postprocess_url(self, url): |
| + # Handle alternative syntax of SSH URLS |
| + if "@" in url and ":" in url and not urlparse.urlsplit(url).scheme: |
| + return "ssh://" + url.replace(":", "/", 1) |
| + return url |
| + |
| repo_types = OrderedDict(( |
| ("hg", Mercurial()), |
| ("git", Git()), |
| @@ -187,10 +196,14 @@ def ensure_repo(parentrepo, target, roots, sourcename): |
| if type is None: |
| raise Exception("No valid source found to create %s" % target) |
| - if os.path.exists(roots[type]): |
| - url = os.path.join(roots[type], sourcename) |
| + postprocess_url = repo_types[type].postprocess_url |
| + root = postprocess_url(roots[type]) |
| + sourcename = postprocess_url(sourcename) |
| + |
| + if os.path.exists(root): |
| + url = os.path.join(root, sourcename) |
| else: |
| - url = urlparse.urljoin(roots[type], sourcename) |
| + url = urlparse.urljoin(root, sourcename) |
| logging.info("Cloning repository %s into %s" % (url, target)) |
| repo_types[type].clone(url, target) |