Mercurial > hg > mercurial-crew-with-dirclash
annotate mercurial/httprepo.py @ 1239:29f17e083e84
Turn hgit into an extension, and add commands supporting the latest gitk
author | mason@suse.com |
---|---|
date | Tue, 13 Sep 2005 19:33:10 -0500 |
parents | 142b5d5ec9cc |
children | 84cf8834efb5 |
rev | line source |
---|---|
1089 | 1 # httprepo.py - HTTP repository proxy classes for mercurial |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
2 # |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
3 # Copyright 2005 Matt Mackall <mpm@selenic.com> |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
4 # |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
5 # This software may be used and distributed according to the terms |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
6 # of the GNU General Public License, incorporated herein by reference. |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
7 |
1089 | 8 import urllib, urllib2, urlparse, os, zlib |
9 from node import * | |
10 from remoterepo import * | |
926 | 11 |
12 class httprepository(remoterepository): | |
60 | 13 def __init__(self, ui, path): |
765 | 14 # fix missing / after hostname |
15 s = urlparse.urlsplit(path) | |
16 partial = s[2] | |
17 if not partial: partial = "/" | |
18 self.url = urlparse.urlunsplit((s[0], s[1], partial, '', '')) | |
60 | 19 self.ui = ui |
321 | 20 no_list = [ "localhost", "127.0.0.1" ] |
21 host = ui.config("http_proxy", "host") | |
424
9294dce4b633
Allow override of HTTP_PROXY, http_proxy and no_proxy; make no_proxy work.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
385
diff
changeset
|
22 if host is None: |
9294dce4b633
Allow override of HTTP_PROXY, http_proxy and no_proxy; make no_proxy work.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
385
diff
changeset
|
23 host = os.environ.get("http_proxy") |
426
8c90ab5644c9
Allow hgrc's proxy host and $http_proxy env var to start with http://
Thomas Arendsen Hein <thomas@intevation.de>
parents:
424
diff
changeset
|
24 if host and host.startswith('http://'): |
8c90ab5644c9
Allow hgrc's proxy host and $http_proxy env var to start with http://
Thomas Arendsen Hein <thomas@intevation.de>
parents:
424
diff
changeset
|
25 host = host[7:] |
321 | 26 user = ui.config("http_proxy", "user") |
27 passwd = ui.config("http_proxy", "passwd") | |
28 no = ui.config("http_proxy", "no") | |
424
9294dce4b633
Allow override of HTTP_PROXY, http_proxy and no_proxy; make no_proxy work.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
385
diff
changeset
|
29 if no is None: |
9294dce4b633
Allow override of HTTP_PROXY, http_proxy and no_proxy; make no_proxy work.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
385
diff
changeset
|
30 no = os.environ.get("no_proxy") |
321 | 31 if no: |
32 no_list = no_list + no.split(",") | |
515 | 33 |
321 | 34 no_proxy = 0 |
35 for h in no_list: | |
36 if (path.startswith("http://" + h + "/") or | |
37 path.startswith("http://" + h + ":") or | |
38 path == "http://" + h): | |
39 no_proxy = 1 | |
40 | |
41 # Note: urllib2 takes proxy values from the environment and those will | |
42 # take precedence | |
424
9294dce4b633
Allow override of HTTP_PROXY, http_proxy and no_proxy; make no_proxy work.
Thomas Arendsen Hein <thomas@intevation.de>
parents:
385
diff
changeset
|
43 for env in ["HTTP_PROXY", "http_proxy", "no_proxy"]: |
859
6390c377a9e6
Trap OSError when deleting env vars
Edouard Gomez <ed.gomez@free.fr>
parents:
856
diff
changeset
|
44 try: |
6390c377a9e6
Trap OSError when deleting env vars
Edouard Gomez <ed.gomez@free.fr>
parents:
856
diff
changeset
|
45 if os.environ.has_key(env): |
6390c377a9e6
Trap OSError when deleting env vars
Edouard Gomez <ed.gomez@free.fr>
parents:
856
diff
changeset
|
46 del os.environ[env] |
6390c377a9e6
Trap OSError when deleting env vars
Edouard Gomez <ed.gomez@free.fr>
parents:
856
diff
changeset
|
47 except OSError: |
6390c377a9e6
Trap OSError when deleting env vars
Edouard Gomez <ed.gomez@free.fr>
parents:
856
diff
changeset
|
48 pass |
321 | 49 |
50 proxy_handler = urllib2.BaseHandler() | |
51 if host and not no_proxy: | |
52 proxy_handler = urllib2.ProxyHandler({"http" : "http://" + host}) | |
53 | |
54 authinfo = None | |
55 if user and passwd: | |
56 passmgr = urllib2.HTTPPasswordMgrWithDefaultRealm() | |
57 passmgr.add_password(None, host, user, passwd) | |
58 authinfo = urllib2.ProxyBasicAuthHandler(passmgr) | |
59 | |
60 opener = urllib2.build_opener(proxy_handler, authinfo) | |
61 urllib2.install_opener(opener) | |
60 | 62 |
634
da5378d39269
Add a repo method to report repo device
Matt Mackall <mpm@selenic.com>
parents:
627
diff
changeset
|
63 def dev(self): |
da5378d39269
Add a repo method to report repo device
Matt Mackall <mpm@selenic.com>
parents:
627
diff
changeset
|
64 return -1 |
da5378d39269
Add a repo method to report repo device
Matt Mackall <mpm@selenic.com>
parents:
627
diff
changeset
|
65 |
60 | 66 def do_cmd(self, cmd, **args): |
83 | 67 self.ui.debug("sending %s command\n" % cmd) |
60 | 68 q = {"cmd": cmd} |
69 q.update(args) | |
70 qs = urllib.urlencode(q) | |
71 cu = "%s?%s" % (self.url, qs) | |
752 | 72 resp = urllib2.urlopen(cu) |
753 | 73 proto = resp.headers['content-type'] |
752 | 74 |
753 | 75 # accept old "text/plain" and "application/hg-changegroup" for now |
76 if not proto.startswith('application/mercurial') and \ | |
77 not proto.startswith('text/plain') and \ | |
78 not proto.startswith('application/hg-changegroup'): | |
752 | 79 raise RepoError("'%s' does not appear to be an hg repository" |
80 % self.url) | |
81 | |
753 | 82 if proto.startswith('application/mercurial'): |
83 version = proto[22:] | |
84 if float(version) > 0.1: | |
85 raise RepoError("'%s' uses newer protocol %s" % | |
86 (self.url, version)) | |
87 | |
752 | 88 return resp |
60 | 89 |
222 | 90 def heads(self): |
91 d = self.do_cmd("heads").read() | |
92 try: | |
93 return map(bin, d[:-1].split(" ")) | |
94 except: | |
95 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n") | |
96 raise | |
97 | |
60 | 98 def branches(self, nodes): |
99 n = " ".join(map(hex, nodes)) | |
752 | 100 d = self.do_cmd("branches", nodes=n).read() |
217 | 101 try: |
102 br = [ tuple(map(bin, b.split(" "))) for b in d.splitlines() ] | |
103 return br | |
104 except: | |
105 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n") | |
106 raise | |
60 | 107 |
108 def between(self, pairs): | |
109 n = "\n".join(["-".join(map(hex, p)) for p in pairs]) | |
752 | 110 d = self.do_cmd("between", pairs=n).read() |
217 | 111 try: |
112 p = [ l and map(bin, l.split(" ")) or [] for l in d.splitlines() ] | |
113 return p | |
114 except: | |
115 self.ui.warn("unexpected response:\n" + d[:400] + "\n...\n") | |
116 raise | |
60 | 117 |
118 def changegroup(self, nodes): | |
119 n = " ".join(map(hex, nodes)) | |
752 | 120 f = self.do_cmd("changegroup", roots=n) |
192 | 121 bytes = 0 |
635
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
122 |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
123 class zread: |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
124 def __init__(self, f): |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
125 self.zd = zlib.decompressobj() |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
126 self.f = f |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
127 self.buf = "" |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
128 def read(self, l): |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
129 while l > len(self.buf): |
751
0b245edec124
When pulling from a non hg repository URL (e.g. http://www.kernel.org/hg)
Muli Ben-Yehuda <mulix@mulix.org>
parents:
741
diff
changeset
|
130 r = self.f.read(4096) |
635
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
131 if r: |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
132 self.buf += self.zd.decompress(r) |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
133 else: |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
134 self.buf += self.zd.flush() |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
135 break |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
136 d, self.buf = self.buf[:l], self.buf[l:] |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
137 return d |
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
138 |
752 | 139 return zread(f) |
635
85e2209d401c
Protocol switch from using generators to stream-like objects.
Matt Mackall <mpm@selenic.com>
parents:
634
diff
changeset
|
140 |
923 | 141 class httpsrepository(httprepository): |
142 pass |