/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
4988.10.3 by John Arbash Meinel
Merge bzr.dev 5007, resolve conflict, update NEWS
1
# Copyright (C) 2005-2010 Canonical Ltd
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
2
#   Authors: Robert Collins <robert.collins@canonical.com>
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
3
#            and others
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
4
#
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
#
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
# GNU General Public License for more details.
14
#
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
4183.7.1 by Sabin Iacob
update FSF mailing address
17
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
18
1442.1.20 by Robert Collins
add some documentation on options
19
"""Configuration that affects the behaviour of Bazaar.
20
21
Currently this configuration resides in ~/.bazaar/bazaar.conf
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
22
and ~/.bazaar/locations.conf, which is written to by bzr.
1442.1.20 by Robert Collins
add some documentation on options
23
1461 by Robert Collins
Typo in config.py (Thanks Fabbione)
24
In bazaar.conf the following options may be set:
1442.1.20 by Robert Collins
add some documentation on options
25
[DEFAULT]
26
editor=name-of-program
27
email=Your Name <your@email.address>
28
check_signatures=require|ignore|check-available(default)
29
create_signatures=always|never|when-required(default)
1442.1.56 by Robert Collins
gpg_signing_command configuration item
30
gpg_signing_command=name-of-program
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
31
log_format=name-of-format
1442.1.20 by Robert Collins
add some documentation on options
32
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
33
in locations.conf, you specify the url of a branch and options for it.
1442.1.20 by Robert Collins
add some documentation on options
34
Wildcards may be used - * and ? as normal in shell completion. Options
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
35
set in both bazaar.conf and locations.conf are overridden by the locations.conf
1442.1.20 by Robert Collins
add some documentation on options
36
setting.
37
[/home/robertc/source]
38
recurse=False|True(default)
39
email= as above
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
40
check_signatures= as above
1442.1.20 by Robert Collins
add some documentation on options
41
create_signatures= as above.
42
43
explanation of options
44
----------------------
45
editor - this option sets the pop up editor to use during commits.
46
email - this option sets the user id bzr will use when committing.
47
check_signatures - this option controls whether bzr will require good gpg
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
48
                   signatures, ignore them, or check them if they are
1442.1.20 by Robert Collins
add some documentation on options
49
                   present.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
50
create_signatures - this option controls whether bzr will always create
1442.1.20 by Robert Collins
add some documentation on options
51
                    gpg signatures, never create them, or create them if the
52
                    branch is configured to require them.
1887.2.1 by Adeodato Simó
Fix some typos and grammar issues.
53
log_format - this option sets the default log format.  Possible values are
54
             long, short, line, or a plugin can register new formats.
1553.6.2 by Erik Bågfors
documentation and NEWS
55
56
In bazaar.conf you can also define aliases in the ALIASES sections, example
57
58
[ALIASES]
59
lastlog=log --line -r-10..-1
60
ll=log --line -r-10..-1
61
h=help
62
up=pull
1442.1.20 by Robert Collins
add some documentation on options
63
"""
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
64
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
65
import os
66
import sys
1474 by Robert Collins
Merge from Aaron Bentley.
67
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
68
from bzrlib.lazy_import import lazy_import
69
lazy_import(globals(), """
1474 by Robert Collins
Merge from Aaron Bentley.
70
import errno
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
71
from fnmatch import fnmatch
72
import re
2900.2.22 by Vincent Ladeuil
Polishing.
73
from cStringIO import StringIO
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
74
75
import bzrlib
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
76
from bzrlib import (
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
77
    debug,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
78
    errors,
2681.1.8 by Aaron Bentley
Add Thunderbird support to bzr send
79
    mail_client,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
80
    osutils,
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
81
    registry,
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
82
    symbol_versioning,
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
83
    trace,
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
84
    ui,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
85
    urlutils,
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
86
    win32utils,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
87
    )
2991.2.4 by Vincent Ladeuil
Various fixes following local testing environment rebuild.
88
from bzrlib.util.configobj import configobj
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
89
""")
90
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
91
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
92
CHECK_IF_POSSIBLE=0
93
CHECK_ALWAYS=1
94
CHECK_NEVER=2
95
96
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
97
SIGN_WHEN_REQUIRED=0
98
SIGN_ALWAYS=1
99
SIGN_NEVER=2
100
101
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
102
POLICY_NONE = 0
103
POLICY_NORECURSE = 1
104
POLICY_APPENDPATH = 2
105
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
106
_policy_name = {
107
    POLICY_NONE: None,
108
    POLICY_NORECURSE: 'norecurse',
109
    POLICY_APPENDPATH: 'appendpath',
110
    }
111
_policy_value = {
112
    None: POLICY_NONE,
113
    'none': POLICY_NONE,
114
    'norecurse': POLICY_NORECURSE,
115
    'appendpath': POLICY_APPENDPATH,
116
    }
2120.6.4 by James Henstridge
add support for specifying policy when storing options
117
118
119
STORE_LOCATION = POLICY_NONE
120
STORE_LOCATION_NORECURSE = POLICY_NORECURSE
121
STORE_LOCATION_APPENDPATH = POLICY_APPENDPATH
122
STORE_BRANCH = 3
123
STORE_GLOBAL = 4
124
3224.5.10 by Andrew Bennetts
Replace some duplication with a different form of hackery.
125
_ConfigObj = None
126
def ConfigObj(*args, **kwargs):
127
    global _ConfigObj
128
    if _ConfigObj is None:
129
        class ConfigObj(configobj.ConfigObj):
130
131
            def get_bool(self, section, key):
132
                return self[section].as_bool(key)
133
134
            def get_value(self, section, name):
135
                # Try [] for the old DEFAULT section.
136
                if section == "DEFAULT":
137
                    try:
138
                        return self[name]
139
                    except KeyError:
140
                        pass
141
                return self[section][name]
142
        _ConfigObj = ConfigObj
143
    return _ConfigObj(*args, **kwargs)
1474 by Robert Collins
Merge from Aaron Bentley.
144
145
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
146
class Config(object):
147
    """A configuration policy - what username, editor, gpg needs etc."""
148
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
149
    def __init__(self):
150
        super(Config, self).__init__()
151
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
152
    def get_editor(self):
153
        """Get the users pop up editor."""
154
        raise NotImplementedError
155
4603.1.10 by Aaron Bentley
Provide change editor via config.
156
    def get_change_editor(self, old_tree, new_tree):
157
        from bzrlib import diff
158
        cmd = self._get_change_editor()
159
        if cmd is None:
160
            return None
161
        return diff.DiffFromTool.from_string(cmd, old_tree, new_tree,
162
                                             sys.stdout)
163
164
2681.1.8 by Aaron Bentley
Add Thunderbird support to bzr send
165
    def get_mail_client(self):
166
        """Get a mail client to use"""
167
        selected_client = self.get_user_option('mail_client')
3638.2.4 by Neil Martinsen-Burrell
Address JAMs review. Dont use register_lazy and dont lazy_import anything other than modules
168
        _registry = mail_client.mail_client_registry
2681.1.10 by Aaron Bentley
Clean up handling of unknown mail clients
169
        try:
3638.2.4 by Neil Martinsen-Burrell
Address JAMs review. Dont use register_lazy and dont lazy_import anything other than modules
170
            mail_client_class = _registry.get(selected_client)
2681.1.10 by Aaron Bentley
Clean up handling of unknown mail clients
171
        except KeyError:
172
            raise errors.UnknownMailClient(selected_client)
173
        return mail_client_class(self)
2681.1.8 by Aaron Bentley
Add Thunderbird support to bzr send
174
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
175
    def _get_signature_checking(self):
176
        """Template method to override signature checking policy."""
177
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
178
    def _get_signing_policy(self):
179
        """Template method to override signature creation policy."""
180
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
181
    def _get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
182
        """Template method to provide a user option."""
183
        return None
184
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
185
    def get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
186
        """Get a generic option - no special process, no default."""
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
187
        return self._get_user_option(option_name)
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
188
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
189
    def get_user_option_as_bool(self, option_name):
