Chris@427
|
1 # -*- coding: utf-8 -*-
|
Chris@427
|
2 #
|
Chris@427
|
3 # EasyMercurial
|
Chris@427
|
4 #
|
Chris@427
|
5 # Based on hgExplorer by Jari Korhonen
|
Chris@427
|
6 # Copyright (c) 2010 Jari Korhonen
|
Chris@427
|
7 # Copyright (c) 2010 Chris Cannam
|
Chris@427
|
8 # Copyright (c) 2010 Queen Mary, University of London
|
Chris@427
|
9 #
|
Chris@427
|
10 # This program is free software; you can redistribute it and/or
|
Chris@427
|
11 # modify it under the terms of the GNU General Public License as
|
Chris@427
|
12 # published by the Free Software Foundation; either version 2 of the
|
Chris@427
|
13 # License, or (at your option) any later version. See the file
|
Chris@427
|
14 # COPYING included with this distribution for more information.
|
Chris@427
|
15
|
Chris@449
|
16 import sys, os, stat, urllib, urllib2, urlparse, platform, hashlib
|
Chris@427
|
17
|
Chris@438
|
18 from mercurial.i18n import _
|
Chris@433
|
19 from mercurial import ui, util, error
|
Chris@427
|
20 try:
|
Chris@427
|
21 from mercurial.url import passwordmgr
|
Chris@427
|
22 except:
|
Chris@427
|
23 from mercurial.httprepo import passwordmgr
|
Chris@427
|
24
|
Chris@427
|
25 # The value assigned here may be modified during installation, by
|
Chris@427
|
26 # replacing its default value with another one. We can't compare
|
Chris@427
|
27 # against its default value, because then the comparison text would
|
Chris@427
|
28 # get modified as well. So, compare using prefix only.
|
Chris@427
|
29 #
|
Chris@427
|
30 easyhg_import_path = 'NO_EASYHG_IMPORT_PATH'
|
Chris@427
|
31 if not easyhg_import_path.startswith('NO_'):
|
Chris@427
|
32 # We have an installation path: append it twice, once with
|
Chris@427
|
33 # the Python version suffixed
|
Chris@440
|
34 version_suffix = 'Py%d.%d' % (sys.version_info[0], sys.version_info[1])
|
Chris@427
|
35 sys.path.append(easyhg_import_path + "/" + version_suffix)
|
Chris@427
|
36 sys.path.append(easyhg_import_path)
|
Chris@427
|
37
|
Chris@427
|
38 # Try to load the PyQt4 module that we need. If this fails, we should
|
Chris@427
|
39 # bail out later (in uisetup), because if we bail out now, Mercurial
|
Chris@427
|
40 # will just continue without us and report success. The invoking
|
Chris@427
|
41 # application needs to be able to discover whether the module load
|
Chris@427
|
42 # succeeded or not, so we need to ensure that Mercurial itself returns
|
Chris@427
|
43 # failure if it didn't.
|
Chris@427
|
44 #
|
Chris@427
|
45 easyhg_pyqt_ok = True
|
Chris@427
|
46 try:
|
Chris@427
|
47 from PyQt4 import Qt, QtGui
|
Chris@427
|
48 except ImportError:
|
Chris@427
|
49 easyhg_pyqt_ok = False
|
Chris@427
|
50 easyhg_qtapp = None
|
Chris@427
|
51
|
Chris@440
|
52 # These imports are optional, we just can't use the authfile (i.e.
|
Chris@438
|
53 # "remember this password") feature without them
|
Chris@438
|
54 #
|
Chris@437
|
55 easyhg_authfile_imports_ok = True
|
Chris@437
|
56 try:
|
Chris@437
|
57 from Crypto.Cipher import AES
|
Chris@437
|
58 import ConfigParser # Mercurial version won't write files
|
Chris@437
|
59 import base64
|
Chris@437
|
60 except ImportError:
|
Chris@445
|
61 print "EasyHg: Failed to import required modules for authfile support"
|
Chris@437
|
62 easyhg_authfile_imports_ok = False
|
Chris@433
|
63
|
Chris@431
|
64
|
Chris@439
|
65 def encrypt_salted(text, key):
|
Chris@439
|
66 salt = os.urandom(8)
|
Chris@439
|
67 text = '%d.%s.%s' % (len(text), base64.b64encode(salt), text)
|
Chris@431
|
68 text += (16 - len(text) % 16) * ' '
|
Chris@446
|
69 cipher = AES.new(key, AES.MODE_CBC)
|
Chris@431
|
70 return base64.b64encode(cipher.encrypt(text))
|
Chris@431
|
71
|
Chris@439
|
72 def decrypt_salted(ctext, key):
|
Chris@446
|
73 cipher = AES.new(key, AES.MODE_CBC)
|
Chris@431
|
74 text = cipher.decrypt(base64.b64decode(ctext))
|
Chris@431
|
75 (tlen, d, text) = text.partition('.')
|
Chris@439
|
76 (salt, d, text) = text.partition('.')
|
Chris@431
|
77 return text[0:int(tlen)]
|
Chris@431
|
78
|
Chris@427
|
79 # from mercurial_keyring by Marcin Kasperski
|
Chris@427
|
80 def canonical_url(authuri):
|
Chris@427
|
81 parsed_url = urlparse.urlparse(authuri)
|
Chris@427
|
82 return "%s://%s%s" % (parsed_url.scheme, parsed_url.netloc,
|
Chris@427
|
83 parsed_url.path)
|
Chris@427
|
84
|
Chris@436
|
85 def load_config(pcfg, pfile):
|
Chris@434
|
86 fp = None
|
Chris@434
|
87 try:
|
Chris@434
|
88 fp = open(pfile)
|
Chris@434
|
89 except:
|
Chris@440
|
90 return
|
Chris@440
|
91 pcfg.readfp(fp)
|
Chris@440
|
92 fp.close()
|
Chris@434
|
93
|
Chris@444
|
94 def save_config(ui, pcfg, pfile):
|
Chris@434
|
95 ofp = None
|
Chris@434
|
96 try:
|
Chris@434
|
97 ofp = open(pfile, 'w')
|
Chris@434
|
98 except:
|
Chris@444
|
99 ui.write("failed to open authfile %s for writing\n" % pfile)
|
Chris@434
|
100 raise
|
Chris@448
|
101 if platform.system() != 'Windows':
|
Chris@448
|
102 try:
|
Chris@448
|
103 os.fchmod(ofp.fileno(), stat.S_IRUSR | stat.S_IWUSR)
|
Chris@448
|
104 except:
|
Chris@448
|
105 ofp.close()
|
Chris@448
|
106 ui.write("failed to set permissions on authfile %s\n" % pfile)
|
Chris@448
|
107 raise
|
Chris@434
|
108 pcfg.write(ofp)
|
Chris@434
|
109 ofp.close()
|
Chris@434
|
110
|
Chris@436
|
111 def get_from_config(pcfg, sect, key):
|
Chris@436
|
112 data = None
|
Chris@436
|
113 try:
|
Chris@436
|
114 data = pcfg.get(sect, key)
|
Chris@436
|
115 except (ConfigParser.NoOptionError, ConfigParser.NoSectionError):
|
Chris@436
|
116 pass
|
Chris@436
|
117 return data
|
Chris@436
|
118
|
Chris@436
|
119 def get_boolean_from_config(pcfg, sect, key, deflt):
|
Chris@436
|
120 data = deflt
|
Chris@436
|
121 try:
|
Chris@436
|
122 data = pcfg.getboolean(sect, key)
|
Chris@436
|
123 except (ConfigParser.NoOptionError, ConfigParser.NoSectionError):
|
Chris@436
|
124 pass
|
Chris@436
|
125 return data
|
Chris@436
|
126
|
Chris@436
|
127 def set_to_config(pcfg, sect, key, data):
|
Chris@436
|
128 if not pcfg.has_section(sect):
|
Chris@436
|
129 pcfg.add_section(sect)
|
Chris@436
|
130 pcfg.set(sect, key, data)
|
Chris@436
|
131
|
Chris@449
|
132 def remote_key(uri, user, key):
|
Chris@440
|
133 # generate a "safe-for-config-file" key representing uri+user
|
Chris@449
|
134 s = '%s@@%s' % (uri, user)
|
Chris@449
|
135 h = hashlib.sha1()
|
Chris@449
|
136 h.update(key)
|
Chris@449
|
137 h.update(s)
|
Chris@449
|
138 return h.hexdigest()
|
Chris@438
|
139
|
Chris@440
|
140
|
Chris@440
|
141 def uisetup(ui):
|
Chris@440
|
142 if not easyhg_pyqt_ok:
|
Chris@440
|
143 raise util.Abort(_('Failed to load PyQt4 module required by easyhg.py'))
|
Chris@440
|
144 global easyhg_qtapp
|
Chris@440
|
145 easyhg_qtapp = QtGui.QApplication([])
|
Chris@440
|
146
|
Chris@440
|
147 def monkeypatch_method(cls):
|
Chris@440
|
148 def decorator(func):
|
Chris@440
|
149 setattr(cls, func.__name__, func)
|
Chris@440
|
150 return func
|
Chris@440
|
151 return decorator
|
Chris@440
|
152
|
Chris@440
|
153 orig_find = passwordmgr.find_user_password
|
Chris@440
|
154
|
Chris@427
|
155 @monkeypatch_method(passwordmgr)
|
Chris@427
|
156 def find_user_password(self, realm, authuri):
|
Chris@427
|
157
|
Chris@427
|
158 if not self.ui.interactive():
|
Chris@427
|
159 return orig_find(self, realm, authuri)
|
Chris@427
|
160 if not easyhg_pyqt_ok:
|
Chris@427
|
161 return orig_find(self, realm, authuri)
|
Chris@427
|
162
|
Chris@427
|
163 authinfo = urllib2.HTTPPasswordMgrWithDefaultRealm.find_user_password(
|
Chris@427
|
164 self, realm, authuri)
|
Chris@427
|
165 user, passwd = authinfo
|
Chris@427
|
166
|
Chris@427
|
167 if user and passwd:
|
Chris@427
|
168 return orig_find(self, realm, authuri)
|
Chris@427
|
169
|
Chris@440
|
170 # self.ui.write("want username and/or password for %s\n" % authuri)
|
Chris@427
|
171
|
Chris@440
|
172 short_uri = canonical_url(authuri)
|
Chris@427
|
173
|
Chris@440
|
174 authkey = self.ui.config('easyhg', 'authkey')
|
Chris@440
|
175 authfile = self.ui.config('easyhg', 'authfile')
|
Chris@440
|
176 use_authfile = (easyhg_authfile_imports_ok and authkey and authfile)
|
Chris@440
|
177 if authfile:
|
Chris@440
|
178 authfile = os.path.expanduser(authfile)
|
Chris@440
|
179 authdata = None
|
Chris@427
|
180
|
Chris@427
|
181 dialog = QtGui.QDialog()
|
Chris@427
|
182 layout = QtGui.QGridLayout()
|
Chris@427
|
183 dialog.setLayout(layout)
|
Chris@427
|
184
|
Chris@440
|
185 layout.addWidget(QtGui.QLabel(_('<h3>Login required</h3><p>Please provide your login details for the repository at<br><code>%s</code>:') % short_uri), 0, 0, 1, 2)
|
Chris@427
|
186
|
Chris@440
|
187 user_field = QtGui.QLineEdit()
|
Chris@427
|
188 if user:
|
Chris@440
|
189 user_field.setText(user)
|
Chris@427
|
190 layout.addWidget(QtGui.QLabel(_('User:')), 1, 0)
|
Chris@440
|
191 layout.addWidget(user_field, 1, 1)
|
Chris@427
|
192
|
Chris@440
|
193 passwd_field = QtGui.QLineEdit()
|
Chris@440
|
194 passwd_field.setEchoMode(QtGui.QLineEdit.Password)
|
Chris@427
|
195 if passwd:
|
Chris@440
|
196 passwd_field.setText(passwd)
|
Chris@427
|
197 layout.addWidget(QtGui.QLabel(_('Password:')), 2, 0)
|
Chris@440
|
198 layout.addWidget(passwd_field, 2, 1)
|
Chris@427
|
199
|
Chris@440
|
200 user_field.connect(user_field, Qt.SIGNAL("textChanged(QString)"),
|
Chris@440
|
201 passwd_field, Qt.SLOT("clear()"))
|
Chris@435
|
202
|
Chris@440
|
203 remember_field = None
|
Chris@440
|
204 remember = False
|
Chris@440
|
205 authconfig = None
|
Chris@433
|
206
|
Chris@436
|
207 if use_authfile:
|
Chris@440
|
208 authconfig = ConfigParser.RawConfigParser()
|
Chris@440
|
209 load_config(authconfig, authfile)
|
Chris@440
|
210 remember = get_boolean_from_config(authconfig, 'preferences',
|
Chris@440
|
211 'remember', False)
|
Chris@440
|
212 authdata = get_from_config(authconfig, 'auth',
|
Chris@449
|
213 remote_key(short_uri, user, authkey))
|
Chris@440
|
214 if authdata:
|
Chris@440
|
215 cachedpwd = decrypt_salted(authdata, authkey)
|
Chris@440
|
216 passwd_field.setText(cachedpwd)
|
Chris@440
|
217 remember_field = QtGui.QCheckBox()
|
Chris@440
|
218 remember_field.setChecked(remember)
|
Chris@444
|
219 remember_field.setText(_('Remember passwords while EasyMercurial is running'))
|
Chris@440
|
220 layout.addWidget(remember_field, 3, 1)
|
Chris@431
|
221
|
Chris@427
|
222 bb = QtGui.QDialogButtonBox()
|
Chris@427
|
223 ok = bb.addButton(bb.Ok)
|
Chris@427
|
224 cancel = bb.addButton(bb.Cancel)
|
Chris@427
|
225 cancel.setDefault(False)
|
Chris@427
|
226 cancel.setAutoDefault(False)
|
Chris@427
|
227 ok.setDefault(True)
|
Chris@427
|
228 bb.connect(ok, Qt.SIGNAL("clicked()"), dialog, Qt.SLOT("accept()"))
|
Chris@427
|
229 bb.connect(cancel, Qt.SIGNAL("clicked()"), dialog, Qt.SLOT("reject()"))
|
Chris@431
|
230 layout.addWidget(bb, 4, 0, 1, 2)
|
Chris@427
|
231
|
Chris@428
|
232 dialog.setWindowTitle(_('EasyMercurial: Login'))
|
Chris@427
|
233 dialog.show()
|
Chris@428
|
234
|
Chris@428
|
235 if not user:
|
Chris@440
|
236 user_field.setFocus(True)
|
Chris@428
|
237 elif not passwd:
|
Chris@440
|
238 passwd_field.setFocus(True)
|
Chris@428
|
239
|
Chris@427
|
240 dialog.raise_()
|
Chris@427
|
241 ok = dialog.exec_()
|
Chris@436
|
242 if not ok:
|
Chris@436
|
243 raise util.Abort(_('password entry cancelled'))
|
Chris@431
|
244
|
Chris@440
|
245 user = user_field.text()
|
Chris@440
|
246 passwd = passwd_field.text()
|
Chris@432
|
247
|
Chris@436
|
248 if use_authfile:
|
Chris@440
|
249 remember = remember_field.isChecked()
|
Chris@440
|
250 set_to_config(authconfig, 'preferences', 'remember', remember)
|
Chris@438
|
251 if user:
|
Chris@440
|
252 if passwd and remember:
|
Chris@440
|
253 authdata = encrypt_salted(passwd, authkey)
|
Chris@449
|
254 set_to_config(authconfig, 'auth', remote_key(short_uri, user, authkey), authdata)
|
Chris@438
|
255 else:
|
Chris@449
|
256 set_to_config(authconfig, 'auth', remote_key(short_uri, user, authkey), '')
|
Chris@444
|
257 save_config(self.ui, authconfig, authfile)
|
Chris@433
|
258
|
Chris@436
|
259 self.add_password(realm, authuri, user, passwd)
|
Chris@427
|
260 return (user, passwd)
|
Chris@427
|
261
|
Chris@427
|
262
|