1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
#!/usr/bin/env python
# Stolen from taskcluster-vcs
# https://github.com/taskcluster/taskcluster-vcs/blob/master/src/vcs/detect_remote.js
from urllib2 import Request, urlopen
from urlparse import urlsplit, urlunsplit
from os.path import exists, join
def first(seq):
return next(iter(filter(lambda x: x, seq)), '')
def all_first(*sequences):
return map(lambda x: first(x), sequences)
# http://codereview.stackexchange.com/questions/13027/joining-url-path-components-intelligently
# I wonder why this is not a builtin feature in Python
def urljoin(*parts):
schemes, netlocs, paths, queries, fragments = zip(*(urlsplit(part) for part in parts))
scheme, netloc, query, fragment = all_first(schemes, netlocs, queries, fragments)
path = '/'.join(p.strip('/') for p in paths if p)
return urlunsplit((scheme, netloc, path, query, fragment))
def _detect_remote(url, content):
try:
response = urlopen(url)
except Exception:
return False
if response.getcode() != 200:
return False
content_type = response.headers.get('content-type', '')
return True if content in content_type else False
def detect_git(url):
location = urljoin(url, '/info/refs?service=git-upload-pack')
req = Request(location, headers={'User-Agent':'git/2.0.1'})
return _detect_remote(req, 'x-git')
def detect_hg(url):
location = urljoin(url, '?cmd=lookup&key=0')
return _detect_remote(location, 'mercurial')
def detect_local(url):
if exists(join(url, '.git')):
return 'git'
if exists(join(url, '.hg')):
return 'hg'
return ''
|