190
        """Get a generic option as a boolean - no special process, no default.
191
192
        :return None if the option doesn't exist or its value can't be
4840.2.4 by Vincent Ladeuil
Implement config.get_user_option_as_list.
193
            interpreted as a boolean. Returns True or False otherwise.
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
194
        """
195
        s = self._get_user_option(option_name)
4989.2.12 by Vincent Ladeuil
Display a warning if an option value is not boolean.
196
        if s is None:
197
            # The option doesn't exist
198
            return None
4989.2.15 by Vincent Ladeuil
Fixed as per Andrew's review.
199
        val = ui.bool_from_string(s)
4989.2.12 by Vincent Ladeuil
Display a warning if an option value is not boolean.
200
        if val is None:
201
            # The value can't be interpreted as a boolean
202
            trace.warning('Value "%s" is not a boolean for "%s"',
203
                          s, option_name)
204
        return val
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
205
4840.2.4 by Vincent Ladeuil
Implement config.get_user_option_as_list.
206
    def get_user_option_as_list(self, option_name):
207
        """Get a generic option as a list - no special process, no default.
208
209
        :return None if the option doesn't exist. Returns the value as a list
210
            otherwise.
211
        """
212
        l = self._get_user_option(option_name)
213
        if isinstance(l, (str, unicode)):
214
            # A single value, most probably the user forgot the final ','
215
            l = [l]
216
        return l
217
1442.1.56 by Robert Collins
gpg_signing_command configuration item
218
    def gpg_signing_command(self):
219
        """What program should be used to sign signatures?"""
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
220
        result = self._gpg_signing_command()
221
        if result is None:
222
            result = "gpg"
223
        return result
224
225
    def _gpg_signing_command(self):
226
        """See gpg_signing_command()."""
227
        return None
1442.1.56 by Robert Collins
gpg_signing_command configuration item
228
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
229
    def log_format(self):
230
        """What log format should be used"""
231
        result = self._log_format()
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
232
        if result is None:
233
            result = "long"
234
        return result
235
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
236
    def _log_format(self):
237
        """See log_format()."""
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
238
        return None
239
1472 by Robert Collins
post commit hook, first pass implementation
240
    def post_commit(self):
241
        """An ordered list of python functions to call.
242
243
        Each function takes branch, rev_id as parameters.
244
        """
245
        return self._post_commit()
246
247
    def _post_commit(self):
248
        """See Config.post_commit."""
249
        return None
250
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
251
    def user_email(self):
252
        """Return just the email component of a username."""
1185.33.31 by Martin Pool
Make annotate cope better with revisions committed without a valid
253
        return extract_email_address(self.username())
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
254
255
    def username(self):
256
        """Return email-style username.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
257
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
258
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
259
5187.2.1 by Parth Malwankar
removed comment about deprecated BZREMAIL.
260
        $BZR_EMAIL can be set to override this, then
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
261
        the concrete policy type is checked, and finally
1185.37.2 by Jamie Wilkinson
Fix a typo and grammar in Config.username() docstring.
262
        $EMAIL is examined.
263
        If none is found, a reasonable default is (hopefully)
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
264
        created.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
265
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
266
        TODO: Check it's reasonably well-formed.
267
        """
1861.4.1 by Matthieu Moy
BZREMAIL renamed to BZR_EMAIL.
268
        v = os.environ.get('BZR_EMAIL')
269
        if v:
3224.5.4 by Andrew Bennetts
Fix test suite, mainly weeding out uses of bzrlib.user_encoding.
270
            return v.decode(osutils.get_user_encoding())
2900.3.1 by Tim Penhey
Removed some annoying trailing whitespace.
271
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
272
        v = self._get_user_id()
273
        if v:
274
            return v
2900.3.1 by Tim Penhey
Removed some annoying trailing whitespace.
275
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
276
        v = os.environ.get('EMAIL')
277
        if v:
3224.5.4 by Andrew Bennetts
Fix test suite, mainly weeding out uses of bzrlib.user_encoding.
278
            return v.decode(osutils.get_user_encoding())
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
279
5187.2.3 by Parth Malwankar
init and init-repo now fail before creating dir if username is not set.
280
        raise errors.BzrCommandError('Unable to determine your name.\n'
281
            "Please, set your name with the 'whoami' command.\n"
282
            'E.g. bzr whoami "Your Name <name@example.com>"')
283
284
    def ensure_username(self):
285
        """Raise BzrCommandError if username is not set.
286
287
        This method relies on the username() function raising the error.
288
        """
289
        self.username()
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
290
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
291
    def signature_checking(self):
292
        """What is the current policy for signature checking?."""
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
293
        policy = self._get_signature_checking()
294
        if policy is not None:
295
            return policy
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
296
        return CHECK_IF_POSSIBLE
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
297
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
298
    def signing_policy(self):
299
        """What is the current policy for signature checking?."""
300
        policy = self._get_signing_policy()
301
        if policy is not None:
302
            return policy
303
        return SIGN_WHEN_REQUIRED
304
1442.1.21 by Robert Collins
create signature_needed() call for commit to trigger creating signatures
305
    def signature_needed(self):
306
        """Is a signature needed when committing ?."""
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
307
        policy = self._get_signing_policy()
308
        if policy is None:
309
            policy = self._get_signature_checking()
310
            if policy is not None:
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
311
                trace.warning("Please use create_signatures,"
312
                              " not check_signatures to set signing policy.")
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
313
            if policy == CHECK_ALWAYS:
314
                return True
315
        elif policy == SIGN_ALWAYS:
1442.1.21 by Robert Collins
create signature_needed() call for commit to trigger creating signatures
316
            return True
317
        return False
318
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
319
    def get_alias(self, value):
320
        return self._get_alias(value)
321
322
    def _get_alias(self, value):
323
        pass
324
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
325
    def get_nickname(self):
326
        return self._get_nickname()
327
328
    def _get_nickname(self):
329
        return None
330
1551.18.17 by Aaron Bentley
Introduce bzr_remote_path configuration variable
331
    def get_bzr_remote_path(self):
332
        try:
333
            return os.environ['BZR_REMOTE_PATH']
334
        except KeyError:
335
            path = self.get_user_option("bzr_remote_path")
336
            if path is None:
337
                path = 'bzr'
338
            return path
339
4840.2.6 by Vincent Ladeuil
Implement config.suppress_warning.
340
    def suppress_warning(self, warning):
341
        """Should the warning be suppressed or emitted.
342
343
        :param warning: The name of the warning being tested.
344
345
        :returns: True if the warning should be suppressed, False otherwise.
346
        """
347
        warnings = self.get_user_option_as_list('suppress_warnings')
348
        if warnings is None or warning not in warnings:
349
            return False
350
        else:
351
            return True
352
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
353
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
354
class IniBasedConfig(Config):
355
    """A configuration policy that draws from ini files."""
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
356
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
357
    def __init__(self, get_filename):
358
        super(IniBasedConfig, self).__init__()
359
        self._get_filename = get_filename
360
        self._parser = None
361
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
362
    def _get_parser(self, file=None):
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
363
        if self._parser is not None:
364
            return self._parser
1185.12.49 by Aaron Bentley
Switched to ConfigObj
365
        if file is None:
366
            input = self._get_filename()
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
367
        else:
1185.12.49 by Aaron Bentley
Switched to ConfigObj
368
            input = file
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
369
        try:
4603.1.21 by Aaron Bentley
Restore ConfigObj interpolation.
370
            self._parser = ConfigObj(input, encoding='utf-8')
1474 by Robert Collins
Merge from Aaron Bentley.
371
        except configobj.ConfigObjError, e:
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
372
            raise errors.ParseConfigError(e.errors, e.config.filename)
1185.12.49 by Aaron Bentley
Switched to ConfigObj
373
        return self._parser
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
374
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
375
    def _get_matching_sections(self):
376
        """Return an ordered list of (section_name, extra_path) pairs.
377
378
        If the section contains inherited configuration, extra_path is
379
        a string containing the additional path components.
380
        """
381
        section = self._get_section()
382
        if section is not None:
383
            return [(section, '')]
384
        else:
385
            return []
386
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
387
    def _get_section(self):
388
        """Override this to define the section used by the config."""
389
        return "DEFAULT"
390
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
391
    def _get_option_policy(self, section, option_name):
392
        """Return the policy for the given (section, option_name) pair."""
393
        return POLICY_NONE
394
4603.1.10 by Aaron Bentley
Provide change editor via config.
395
    def _get_change_editor(self):
396
        return self.get_user_option('change_editor')
397
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
398
    def _get_signature_checking(self):
399
        """See Config._get_signature_checking."""
1474 by Robert Collins
Merge from Aaron Bentley.
400
        policy = self._get_user_option('check_signatures')
401
        if policy:
402
            return self._string_to_signature_policy(policy)
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
403
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
404
    def _get_signing_policy(self):
1773.4.3 by Martin Pool
[merge] bzr.dev
405
        """See Config._get_signing_policy"""
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
406
        policy = self._get_user_option('create_signatures')
407
        if policy:
408
            return self._string_to_signing_policy(policy)
409
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
410
    def _get_user_id(self):
411
        """Get the user id from the 'email' key in the current section."""
1474 by Robert Collins
Merge from Aaron Bentley.
412
        return self._get_user_option('email')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
413
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
414
    def _get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
415
        """See Config._get_user_option."""
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
416
        for (section, extra_path) in self._get_matching_sections():
417
            try:
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
418
                value = self._get_parser().get_value(section, option_name)
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
419
            except KeyError:
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
420
                continue
421
            policy = self._get_option_policy(section, option_name)
422
            if policy == POLICY_NONE:
423
                return value
424
            elif policy == POLICY_NORECURSE:
425
                # norecurse items only apply to the exact path
426
                if extra_path:
427
                    continue
428
                else:
429
                    return value
430
            elif policy == POLICY_APPENDPATH:
2120.6.3 by James Henstridge
add some more tests for getting policy options, and behaviour of get_user_option in the presence of config policies
431
                if extra_path:
432
                    value = urlutils.join(value, extra_path)
433
                return value
2120.6.6 by James Henstridge
fix test_set_push_location test
434
            else:
435
                raise AssertionError('Unexpected config policy %r' % policy)
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
436
        else:
1993.3.1 by James Henstridge
first go at making location config lookup recursive
437
            return None
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
438
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
439
    def _gpg_signing_command(self):
1442.1.56 by Robert Collins
gpg_signing_command configuration item
440
        """See Config.gpg_signing_command."""
1472 by Robert Collins
post commit hook, first pass implementation
441
        return self._get_user_option('gpg_signing_command')
1442.1.56 by Robert Collins
gpg_signing_command configuration item
442
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
443
    def _log_format(self):
444
        """See Config.log_format."""
445
        return self._get_user_option('log_format')
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
446
1472 by Robert Collins
post commit hook, first pass implementation
447
    def _post_commit(self):
448
        """See Config.post_commit."""
449
        return self._get_user_option('post_commit')
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
450
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
451
    def _string_to_signature_policy(self, signature_string):
452
        """Convert a string to a signing policy."""
1442.1.17 by Robert Collins
allow global overriding of signature policy to force checking, or (pointless but allowed) to set auto checking
453
        if signature_string.lower() == 'check-available':
454
            return CHECK_IF_POSSIBLE
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
455
        if signature_string.lower() == 'ignore':
456
            return CHECK_NEVER
1442.1.17 by Robert Collins
allow global overriding of signature policy to force checking, or (pointless but allowed) to set auto checking
457
        if signature_string.lower() == 'require':
458
            return CHECK_ALWAYS
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
459
        raise errors.BzrError("Invalid signatures policy '%s'"
460
                              % signature_string)
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
461
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
462
    def _string_to_signing_policy(self, signature_string):
463
        """Convert a string to a signing policy."""
464
        if signature_string.lower() == 'when-required':
465
            return SIGN_WHEN_REQUIRED
466
        if signature_string.lower() == 'never':
467
            return SIGN_NEVER
468
        if signature_string.lower() == 'always':
469
            return SIGN_ALWAYS
470
        raise errors.BzrError("Invalid signing policy '%s'"
471
                              % signature_string)
472
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
473
    def _get_alias(self, value):
474
        try:
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
475
            return self._get_parser().get_value("ALIASES",
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
476
                                                value)
477
        except KeyError:
478
            pass
479
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
480
    def _get_nickname(self):
481
        return self.get_user_option('nickname')
482
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
483
484
class GlobalConfig(IniBasedConfig):
485
    """The configuration that should be used for a specific location."""
486
487
    def get_editor(self):
1474 by Robert Collins
Merge from Aaron Bentley.
488
        return self._get_user_option('editor')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
489
490
    def __init__(self):
491
        super(GlobalConfig, self).__init__(config_filename)
492
1816.2.1 by Robey Pointer
add set_user_option to GlobalConfig, and make /etc/passwd username lookup try harder with encodings
493
    def set_user_option(self, option, value):
494
        """Save option and its value in the configuration."""
2900.3.2 by Tim Penhey
A working alias command.
495
        self._set_option(option, value, 'DEFAULT')
496
497
    def get_aliases(self):
498
        """Return the aliases section."""
499
        if 'ALIASES' in self._get_parser():
500
            return self._get_parser()['ALIASES']
501
        else:
502
            return {}
503
504
    def set_alias(self, alias_name, alias_command):
505
        """Save the alias in the configuration."""
506
        self._set_option(alias_name, alias_command, 'ALIASES')
507
508
    def unset_alias(self, alias_name):
509
        """Unset an existing alias."""
510
        aliases = self._get_parser().get('ALIASES')
2900.3.7 by Tim Penhey
Updates from Aaron's review.
511
        if not aliases or alias_name not in aliases:
512
            raise errors.NoSuchAlias(alias_name)
2900.3.2 by Tim Penhey
A working alias command.
513
        del aliases[alias_name]
2900.3.12 by Tim Penhey
Final review comments.
514
        self._write_config_file()
2900.3.2 by Tim Penhey
A working alias command.
515
516
    def _set_option(self, option, value, section):
1816.2.1 by Robey Pointer
add set_user_option to GlobalConfig, and make /etc/passwd username lookup try harder with encodings
517
        # FIXME: RBC 20051029 This should refresh the parser and also take a
518
        # file lock on bazaar.conf.
519
        conf_dir = os.path.dirname(self._get_filename())
520
        ensure_config_dir_exists(conf_dir)
2900.3.7 by Tim Penhey
Updates from Aaron's review.
521
        self._get_parser().setdefault(section, {})[option] = value
2900.3.12 by Tim Penhey
Final review comments.
522
        self._write_config_file()
2900.3.2 by Tim Penhey
A working alias command.
523
2900.3.12 by Tim Penhey
Final review comments.
524
    def _write_config_file(self):
4634.143.1 by Parth Malwankar
default .bazaar, .bzr.log and .bazaar/bazaar.conf retain
525
        path = self._get_filename()
5116.2.3 by Parth Malwankar
updated log file creation to avoid race based on implementation by Martin [gz]
526
        f = open(path, 'wb')
5116.2.6 by Parth Malwankar
renamed copy_ownership to copy_ownership_from_path.
527
        osutils.copy_ownership_from_path(path)
1816.2.1 by Robey Pointer
add set_user_option to GlobalConfig, and make /etc/passwd username lookup try harder with encodings
528
        self._get_parser().write(f)
529
        f.close()
530
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
531
532
class LocationConfig(IniBasedConfig):
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
533
    """A configuration object that gives the policy for a location."""
534
535
    def __init__(self, location):
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
536
        name_generator = locations_config_filename
2900.3.1 by Tim Penhey
Removed some annoying trailing whitespace.
537
        if (not os.path.exists(name_generator()) and
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
538
                os.path.exists(branches_config_filename())):
1830.2.1 by John Arbash Meinel
Make it clearer what config file needs to be renamed.
539
            if sys.platform == 'win32':
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
540
                trace.warning('Please rename %s to %s'
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
541
                              % (branches_config_filename(),
542
                                 locations_config_filename()))
1830.2.1 by John Arbash Meinel
Make it clearer what config file needs to be renamed.
543
            else:
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
544
                trace.warning('Please rename ~/.bazaar/branches.conf'
545
                              ' to ~/.bazaar/locations.conf')
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
546
            name_generator = branches_config_filename
547
        super(LocationConfig, self).__init__(name_generator)
1878.1.1 by John Arbash Meinel
Entries in locations.conf should prefer local paths if available (bug #53653)
548
        # local file locations are looked up by local path, rather than
549
        # by file url. This is because the config file is a user
550
        # file, and we would rather not expose the user to file urls.
551
        if location.startswith('file://'):
552
            location = urlutils.local_path_from_url(location)
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
553
        self.location = location
554
1993.3.1 by James Henstridge
first go at making location config lookup recursive
555
    def _get_matching_sections(self):
556
        """Return an ordered list of section names matching this location."""
1185.12.49 by Aaron Bentley
Switched to ConfigObj
557
        sections = self._get_parser()
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
558
        location_names = self.location.split('/')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
559
        if self.location.endswith('/'):
560
            del location_names[-1]
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
561
        matches=[]
1442.1.10 by Robert Collins
explicit over glob test passes
562
        for section in sections:
1878.1.1 by John Arbash Meinel
Entries in locations.conf should prefer local paths if available (bug #53653)
563
            # location is a local path if possible, so we need
564
            # to convert 'file://' urls to local paths if necessary.
565
            # This also avoids having file:///path be a more exact
566
            # match than '/path'.
567
            if section.startswith('file://'):
568
                section_path = urlutils.local_path_from_url(section)
569
            else:
570
                section_path = section
571
            section_names = section_path.split('/')
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
572
            if section.endswith('/'):
573
                del section_names[-1]
574
            names = zip(location_names, section_names)
575
            matched = True
576
            for name in names:
577
                if not fnmatch(name[0], name[1]):
578
                    matched = False
579
                    break
580
            if not matched:
581
                continue
582
            # so, for the common prefix they matched.
583
            # if section is longer, no match.
584
            if len(section_names) > len(location_names):
585
                continue
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
586
            matches.append((len(section_names), section,
587
                            '/'.join(location_names[len(section_names):])))
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
588
        matches.sort(reverse=True)
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
589
        sections = []
590
        for (length, section, extra_path) in matches:
591
            sections.append((section, extra_path))
592
            # should we stop looking for parent configs here?
1993.3.1 by James Henstridge
first go at making location config lookup recursive
593
            try:
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
594
                if self._get_parser()[section].as_bool('ignore_parents'):
595
                    break
1993.3.1 by James Henstridge
first go at making location config lookup recursive
596
            except KeyError:
597
                pass
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
598
        return sections
1442.1.9 by Robert Collins
exact section test passes
599
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
600
    def _get_option_policy(self, section, option_name):
601
        """Return the policy for the given (section, option_name) pair."""
602
        # check for the old 'recurse=False' flag
603
        try:
604
            recurse = self._get_parser()[section].as_bool('recurse')
605
        except KeyError:
606
            recurse = True
607
        if not recurse:
608
            return POLICY_NORECURSE
609
2120.6.10 by James Henstridge
Catch another deprecation warning, and more cleanup
610
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
611
        try:
612
            policy_name = self._get_parser()[section][policy_key]
613
        except KeyError:
614
            policy_name = None
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
615
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
616
        return _policy_value[policy_name]
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
617
2120.6.4 by James Henstridge
add support for specifying policy when storing options
618
    def _set_option_policy(self, section, option_name, option_policy):
619
        """Set the policy for the given option name in the given section."""
620
        # The old recurse=False option affects all options in the
621
        # section.  To handle multiple policies in the section, we
622
        # need to convert it to a policy_norecurse key.
623
        try:
624
            recurse = self._get_parser()[section].as_bool('recurse')
625
        except KeyError:
626
            pass
627
        else:
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
628
            symbol_versioning.warn(
2120.6.11 by James Henstridge
s/0.13/0.14/ in deprecation warning
629
                'The recurse option is deprecated as of 0.14.  '
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
630
                'The section "%s" has been converted to use policies.'
631
                % section,
632
                DeprecationWarning)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
633
            del self._get_parser()[section]['recurse']
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
634
            if not recurse:
635
                for key in self._get_parser()[section].keys():
636
                    if not key.endswith(':policy'):
637
                        self._get_parser()[section][key +
638
                                                    ':policy'] = 'norecurse'
2120.6.4 by James Henstridge
add support for specifying policy when storing options
639
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
640
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
641
        policy_name = _policy_name[option_policy]
642
        if policy_name is not None:
643
            self._get_parser()[section][policy_key] = policy_name
644
        else:
645
            if policy_key in self._get_parser()[section]:
646
                del self._get_parser()[section][policy_key]
2120.6.4 by James Henstridge
add support for specifying policy when storing options
647
648
    def set_user_option(self, option, value, store=STORE_LOCATION):
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
649
        """Save option and its value in the configuration."""
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
650
        if store not in [STORE_LOCATION,
2120.6.4 by James Henstridge
add support for specifying policy when storing options
651
                         STORE_LOCATION_NORECURSE,
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
652
                         STORE_LOCATION_APPENDPATH]:
653
            raise ValueError('bad storage policy %r for %r' %
654
                (store, option))
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
655
        # FIXME: RBC 20051029 This should refresh the parser and also take a
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
656
        # file lock on locations.conf.
1185.31.39 by John Arbash Meinel
Replacing os.getcwdu() with osutils.getcwd(),
657
        conf_dir = os.path.dirname(self._get_filename())
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
658
        ensure_config_dir_exists(conf_dir)
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
659
        location = self.location
660
        if location.endswith('/'):
661
            location = location[:-1]
662
        if (not location in self._get_parser() and
663
            not location + '/' in self._get_parser()):
664
            self._get_parser()[location]={}
665
        elif location + '/' in self._get_parser():
666
            location = location + '/'
667
        self._get_parser()[location][option]=value
2120.6.4 by James Henstridge
add support for specifying policy when storing options
668
        # the allowed values of store match the config policies
669
        self._set_option_policy(location, option, store)
1551.2.49 by abentley
Made ConfigObj output binary-identical files on win32 and *nix
670
        self._get_parser().write(file(self._get_filename(), 'wb'))
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
671
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
672
673
class BranchConfig(Config):
674
    """A configuration object giving the policy for a branch."""
675
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
676
    def _get_branch_data_config(self):
677
        if self._branch_data_config is None:
678
            self._branch_data_config = TreeConfig(self.branch)
679
        return self._branch_data_config
680
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
681
    def _get_location_config(self):
682
        if self._location_config is None:
683
            self._location_config = LocationConfig(self.branch.base)
684
        return self._location_config
685
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
686
    def _get_global_config(self):
687
        if self._global_config is None:
688
            self._global_config = GlobalConfig()
689
        return self._global_config
690
691
    def _get_best_value(self, option_name):
692
        """This returns a user option from local, tree or global config.
693
694
        They are tried in that order.  Use get_safe_value if trusted values
695
        are necessary.
696
        """
697
        for source in self.option_sources:
698
            value = getattr(source(), option_name)()
699
            if value is not None:
700
                return value
701
        return None
702
703
    def _get_safe_value(self, option_name):
704
        """This variant of get_best_value never returns untrusted values.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
705
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
706
        It does not return values from the branch data, because the branch may
707
        not be controlled by the user.
708
709
        We may wish to allow locations.conf to control whether branches are
710
        trusted in the future.
711
        """
712
        for source in (self._get_location_config, self._get_global_config):
713
            value = getattr(source(), option_name)()
714
            if value is not None:
715
                return value
716
        return None
717
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
718
    def _get_user_id(self):
719
        """Return the full user id for the branch.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
720
3407.2.14 by Martin Pool
Remove more cases of getting transport via control_files
721
        e.g. "John Hacker <jhacker@example.com>"
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
722
        This is looked up in the email controlfile for the branch.
723
        """
724
        try:
3407.2.16 by Martin Pool
Remove RemoteBranch reliance on control_files._transport
725
            return (self.branch._transport.get_bytes("email")
3224.5.4 by Andrew Bennetts
Fix test suite, mainly weeding out uses of bzrlib.user_encoding.
726
                    .decode(osutils.get_user_encoding())
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
727
                    .rstrip("\r\n"))
728
        except errors.NoSuchFile, e:
729
            pass
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
730
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
731
        return self._get_best_value('_get_user_id')
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
732
4603.1.10 by Aaron Bentley
Provide change editor via config.
733
    def _get_change_editor(self):
734
        return self._get_best_value('_get_change_editor')
735
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
736
    def _get_signature_checking(self):
737
        """See Config._get_signature_checking."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
738
        return self._get_best_value('_get_signature_checking')
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
739
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
740
    def _get_signing_policy(self):
741
        """See Config._get_signing_policy."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
742
        return self._get_best_value('_get_signing_policy')
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
743
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
744
    def _get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
745
        """See Config._get_user_option."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
746
        for source in self.option_sources:
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
747
            value = source()._get_user_option(option_name)
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
748
            if value is not None:
749
                return value
750
        return None
751
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
752
    def set_user_option(self, name, value, store=STORE_BRANCH,
753
        warn_masked=False):
2120.6.4 by James Henstridge
add support for specifying policy when storing options
754
        if store == STORE_BRANCH:
1770.2.6 by Aaron Bentley
Ensure branch.conf works properly
755
            self._get_branch_data_config().set_option(value, name)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
756
        elif store == STORE_GLOBAL:
2120.6.7 by James Henstridge
Fix GlobalConfig.set_user_option() call
757
            self._get_global_config().set_user_option(name, value)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
758
        else:
759
            self._get_location_config().set_user_option(name, value, store)
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
760
        if not warn_masked:
761
            return
762
        if store in (STORE_GLOBAL, STORE_BRANCH):
763
            mask_value = self._get_location_config().get_user_option(name)
764
            if mask_value is not None:
765
                trace.warning('Value "%s" is masked by "%s" from'
766
                              ' locations.conf', value, mask_value)
767
            else:
768
                if store == STORE_GLOBAL:
769
                    branch_config = self._get_branch_data_config()
770
                    mask_value = branch_config.get_user_option(name)
771
                    if mask_value is not None:
772
                        trace.warning('Value "%s" is masked by "%s" from'
1551.15.37 by Aaron Bentley
Don't treat a format string as a normal string
773
                                      ' branch.conf', value, mask_value)
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
774
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
775
    def _gpg_signing_command(self):
1442.1.56 by Robert Collins
gpg_signing_command configuration item
776
        """See Config.gpg_signing_command."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
777
        return self._get_safe_value('_gpg_signing_command')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
778
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
779
    def __init__(self, branch):
780
        super(BranchConfig, self).__init__()
781
        self._location_config = None
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
782
        self._branch_data_config = None
783
        self._global_config = None
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
784
        self.branch = branch
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
785
        self.option_sources = (self._get_location_config,
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
786
                               self._get_branch_data_config,
787
                               self._get_global_config)
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
788
1472 by Robert Collins
post commit hook, first pass implementation
789
    def _post_commit(self):
790
        """See Config.post_commit."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
791
        return self._get_safe_value('_post_commit')
1472 by Robert Collins
post commit hook, first pass implementation
792
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
793
    def _get_nickname(self):
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
794
        value = self._get_explicit_nickname()
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
795
        if value is not None:
796
            return value
2120.5.2 by Alexander Belchenko
(jam) Fix for bug #66857
797
        return urlutils.unescape(self.branch.base.split('/')[-2])
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
798
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
799
    def has_explicit_nickname(self):
800
        """Return true if a nickname has been explicitly assigned."""
801
        return self._get_explicit_nickname() is not None
802
803
    def _get_explicit_nickname(self):
804
        return self._get_best_value('_get_nickname')
805
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
806
    def _log_format(self):
807
        """See Config.log_format."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
808
        return self._get_best_value('_log_format')
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
809
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
810
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
811
def ensure_config_dir_exists(path=None):
812
    """Make sure a configuration directory exists.
813
    This makes sure that the directory exists.
814
    On windows, since configuration directories are 2 levels deep,
815
    it makes sure both the directory and the parent directory exists.
816
    """
817
    if path is None:
818
        path = config_dir()
819
    if not os.path.isdir(path):
820
        if sys.platform == 'win32':
821
            parent_dir = os.path.dirname(path)
822
            if not os.path.isdir(parent_dir):
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
823
                trace.mutter('creating config parent directory: %r', parent_dir)
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
824
            os.mkdir(parent_dir)
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
825
        trace.mutter('creating config directory: %r', path)
5116.2.4 by Parth Malwankar
removed mkdir_with_ownership as its probably cleaner to just use copy_ownership
826
        os.mkdir(path)
5116.2.6 by Parth Malwankar
renamed copy_ownership to copy_ownership_from_path.
827
        osutils.copy_ownership_from_path(path)
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
828
1532 by Robert Collins
Merge in John Meinels integration branch.
829
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
830
def config_dir():
831
    """Return per-user configuration directory.
832
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
833
    By default this is ~/.bazaar/
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
834
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
835
    TODO: Global option --config-dir to override this.
836
    """
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
837
    base = os.environ.get('BZR_HOME', None)
838
    if sys.platform == 'win32':
839
        if base is None:
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
840
            base = win32utils.get_appdata_location_unicode()
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
841
        if base is None:
842
            base = os.environ.get('HOME', None)
843
        if base is None:
2991.2.2 by Vincent Ladeuil
No tests worth adding after upgrading to configobj-4.4.0.
844
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
845
                                  ' or HOME set')
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
846
        return osutils.pathjoin(base, 'bazaar', '2.0')
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
847
    else:
848
        # cygwin, linux, and darwin all have a $HOME directory
849
        if base is None:
850
            base = os.path.expanduser("~")
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
851
        return osutils.pathjoin(base, ".bazaar")
1185.31.32 by John Arbash Meinel
Updated the bzr sourcecode to use bzrlib.osutils.pathjoin rather than os.path.join to enforce internal use of / instead of \
852
853
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
854
def config_filename():
855
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
856
    return osutils.pathjoin(config_dir(), 'bazaar.conf')
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
857
858
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
859
def branches_config_filename():
860
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
861
    return osutils.pathjoin(config_dir(), 'branches.conf')
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
862
1830.2.1 by John Arbash Meinel
Make it clearer what config file needs to be renamed.
863
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
864
def locations_config_filename():
865
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
866
    return osutils.pathjoin(config_dir(), 'locations.conf')
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
867
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
868
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
869
def authentication_config_filename():
870
    """Return per-user authentication ini file filename."""
871
    return osutils.pathjoin(config_dir(), 'authentication.conf')
872
873
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
874
def user_ignore_config_filename():
875
    """Return the user default ignore filename"""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
876
    return osutils.pathjoin(config_dir(), 'ignore')
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
877
878
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
879
def crash_dir():
880
    """Return the directory name to store crash files.
881
882
    This doesn't implicitly create it.
883
4634.128.2 by Martin Pool
Write crash files into /var/crash where apport can see them.
884
    On Windows it's in the config directory; elsewhere it's /var/crash
4634.128.18 by Martin Pool
Update apport crash tests
885
    which may be monitored by apport.  It can be overridden by
886
    $APPORT_CRASH_DIR.
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
887
    """
888
    if sys.platform == 'win32':
889
        return osutils.pathjoin(config_dir(), 'Crash')
890
    else:
4634.128.2 by Martin Pool
Write crash files into /var/crash where apport can see them.
891
        # XXX: hardcoded in apport_python_hook.py; therefore here too -- mbp
892
        # 2010-01-31
4634.128.18 by Martin Pool
Update apport crash tests
893
        return os.environ.get('APPORT_CRASH_DIR', '/var/crash')
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
894
895
896
def xdg_cache_dir():
4584.3.23 by Martin Pool
Correction to xdg_cache_dir and add a simple test
897
    # See http://standards.freedesktop.org/basedir-spec/latest/ar01s03.html
898
    # Possibly this should be different on Windows?
899
    e = os.environ.get('XDG_CACHE_DIR', None)
900
    if e:
901
        return e
902
    else:
903
        return os.path.expanduser('~/.cache')
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
904
905
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
906
def parse_username(username):
907
    """Parse e-mail username and return a (name, address) tuple."""
908
    match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
909
    if match is None:
910
        return (username, '')
911
    else:
912
        return (match.group(1), match.group(2))
913
914
1185.16.52 by Martin Pool
- add extract_email_address
915
def extract_email_address(e):
916
    """Return just the address part of an email string.
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
917
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
918
    That is just the user@domain part, nothing else.
1185.16.52 by Martin Pool
- add extract_email_address
919
    This part is required to contain only ascii characters.
920
    If it can't be extracted, raises an error.
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
921
1185.16.52 by Martin Pool
- add extract_email_address
922
    >>> extract_email_address('Jane Tester <jane@test.com>')
923
    "jane@test.com"
924
    """
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
925
    name, email = parse_username(e)
926
    if not email:
2055.2.2 by John Arbash Meinel
Switch extract_email_address() to use a more specific exception
927
        raise errors.NoEmailInUsername(e)
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
928
    return email
1185.35.11 by Aaron Bentley
Added support for branch nicks
929
1185.85.30 by John Arbash Meinel
Fixing 'bzr push' exposed that IniBasedConfig didn't handle unicode.
930
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
931
class TreeConfig(IniBasedConfig):
1185.35.11 by Aaron Bentley
Added support for branch nicks
932
    """Branch configuration data associated with its contents, not location"""
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
933
3408.3.1 by Martin Pool
Remove erroneous handling of branch.conf for RemoteBranch
934
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
935
1185.35.11 by Aaron Bentley
Added support for branch nicks
936
    def __init__(self, branch):
4226.1.5 by Robert Collins
Reinstate the use of the Branch.get_config_file verb.
937
        self._config = branch._get_config()
1185.35.11 by Aaron Bentley
Added support for branch nicks
938
        self.branch = branch
939
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
940
    def _get_parser(self, file=None):
941
        if file is not None:
942
            return IniBasedConfig._get_parser(file)
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
943
        return self._config._get_configobj()
1185.35.11 by Aaron Bentley
Added support for branch nicks
944
945
    def get_option(self, name, section=None, default=None):
946
        self.branch.lock_read()
947
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
948
            return self._config.get_option(name, section, default)
1185.35.11 by Aaron Bentley
Added support for branch nicks
949
        finally:
950
            self.branch.unlock()
951
952
    def set_option(self, value, name, section=None):
953
        """Set a per-branch configuration option"""
954
        self.branch.lock_write()
955
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
956
            self._config.set_option(value, name, section)
1185.35.11 by Aaron Bentley
Added support for branch nicks
957
        finally:
958
            self.branch.unlock()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
959
960
961
class AuthenticationConfig(object):
962
    """The authentication configuration file based on a ini file.
963
964
    Implements the authentication.conf file described in
965
    doc/developers/authentication-ring.txt.
966
    """
967
968
    def __init__(self, _file=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
969
        self._config = None # The ConfigObj
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
970
        if _file is None:
2900.2.24 by Vincent Ladeuil
Review feedback.
971
            self._filename = authentication_config_filename()
972
            self._input = self._filename = authentication_config_filename()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
973
        else:
2900.2.24 by Vincent Ladeuil
Review feedback.
974
            # Tests can provide a string as _file
975
            self._filename = None
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
976
            self._input = _file
977
978
    def _get_config(self):
979
        if self._config is not None:
980
            return self._config
981
        try:
2900.2.22 by Vincent Ladeuil
Polishing.
982
            # FIXME: Should we validate something here ? Includes: empty
983
            # sections are useless, at least one of
984
            # user/password/password_encoding should be defined, etc.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
985
986
            # Note: the encoding below declares that the file itself is utf-8
987
            # encoded, but the values in the ConfigObj are always Unicode.
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
988
            self._config = ConfigObj(self._input, encoding='utf-8')
989
        except configobj.ConfigObjError, e:
990
            raise errors.ParseConfigError(e.errors, e.config.filename)
991
        return self._config
992
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
993
    def _save(self):
994
        """Save the config file, only tests should use it for now."""
2900.2.26 by Vincent Ladeuil
Fix forgotten reference to _get_filename and duplicated code.
995
        conf_dir = os.path.dirname(self._filename)
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
996
        ensure_config_dir_exists(conf_dir)
2900.2.26 by Vincent Ladeuil
Fix forgotten reference to _get_filename and duplicated code.
997
        self._get_config().write(file(self._filename, 'wb'))
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
998
999
    def _set_option(self, section_name, option_name, value):
1000
        """Set an authentication configuration option"""
1001
        conf = self._get_config()
1002
        section = conf.get(section_name)
1003
        if section is None:
1004
            conf[section] = {}
1005
            section = conf[section]
1006
        section[option_name] = value
1007
        self._save()
1008
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1009
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
1010
                        realm=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1011
        """Returns the matching credentials from authentication.conf file.
1012
1013
        :param scheme: protocol
1014
1015
        :param host: the server address
1016
1017
        :param port: the associated port (optional)
1018
1019
        :param user: login (optional)
1020
1021
        :param path: the absolute path on the server (optional)
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1022
        
1023
        :param realm: the http authentication realm (optional)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1024
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1025
        :return: A dict containing the matching credentials or None.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1026
           This includes:
1027
           - name: the section name of the credentials in the
1028
             authentication.conf file,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1029
           - user: can't be different from the provided user if any,
4107.1.7 by Jean-Francois Roy
No longer deleting the extra credentials keys in get_credentials.
1030
           - scheme: the server protocol,
1031
           - host: the server address,
1032
           - port: the server port (can be None),
1033
           - path: the absolute server path (can be None),
1034
           - realm: the http specific authentication realm (can be None),
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1035
           - password: the decoded password, could be None if the credential
1036
             defines only the user
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1037
           - verify_certificates: https specific, True if the server
1038
             certificate should be verified, False otherwise.
1039
        """
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1040
        credentials = None
1041
        for auth_def_name, auth_def in self._get_config().items():
3418.2.1 by Vincent Ladeuil
Fix #217650 by catching declarations outside sections.
1042
            if type(auth_def) is not configobj.Section:
1043
                raise ValueError("%s defined outside a section" % auth_def_name)
1044
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1045
            a_scheme, a_host, a_user, a_path = map(
1046
                auth_def.get, ['scheme', 'host', 'user', 'path'])
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1047
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1048
            try:
1049
                a_port = auth_def.as_int('port')
1050
            except KeyError:
1051
                a_port = None
2900.2.22 by Vincent Ladeuil
Polishing.
1052
            except ValueError:
1053
                raise ValueError("'port' not numeric in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1054
            try:
1055
                a_verify_certificates = auth_def.as_bool('verify_certificates')
1056
            except KeyError:
1057
                a_verify_certificates = True
2900.2.22 by Vincent Ladeuil
Polishing.
1058
            except ValueError:
1059
                raise ValueError(
1060
                    "'verify_certificates' not boolean in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1061
1062
            # Attempt matching
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1063
            if a_scheme is not None and scheme != a_scheme:
1064
                continue
1065
            if a_host is not None:
1066
                if not (host == a_host
1067
                        or (a_host.startswith('.') and host.endswith(a_host))):
1068
                    continue
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1069
            if a_port is not None and port != a_port:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1070
                continue
1071
            if (a_path is not None and path is not None
1072
                and not path.startswith(a_path)):
1073
                continue
1074
            if (a_user is not None and user is not None
1075
                and a_user != user):
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1076
                # Never contradict the caller about the user to be used
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1077
                continue
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1078
            if a_user is None:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1079
                # Can't find a user
1080
                continue
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1081
            # Prepare a credentials dictionary with additional keys
1082
            # for the credential providers
2900.2.24 by Vincent Ladeuil
Review feedback.
1083
            credentials = dict(name=auth_def_name,
3418.4.2 by Vincent Ladeuil
Fix bug #199440 by taking into account that a section may not
1084
                               user=a_user,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1085
                               scheme=a_scheme,
1086
                               host=host,
1087
                               port=port,
1088
                               path=path,
1089
                               realm=realm,
3418.4.2 by Vincent Ladeuil
Fix bug #199440 by taking into account that a section may not
1090
                               password=auth_def.get('password', None),
2900.2.24 by Vincent Ladeuil
Review feedback.
1091
                               verify_certificates=a_verify_certificates)
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1092
            # Decode the password in the credentials (or get one)
2900.2.22 by Vincent Ladeuil
Polishing.
1093
            self.decode_password(credentials,
1094
                                 auth_def.get('password_encoding', None))
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1095
            if 'auth' in debug.debug_flags:
1096
                trace.mutter("Using authentication section: %r", auth_def_name)
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1097
            break
1098
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1099
        if credentials is None:
1100
            # No credentials were found in authentication.conf, try the fallback
1101
            # credentials stores.
1102
            credentials = credential_store_registry.get_fallback_credentials(
1103
                scheme, host, port, user, path, realm)
1104
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1105
        return credentials
1106
3777.3.2 by Aaron Bentley
Reverse order of scheme and password
1107
    def set_credentials(self, name, host, user, scheme=None, password=None,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1108
                        port=None, path=None, verify_certificates=None,
1109
                        realm=None):
3777.3.1 by Aaron Bentley
Update docs
1110
        """Set authentication credentials for a host.
1111
1112
        Any existing credentials with matching scheme, host, port and path
1113
        will be deleted, regardless of name.
1114
1115
        :param name: An arbitrary name to describe this set of credentials.
1116
        :param host: Name of the host that accepts these credentials.
1117
        :param user: The username portion of these credentials.
1118
        :param scheme: The URL scheme (e.g. ssh, http) the credentials apply
1119
            to.
3777.3.2 by Aaron Bentley
Reverse order of scheme and password
1120
        :param password: Password portion of these credentials.
3777.3.1 by Aaron Bentley
Update docs
1121
        :param port: The IP port on the host that these credentials apply to.
1122
        :param path: A filesystem path on the host that these credentials
1123
            apply to.
1124
        :param verify_certificates: On https, verify server certificates if
1125
            True.
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1126
        :param realm: The http authentication realm (optional).
3777.3.1 by Aaron Bentley
Update docs
1127
        """
3777.1.8 by Aaron Bentley
Commit work-in-progress
1128
        values = {'host': host, 'user': user}
1129
        if password is not None:
1130
            values['password'] = password
1131
        if scheme is not None:
1132
            values['scheme'] = scheme
1133
        if port is not None:
1134
            values['port'] = '%d' % port
1135
        if path is not None:
1136
            values['path'] = path
3777.1.10 by Aaron Bentley
Ensure credentials are stored
1137
        if verify_certificates is not None:
1138
            values['verify_certificates'] = str(verify_certificates)
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1139
        if realm is not None:
1140
            values['realm'] = realm
3777.1.11 by Aaron Bentley
Ensure changed-name updates clear old values
1141
        config = self._get_config()
1142
        for_deletion = []
1143
        for section, existing_values in config.items():
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1144
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
3777.1.11 by Aaron Bentley
Ensure changed-name updates clear old values
1145
                if existing_values.get(key) != values.get(key):
1146
                    break
1147
            else:
1148
                del config[section]
1149
        config.update({name: values})
3777.1.10 by Aaron Bentley
Ensure credentials are stored
1150
        self._save()
3777.1.8 by Aaron Bentley
Commit work-in-progress
1151
4304.2.1 by Vincent Ladeuil
Fix bug #367726 by reverting some default user handling introduced
1152
    def get_user(self, scheme, host, port=None, realm=None, path=None,
4222.3.10 by Jelmer Vernooij
Avoid using the default username in the case of SMTP.
1153
                 prompt=None, ask=False, default=None):
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1154
        """Get a user from authentication file.
1155
1156
        :param scheme: protocol
1157
1158
        :param host: the server address
1159
1160
        :param port: the associated port (optional)
1161
1162
        :param realm: the realm sent by the server (optional)
1163
1164
        :param path: the absolute path on the server (optional)
1165
4222.3.4 by Jelmer Vernooij
Default to getpass.getuser() in AuthenticationConfig.get_user(), but allow
1166
        :param ask: Ask the user if there is no explicitly configured username 
1167
                    (optional)
1168
4304.2.1 by Vincent Ladeuil
Fix bug #367726 by reverting some default user handling introduced
1169
        :param default: The username returned if none is defined (optional).
1170
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1171
        :return: The found user.
1172
        """
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1173
        credentials = self.get_credentials(scheme, host, port, user=None,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1174
                                           path=path, realm=realm)
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1175
        if credentials is not None:
1176
            user = credentials['user']
1177
        else:
1178
            user = None
4222.3.2 by Jelmer Vernooij
Prompt for user names if they are not in the configuration.
1179
        if user is None:
4222.3.4 by Jelmer Vernooij
Default to getpass.getuser() in AuthenticationConfig.get_user(), but allow
1180
            if ask:
1181
                if prompt is None:
1182
                    # Create a default prompt suitable for most cases
1183
                    prompt = scheme.upper() + ' %(host)s username'
1184
                # Special handling for optional fields in the prompt
1185
                if port is not None:
1186
                    prompt_host = '%s:%d' % (host, port)
1187
                else:
1188
                    prompt_host = host
1189
                user = ui.ui_factory.get_username(prompt, host=prompt_host)
4222.3.2 by Jelmer Vernooij
Prompt for user names if they are not in the configuration.
1190
            else:
4222.3.10 by Jelmer Vernooij
Avoid using the default username in the case of SMTP.
1191
                user = default
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1192
        return user
1193
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1194
    def get_password(self, scheme, host, user, port=None,
1195
                     realm=None, path=None, prompt=None):
1196
        """Get a password from authentication file or prompt the user for one.
1197
1198
        :param scheme: protocol
1199
1200
        :param host: the server address
1201
1202
        :param port: the associated port (optional)
1203
1204
        :param user: login
1205
1206
        :param realm: the realm sent by the server (optional)
1207
1208
        :param path: the absolute path on the server (optional)
1209
1210
        :return: The found password or the one entered by the user.
1211
        """
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1212
        credentials = self.get_credentials(scheme, host, port, user, path,
1213
                                           realm)
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1214
        if credentials is not None:
1215
            password = credentials['password']
3420.1.3 by Vincent Ladeuil
John's review feedback.
1216
            if password is not None and scheme is 'ssh':
3420.1.2 by Vincent Ladeuil
Fix bug #203186 by ignoring passwords for ssh and warning user.
1217
                trace.warning('password ignored in section [%s],'
1218
                              ' use an ssh agent instead'
1219
                              % credentials['name'])
1220
                password = None
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1221
        else:
1222
            password = None
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1223
        # Prompt user only if we could't find a password
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1224
        if password is None:
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1225
            if prompt is None:
3420.1.2 by Vincent Ladeuil
Fix bug #203186 by ignoring passwords for ssh and warning user.
1226
                # Create a default prompt suitable for most cases
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1227
                prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1228
            # Special handling for optional fields in the prompt
1229
            if port is not None:
1230
                prompt_host = '%s:%d' % (host, port)
1231
            else:
1232
                prompt_host = host
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1233
            password = ui.ui_factory.get_password(prompt,
1234
                                                  host=prompt_host, user=user)
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1235
        return password
1236
2900.2.22 by Vincent Ladeuil
Polishing.
1237
    def decode_password(self, credentials, encoding):
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1238
        try:
1239
            cs = credential_store_registry.get_credential_store(encoding)
1240
        except KeyError:
1241
            raise ValueError('%r is not a known password_encoding' % encoding)
1242
        credentials['password'] = cs.decode_password(credentials)
2900.2.22 by Vincent Ladeuil
Polishing.
1243
        return credentials
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1244
3242.3.17 by Aaron Bentley
Whitespace cleanup
1245
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1246
class CredentialStoreRegistry(registry.Registry):
1247
    """A class that registers credential stores.
1248
1249
    A credential store provides access to credentials via the password_encoding
1250
    field in authentication.conf sections.
1251
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1252
    Except for stores provided by bzr itself, most stores are expected to be
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1253
    provided by plugins that will therefore use
1254
    register_lazy(password_encoding, module_name, member_name, help=help,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1255
    fallback=fallback) to install themselves.
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1256
1257
    A fallback credential store is one that is queried if no credentials can be
1258
    found via authentication.conf.
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1259
    """
1260
1261
    def get_credential_store(self, encoding=None):
1262
        cs = self.get(encoding)
1263
        if callable(cs):
1264
            cs = cs()
1265
        return cs
1266
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1267
    def is_fallback(self, name):
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1268
        """Check if the named credentials store should be used as fallback."""
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1269
        return self.get_info(name)
1270
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1271
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1272
                                 path=None, realm=None):
1273
        """Request credentials from all fallback credentials stores.
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1274
1275
        The first credentials store that can provide credentials wins.
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1276
        """
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1277
        credentials = None
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1278
        for name in self.keys():
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1279
            if not self.is_fallback(name):
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1280
                continue
1281
            cs = self.get_credential_store(name)
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1282
            credentials = cs.get_credentials(scheme, host, port, user,
1283
                                             path, realm)
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1284
            if credentials is not None:
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1285
                # We found some credentials
1286
                break
1287
        return credentials
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1288
1289
    def register(self, key, obj, help=None, override_existing=False,
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1290
                 fallback=False):
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1291
        """Register a new object to a name.
1292
1293
        :param key: This is the key to use to request the object later.
1294
        :param obj: The object to register.
1295
        :param help: Help text for this entry. This may be a string or
1296
                a callable. If it is a callable, it should take two
1297
                parameters (registry, key): this registry and the key that
1298
                the help was registered under.
1299
        :param override_existing: Raise KeyErorr if False and something has
1300
                already been registered for that key. If True, ignore if there
1301
                is an existing key (always register the new value).
1302
        :param fallback: Whether this credential store should be 
1303
                used as fallback.
1304
        """
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1305
        return super(CredentialStoreRegistry,
1306
                     self).register(key, obj, help, info=fallback,
1307
                                    override_existing=override_existing)
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1308
1309
    def register_lazy(self, key, module_name, member_name,
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1310
                      help=None, override_existing=False,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1311
                      fallback=False):
1312
        """Register a new credential store to be loaded on request.
1313
1314
        :param module_name: The python path to the module. Such as 'os.path'.
1315
        :param member_name: The member of the module to return.  If empty or
1316
                None, get() will return the module itself.
1317
        :param help: Help text for this entry. This may be a string or
1318
                a callable.
1319
        :param override_existing: If True, replace the existing object
1320
                with the new one. If False, if there is already something
1321
                registered with the same key, raise a KeyError
1322
        :param fallback: Whether this credential store should be 
1323
                used as fallback.
1324
        """
1325
        return super(CredentialStoreRegistry, self).register_lazy(
1326
            key, module_name, member_name, help,
1327
            info=fallback, override_existing=override_existing)
1328
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1329
1330
credential_store_registry = CredentialStoreRegistry()
1331
1332
1333
class CredentialStore(object):
1334
    """An abstract class to implement storage for credentials"""
1335
1336
    def decode_password(self, credentials):
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1337
        """Returns a clear text password for the provided credentials."""
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1338
        raise NotImplementedError(self.decode_password)
1339
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1340
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1341
                        realm=None):
1342
        """Return the matching credentials from this credential store.
1343
1344
        This method is only called on fallback credential stores.
1345
        """
1346
        raise NotImplementedError(self.get_credentials)
1347
1348
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1349
1350
class PlainTextCredentialStore(CredentialStore):
5131.2.1 by Martin
Permit bzrlib to run under python -OO by explictly assigning to __doc__ for user-visible docstrings
1351
    __doc__ = """Plain text credential store for the authentication.conf file"""
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1352
1353
    def decode_password(self, credentials):
1354
        """See CredentialStore.decode_password."""
1355
        return credentials['password']
1356
1357
1358
credential_store_registry.register('plain', PlainTextCredentialStore,
1359
                                   help=PlainTextCredentialStore.__doc__)
1360
credential_store_registry.default_key = 'plain'
1361
1362
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1363
class BzrDirConfig(object):
1364
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
1365
    def __init__(self, bzrdir):
1366
        self._bzrdir = bzrdir
1367
        self._config = bzrdir._get_config()
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1368
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
1369
    def set_default_stack_on(self, value):
1370
        """Set the default stacking location.
1371
1372
        It may be set to a location, or None.
1373
1374
        This policy affects all branches contained by this bzrdir, except for
1375
        those under repositories.
1376
        """
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
1377
        if self._config is None:
1378
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
1379
        if value is None:
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1380
            self._config.set_option('', 'default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
1381
        else:
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1382
            self._config.set_option(value, 'default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
1383
1384
    def get_default_stack_on(self):
1385
        """Return the default stacking location.
1386
1387
        This will either be a location, or None.
1388
1389
        This policy affects all branches contained by this bzrdir, except for
1390
        those under repositories.
1391
        """
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
1392
        if self._config is None:
1393
            return None
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1394
        value = self._config.get_option('default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
1395
        if value == '':
1396
            value = None
1397
        return value
1398
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1399
1400
class TransportConfig(object):
3242.1.5 by Aaron Bentley
Update per review comments
1401
    """A Config that reads/writes a config file on a Transport.
3242.1.4 by Aaron Bentley
Clean-up
1402
1403
    It is a low-level object that considers config data to be name/value pairs
1404
    that may be associated with a section.  Assigning meaning to the these
1405
    values is done at higher levels like TreeConfig.
1406
    """
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1407
1408
    def __init__(self, transport, filename):
1409
        self._transport = transport
1410
        self._filename = filename
1411
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1412
    def get_option(self, name, section=None, default=None):
1413
        """Return the value associated with a named option.
1414
1415
        :param name: The name of the value
1416
        :param section: The section the option is in (if any)
1417
        :param default: The value to return if the value is not set
1418
        :return: The value or default value
1419
        """
1420
        configobj = self._get_configobj()
1421
        if section is None:
1422
            section_obj = configobj
1423
        else:
1424
            try:
1425
                section_obj = configobj[section]
1426
            except KeyError:
1427
                return default
1428
        return section_obj.get(name, default)
1429
1430
    def set_option(self, value, name, section=None):
1431
        """Set the value associated with a named option.
1432
1433
        :param value: The value to set
1434
        :param name: The name of the value to set
1435
        :param section: The section the option is in (if any)
1436
        """
1437
        configobj = self._get_configobj()
1438
        if section is None:
1439
            configobj[name] = value
1440
        else:
1441
            configobj.setdefault(section, {})[name] = value
1442
        self._set_configobj(configobj)
1443
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1444
    def _get_config_file(self):
1445
        try:
4852.1.10 by John Arbash Meinel
Use a StringIO instead, otherwise we get failures with smart server requests.
1446
            return StringIO(self._transport.get_bytes(self._filename))
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1447
        except errors.NoSuchFile:
1448
            return StringIO()
1449
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1450
    def _get_configobj(self):
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
1451
        return ConfigObj(self._get_config_file(), encoding='utf-8')
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1452
1453
    def _set_configobj(self, configobj):
1454
        out_file = StringIO()
1455
        configobj.write(out_file)
1456
        out_file.seek(0)
1457
        self._transport.put_file(self._filename, out_file)