/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
5598.2.1 by John Arbash Meinel
Decode windows env vars using mbcs rather than assuming the 8-bit string is ok.
1
# Copyright (C) 2005-2011 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
5050.72.1 by Martin Pool
Set email address from /etc/mailname if possible
66
import string
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
67
import sys
1474 by Robert Collins
Merge from Aaron Bentley.
68
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
69
from bzrlib import commands
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
70
from bzrlib.decorators import needs_write_lock
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
71
from bzrlib.lazy_import import lazy_import
72
lazy_import(globals(), """
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
73
import fnmatch
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
74
import re
2900.2.22 by Vincent Ladeuil
Polishing.
75
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.
76
77
import bzrlib
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
78
from bzrlib import (
4797.59.2 by Vincent Ladeuil
Use AtomicFile and avoid all unicode/encoding issues around transport (thanks jam).
79
    atomicfile,
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
80
    bzrdir,
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
81
    debug,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
82
    errors,
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
83
    lockdir,
2681.1.8 by Aaron Bentley
Add Thunderbird support to bzr send
84
    mail_client,
5321.1.88 by Gordon Tyler
Moved mergetools config functions into bzrlib.config.Config.
85
    mergetools,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
86
    osutils,
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
87
    registry,
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
88
    symbol_versioning,
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
89
    trace,
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
90
    transport,
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
91
    ui,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
92
    urlutils,
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
93
    win32utils,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
94
    )
2991.2.4 by Vincent Ladeuil
Various fixes following local testing environment rebuild.
95
from bzrlib.util.configobj import configobj
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
96
""")
97
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
98
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
99
CHECK_IF_POSSIBLE=0
100
CHECK_ALWAYS=1
101
CHECK_NEVER=2
102
103
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
104
SIGN_WHEN_REQUIRED=0
105
SIGN_ALWAYS=1
106
SIGN_NEVER=2
107
108
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
109
POLICY_NONE = 0
110
POLICY_NORECURSE = 1
111
POLICY_APPENDPATH = 2
112
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
113
_policy_name = {
114
    POLICY_NONE: None,
115
    POLICY_NORECURSE: 'norecurse',
116
    POLICY_APPENDPATH: 'appendpath',
117
    }
118
_policy_value = {
119
    None: POLICY_NONE,
120
    'none': POLICY_NONE,
121
    'norecurse': POLICY_NORECURSE,
122
    'appendpath': POLICY_APPENDPATH,
123
    }
2120.6.4 by James Henstridge
add support for specifying policy when storing options
124
125
126
STORE_LOCATION = POLICY_NONE
127
STORE_LOCATION_NORECURSE = POLICY_NORECURSE
128
STORE_LOCATION_APPENDPATH = POLICY_APPENDPATH
129
STORE_BRANCH = 3
130
STORE_GLOBAL = 4
131
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
132
133
class ConfigObj(configobj.ConfigObj):
134
135
    def __init__(self, infile=None, **kwargs):
136
        # We define our own interpolation mechanism calling it option expansion
137
        super(ConfigObj, self).__init__(infile=infile,
138
                                        interpolation=False,
139
                                        **kwargs)
140
141
142
    def get_bool(self, section, key):
143
        return self[section].as_bool(key)
144
145
    def get_value(self, section, name):
146
        # Try [] for the old DEFAULT section.
147
        if section == "DEFAULT":
148
            try:
149
                return self[name]
150
            except KeyError:
151
                pass
152
        return self[section][name]
153
154
155
# FIXME: Until we can guarantee that each config file is loaded once and and
156
# only once for a given bzrlib session, we don't want to re-read the file every
157
# time we query for an option so we cache the value (bad ! watch out for tests
158
# needing to restore the proper value).This shouldn't be part of 2.4.0 final,
159
# yell at mgz^W vila and the RM if this is still present at that time
160
# -- vila 20110219
161
_expand_default_value = None
162
def _get_expand_default_value():
163
    global _expand_default_value
164
    if _expand_default_value is not None:
165
        return _expand_default_value
166
    conf = GlobalConfig()
167
    # Note that we must not use None for the expand value below or we'll run
168
    # into infinite recursion. Using False really would be quite silly ;)
169
    expand = conf.get_user_option_as_bool('bzr.config.expand', expand=True)
170
    if expand is None:
171
        # This is an opt-in feature, you *really* need to clearly say you want
172
        # to activate it !
173
        expand = False
174
    _expand_default_value = expand
175
    return expand
5549.1.31 by Vincent Ladeuil
Implement a default value for config option expansion (what ? No tests ?).
176
5549.1.19 by Vincent Ladeuil
Push down interpolation at the config level (make tests slightly less
177
178
class Config(object):
179
    """A configuration policy - what username, editor, gpg needs etc."""
180
181
    def __init__(self):
182
        super(Config, self).__init__()
183
184
    def config_id(self):
185
        """Returns a unique ID for the config."""
186
        raise NotImplementedError(self.config_id)
187
188
    def get_editor(self):
189
        """Get the users pop up editor."""
190
        raise NotImplementedError
191
192
    def get_change_editor(self, old_tree, new_tree):
193
        from bzrlib import diff
194
        cmd = self._get_change_editor()
195
        if cmd is None:
196
            return None
197
        return diff.DiffFromTool.from_string(cmd, old_tree, new_tree,
198
                                             sys.stdout)
199
200
201
    def get_mail_client(self):
202
        """Get a mail client to use"""
203
        selected_client = self.get_user_option('mail_client')
204
        _registry = mail_client.mail_client_registry
205
        try:
206
            mail_client_class = _registry.get(selected_client)
207
        except KeyError:
208
            raise errors.UnknownMailClient(selected_client)
209
        return mail_client_class(self)
210
211
    def _get_signature_checking(self):
212
        """Template method to override signature checking policy."""
213
214
    def _get_signing_policy(self):
215
        """Template method to override signature creation policy."""
216
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
217
    option_ref_re = None
218
219
    def expand_options(self, string, env=None):
220
        """Expand option references in the string in the configuration context.
221
222
        :param string: The string containing option to expand.
223
224
        :param env: An option dict defining additional configuration options or
225
            overriding existing ones.
226
227
        :returns: The expanded string.
228
        """
229
        return self._expand_options_in_string(string, env)
230
231
    def _expand_options_in_list(self, slist, env=None, _ref_stack=None):
232
        """Expand options in  a list of strings in the configuration context.
233
234
        :param slist: A list of strings.
235
236
        :param env: An option dict defining additional configuration options or
237
            overriding existing ones.
238
239
        :param _ref_stack: Private list containing the options being
240
            expanded to detect loops.
241
242
        :returns: The flatten list of expanded strings.
243
        """
244
        # expand options in each value separately flattening lists
245
        result = []
246
        for s in slist:
247
            value = self._expand_options_in_string(s, env, _ref_stack)
248
            if isinstance(value, list):
249
                result.extend(value)
250
            else:
251
                result.append(value)
252
        return result
253
254
    def _expand_options_in_string(self, string, env=None, _ref_stack=None):
255
        """Expand options in the string in the configuration context.
256
257
        :param string: The string to be expanded.
258
259
        :param env: An option dict defining additional configuration options or
260
            overriding existing ones.
261
262
        :param _ref_stack: Private list containing the options being
263
            expanded to detect loops.
264
265
        :returns: The expanded string.
266
        """
267
        if string is None:
268
            # Not much to expand there
269
            return None
270
        if _ref_stack is None:
271
            # What references are currently resolved (to detect loops)
272
            _ref_stack = []
273
        if self.option_ref_re is None:
274
            # We want to match the most embedded reference first (i.e. for
275
            # '{{foo}}' we will get '{foo}',
276
            # for '{bar{baz}}' we will get '{baz}'
277
            self.option_ref_re = re.compile('({[^{}]+})')
278
        result = string
279
        # We need to iterate until no more refs appear ({{foo}} will need two
280
        # iterations for example).
281
        while True:
5745.1.1 by Vincent Ladeuil
Remove debug code
282
            raw_chunks = self.option_ref_re.split(result)
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
283
            if len(raw_chunks) == 1:
284
                # Shorcut the trivial case: no refs
285
                return result
286
            chunks = []
287
            list_value = False
288
            # Split will isolate refs so that every other chunk is a ref
289
            chunk_is_ref = False
290
            for chunk in raw_chunks:
291
                if not chunk_is_ref:
292
                    if chunk:
293
                        # Keep only non-empty strings (or we get bogus empty
294
                        # slots when a list value is involved).
295
                        chunks.append(chunk)
296
                    chunk_is_ref = True
297
                else:
298
                    name = chunk[1:-1]
299
                    if name in _ref_stack:
300
                        raise errors.OptionExpansionLoop(string, _ref_stack)
301
                    _ref_stack.append(name)
302
                    value = self._expand_option(name, env, _ref_stack)
303
                    if value is None:
304
                        raise errors.ExpandingUnknownOption(name, string)
305
                    if isinstance(value, list):
306
                        list_value = True
307
                        chunks.extend(value)
308
                    else:
309
                        chunks.append(value)
310
                    _ref_stack.pop()
311
                    chunk_is_ref = False
312
            if list_value:
313
                # Once a list appears as the result of an expansion, all
314
                # callers will get a list result. This allows a consistent
315
                # behavior even when some options in the expansion chain
316
                # defined as strings (no comma in their value) but their
317
                # expanded value is a list.
318
                return self._expand_options_in_list(chunks, env, _ref_stack)
319
            else:
320
                result = ''.join(chunks)
321
        return result
322
323
    def _expand_option(self, name, env, _ref_stack):
324
        if env is not None and name in env:
325
            # Special case, values provided in env takes precedence over
326
            # anything else
327
            value = env[name]
328
        else:
329
            # FIXME: This is a limited implementation, what we really need is a
330
            # way to query the bzr config for the value of an option,
331
            # respecting the scope rules (That is, once we implement fallback
332
            # configs, getting the option value should restart from the top
333
            # config, not the current one) -- vila 20101222
334
            value = self.get_user_option(name, expand=False)
335
            if isinstance(value, list):
336
                value = self._expand_options_in_list(value, env, _ref_stack)
337
            else:
338
                value = self._expand_options_in_string(value, env, _ref_stack)
339
        return value
340
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
341
    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.
342
        """Template method to provide a user option."""
343
        return None
344
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
345
    def get_user_option(self, option_name, expand=None):
346
        """Get a generic option - no special process, no default.
347
348
        :param option_name: The queried option.
349
350
        :param expand: Whether options references should be expanded.
351
352
        :returns: The value of the option.
353
        """
354
        if expand is None:
355
            expand = _get_expand_default_value()
356
        value = self._get_user_option(option_name)
357
        if expand:
358
            if isinstance(value, list):
359
                value = self._expand_options_in_list(value)
360
            elif isinstance(value, dict):
361
                trace.warning('Cannot expand "%s":'
362
                              ' Dicts do not support option expansion'
363
                              % (option_name,))
364
            else:
365
                value = self._expand_options_in_string(value)
366
        return value
367
368
    def get_user_option_as_bool(self, option_name, expand=None):
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
369
        """Get a generic option as a boolean - no special process, no default.
370
371
        :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.
372
            interpreted as a boolean. Returns True or False otherwise.
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
373
        """
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
374
        s = self.get_user_option(option_name, expand=expand)
4989.2.12 by Vincent Ladeuil
Display a warning if an option value is not boolean.
375
        if s is None:
376
            # The option doesn't exist
377
            return None
4989.2.15 by Vincent Ladeuil
Fixed as per Andrew's review.
378
        val = ui.bool_from_string(s)
4989.2.12 by Vincent Ladeuil
Display a warning if an option value is not boolean.
379
        if val is None:
380
            # The value can't be interpreted as a boolean
381
            trace.warning('Value "%s" is not a boolean for "%s"',
382
                          s, option_name)
383
        return val
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
384
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
385
    def get_user_option_as_list(self, option_name, expand=None):
4840.2.4 by Vincent Ladeuil
Implement config.get_user_option_as_list.
386
        """Get a generic option as a list - no special process, no default.
387
388
        :return None if the option doesn't exist. Returns the value as a list
389
            otherwise.
390
        """
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
391
        l = self.get_user_option(option_name, expand=expand)
4840.2.4 by Vincent Ladeuil
Implement config.get_user_option_as_list.
392
        if isinstance(l, (str, unicode)):
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
393
            # A single value, most probably the user forgot (or didn't care to
394
            # add) the final ','
4840.2.4 by Vincent Ladeuil
Implement config.get_user_option_as_list.
395
            l = [l]
396
        return l
397
1442.1.56 by Robert Collins
gpg_signing_command configuration item
398
    def gpg_signing_command(self):
399
        """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.
400
        result = self._gpg_signing_command()
401
        if result is None:
402
            result = "gpg"
403
        return result
404
405
    def _gpg_signing_command(self):
406
        """See gpg_signing_command()."""
407
        return None
1442.1.56 by Robert Collins
gpg_signing_command configuration item
408
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
409
    def log_format(self):
410
        """What log format should be used"""
411
        result = self._log_format()
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
412
        if result is None:
413
            result = "long"
414
        return result
415
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
416
    def _log_format(self):
417
        """See log_format()."""
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
418
        return None
419
1472 by Robert Collins
post commit hook, first pass implementation
420
    def post_commit(self):
421
        """An ordered list of python functions to call.
422
423
        Each function takes branch, rev_id as parameters.
424
        """
425
        return self._post_commit()
426
427
    def _post_commit(self):
428
        """See Config.post_commit."""
429
        return None
430
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
431
    def user_email(self):
432
        """Return just the email component of a username."""
1185.33.31 by Martin Pool
Make annotate cope better with revisions committed without a valid
433
        return extract_email_address(self.username())
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
434
435
    def username(self):
436
        """Return email-style username.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
437
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
438
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
439
5187.2.1 by Parth Malwankar
removed comment about deprecated BZREMAIL.
440
        $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
441
        the concrete policy type is checked, and finally
1185.37.2 by Jamie Wilkinson
Fix a typo and grammar in Config.username() docstring.
442
        $EMAIL is examined.
5187.2.12 by Parth Malwankar
trivial clarification in docstring.
443
        If no username can be found, errors.NoWhoami exception is raised.
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
444
        """
1861.4.1 by Matthieu Moy
BZREMAIL renamed to BZR_EMAIL.
445
        v = os.environ.get('BZR_EMAIL')
446
        if v:
3224.5.4 by Andrew Bennetts
Fix test suite, mainly weeding out uses of bzrlib.user_encoding.
447
            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
448
        v = self._get_user_id()
449
        if v:
450
            return v
451
        v = os.environ.get('EMAIL')
452
        if v:
3224.5.4 by Andrew Bennetts
Fix test suite, mainly weeding out uses of bzrlib.user_encoding.
453
            return v.decode(osutils.get_user_encoding())
5050.72.1 by Martin Pool
Set email address from /etc/mailname if possible
454
        name, email = _auto_user_id()
455
        if name and email:
456
            return '%s <%s>' % (name, email)
457
        elif email:
458
            return email
5187.2.6 by Parth Malwankar
lockdir no long mandates whoami but uses unicode version of getuser
459
        raise errors.NoWhoami()
5187.2.3 by Parth Malwankar
init and init-repo now fail before creating dir if username is not set.
460
461
    def ensure_username(self):
5187.2.11 by Parth Malwankar
documentation updates
462
        """Raise errors.NoWhoami if username is not set.
5187.2.3 by Parth Malwankar
init and init-repo now fail before creating dir if username is not set.
463
464
        This method relies on the username() function raising the error.
465
        """
466
        self.username()
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
467
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
468
    def signature_checking(self):
469
        """What is the current policy for signature checking?."""
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
470
        policy = self._get_signature_checking()
471
        if policy is not None:
472
            return policy
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
473
        return CHECK_IF_POSSIBLE
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
474
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
475
    def signing_policy(self):
476
        """What is the current policy for signature checking?."""
477
        policy = self._get_signing_policy()
478
        if policy is not None:
479
            return policy
480
        return SIGN_WHEN_REQUIRED
481
1442.1.21 by Robert Collins
create signature_needed() call for commit to trigger creating signatures
482
    def signature_needed(self):
483
        """Is a signature needed when committing ?."""
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
484
        policy = self._get_signing_policy()
485
        if policy is None:
486
            policy = self._get_signature_checking()
487
            if policy is not None:
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
488
                trace.warning("Please use create_signatures,"
489
                              " not check_signatures to set signing policy.")
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
490
            if policy == CHECK_ALWAYS:
491
                return True
492
        elif policy == SIGN_ALWAYS:
1442.1.21 by Robert Collins
create signature_needed() call for commit to trigger creating signatures
493
            return True
494
        return False
495
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
496
    def get_alias(self, value):
497
        return self._get_alias(value)
498
499
    def _get_alias(self, value):
500
        pass
501
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
502
    def get_nickname(self):
503
        return self._get_nickname()
504
505
    def _get_nickname(self):
506
        return None
507
1551.18.17 by Aaron Bentley
Introduce bzr_remote_path configuration variable
508
    def get_bzr_remote_path(self):
509
        try:
510
            return os.environ['BZR_REMOTE_PATH']
511
        except KeyError:
512
            path = self.get_user_option("bzr_remote_path")
513
            if path is None:
514
                path = 'bzr'
515
            return path
516
4840.2.6 by Vincent Ladeuil
Implement config.suppress_warning.
517
    def suppress_warning(self, warning):
518
        """Should the warning be suppressed or emitted.
519
520
        :param warning: The name of the warning being tested.
521
522
        :returns: True if the warning should be suppressed, False otherwise.
523
        """
524
        warnings = self.get_user_option_as_list('suppress_warnings')
525
        if warnings is None or warning not in warnings:
526
            return False
527
        else:
528
            return True
529
5321.1.88 by Gordon Tyler
Moved mergetools config functions into bzrlib.config.Config.
530
    def get_merge_tools(self):
5321.1.108 by Gordon Tyler
Changed known merge tools into a default set of merge tools that are always defined but can be overridden by user-defined merge tools.
531
        tools = {}
5321.1.99 by Gordon Tyler
Fixes for changes to Config._get_options().
532
        for (oname, value, section, conf_id, parser) in self._get_options():
5321.2.3 by Vincent Ladeuil
Prefix mergetools option names with 'bzr.'.
533
            if oname.startswith('bzr.mergetool.'):
5321.1.108 by Gordon Tyler
Changed known merge tools into a default set of merge tools that are always defined but can be overridden by user-defined merge tools.
534
                tool_name = oname[len('bzr.mergetool.'):]
5321.1.116 by Gordon Tyler
Simplified mergetools module down to functions which deal with command lines -- no MergeTool class.
535
                tools[tool_name] = value
5321.1.88 by Gordon Tyler
Moved mergetools config functions into bzrlib.config.Config.
536
        trace.mutter('loaded merge tools: %r' % tools)
5321.1.116 by Gordon Tyler
Simplified mergetools module down to functions which deal with command lines -- no MergeTool class.
537
        return tools
5321.1.88 by Gordon Tyler
Moved mergetools config functions into bzrlib.config.Config.
538
5321.1.103 by Gordon Tyler
Renamed _find_merge_tool back to find_merge_tool since it must be public for UI code to lookup merge tools by name, and added tests for it.
539
    def find_merge_tool(self, name):
5321.1.111 by Gordon Tyler
Remove predefined merge tools from list returned by get_merge_tools.
540
        # We fake a defaults mechanism here by checking if the given name can 
541
        # be found in the known_merge_tools if it's not found in the config.
542
        # This should be done through the proposed config defaults mechanism
543
        # when it becomes available in the future.
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
544
        command_line = (self.get_user_option('bzr.mergetool.%s' % name,
545
                                             expand=False)
546
                        or mergetools.known_merge_tools.get(name, None))
5321.1.116 by Gordon Tyler
Simplified mergetools module down to functions which deal with command lines -- no MergeTool class.
547
        return command_line
5321.1.88 by Gordon Tyler
Moved mergetools config functions into bzrlib.config.Config.
548
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
549
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
550
class IniBasedConfig(Config):
551
    """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
552
5345.1.1 by Vincent Ladeuil
Deprecate the get_filename parameter in IniBasedConfig.
553
    def __init__(self, get_filename=symbol_versioning.DEPRECATED_PARAMETER,
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
554
                 file_name=None):
5345.2.5 by Vincent Ladeuil
Add docstring.
555
        """Base class for configuration files using an ini-like syntax.
556
557
        :param file_name: The configuration file path.
558
        """
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
559
        super(IniBasedConfig, self).__init__()
5345.1.8 by Vincent Ladeuil
Make the test_listen_to_the_last_speaker pass and fix fallouts.
560
        self.file_name = file_name
5345.1.1 by Vincent Ladeuil
Deprecate the get_filename parameter in IniBasedConfig.
561
        if symbol_versioning.deprecated_passed(get_filename):
562
            symbol_versioning.warn(
563
                'IniBasedConfig.__init__(get_filename) was deprecated in 2.3.'
564
                ' Use file_name instead.',
565
                DeprecationWarning,
566
                stacklevel=2)
5345.1.8 by Vincent Ladeuil
Make the test_listen_to_the_last_speaker pass and fix fallouts.
567
            if get_filename is not None:
5345.1.1 by Vincent Ladeuil
Deprecate the get_filename parameter in IniBasedConfig.
568
                self.file_name = get_filename()
569
        else:
570
            self.file_name = file_name
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
571
        self._content = None
4503.2.2 by Vincent Ladeuil
Get a bool or none from a config file.
572
        self._parser = None
573
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
574
    @classmethod
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
575
    def from_string(cls, str_or_unicode, file_name=None, save=False):
5345.5.13 by Vincent Ladeuil
Merge simplify-test-config-building into lockable-config-files resolving conflicts
576
        """Create a config object from a string.
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
577
5345.2.9 by Vincent Ladeuil
Rename IniBaseConfig.from_bytes to from_string.
578
        :param str_or_unicode: A string representing the file content. This will
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
579
            be utf-8 encoded.
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
580
581
        :param file_name: The configuration file path.
582
583
        :param _save: Whether the file should be saved upon creation.
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
584
        """
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
585
        conf = cls(file_name=file_name)
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
586
        conf._create_from_string(str_or_unicode, save)
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
587
        return conf
588
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
589
    def _create_from_string(self, str_or_unicode, save):
5345.5.13 by Vincent Ladeuil
Merge simplify-test-config-building into lockable-config-files resolving conflicts
590
        self._content = StringIO(str_or_unicode.encode('utf-8'))
5345.1.16 by Vincent Ladeuil
Allows tests to save the config file at build time.
591
        # Some tests use in-memory configs, some other always need the config
592
        # file to exist on disk.
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
593
        if save:
5345.1.16 by Vincent Ladeuil
Allows tests to save the config file at build time.
594
            self._write_config_file()
5345.5.12 by Vincent Ladeuil
Fix fallouts from replacing '_content' by 'from_bytes' for config files.
595
5345.1.4 by Vincent Ladeuil
Deprecate the ``file`` parameter of the ``config._get_parser()`` method.
596
    def _get_parser(self, file=symbol_versioning.DEPRECATED_PARAMETER):
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
597
        if self._parser is not None:
598
            return self._parser
5345.1.4 by Vincent Ladeuil
Deprecate the ``file`` parameter of the ``config._get_parser()`` method.
599
        if symbol_versioning.deprecated_passed(file):
600
            symbol_versioning.warn(
601
                'IniBasedConfig._get_parser(file=xxx) was deprecated in 2.3.'
5345.1.5 by Vincent Ladeuil
Fix fallouts by slightly editing the tests. More refactoring avoided to keep the review light.
602
                ' Use IniBasedConfig(_content=xxx) instead.',
5345.1.4 by Vincent Ladeuil
Deprecate the ``file`` parameter of the ``config._get_parser()`` method.
603
                DeprecationWarning,
604
                stacklevel=2)
605
        if self._content is not None:
606
            co_input = self._content
607
        elif self.file_name is None:
608
            raise AssertionError('We have no content to create the config')
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
609
        else:
5345.1.4 by Vincent Ladeuil
Deprecate the ``file`` parameter of the ``config._get_parser()`` method.
610
            co_input = self.file_name
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
611
        try:
5345.1.4 by Vincent Ladeuil
Deprecate the ``file`` parameter of the ``config._get_parser()`` method.
612
            self._parser = ConfigObj(co_input, encoding='utf-8')
1474 by Robert Collins
Merge from Aaron Bentley.
613
        except configobj.ConfigObjError, e:
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
614
            raise errors.ParseConfigError(e.errors, e.config.filename)
5345.5.1 by Vincent Ladeuil
Implement config.reload and make sure we have a file name when using it.
615
        # Make sure self.reload() will use the right file name
5345.1.8 by Vincent Ladeuil
Make the test_listen_to_the_last_speaker pass and fix fallouts.
616
        self._parser.filename = self.file_name
1185.12.49 by Aaron Bentley
Switched to ConfigObj
617
        return self._parser
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
618
5345.5.1 by Vincent Ladeuil
Implement config.reload and make sure we have a file name when using it.
619
    def reload(self):
620
        """Reload the config file from disk."""
621
        if self.file_name is None:
622
            raise AssertionError('We need a file name to reload the config')
623
        if self._parser is not None:
624
            self._parser.reload()
625
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
626
    def _get_matching_sections(self):
627
        """Return an ordered list of (section_name, extra_path) pairs.
628
629
        If the section contains inherited configuration, extra_path is
630
        a string containing the additional path components.
631
        """
632
        section = self._get_section()
633
        if section is not None:
634
            return [(section, '')]
635
        else:
636
            return []
637
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
638
    def _get_section(self):
639
        """Override this to define the section used by the config."""
640
        return "DEFAULT"
641
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
642
    def _get_sections(self, name=None):
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
643
        """Returns an iterator of the sections specified by ``name``.
644
645
        :param name: The section name. If None is supplied, the default
646
            configurations are yielded.
647
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
648
        :return: A tuple (name, section, config_id) for all sections that will
649
            be walked by user_get_option() in the 'right' order. The first one
650
            is where set_user_option() will update the value.
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
651
        """
652
        parser = self._get_parser()
653
        if name is not None:
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
654
            yield (name, parser[name], self.config_id())
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
655
        else:
656
            # No section name has been given so we fallback to the configobj
657
            # itself which holds the variables defined outside of any section.
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
658
            yield (None, parser, self.config_id())
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
659
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
660
    def _get_options(self, sections=None):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
661
        """Return an ordered list of (name, value, section, config_id) tuples.
662
663
        All options are returned with their associated value and the section
664
        they appeared in. ``config_id`` is a unique identifier for the
665
        configuration file the option is defined in.
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
666
667
        :param sections: Default to ``_get_matching_sections`` if not
668
            specified. This gives a better control to daughter classes about
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
669
            which sections should be searched. This is a list of (name,
670
            configobj) tuples.
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
671
        """
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
672
        opts = []
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
673
        if sections is None:
674
            parser = self._get_parser()
675
            sections = []
676
            for (section_name, _) in self._get_matching_sections():
677
                try:
678
                    section = parser[section_name]
679
                except KeyError:
680
                    # This could happen for an empty file for which we define a
681
                    # DEFAULT section. FIXME: Force callers to provide sections
682
                    # instead ? -- vila 20100930
683
                    continue
684
                sections.append((section_name, section))
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
685
        config_id = self.config_id()
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
686
        for (section_name, section) in sections:
687
            for (name, value) in section.iteritems():
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
688
                yield (name, parser._quote(value), section_name,
689
                       config_id, parser)
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
690
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
691
    def _get_option_policy(self, section, option_name):
692
        """Return the policy for the given (section, option_name) pair."""
693
        return POLICY_NONE
694
4603.1.10 by Aaron Bentley
Provide change editor via config.
695
    def _get_change_editor(self):
696
        return self.get_user_option('change_editor')
697
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
698
    def _get_signature_checking(self):
699
        """See Config._get_signature_checking."""
1474 by Robert Collins
Merge from Aaron Bentley.
700
        policy = self._get_user_option('check_signatures')
701
        if policy:
702
            return self._string_to_signature_policy(policy)
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
703
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
704
    def _get_signing_policy(self):
1773.4.3 by Martin Pool
[merge] bzr.dev
705
        """See Config._get_signing_policy"""
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
706
        policy = self._get_user_option('create_signatures')
707
        if policy:
708
            return self._string_to_signing_policy(policy)
709
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
710
    def _get_user_id(self):
711
        """Get the user id from the 'email' key in the current section."""
1474 by Robert Collins
Merge from Aaron Bentley.
712
        return self._get_user_option('email')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
713
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
714
    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.
715
        """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
716
        for (section, extra_path) in self._get_matching_sections():
717
            try:
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
718
                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
719
            except KeyError:
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
720
                continue
721
            policy = self._get_option_policy(section, option_name)
722
            if policy == POLICY_NONE:
723
                return value
724
            elif policy == POLICY_NORECURSE:
725
                # norecurse items only apply to the exact path
726
                if extra_path:
727
                    continue
728
                else:
729
                    return value
730
            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
731
                if extra_path:
732
                    value = urlutils.join(value, extra_path)
733
                return value
2120.6.6 by James Henstridge
fix test_set_push_location test
734
            else:
735
                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
736
        else:
1993.3.1 by James Henstridge
first go at making location config lookup recursive
737
            return None
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
738
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
739
    def _gpg_signing_command(self):
1442.1.56 by Robert Collins
gpg_signing_command configuration item
740
        """See Config.gpg_signing_command."""
1472 by Robert Collins
post commit hook, first pass implementation
741
        return self._get_user_option('gpg_signing_command')
1442.1.56 by Robert Collins
gpg_signing_command configuration item
742
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
743
    def _log_format(self):
744
        """See Config.log_format."""
745
        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
746
1472 by Robert Collins
post commit hook, first pass implementation
747
    def _post_commit(self):
748
        """See Config.post_commit."""
749
        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
750
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
751
    def _string_to_signature_policy(self, signature_string):
752
        """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
753
        if signature_string.lower() == 'check-available':
754
            return CHECK_IF_POSSIBLE
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
755
        if signature_string.lower() == 'ignore':
756
            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
757
        if signature_string.lower() == 'require':
758
            return CHECK_ALWAYS
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
759
        raise errors.BzrError("Invalid signatures policy '%s'"
760
                              % signature_string)
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
761
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
762
    def _string_to_signing_policy(self, signature_string):
763
        """Convert a string to a signing policy."""
764
        if signature_string.lower() == 'when-required':
765
            return SIGN_WHEN_REQUIRED
766
        if signature_string.lower() == 'never':
767
            return SIGN_NEVER
768
        if signature_string.lower() == 'always':
769
            return SIGN_ALWAYS
770
        raise errors.BzrError("Invalid signing policy '%s'"
771
                              % signature_string)
772
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
773
    def _get_alias(self, value):
774
        try:
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
775
            return self._get_parser().get_value("ALIASES",
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
776
                                                value)
777
        except KeyError:
778
            pass
779
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
780
    def _get_nickname(self):
781
        return self.get_user_option('nickname')
782
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
783
    def remove_user_option(self, option_name, section_name=None):
784
        """Remove a user option and save the configuration file.
785
786
        :param option_name: The option to be removed.
787
788
        :param section_name: The section the option is defined in, default to
789
            the default section.
790
        """
791
        self.reload()
792
        parser = self._get_parser()
793
        if section_name is None:
794
            section = parser
795
        else:
796
            section = parser[section_name]
797
        try:
798
            del section[option_name]
799
        except KeyError:
800
            raise errors.NoSuchConfigOption(option_name)
801
        self._write_config_file()
802
4708.2.1 by Martin
Ensure all files opened by bazaar proper are explicitly closed
803
    def _write_config_file(self):
5345.1.1 by Vincent Ladeuil
Deprecate the get_filename parameter in IniBasedConfig.
804
        if self.file_name is None:
805
            raise AssertionError('We cannot save, self.file_name is None')
5345.1.9 by Vincent Ladeuil
Refactor config dir check.
806
        conf_dir = os.path.dirname(self.file_name)
807
        ensure_config_dir_exists(conf_dir)
5345.1.1 by Vincent Ladeuil
Deprecate the get_filename parameter in IniBasedConfig.
808
        atomic_file = atomicfile.AtomicFile(self.file_name)
5050.6.1 by Vincent Ladeuil
Merge 2.1 into 2.2 including fixes for bug #525571 and bug #494221
809
        self._get_parser().write(atomic_file)
810
        atomic_file.commit()
811
        atomic_file.close()
5345.3.3 by Vincent Ladeuil
Merge bzr.dev into deprecate-get-filename resolving conflicts
812
        osutils.copy_ownership_from_path(self.file_name)
4708.2.1 by Martin
Ensure all files opened by bazaar proper are explicitly closed
813
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
814
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
815
class LockableConfig(IniBasedConfig):
816
    """A configuration needing explicit locking for access.
817
818
    If several processes try to write the config file, the accesses need to be
819
    serialized.
5345.5.8 by Vincent Ladeuil
More doc and ensure that the config is locked when _write_config_file is called.
820
821
    Daughter classes should decorate all methods that update a config with the
822
    ``@needs_write_lock`` decorator (they call, directly or indirectly, the
823
    ``_write_config_file()`` method. These methods (typically ``set_option()``
824
    and variants must reload the config file from disk before calling
825
    ``_write_config_file()``), this can be achieved by calling the
826
    ``self.reload()`` method. Note that the lock scope should cover both the
827
    reading and the writing of the config file which is why the decorator can't
828
    be applied to ``_write_config_file()`` only.
829
830
    This should be enough to implement the following logic:
831
    - lock for exclusive write access,
832
    - reload the config file from disk,
833
    - set the new value
834
    - unlock
835
836
    This logic guarantees that a writer can update a value without erasing an
837
    update made by another writer.
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
838
    """
839
5345.5.5 by Vincent Ladeuil
Make bb.test_version.TestVersionUnicodeOutput.test_unicode_bzr_home pass.
840
    lock_name = 'lock'
841
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
842
    def __init__(self, file_name):
843
        super(LockableConfig, self).__init__(file_name=file_name)
5345.5.5 by Vincent Ladeuil
Make bb.test_version.TestVersionUnicodeOutput.test_unicode_bzr_home pass.
844
        self.dir = osutils.dirname(osutils.safe_unicode(self.file_name))
5676.1.4 by Jelmer Vernooij
merge bzr.dev.
845
        # FIXME: It doesn't matter that we don't provide possible_transports
846
        # below since this is currently used only for local config files ;
847
        # local transports are not shared. But if/when we start using
848
        # LockableConfig for other kind of transports, we will need to reuse
849
        # whatever connection is already established -- vila 20100929
5345.5.5 by Vincent Ladeuil
Make bb.test_version.TestVersionUnicodeOutput.test_unicode_bzr_home pass.
850
        self.transport = transport.get_transport(self.dir)
5743.4.9 by Vincent Ladeuil
Implement a LockableConfigObjStore to be able to mimick the actual behaviour.
851
        self._lock = lockdir.LockDir(self.transport, self.lock_name)
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
852
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
853
    def _create_from_string(self, unicode_bytes, save):
854
        super(LockableConfig, self)._create_from_string(unicode_bytes, False)
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
855
        if save:
5345.1.24 by Vincent Ladeuil
Implement _save for LockableConfig too.
856
            # We need to handle the saving here (as opposed to IniBasedConfig)
857
            # to be able to lock
858
            self.lock_write()
859
            self._write_config_file()
860
            self.unlock()
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
861
862
    def lock_write(self, token=None):
5345.5.8 by Vincent Ladeuil
More doc and ensure that the config is locked when _write_config_file is called.
863
        """Takes a write lock in the directory containing the config file.
864
865
        If the directory doesn't exist it is created.
866
        """
5345.5.5 by Vincent Ladeuil
Make bb.test_version.TestVersionUnicodeOutput.test_unicode_bzr_home pass.
867
        ensure_config_dir_exists(self.dir)
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
868
        return self._lock.lock_write(token)
869
870
    def unlock(self):
871
        self._lock.unlock()
872
5345.5.9 by Vincent Ladeuil
Implements 'bzr lock --config <file>'.
873
    def break_lock(self):
874
        self._lock.break_lock()
875
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
876
    @needs_write_lock
877
    def remove_user_option(self, option_name, section_name=None):
878
        super(LockableConfig, self).remove_user_option(option_name,
879
                                                       section_name)
880
5345.5.8 by Vincent Ladeuil
More doc and ensure that the config is locked when _write_config_file is called.
881
    def _write_config_file(self):
882
        if self._lock is None or not self._lock.is_held:
883
            # NB: if the following exception is raised it probably means a
884
            # missing @needs_write_lock decorator on one of the callers.
885
            raise errors.ObjectNotLocked(self)
886
        super(LockableConfig, self)._write_config_file()
887
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
888
889
class GlobalConfig(LockableConfig):
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
890
    """The configuration that should be used for a specific location."""
891
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
892
    def __init__(self):
893
        super(GlobalConfig, self).__init__(file_name=config_filename())
5345.1.1 by Vincent Ladeuil
Deprecate the get_filename parameter in IniBasedConfig.
894
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
895
    def config_id(self):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
896
        return 'bazaar'
897
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
898
    @classmethod
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
899
    def from_string(cls, str_or_unicode, save=False):
5345.5.13 by Vincent Ladeuil
Merge simplify-test-config-building into lockable-config-files resolving conflicts
900
        """Create a config object from a string.
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
901
5345.5.13 by Vincent Ladeuil
Merge simplify-test-config-building into lockable-config-files resolving conflicts
902
        :param str_or_unicode: A string representing the file content. This
903
            will be utf-8 encoded.
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
904
905
        :param save: Whether the file should be saved upon creation.
906
        """
907
        conf = cls()
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
908
        conf._create_from_string(str_or_unicode, save)
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
909
        return conf
5345.5.12 by Vincent Ladeuil
Fix fallouts from replacing '_content' by 'from_bytes' for config files.
910
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
911
    def get_editor(self):
1474 by Robert Collins
Merge from Aaron Bentley.
912
        return self._get_user_option('editor')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
913
5345.5.4 by Vincent Ladeuil
Start implementing config files locking.
914
    @needs_write_lock
1816.2.1 by Robey Pointer
add set_user_option to GlobalConfig, and make /etc/passwd username lookup try harder with encodings
915
    def set_user_option(self, option, value):
916
        """Save option and its value in the configuration."""
2900.3.2 by Tim Penhey
A working alias command.
917
        self._set_option(option, value, 'DEFAULT')
918
919
    def get_aliases(self):
920
        """Return the aliases section."""
921
        if 'ALIASES' in self._get_parser():
922
            return self._get_parser()['ALIASES']
923
        else:
924
            return {}
925
5345.5.8 by Vincent Ladeuil
More doc and ensure that the config is locked when _write_config_file is called.
926
    @needs_write_lock
2900.3.2 by Tim Penhey
A working alias command.
927
    def set_alias(self, alias_name, alias_command):
928
        """Save the alias in the configuration."""
929
        self._set_option(alias_name, alias_command, 'ALIASES')
930
5345.5.8 by Vincent Ladeuil
More doc and ensure that the config is locked when _write_config_file is called.
931
    @needs_write_lock
2900.3.2 by Tim Penhey
A working alias command.
932
    def unset_alias(self, alias_name):
933
        """Unset an existing alias."""
5345.5.10 by Vincent Ladeuil
Add a missing config.reload().
934
        self.reload()
2900.3.2 by Tim Penhey
A working alias command.
935
        aliases = self._get_parser().get('ALIASES')
2900.3.7 by Tim Penhey
Updates from Aaron's review.
936
        if not aliases or alias_name not in aliases:
937
            raise errors.NoSuchAlias(alias_name)
2900.3.2 by Tim Penhey
A working alias command.
938
        del aliases[alias_name]
2900.3.12 by Tim Penhey
Final review comments.
939
        self._write_config_file()
2900.3.2 by Tim Penhey
A working alias command.
940
941
    def _set_option(self, option, value, section):
5345.5.1 by Vincent Ladeuil
Implement config.reload and make sure we have a file name when using it.
942
        self.reload()
2900.3.7 by Tim Penhey
Updates from Aaron's review.
943
        self._get_parser().setdefault(section, {})[option] = value
2900.3.12 by Tim Penhey
Final review comments.
944
        self._write_config_file()
2900.3.2 by Tim Penhey
A working alias command.
945
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
946
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
947
    def _get_sections(self, name=None):
948
        """See IniBasedConfig._get_sections()."""
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
949
        parser = self._get_parser()
950
        # We don't give access to options defined outside of any section, we
951
        # used the DEFAULT section by... default.
952
        if name in (None, 'DEFAULT'):
953
            # This could happen for an empty file where the DEFAULT section
954
            # doesn't exist yet. So we force DEFAULT when yielding
955
            name = 'DEFAULT'
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
956
            if 'DEFAULT' not in parser:
957
               parser['DEFAULT']= {}
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
958
        yield (name, parser[name], self.config_id())
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
959
960
    @needs_write_lock
961
    def remove_user_option(self, option_name, section_name=None):
962
        if section_name is None:
963
            # We need to force the default section.
964
            section_name = 'DEFAULT'
965
        # We need to avoid the LockableConfig implementation or we'll lock
966
        # twice
967
        super(LockableConfig, self).remove_user_option(option_name,
968
                                                       section_name)
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
969
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
970
def _iter_for_location_by_parts(sections, location):
5764.1.3 by Vincent Ladeuil
Add a doctrsing and address the location being split for all iterations by making letting the function iterate over all sections.
971
    """Keep only the sessions matching the specified location.
972
973
    :param sections: An iterable of section names.
974
975
    :param location: An url or a local path to match against.
976
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
977
    :returns: An iterator of (section, extra_path, nb_parts) where nb is the
978
        number of path components in the section name, section is the section
979
        name and extra_path is the difference between location and the section
980
        name.
5743.6.19 by Vincent Ladeuil
Clarify comments about section names for Location-related objects (also fix LocationMatcher and add tests).
981
982
    ``location`` will always be a local path and never a 'file://' url but the
983
    section names themselves can be in either form.
5764.1.3 by Vincent Ladeuil
Add a doctrsing and address the location being split for all iterations by making letting the function iterate over all sections.
984
    """
5764.1.2 by Vincent Ladeuil
This put a common processing into the loop to avoid bad inputs. The
985
    location_parts = location.rstrip('/').split('/')
986
5764.1.3 by Vincent Ladeuil
Add a doctrsing and address the location being split for all iterations by making letting the function iterate over all sections.
987
    for section in sections:
5743.6.19 by Vincent Ladeuil
Clarify comments about section names for Location-related objects (also fix LocationMatcher and add tests).
988
        # location is a local path if possible, so we need to convert 'file://'
989
        # urls in section names to local paths if necessary.
5764.1.3 by Vincent Ladeuil
Add a doctrsing and address the location being split for all iterations by making letting the function iterate over all sections.
990
991
        # This also avoids having file:///path be a more exact
992
        # match than '/path'.
993
5743.6.19 by Vincent Ladeuil
Clarify comments about section names for Location-related objects (also fix LocationMatcher and add tests).
994
        # FIXME: This still raises an issue if a user defines both file:///path
995
        # *and* /path. Should we raise an error in this case -- vila 20110505
996
5764.1.3 by Vincent Ladeuil
Add a doctrsing and address the location being split for all iterations by making letting the function iterate over all sections.
997
        if section.startswith('file://'):
998
            section_path = urlutils.local_path_from_url(section)
999
        else:
1000
            section_path = section
1001
        section_parts = section_path.rstrip('/').split('/')
1002
1003
        matched = True
1004
        if len(section_parts) > len(location_parts):
1005
            # More path components in the section, they can't match
1006
            matched = False
1007
        else:
1008
            # Rely on zip truncating in length to the length of the shortest
1009
            # argument sequence.
1010
            names = zip(location_parts, section_parts)
1011
            for name in names:
1012
                if not fnmatch.fnmatch(name[0], name[1]):
1013
                    matched = False
1014
                    break
1015
        if not matched:
1016
            continue
5764.1.1 by Vincent Ladeuil
Extract _match_section_by_parts from LocationConfig._get_matching_sections and more comments to explain the behavior.
1017
        # build the path difference between the section and the location
5764.1.3 by Vincent Ladeuil
Add a doctrsing and address the location being split for all iterations by making letting the function iterate over all sections.
1018
        extra_path = '/'.join(location_parts[len(section_parts):])
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
1019
        yield section, extra_path, len(section_parts)
5764.1.1 by Vincent Ladeuil
Extract _match_section_by_parts from LocationConfig._get_matching_sections and more comments to explain the behavior.
1020
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1021
5345.5.7 by Vincent Ladeuil
Make LocationConfig use a lock too.
1022
class LocationConfig(LockableConfig):
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1023
    """A configuration object that gives the policy for a location."""
1024
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1025
    def __init__(self, location):
5345.1.2 by Vincent Ladeuil
Get rid of 'branches.conf' references.
1026
        super(LocationConfig, self).__init__(
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1027
            file_name=locations_config_filename())
1878.1.1 by John Arbash Meinel
Entries in locations.conf should prefer local paths if available (bug #53653)
1028
        # local file locations are looked up by local path, rather than
1029
        # by file url. This is because the config file is a user
1030
        # file, and we would rather not expose the user to file urls.
1031
        if location.startswith('file://'):
1032
            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
1033
        self.location = location
1034
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1035
    def config_id(self):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1036
        return 'locations'
1037
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1038
    @classmethod
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
1039
    def from_string(cls, str_or_unicode, location, save=False):
1040
        """Create a config object from a string.
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1041
5345.2.9 by Vincent Ladeuil
Rename IniBaseConfig.from_bytes to from_string.
1042
        :param str_or_unicode: A string representing the file content. This will
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1043
            be utf-8 encoded.
1044
1045
        :param location: The location url to filter the configuration.
5345.1.25 by Vincent Ladeuil
Move the '_save' parameter from '__init__' to 'from_bytes', fix fallouts.
1046
1047
        :param save: Whether the file should be saved upon creation.
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1048
        """
1049
        conf = cls(location)
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
1050
        conf._create_from_string(str_or_unicode, save)
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1051
        return conf
1052
1993.3.1 by James Henstridge
first go at making location config lookup recursive
1053
    def _get_matching_sections(self):
1054
        """Return an ordered list of section names matching this location."""
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
1055
        matches = list(_iter_for_location_by_parts(self._get_parser(),
1056
                                                   self.location))
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
1057
        # put the longest (aka more specific) locations first
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
1058
        matches.sort(
1059
            key=lambda (section, extra_path, length): (length, section),
1060
            reverse=True)
1061
        for (section, extra_path, length) in matches:
1062
            yield section, extra_path
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
1063
            # should we stop looking for parent configs here?
1993.3.1 by James Henstridge
first go at making location config lookup recursive
1064
            try:
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
1065
                if self._get_parser()[section].as_bool('ignore_parents'):
1066
                    break
1993.3.1 by James Henstridge
first go at making location config lookup recursive
1067
            except KeyError:
1068
                pass
1442.1.9 by Robert Collins
exact section test passes
1069
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
1070
    def _get_sections(self, name=None):
1071
        """See IniBasedConfig._get_sections()."""
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1072
        # We ignore the name here as the only sections handled are named with
1073
        # the location path and we don't expose embedded sections either.
1074
        parser = self._get_parser()
1075
        for name, extra_path in self._get_matching_sections():
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1076
            yield (name, parser[name], self.config_id())
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1077
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
1078
    def _get_option_policy(self, section, option_name):
1079
        """Return the policy for the given (section, option_name) pair."""
1080
        # check for the old 'recurse=False' flag
1081
        try:
1082
            recurse = self._get_parser()[section].as_bool('recurse')
1083
        except KeyError:
1084
            recurse = True
1085
        if not recurse:
1086
            return POLICY_NORECURSE
1087
2120.6.10 by James Henstridge
Catch another deprecation warning, and more cleanup
1088
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
1089
        try:
1090
            policy_name = self._get_parser()[section][policy_key]
1091
        except KeyError:
1092
            policy_name = None
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
1093
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
1094
        return _policy_value[policy_name]
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
1095
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1096
    def _set_option_policy(self, section, option_name, option_policy):
1097
        """Set the policy for the given option name in the given section."""
1098
        # The old recurse=False option affects all options in the
1099
        # section.  To handle multiple policies in the section, we
1100
        # need to convert it to a policy_norecurse key.
1101
        try:
1102
            recurse = self._get_parser()[section].as_bool('recurse')
1103
        except KeyError:
1104
            pass
1105
        else:
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1106
            symbol_versioning.warn(
2120.6.11 by James Henstridge
s/0.13/0.14/ in deprecation warning
1107
                'The recurse option is deprecated as of 0.14.  '
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1108
                'The section "%s" has been converted to use policies.'
1109
                % section,
1110
                DeprecationWarning)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1111
            del self._get_parser()[section]['recurse']
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1112
            if not recurse:
1113
                for key in self._get_parser()[section].keys():
1114
                    if not key.endswith(':policy'):
1115
                        self._get_parser()[section][key +
1116
                                                    ':policy'] = 'norecurse'
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1117
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1118
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
1119
        policy_name = _policy_name[option_policy]
1120
        if policy_name is not None:
1121
            self._get_parser()[section][policy_key] = policy_name
1122
        else:
1123
            if policy_key in self._get_parser()[section]:
1124
                del self._get_parser()[section][policy_key]
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1125
5345.5.7 by Vincent Ladeuil
Make LocationConfig use a lock too.
1126
    @needs_write_lock
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1127
    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.
1128
        """Save option and its value in the configuration."""
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
1129
        if store not in [STORE_LOCATION,
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1130
                         STORE_LOCATION_NORECURSE,
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
1131
                         STORE_LOCATION_APPENDPATH]:
1132
            raise ValueError('bad storage policy %r for %r' %
1133
                (store, option))
5345.5.1 by Vincent Ladeuil
Implement config.reload and make sure we have a file name when using it.
1134
        self.reload()
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
1135
        location = self.location
1136
        if location.endswith('/'):
1137
            location = location[:-1]
5345.1.24 by Vincent Ladeuil
Implement _save for LockableConfig too.
1138
        parser = self._get_parser()
5345.1.21 by Vincent Ladeuil
Slight rewrite to make the method more readable.
1139
        if not location in parser and not location + '/' in parser:
1140
            parser[location] = {}
1141
        elif location + '/' in parser:
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
1142
            location = location + '/'
5345.1.21 by Vincent Ladeuil
Slight rewrite to make the method more readable.
1143
        parser[location][option]=value
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1144
        # the allowed values of store match the config policies
1145
        self._set_option_policy(location, option, store)
4708.2.1 by Martin
Ensure all files opened by bazaar proper are explicitly closed
1146
        self._write_config_file()
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
1147
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1148
1149
class BranchConfig(Config):
1150
    """A configuration object giving the policy for a branch."""
1151
5345.1.3 by Vincent Ladeuil
Make __init__ the first method in the BranchConfig class.
1152
    def __init__(self, branch):
1153
        super(BranchConfig, self).__init__()
1154
        self._location_config = None
1155
        self._branch_data_config = None
1156
        self._global_config = None
1157
        self.branch = branch
1158
        self.option_sources = (self._get_location_config,
1159
                               self._get_branch_data_config,
1160
                               self._get_global_config)
1161
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1162
    def config_id(self):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1163
        return 'branch'
1164
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1165
    def _get_branch_data_config(self):
1166
        if self._branch_data_config is None:
1167
            self._branch_data_config = TreeConfig(self.branch)
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1168
            self._branch_data_config.config_id = self.config_id
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1169
        return self._branch_data_config
1170
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1171
    def _get_location_config(self):
1172
        if self._location_config is None:
1173
            self._location_config = LocationConfig(self.branch.base)
1174
        return self._location_config
1175
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1176
    def _get_global_config(self):
1177
        if self._global_config is None:
1178
            self._global_config = GlobalConfig()
1179
        return self._global_config
1180
1181
    def _get_best_value(self, option_name):
1182
        """This returns a user option from local, tree or global config.
1183
1184
        They are tried in that order.  Use get_safe_value if trusted values
1185
        are necessary.
1186
        """
1187
        for source in self.option_sources:
1188
            value = getattr(source(), option_name)()
1189
            if value is not None:
1190
                return value
1191
        return None
1192
1193
    def _get_safe_value(self, option_name):
1194
        """This variant of get_best_value never returns untrusted values.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1195
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1196
        It does not return values from the branch data, because the branch may
1197
        not be controlled by the user.
1198
1199
        We may wish to allow locations.conf to control whether branches are
1200
        trusted in the future.
1201
        """
1202
        for source in (self._get_location_config, self._get_global_config):
1203
            value = getattr(source(), option_name)()
1204
            if value is not None:
1205
                return value
1206
        return None
1207
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1208
    def _get_user_id(self):
1209
        """Return the full user id for the branch.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1210
3407.2.14 by Martin Pool
Remove more cases of getting transport via control_files
1211
        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
1212
        This is looked up in the email controlfile for the branch.
1213
        """
1214
        try:
3407.2.16 by Martin Pool
Remove RemoteBranch reliance on control_files._transport
1215
            return (self.branch._transport.get_bytes("email")
3224.5.4 by Andrew Bennetts
Fix test suite, mainly weeding out uses of bzrlib.user_encoding.
1216
                    .decode(osutils.get_user_encoding())
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1217
                    .rstrip("\r\n"))
1218
        except errors.NoSuchFile, e:
1219
            pass
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1220
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1221
        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
1222
4603.1.10 by Aaron Bentley
Provide change editor via config.
1223
    def _get_change_editor(self):
1224
        return self._get_best_value('_get_change_editor')
1225
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
1226
    def _get_signature_checking(self):
1227
        """See Config._get_signature_checking."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1228
        return self._get_best_value('_get_signature_checking')
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
1229
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
1230
    def _get_signing_policy(self):
1231
        """See Config._get_signing_policy."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1232
        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
1233
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
1234
    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.
1235
        """See Config._get_user_option."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1236
        for source in self.option_sources:
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
1237
            value = source()._get_user_option(option_name)
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1238
            if value is not None:
1239
                return value
1240
        return None
1241
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
1242
    def _get_sections(self, name=None):
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1243
        """See IniBasedConfig.get_sections()."""
1244
        for source in self.option_sources:
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
1245
            for section in source()._get_sections(name):
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1246
                yield section
1247
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
1248
    def _get_options(self, sections=None):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1249
        opts = []
1250
        # First the locations options
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
1251
        for option in self._get_location_config()._get_options():
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1252
            yield option
1253
        # Then the branch options
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
1254
        branch_config = self._get_branch_data_config()
1255
        if sections is None:
1256
            sections = [('DEFAULT', branch_config._get_parser())]
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1257
        # FIXME: We shouldn't have to duplicate the code in IniBasedConfig but
1258
        # Config itself has no notion of sections :( -- vila 20101001
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1259
        config_id = self.config_id()
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1260
        for (section_name, section) in sections:
1261
            for (name, value) in section.iteritems():
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
1262
                yield (name, value, section_name,
1263
                       config_id, branch_config._get_parser())
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1264
        # Then the global options
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
1265
        for option in self._get_global_config()._get_options():
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1266
            yield option
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
1267
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
1268
    def set_user_option(self, name, value, store=STORE_BRANCH,
1269
        warn_masked=False):
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1270
        if store == STORE_BRANCH:
1770.2.6 by Aaron Bentley
Ensure branch.conf works properly
1271
            self._get_branch_data_config().set_option(value, name)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1272
        elif store == STORE_GLOBAL:
2120.6.7 by James Henstridge
Fix GlobalConfig.set_user_option() call
1273
            self._get_global_config().set_user_option(name, value)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1274
        else:
1275
            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)
1276
        if not warn_masked:
1277
            return
1278
        if store in (STORE_GLOBAL, STORE_BRANCH):
1279
            mask_value = self._get_location_config().get_user_option(name)
1280
            if mask_value is not None:
1281
                trace.warning('Value "%s" is masked by "%s" from'
1282
                              ' locations.conf', value, mask_value)
1283
            else:
1284
                if store == STORE_GLOBAL:
1285
                    branch_config = self._get_branch_data_config()
1286
                    mask_value = branch_config.get_user_option(name)
1287
                    if mask_value is not None:
1288
                        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
1289
                                      ' branch.conf', value, mask_value)
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
1290
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
1291
    def remove_user_option(self, option_name, section_name=None):
1292
        self._get_branch_data_config().remove_option(option_name, section_name)
1293
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
1294
    def _gpg_signing_command(self):
1442.1.56 by Robert Collins
gpg_signing_command configuration item
1295
        """See Config.gpg_signing_command."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1296
        return self._get_safe_value('_gpg_signing_command')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1297
1472 by Robert Collins
post commit hook, first pass implementation
1298
    def _post_commit(self):
1299
        """See Config.post_commit."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1300
        return self._get_safe_value('_post_commit')
1472 by Robert Collins
post commit hook, first pass implementation
1301
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
1302
    def _get_nickname(self):
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
1303
        value = self._get_explicit_nickname()
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
1304
        if value is not None:
1305
            return value
2120.5.2 by Alexander Belchenko
(jam) Fix for bug #66857
1306
        return urlutils.unescape(self.branch.base.split('/')[-2])
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
1307
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
1308
    def has_explicit_nickname(self):
1309
        """Return true if a nickname has been explicitly assigned."""
1310
        return self._get_explicit_nickname() is not None
1311
1312
    def _get_explicit_nickname(self):
1313
        return self._get_best_value('_get_nickname')
1314
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
1315
    def _log_format(self):
1316
        """See Config.log_format."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1317
        return self._get_best_value('_log_format')
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
1318
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
1319
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
1320
def ensure_config_dir_exists(path=None):
5519.4.4 by Neil Martinsen-Burrell
restore ensure_config_dir since XDG_CONFIG_HOME is optional
1321
    """Make sure a configuration directory exists.
1322
    This makes sure that the directory exists.
1323
    On windows, since configuration directories are 2 levels deep,
1324
    it makes sure both the directory and the parent directory exists.
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
1325
    """
1326
    if path is None:
1327
        path = config_dir()
1328
    if not os.path.isdir(path):
5519.4.4 by Neil Martinsen-Burrell
restore ensure_config_dir since XDG_CONFIG_HOME is optional
1329
        if sys.platform == 'win32':
1330
            parent_dir = os.path.dirname(path)
1331
            if not os.path.isdir(parent_dir):
1332
                trace.mutter('creating config parent directory: %r', parent_dir)
1333
                os.mkdir(parent_dir)
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1334
        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
1335
        os.mkdir(path)
5116.2.6 by Parth Malwankar
renamed copy_ownership to copy_ownership_from_path.
1336
        osutils.copy_ownership_from_path(path)
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
1337
1532 by Robert Collins
Merge in John Meinels integration branch.
1338
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
1339
def config_dir():
1340
    """Return per-user configuration directory.
1341
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1342
    By default this is %APPDATA%/bazaar/2.0 on Windows, ~/.bazaar on Mac OS X
5519.4.3 by Neil Martinsen-Burrell
be permissive about using $XDG_CONFIG_HOME/bazaar, but dont complain
1343
    and Linux.  On Linux, if there is a $XDG_CONFIG_HOME/bazaar directory,
1344
    that will be used instead.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1345
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
1346
    TODO: Global option --config-dir to override this.
1347
    """
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1348
    base = os.environ.get('BZR_HOME', None)
1349
    if sys.platform == 'win32':
5598.2.2 by John Arbash Meinel
Change the comment slightly
1350
        # environ variables on Windows are in user encoding/mbcs. So decode
1351
        # before using one
5598.2.1 by John Arbash Meinel
Decode windows env vars using mbcs rather than assuming the 8-bit string is ok.
1352
        if base is not None:
1353
            base = base.decode('mbcs')
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1354
        if base is None:
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
1355
            base = win32utils.get_appdata_location_unicode()
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1356
        if base is None:
1357
            base = os.environ.get('HOME', None)
5598.2.1 by John Arbash Meinel
Decode windows env vars using mbcs rather than assuming the 8-bit string is ok.
1358
            if base is not None:
1359
                base = base.decode('mbcs')
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1360
        if base is None:
2991.2.2 by Vincent Ladeuil
No tests worth adding after upgrading to configobj-4.4.0.
1361
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
1362
                                  ' or HOME set')
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1363
        return osutils.pathjoin(base, 'bazaar', '2.0')
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1364
    elif sys.platform == 'darwin':
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1365
        if base is None:
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1366
            # this takes into account $HOME
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1367
            base = os.path.expanduser("~")
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1368
        return osutils.pathjoin(base, '.bazaar')
1369
    else:
1370
        if base is None:
5519.4.3 by Neil Martinsen-Burrell
be permissive about using $XDG_CONFIG_HOME/bazaar, but dont complain
1371
1372
            xdg_dir = os.environ.get('XDG_CONFIG_HOME', None)
1373
            if xdg_dir is None:
1374
                xdg_dir = osutils.pathjoin(os.path.expanduser("~"), ".config")
1375
            xdg_dir = osutils.pathjoin(xdg_dir, 'bazaar')
1376
            if osutils.isdir(xdg_dir):
1377
                trace.mutter(
1378
                    "Using configuration in XDG directory %s." % xdg_dir)
1379
                return xdg_dir
1380
1381
            base = os.path.expanduser("~")
5519.4.4 by Neil Martinsen-Burrell
restore ensure_config_dir since XDG_CONFIG_HOME is optional
1382
        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 \
1383
1384
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
1385
def config_filename():
1386
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1387
    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.
1388
1389
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
1390
def locations_config_filename():
1391
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1392
    return osutils.pathjoin(config_dir(), 'locations.conf')
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
1393
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
1394
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1395
def authentication_config_filename():
1396
    """Return per-user authentication ini file filename."""
1397
    return osutils.pathjoin(config_dir(), 'authentication.conf')
1398
1399
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
1400
def user_ignore_config_filename():
1401
    """Return the user default ignore filename"""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1402
    return osutils.pathjoin(config_dir(), 'ignore')
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
1403
1404
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1405
def crash_dir():
1406
    """Return the directory name to store crash files.
1407
1408
    This doesn't implicitly create it.
1409
4634.128.2 by Martin Pool
Write crash files into /var/crash where apport can see them.
1410
    On Windows it's in the config directory; elsewhere it's /var/crash
4634.128.18 by Martin Pool
Update apport crash tests
1411
    which may be monitored by apport.  It can be overridden by
1412
    $APPORT_CRASH_DIR.
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1413
    """
1414
    if sys.platform == 'win32':
1415
        return osutils.pathjoin(config_dir(), 'Crash')
1416
    else:
4634.128.2 by Martin Pool
Write crash files into /var/crash where apport can see them.
1417
        # XXX: hardcoded in apport_python_hook.py; therefore here too -- mbp
1418
        # 2010-01-31
4634.128.18 by Martin Pool
Update apport crash tests
1419
        return os.environ.get('APPORT_CRASH_DIR', '/var/crash')
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1420
1421
1422
def xdg_cache_dir():
4584.3.23 by Martin Pool
Correction to xdg_cache_dir and add a simple test
1423
    # See http://standards.freedesktop.org/basedir-spec/latest/ar01s03.html
1424
    # Possibly this should be different on Windows?
1425
    e = os.environ.get('XDG_CACHE_DIR', None)
1426
    if e:
1427
        return e
1428
    else:
1429
        return os.path.expanduser('~/.cache')
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1430
1431
5050.72.1 by Martin Pool
Set email address from /etc/mailname if possible
1432
def _get_default_mail_domain():
1433
    """If possible, return the assumed default email domain.
1434
1435
    :returns: string mail domain, or None.
1436
    """
1437
    if sys.platform == 'win32':
1438
        # No implementation yet; patches welcome
1439
        return None
1440
    try:
1441
        f = open('/etc/mailname')
1442
    except (IOError, OSError), e:
1443
        return None
1444
    try:
1445
        domain = f.read().strip()
1446
        return domain
1447
    finally:
1448
        f.close()
1449
1450
1451
def _auto_user_id():
1452
    """Calculate automatic user identification.
1453
1454
    :returns: (realname, email), either of which may be None if they can't be
1455
    determined.
1456
1457
    Only used when none is set in the environment or the id file.
1458
1459
    This only returns an email address if we can be fairly sure the 
1460
    address is reasonable, ie if /etc/mailname is set on unix.
1461
1462
    This doesn't use the FQDN as the default domain because that may be 
1463
    slow, and it doesn't use the hostname alone because that's not normally 
1464
    a reasonable address.
1465
    """
1466
    if sys.platform == 'win32':
1467
        # No implementation to reliably determine Windows default mail
1468
        # address; please add one.
1469
        return None, None
1470
1471
    default_mail_domain = _get_default_mail_domain()
1472
    if not default_mail_domain:
1473
        return None, None
1474
1475
    import pwd
1476
    uid = os.getuid()
1477
    try:
1478
        w = pwd.getpwuid(uid)
1479
    except KeyError:
1480
        mutter('no passwd entry for uid %d?' % uid)
1481
        return None, None
1482
1483
    # we try utf-8 first, because on many variants (like Linux),
1484
    # /etc/passwd "should" be in utf-8, and because it's unlikely to give
1485
    # false positives.  (many users will have their user encoding set to
1486
    # latin-1, which cannot raise UnicodeError.)
1487
    try:
1488
        gecos = w.pw_gecos.decode('utf-8')
1489
        encoding = 'utf-8'
1490
    except UnicodeError:
1491
        try:
1492
            encoding = osutils.get_user_encoding()
1493
            gecos = w.pw_gecos.decode(encoding)
1494
        except UnicodeError, e:
1495
            mutter("cannot decode passwd entry %s" % w)
1496
            return None, None
1497
    try:
1498
        username = w.pw_name.decode(encoding)
1499
    except UnicodeError, e:
1500
        mutter("cannot decode passwd entry %s" % w)
1501
        return None, None
1502
1503
    comma = gecos.find(',')
1504
    if comma == -1:
1505
        realname = gecos
1506
    else:
1507
        realname = gecos[:comma]
1508
1509
    return realname, (username + '@' + default_mail_domain)
1510
1511
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
1512
def parse_username(username):
1513
    """Parse e-mail username and return a (name, address) tuple."""
1514
    match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
1515
    if match is None:
1516
        return (username, '')
1517
    else:
1518
        return (match.group(1), match.group(2))
1519
1520
1185.16.52 by Martin Pool
- add extract_email_address
1521
def extract_email_address(e):
1522
    """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.
1523
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1524
    That is just the user@domain part, nothing else.
1185.16.52 by Martin Pool
- add extract_email_address
1525
    This part is required to contain only ascii characters.
1526
    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.
1527
1185.16.52 by Martin Pool
- add extract_email_address
1528
    >>> extract_email_address('Jane Tester <jane@test.com>')
1529
    "jane@test.com"
1530
    """
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
1531
    name, email = parse_username(e)
1532
    if not email:
2055.2.2 by John Arbash Meinel
Switch extract_email_address() to use a more specific exception
1533
        raise errors.NoEmailInUsername(e)
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
1534
    return email
1185.35.11 by Aaron Bentley
Added support for branch nicks
1535
1185.85.30 by John Arbash Meinel
Fixing 'bzr push' exposed that IniBasedConfig didn't handle unicode.
1536
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1537
class TreeConfig(IniBasedConfig):
1185.35.11 by Aaron Bentley
Added support for branch nicks
1538
    """Branch configuration data associated with its contents, not location"""
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1539
3408.3.1 by Martin Pool
Remove erroneous handling of branch.conf for RemoteBranch
1540
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
1541
1185.35.11 by Aaron Bentley
Added support for branch nicks
1542
    def __init__(self, branch):
4226.1.5 by Robert Collins
Reinstate the use of the Branch.get_config_file verb.
1543
        self._config = branch._get_config()
1185.35.11 by Aaron Bentley
Added support for branch nicks
1544
        self.branch = branch
1545
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1546
    def _get_parser(self, file=None):
1547
        if file is not None:
1548
            return IniBasedConfig._get_parser(file)
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1549
        return self._config._get_configobj()
1185.35.11 by Aaron Bentley
Added support for branch nicks
1550
1551
    def get_option(self, name, section=None, default=None):
1552
        self.branch.lock_read()
1553
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1554
            return self._config.get_option(name, section, default)
1185.35.11 by Aaron Bentley
Added support for branch nicks
1555
        finally:
1556
            self.branch.unlock()
1557
1558
    def set_option(self, value, name, section=None):
1559
        """Set a per-branch configuration option"""
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
1560
        # FIXME: We shouldn't need to lock explicitly here but rather rely on
1561
        # higher levels providing the right lock -- vila 20101004
1185.35.11 by Aaron Bentley
Added support for branch nicks
1562
        self.branch.lock_write()
1563
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1564
            self._config.set_option(value, name, section)
1185.35.11 by Aaron Bentley
Added support for branch nicks
1565
        finally:
1566
            self.branch.unlock()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1567
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
1568
    def remove_option(self, option_name, section_name=None):
1569
        # FIXME: We shouldn't need to lock explicitly here but rather rely on
1570
        # higher levels providing the right lock -- vila 20101004
1571
        self.branch.lock_write()
1572
        try:
1573
            self._config.remove_option(option_name, section_name)
1574
        finally:
1575
            self.branch.unlock()
1576
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1577
1578
class AuthenticationConfig(object):
1579
    """The authentication configuration file based on a ini file.
1580
1581
    Implements the authentication.conf file described in
1582
    doc/developers/authentication-ring.txt.
1583
    """
1584
1585
    def __init__(self, _file=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1586
        self._config = None # The ConfigObj
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1587
        if _file is None:
2900.2.24 by Vincent Ladeuil
Review feedback.
1588
            self._filename = authentication_config_filename()
1589
            self._input = self._filename = authentication_config_filename()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1590
        else:
2900.2.24 by Vincent Ladeuil
Review feedback.
1591
            # Tests can provide a string as _file
1592
            self._filename = None
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1593
            self._input = _file
1594
1595
    def _get_config(self):
1596
        if self._config is not None:
1597
            return self._config
1598
        try:
2900.2.22 by Vincent Ladeuil
Polishing.
1599
            # FIXME: Should we validate something here ? Includes: empty
1600
            # sections are useless, at least one of
1601
            # user/password/password_encoding should be defined, etc.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1602
1603
            # Note: the encoding below declares that the file itself is utf-8
1604
            # encoded, but the values in the ConfigObj are always Unicode.
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1605
            self._config = ConfigObj(self._input, encoding='utf-8')
1606
        except configobj.ConfigObjError, e:
1607
            raise errors.ParseConfigError(e.errors, e.config.filename)
1608
        return self._config
1609
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1610
    def _save(self):
1611
        """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.
1612
        conf_dir = os.path.dirname(self._filename)
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1613
        ensure_config_dir_exists(conf_dir)
4708.2.2 by Martin
Workingtree changes sitting around since November, more explict closing of files in bzrlib
1614
        f = file(self._filename, 'wb')
1615
        try:
1616
            self._get_config().write(f)
1617
        finally:
1618
            f.close()
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1619
1620
    def _set_option(self, section_name, option_name, value):
1621
        """Set an authentication configuration option"""
1622
        conf = self._get_config()
1623
        section = conf.get(section_name)
1624
        if section is None:
1625
            conf[section] = {}
1626
            section = conf[section]
1627
        section[option_name] = value
1628
        self._save()
1629
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1630
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
1631
                        realm=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1632
        """Returns the matching credentials from authentication.conf file.
1633
1634
        :param scheme: protocol
1635
1636
        :param host: the server address
1637
1638
        :param port: the associated port (optional)
1639
1640
        :param user: login (optional)
1641
1642
        :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
1643
        
1644
        :param realm: the http authentication realm (optional)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1645
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1646
        :return: A dict containing the matching credentials or None.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1647
           This includes:
1648
           - name: the section name of the credentials in the
1649
             authentication.conf file,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1650
           - 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.
1651
           - scheme: the server protocol,
1652
           - host: the server address,
1653
           - port: the server port (can be None),
1654
           - path: the absolute server path (can be None),
1655
           - realm: the http specific authentication realm (can be None),
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1656
           - password: the decoded password, could be None if the credential
1657
             defines only the user
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1658
           - verify_certificates: https specific, True if the server
1659
             certificate should be verified, False otherwise.
1660
        """
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1661
        credentials = None
1662
        for auth_def_name, auth_def in self._get_config().items():
3418.2.1 by Vincent Ladeuil
Fix #217650 by catching declarations outside sections.
1663
            if type(auth_def) is not configobj.Section:
1664
                raise ValueError("%s defined outside a section" % auth_def_name)
1665
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1666
            a_scheme, a_host, a_user, a_path = map(
1667
                auth_def.get, ['scheme', 'host', 'user', 'path'])
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1668
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1669
            try:
1670
                a_port = auth_def.as_int('port')
1671
            except KeyError:
1672
                a_port = None
2900.2.22 by Vincent Ladeuil
Polishing.
1673
            except ValueError:
1674
                raise ValueError("'port' not numeric in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1675
            try:
1676
                a_verify_certificates = auth_def.as_bool('verify_certificates')
1677
            except KeyError:
1678
                a_verify_certificates = True
2900.2.22 by Vincent Ladeuil
Polishing.
1679
            except ValueError:
1680
                raise ValueError(
1681
                    "'verify_certificates' not boolean in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1682
1683
            # Attempt matching
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1684
            if a_scheme is not None and scheme != a_scheme:
1685
                continue
1686
            if a_host is not None:
1687
                if not (host == a_host
1688
                        or (a_host.startswith('.') and host.endswith(a_host))):
1689
                    continue
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1690
            if a_port is not None and port != a_port:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1691
                continue
1692
            if (a_path is not None and path is not None
1693
                and not path.startswith(a_path)):
1694
                continue
1695
            if (a_user is not None and user is not None
1696
                and a_user != user):
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1697
                # Never contradict the caller about the user to be used
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1698
                continue
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1699
            if a_user is None:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1700
                # Can't find a user
1701
                continue
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1702
            # Prepare a credentials dictionary with additional keys
1703
            # for the credential providers
2900.2.24 by Vincent Ladeuil
Review feedback.
1704
            credentials = dict(name=auth_def_name,
3418.4.2 by Vincent Ladeuil
Fix bug #199440 by taking into account that a section may not
1705
                               user=a_user,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1706
                               scheme=a_scheme,
1707
                               host=host,
1708
                               port=port,
1709
                               path=path,
1710
                               realm=realm,
3418.4.2 by Vincent Ladeuil
Fix bug #199440 by taking into account that a section may not
1711
                               password=auth_def.get('password', None),
2900.2.24 by Vincent Ladeuil
Review feedback.
1712
                               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
1713
            # Decode the password in the credentials (or get one)
2900.2.22 by Vincent Ladeuil
Polishing.
1714
            self.decode_password(credentials,
1715
                                 auth_def.get('password_encoding', None))
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1716
            if 'auth' in debug.debug_flags:
1717
                trace.mutter("Using authentication section: %r", auth_def_name)
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1718
            break
1719
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1720
        if credentials is None:
1721
            # No credentials were found in authentication.conf, try the fallback
1722
            # credentials stores.
1723
            credentials = credential_store_registry.get_fallback_credentials(
1724
                scheme, host, port, user, path, realm)
1725
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1726
        return credentials
1727
3777.3.2 by Aaron Bentley
Reverse order of scheme and password
1728
    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
1729
                        port=None, path=None, verify_certificates=None,
1730
                        realm=None):
3777.3.1 by Aaron Bentley
Update docs
1731
        """Set authentication credentials for a host.
1732
1733
        Any existing credentials with matching scheme, host, port and path
1734
        will be deleted, regardless of name.
1735
1736
        :param name: An arbitrary name to describe this set of credentials.
1737
        :param host: Name of the host that accepts these credentials.
1738
        :param user: The username portion of these credentials.
1739
        :param scheme: The URL scheme (e.g. ssh, http) the credentials apply
1740
            to.
3777.3.2 by Aaron Bentley
Reverse order of scheme and password
1741
        :param password: Password portion of these credentials.
3777.3.1 by Aaron Bentley
Update docs
1742
        :param port: The IP port on the host that these credentials apply to.
1743
        :param path: A filesystem path on the host that these credentials
1744
            apply to.
1745
        :param verify_certificates: On https, verify server certificates if
1746
            True.
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1747
        :param realm: The http authentication realm (optional).
3777.3.1 by Aaron Bentley
Update docs
1748
        """
3777.1.8 by Aaron Bentley
Commit work-in-progress
1749
        values = {'host': host, 'user': user}
1750
        if password is not None:
1751
            values['password'] = password
1752
        if scheme is not None:
1753
            values['scheme'] = scheme
1754
        if port is not None:
1755
            values['port'] = '%d' % port
1756
        if path is not None:
1757
            values['path'] = path
3777.1.10 by Aaron Bentley
Ensure credentials are stored
1758
        if verify_certificates is not None:
1759
            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
1760
        if realm is not None:
1761
            values['realm'] = realm
3777.1.11 by Aaron Bentley
Ensure changed-name updates clear old values
1762
        config = self._get_config()
1763
        for_deletion = []
1764
        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
1765
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
3777.1.11 by Aaron Bentley
Ensure changed-name updates clear old values
1766
                if existing_values.get(key) != values.get(key):
1767
                    break
1768
            else:
1769
                del config[section]
1770
        config.update({name: values})
3777.1.10 by Aaron Bentley
Ensure credentials are stored
1771
        self._save()
3777.1.8 by Aaron Bentley
Commit work-in-progress
1772
4304.2.1 by Vincent Ladeuil
Fix bug #367726 by reverting some default user handling introduced
1773
    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.
1774
                 prompt=None, ask=False, default=None):
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1775
        """Get a user from authentication file.
1776
1777
        :param scheme: protocol
1778
1779
        :param host: the server address
1780
1781
        :param port: the associated port (optional)
1782
1783
        :param realm: the realm sent by the server (optional)
1784
1785
        :param path: the absolute path on the server (optional)
1786
4222.3.4 by Jelmer Vernooij
Default to getpass.getuser() in AuthenticationConfig.get_user(), but allow
1787
        :param ask: Ask the user if there is no explicitly configured username 
1788
                    (optional)
1789
4304.2.1 by Vincent Ladeuil
Fix bug #367726 by reverting some default user handling introduced
1790
        :param default: The username returned if none is defined (optional).
1791
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1792
        :return: The found user.
1793
        """
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1794
        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
1795
                                           path=path, realm=realm)
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1796
        if credentials is not None:
1797
            user = credentials['user']
1798
        else:
1799
            user = None
4222.3.2 by Jelmer Vernooij
Prompt for user names if they are not in the configuration.
1800
        if user is None:
4222.3.4 by Jelmer Vernooij
Default to getpass.getuser() in AuthenticationConfig.get_user(), but allow
1801
            if ask:
1802
                if prompt is None:
1803
                    # Create a default prompt suitable for most cases
1804
                    prompt = scheme.upper() + ' %(host)s username'
1805
                # Special handling for optional fields in the prompt
1806
                if port is not None:
1807
                    prompt_host = '%s:%d' % (host, port)
1808
                else:
1809
                    prompt_host = host
1810
                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.
1811
            else:
4222.3.10 by Jelmer Vernooij
Avoid using the default username in the case of SMTP.
1812
                user = default
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1813
        return user
1814
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1815
    def get_password(self, scheme, host, user, port=None,
1816
                     realm=None, path=None, prompt=None):
1817
        """Get a password from authentication file or prompt the user for one.
1818
1819
        :param scheme: protocol
1820
1821
        :param host: the server address
1822
1823
        :param port: the associated port (optional)
1824
1825
        :param user: login
1826
1827
        :param realm: the realm sent by the server (optional)
1828
1829
        :param path: the absolute path on the server (optional)
1830
1831
        :return: The found password or the one entered by the user.
1832
        """
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1833
        credentials = self.get_credentials(scheme, host, port, user, path,
1834
                                           realm)
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1835
        if credentials is not None:
1836
            password = credentials['password']
3420.1.3 by Vincent Ladeuil
John's review feedback.
1837
            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.
1838
                trace.warning('password ignored in section [%s],'
1839
                              ' use an ssh agent instead'
1840
                              % credentials['name'])
1841
                password = None
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1842
        else:
1843
            password = None
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1844
        # 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).
1845
        if password is None:
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1846
            if prompt is None:
3420.1.2 by Vincent Ladeuil
Fix bug #203186 by ignoring passwords for ssh and warning user.
1847
                # Create a default prompt suitable for most cases
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1848
                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
1849
            # Special handling for optional fields in the prompt
1850
            if port is not None:
1851
                prompt_host = '%s:%d' % (host, port)
1852
            else:
1853
                prompt_host = host
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1854
            password = ui.ui_factory.get_password(prompt,
1855
                                                  host=prompt_host, user=user)
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1856
        return password
1857
2900.2.22 by Vincent Ladeuil
Polishing.
1858
    def decode_password(self, credentials, encoding):
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1859
        try:
1860
            cs = credential_store_registry.get_credential_store(encoding)
1861
        except KeyError:
1862
            raise ValueError('%r is not a known password_encoding' % encoding)
1863
        credentials['password'] = cs.decode_password(credentials)
2900.2.22 by Vincent Ladeuil
Polishing.
1864
        return credentials
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1865
3242.3.17 by Aaron Bentley
Whitespace cleanup
1866
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1867
class CredentialStoreRegistry(registry.Registry):
1868
    """A class that registers credential stores.
1869
1870
    A credential store provides access to credentials via the password_encoding
1871
    field in authentication.conf sections.
1872
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1873
    Except for stores provided by bzr itself, most stores are expected to be
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1874
    provided by plugins that will therefore use
1875
    register_lazy(password_encoding, module_name, member_name, help=help,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1876
    fallback=fallback) to install themselves.
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1877
1878
    A fallback credential store is one that is queried if no credentials can be
1879
    found via authentication.conf.
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1880
    """
1881
1882
    def get_credential_store(self, encoding=None):
1883
        cs = self.get(encoding)
1884
        if callable(cs):
1885
            cs = cs()
1886
        return cs
1887
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1888
    def is_fallback(self, name):
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1889
        """Check if the named credentials store should be used as fallback."""
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1890
        return self.get_info(name)
1891
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1892
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1893
                                 path=None, realm=None):
1894
        """Request credentials from all fallback credentials stores.
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1895
1896
        The first credentials store that can provide credentials wins.
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1897
        """
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1898
        credentials = None
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1899
        for name in self.keys():
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1900
            if not self.is_fallback(name):
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1901
                continue
1902
            cs = self.get_credential_store(name)
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1903
            credentials = cs.get_credentials(scheme, host, port, user,
1904
                                             path, realm)
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1905
            if credentials is not None:
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1906
                # We found some credentials
1907
                break
1908
        return credentials
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1909
1910
    def register(self, key, obj, help=None, override_existing=False,
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1911
                 fallback=False):
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1912
        """Register a new object to a name.
1913
1914
        :param key: This is the key to use to request the object later.
1915
        :param obj: The object to register.
1916
        :param help: Help text for this entry. This may be a string or
1917
                a callable. If it is a callable, it should take two
1918
                parameters (registry, key): this registry and the key that
1919
                the help was registered under.
1920
        :param override_existing: Raise KeyErorr if False and something has
1921
                already been registered for that key. If True, ignore if there
1922
                is an existing key (always register the new value).
1923
        :param fallback: Whether this credential store should be 
1924
                used as fallback.
1925
        """
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1926
        return super(CredentialStoreRegistry,
1927
                     self).register(key, obj, help, info=fallback,
1928
                                    override_existing=override_existing)
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1929
1930
    def register_lazy(self, key, module_name, member_name,
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1931
                      help=None, override_existing=False,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1932
                      fallback=False):
1933
        """Register a new credential store to be loaded on request.
1934
1935
        :param module_name: The python path to the module. Such as 'os.path'.
1936
        :param member_name: The member of the module to return.  If empty or
1937
                None, get() will return the module itself.
1938
        :param help: Help text for this entry. This may be a string or
1939
                a callable.
1940
        :param override_existing: If True, replace the existing object
1941
                with the new one. If False, if there is already something
1942
                registered with the same key, raise a KeyError
1943
        :param fallback: Whether this credential store should be 
1944
                used as fallback.
1945
        """
1946
        return super(CredentialStoreRegistry, self).register_lazy(
1947
            key, module_name, member_name, help,
1948
            info=fallback, override_existing=override_existing)
1949
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1950
1951
credential_store_registry = CredentialStoreRegistry()
1952
1953
1954
class CredentialStore(object):
1955
    """An abstract class to implement storage for credentials"""
1956
1957
    def decode_password(self, credentials):
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1958
        """Returns a clear text password for the provided credentials."""
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1959
        raise NotImplementedError(self.decode_password)
1960
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1961
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1962
                        realm=None):
1963
        """Return the matching credentials from this credential store.
1964
1965
        This method is only called on fallback credential stores.
1966
        """
1967
        raise NotImplementedError(self.get_credentials)
1968
1969
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1970
1971
class PlainTextCredentialStore(CredentialStore):
5131.2.1 by Martin
Permit bzrlib to run under python -OO by explictly assigning to __doc__ for user-visible docstrings
1972
    __doc__ = """Plain text credential store for the authentication.conf file"""
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1973
1974
    def decode_password(self, credentials):
1975
        """See CredentialStore.decode_password."""
1976
        return credentials['password']
1977
1978
1979
credential_store_registry.register('plain', PlainTextCredentialStore,
1980
                                   help=PlainTextCredentialStore.__doc__)
1981
credential_store_registry.default_key = 'plain'
1982
1983
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1984
class BzrDirConfig(object):
1985
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
1986
    def __init__(self, bzrdir):
1987
        self._bzrdir = bzrdir
1988
        self._config = bzrdir._get_config()
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1989
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
1990
    def set_default_stack_on(self, value):
1991
        """Set the default stacking location.
1992
1993
        It may be set to a location, or None.
1994
1995
        This policy affects all branches contained by this bzrdir, except for
1996
        those under repositories.
1997
        """
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
1998
        if self._config is None:
1999
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2000
        if value is None:
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2001
            self._config.set_option('', 'default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2002
        else:
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2003
            self._config.set_option(value, 'default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2004
2005
    def get_default_stack_on(self):
2006
        """Return the default stacking location.
2007
2008
        This will either be a location, or None.
2009
2010
        This policy affects all branches contained by this bzrdir, except for
2011
        those under repositories.
2012
        """
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
2013
        if self._config is None:
2014
            return None
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2015
        value = self._config.get_option('default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2016
        if value == '':
2017
            value = None
2018
        return value
2019
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2020
2021
class TransportConfig(object):
3242.1.5 by Aaron Bentley
Update per review comments
2022
    """A Config that reads/writes a config file on a Transport.
3242.1.4 by Aaron Bentley
Clean-up
2023
2024
    It is a low-level object that considers config data to be name/value pairs
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
2025
    that may be associated with a section.  Assigning meaning to these values
2026
    is done at higher levels like TreeConfig.
3242.1.4 by Aaron Bentley
Clean-up
2027
    """
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2028
2029
    def __init__(self, transport, filename):
2030
        self._transport = transport
2031
        self._filename = filename
2032
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
2033
    def get_option(self, name, section=None, default=None):
2034
        """Return the value associated with a named option.
2035
2036
        :param name: The name of the value
2037
        :param section: The section the option is in (if any)
2038
        :param default: The value to return if the value is not set
2039
        :return: The value or default value
2040
        """
2041
        configobj = self._get_configobj()
2042
        if section is None:
2043
            section_obj = configobj
2044
        else:
2045
            try:
2046
                section_obj = configobj[section]
2047
            except KeyError:
2048
                return default
2049
        return section_obj.get(name, default)
2050
2051
    def set_option(self, value, name, section=None):
2052
        """Set the value associated with a named option.
2053
2054
        :param value: The value to set
2055
        :param name: The name of the value to set
2056
        :param section: The section the option is in (if any)
2057
        """
2058
        configobj = self._get_configobj()
2059
        if section is None:
2060
            configobj[name] = value
2061
        else:
2062
            configobj.setdefault(section, {})[name] = value
2063
        self._set_configobj(configobj)
2064
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2065
    def remove_option(self, option_name, section_name=None):
2066
        configobj = self._get_configobj()
2067
        if section_name is None:
2068
            del configobj[option_name]
2069
        else:
2070
            del configobj[section_name][option_name]
2071
        self._set_configobj(configobj)
2072
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
2073
    def _get_config_file(self):
2074
        try:
4852.1.10 by John Arbash Meinel
Use a StringIO instead, otherwise we get failures with smart server requests.
2075
            return StringIO(self._transport.get_bytes(self._filename))
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
2076
        except errors.NoSuchFile:
2077
            return StringIO()
2078
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
2079
    def _get_configobj(self):
4708.2.1 by Martin
Ensure all files opened by bazaar proper are explicitly closed
2080
        f = self._get_config_file()
2081
        try:
2082
            return ConfigObj(f, encoding='utf-8')
2083
        finally:
2084
            f.close()
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
2085
2086
    def _set_configobj(self, configobj):
2087
        out_file = StringIO()
2088
        configobj.write(out_file)
2089
        out_file.seek(0)
2090
        self._transport.put_file(self._filename, out_file)
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2091
2092
5743.3.11 by Vincent Ladeuil
Config sections only implement read access.
2093
class Section(object):
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2094
    """A section defines a dict of options.
2095
2096
    This is merely a read-only dict which can add some knowledge about the
5743.3.10 by Vincent Ladeuil
Fix typos mentioned in reviews.
2097
    options. It is *not* a python dict object though and doesn't try to mimic
2098
    its API.
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2099
    """
2100
2101
    def __init__(self, section_id, options):
2102
        self.id = section_id
2103
        # We re-use the dict-like object received
2104
        self.options = options
2105
2106
    def get(self, name, default=None):
2107
        return self.options.get(name, default)
2108
5743.3.12 by Vincent Ladeuil
Add an ad-hoc __repr__.
2109
    def __repr__(self):
2110
        # Mostly for debugging use
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2111
        return "<config.%s id=%s>" % (self.__class__.__name__, self.id)
5743.3.12 by Vincent Ladeuil
Add an ad-hoc __repr__.
2112
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2113
5743.3.6 by Vincent Ladeuil
Use a name less likely to be reused.
2114
_NewlyCreatedOption = object()
5743.2.2 by Vincent Ladeuil
Add tests for remove.
2115
"""Was the option created during the MutableSection lifetime"""
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2116
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2117
5743.3.11 by Vincent Ladeuil
Config sections only implement read access.
2118
class MutableSection(Section):
5743.3.1 by Vincent Ladeuil
Add a docstring and dates to FIXMEs.
2119
    """A section allowing changes and keeping track of the original values."""
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2120
2121
    def __init__(self, section_id, options):
2122
        super(MutableSection, self).__init__(section_id, options)
2123
        self.orig = {}
2124
2125
    def set(self, name, value):
5743.2.2 by Vincent Ladeuil
Add tests for remove.
2126
        if name not in self.options:
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2127
            # This is a new option
5743.3.6 by Vincent Ladeuil
Use a name less likely to be reused.
2128
            self.orig[name] = _NewlyCreatedOption
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2129
        elif name not in self.orig:
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2130
            self.orig[name] = self.get(name, None)
2131
        self.options[name] = value
2132
2133
    def remove(self, name):
2134
        if name not in self.orig:
2135
            self.orig[name] = self.get(name, None)
2136
        del self.options[name]
2137
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2138
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2139
class Store(object):
2140
    """Abstract interface to persistent storage for configuration options."""
2141
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2142
    readonly_section_class = Section
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2143
    mutable_section_class = MutableSection
5743.4.21 by Vincent Ladeuil
All stores should provide _load_from_string to reuse the existing tests.
2144
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2145
    def is_loaded(self):
2146
        """Returns True if the Store has been loaded.
2147
2148
        This is used to implement lazy loading and ensure the persistent
2149
        storage is queried only when needed.
2150
        """
2151
        raise NotImplementedError(self.is_loaded)
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2152
2153
    def load(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2154
        """Loads the Store from persistent storage."""
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2155
        raise NotImplementedError(self.load)
2156
5743.4.21 by Vincent Ladeuil
All stores should provide _load_from_string to reuse the existing tests.
2157
    def _load_from_string(self, str_or_unicode):
2158
        """Create a store from a string in configobj syntax.
2159
2160
        :param str_or_unicode: A string representing the file content. This will
2161
            be encoded to suit store needs internally.
2162
2163
        This is for tests and should not be used in production unless a
2164
        convincing use case can be demonstrated :)
2165
        """
2166
        raise NotImplementedError(self._load_from_string)
2167
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2168
    def save(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2169
        """Saves the Store to persistent storage."""
5743.4.10 by Vincent Ladeuil
Fix copy/paste, bad.
2170
        raise NotImplementedError(self.save)
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2171
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2172
    def external_url(self):
2173
        raise NotImplementedError(self.external_url)
2174
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2175
    def get_sections(self):
2176
        """Returns an ordered iterable of existing sections.
2177
2178
        :returns: An iterable of (name, dict).
2179
        """
2180
        raise NotImplementedError(self.get_sections)
2181
5743.4.2 by Vincent Ladeuil
Stores don't implement set_option, they just provide a mutable section.
2182
    def get_mutable_section(self, section_name=None):
2183
        """Returns the specified mutable section.
2184
2185
        :param section_name: The section identifier
2186
        """
2187
        raise NotImplementedError(self.get_mutable_section)
5743.2.12 by Vincent Ladeuil
Rename store.set to store.set_option as it's clearer in this context and will act as a safe-guard against unintended uses (set() will be used for stacks).
2188
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2189
    def __repr__(self):
2190
        # Mostly for debugging use
5743.5.17 by Vincent Ladeuil
Use external_url to identify stores.
2191
        return "<config.%s(%s)>" % (self.__class__.__name__,
5743.5.18 by Vincent Ladeuil
Fix typo.
2192
                                    self.external_url())
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2193
2194
2195
class IniFileStore(Store):
2196
    """A config Store using ConfigObj for storage.
2197
2198
    :ivar transport: The transport object where the config file is located.
2199
2200
    :ivar file_name: The config file basename in the transport directory.
2201
2202
    :ivar _config_obj: Private member to hold the ConfigObj instance used to
2203
        serialize/deserialize the config file.
2204
    """
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2205
2206
    def __init__(self, transport, file_name):
2207
        """A config Store using ConfigObj for storage.
2208
2209
        :param transport: The transport object where the config file is located.
2210
2211
        :param file_name: The config file basename in the transport directory.
2212
        """
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2213
        super(IniFileStore, self).__init__()
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2214
        self.transport = transport
2215
        self.file_name = file_name
5743.4.18 by Vincent Ladeuil
Replace class.from_string with self._load_from_string to all stores can use it.
2216
        self._config_obj = None
2217
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2218
    def is_loaded(self):
5743.4.18 by Vincent Ladeuil
Replace class.from_string with self._load_from_string to all stores can use it.
2219
        return self._config_obj != None
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2220
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2221
    def load(self):
2222
        """Load the store from the associated file."""
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2223
        if self.is_loaded():
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2224
            return
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2225
        content = self.transport.get_bytes(self.file_name)
5743.4.18 by Vincent Ladeuil
Replace class.from_string with self._load_from_string to all stores can use it.
2226
        self._load_from_string(content)
2227
2228
    def _load_from_string(self, str_or_unicode):
2229
        """Create a config store from a string.
2230
2231
        :param str_or_unicode: A string representing the file content. This will
2232
            be utf-8 encoded internally.
2233
2234
        This is for tests and should not be used in production unless a
2235
        convincing use case can be demonstrated :)
2236
        """
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2237
        if self.is_loaded():
5743.4.18 by Vincent Ladeuil
Replace class.from_string with self._load_from_string to all stores can use it.
2238
            raise AssertionError('Already loaded: %r' % (self._config_obj,))
2239
        co_input = StringIO(str_or_unicode.encode('utf-8'))
5743.4.3 by Vincent Ladeuil
Implement get_mutable_section.
2240
        try:
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2241
            # The config files are always stored utf8-encoded
2242
            self._config_obj = ConfigObj(co_input, encoding='utf-8')
2243
        except configobj.ConfigObjError, e:
5743.4.18 by Vincent Ladeuil
Replace class.from_string with self._load_from_string to all stores can use it.
2244
            self._config_obj = None
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2245
            raise errors.ParseConfigError(e.errors, self.external_url())
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2246
5743.2.9 by Vincent Ladeuil
Implement and test store.save() and remove the 'save' parameter from store.from_string() as this won't scale well when adding class specific parameters.
2247
    def save(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2248
        if not self.is_loaded():
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2249
            # Nothing to save
2250
            return
5743.2.9 by Vincent Ladeuil
Implement and test store.save() and remove the 'save' parameter from store.from_string() as this won't scale well when adding class specific parameters.
2251
        out = StringIO()
2252
        self._config_obj.write(out)
2253
        self.transport.put_bytes(self.file_name, out.getvalue())
2254
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2255
    def external_url(self):
2256
        # FIXME: external_url should really accepts an optional relpath
2257
        # parameter (bug #750169) :-/ -- vila 2011-04-04
2258
        # The following will do in the interim but maybe we don't want to
2259
        # expose a path here but rather a config ID and its associated
2260
        # object </hand wawe>.
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2261
        return urlutils.join(self.transport.external_url(), self.file_name)
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2262
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2263
    def get_sections(self):
2264
        """Get the configobj section in the file order.
2265
2266
        :returns: An iterable of (name, dict).
2267
        """
2268
        # We need a loaded store
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2269
        try:
2270
            self.load()
2271
        except errors.NoSuchFile:
2272
            # If the file doesn't exist, there is no sections
2273
            return
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2274
        cobj = self._config_obj
2275
        if cobj.scalars:
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2276
            yield self.readonly_section_class(None, cobj)
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2277
        for section_name in cobj.sections:
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2278
            yield self.readonly_section_class(section_name, cobj[section_name])
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2279
5743.4.2 by Vincent Ladeuil
Stores don't implement set_option, they just provide a mutable section.
2280
    def get_mutable_section(self, section_name=None):
5743.2.11 by Vincent Ladeuil
Basic store.set implementation.
2281
        # We need a loaded store
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2282
        try:
2283
            self.load()
2284
        except errors.NoSuchFile:
2285
            # The file doesn't exist, let's pretend it was empty
2286
            self._load_from_string('')
5743.2.11 by Vincent Ladeuil
Basic store.set implementation.
2287
        if section_name is None:
2288
            section = self._config_obj
2289
        else:
2290
            section = self._config_obj.setdefault(section_name, {})
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2291
        return self.mutable_section_class(section_name, section)
5743.2.11 by Vincent Ladeuil
Basic store.set implementation.
2292
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2293
5743.4.16 by Vincent Ladeuil
Some doc for the stores.
2294
# Note that LockableConfigObjStore inherits from ConfigObjStore because we need
2295
# unlockable stores for use with objects that can already ensure the locking
2296
# (think branches). If different stores (not based on ConfigObj) are created,
2297
# they may face the same issue.
2298
2299
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2300
class LockableIniFileStore(IniFileStore):
5743.4.9 by Vincent Ladeuil
Implement a LockableConfigObjStore to be able to mimick the actual behaviour.
2301
    """A ConfigObjStore using locks on save to ensure store integrity."""
2302
2303
    def __init__(self, transport, file_name, lock_dir_name=None):
2304
        """A config Store using ConfigObj for storage.
2305
2306
        :param transport: The transport object where the config file is located.
2307
2308
        :param file_name: The config file basename in the transport directory.
2309
        """
2310
        if lock_dir_name is None:
2311
            lock_dir_name = 'lock'
2312
        self.lock_dir_name = lock_dir_name
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2313
        super(LockableIniFileStore, self).__init__(transport, file_name)
5743.4.9 by Vincent Ladeuil
Implement a LockableConfigObjStore to be able to mimick the actual behaviour.
2314
        self._lock = lockdir.LockDir(self.transport, self.lock_dir_name)
2315
2316
    def lock_write(self, token=None):
2317
        """Takes a write lock in the directory containing the config file.
2318
2319
        If the directory doesn't exist it is created.
2320
        """
2321
        # FIXME: This doesn't check the ownership of the created directories as
2322
        # ensure_config_dir_exists does. It should if the transport is local
2323
        # -- vila 2011-04-06
2324
        self.transport.create_prefix()
2325
        return self._lock.lock_write(token)
2326
2327
    def unlock(self):
2328
        self._lock.unlock()
2329
2330
    def break_lock(self):
2331
        self._lock.break_lock()
2332
2333
    @needs_write_lock
2334
    def save(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2335
        super(LockableIniFileStore, self).save()
5743.4.9 by Vincent Ladeuil
Implement a LockableConfigObjStore to be able to mimick the actual behaviour.
2336
2337
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2338
# FIXME: global, bazaar, shouldn't that be 'user' instead or even
2339
# 'user_defaults' as opposed to 'user_overrides', 'system_defaults'
2340
# (/etc/bzr/bazaar.conf) and 'system_overrides' ? -- vila 2011-04-05
5743.5.15 by Vincent Ladeuil
Mention poolie's point about focusing tests.
2341
2342
# FIXME: Moreover, we shouldn't need classes for these stores either, factory
2343
# functions or a registry will make it easier and clearer for tests, focusing
2344
# on the relevant parts of the API that needs testing -- vila 20110503 (based
2345
# on a poolie's remark)
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2346
class GlobalStore(LockableIniFileStore):
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2347
2348
    def __init__(self, possible_transports=None):
2349
        t = transport.get_transport(config_dir(),
2350
                                    possible_transports=possible_transports)
2351
        super(GlobalStore, self).__init__(t, 'bazaar.conf')
2352
2353
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2354
class LocationStore(LockableIniFileStore):
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2355
2356
    def __init__(self, possible_transports=None):
2357
        t = transport.get_transport(config_dir(),
2358
                                    possible_transports=possible_transports)
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2359
        super(LocationStore, self).__init__(t, 'locations.conf')
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2360
2361
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2362
class BranchStore(IniFileStore):
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2363
2364
    def __init__(self, branch):
2365
        super(BranchStore, self).__init__(branch.control_transport,
2366
                                          'branch.conf')
2367
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2368
class SectionMatcher(object):
2369
    """Select sections into a given Store.
2370
2371
    This intended to be used to postpone getting an iterable of sections from a
2372
    store.
2373
    """
2374
2375
    def __init__(self, store):
2376
        self.store = store
2377
2378
    def get_sections(self):
5743.2.29 by Vincent Ladeuil
Add doc for the section matchers.
2379
        # This is where we require loading the store so we can see all defined
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2380
        # sections.
2381
        sections = self.store.get_sections()
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2382
        # Walk the revisions in the order provided
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2383
        for s in sections:
2384
            if self.match(s):
2385
                yield s
2386
2387
    def match(self, secion):
2388
        raise NotImplementedError(self.match)
2389
2390
5743.2.37 by Vincent Ladeuil
Merge config-concrete-stores into config-section-matchers resolving conflicts
2391
class LocationSection(Section):
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2392
2393
    def __init__(self, section, length, extra_path):
2394
        super(LocationSection, self).__init__(section.id, section.options)
2395
        self.length = length
2396
        self.extra_path = extra_path
2397
2398
    def get(self, name, default=None):
2399
        value = super(LocationSection, self).get(name, default)
2400
        if value is not None:
2401
            policy_name = self.get(name + ':policy', None)
2402
            policy = _policy_value.get(policy_name, POLICY_NONE)
2403
            if policy == POLICY_APPENDPATH:
2404
                value = urlutils.join(value, self.extra_path)
2405
        return value
2406
2407
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2408
class LocationMatcher(SectionMatcher):
2409
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2410
    def __init__(self, store, location):
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2411
        super(LocationMatcher, self).__init__(store)
5743.6.19 by Vincent Ladeuil
Clarify comments about section names for Location-related objects (also fix LocationMatcher and add tests).
2412
        if location.startswith('file://'):
2413
            location = urlutils.local_path_from_url(location)
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2414
        self.location = location
2415
5743.6.15 by Vincent Ladeuil
Don't pollute _iter_for_location_by_parts.
2416
    def _get_matching_sections(self):
2417
        """Get all sections matching ``location``."""
2418
        # We slightly diverge from LocalConfig here by allowing the no-name
2419
        # section as the most generic one and the lower priority.
2420
        no_name_section = None
2421
        sections = []
2422
        # Filter out the no_name_section so _iter_for_location_by_parts can be
2423
        # used (it assumes all sections have a name).
2424
        for section in self.store.get_sections():
2425
            if section.id is None:
2426
                no_name_section = section
2427
            else:
2428
                sections.append(section)
2429
        # Unfortunately _iter_for_location_by_parts deals with section names so
2430
        # we have to resync.
5743.2.31 by Vincent Ladeuil
Both the length and the section id should be used to sort.
2431
        filtered_sections = _iter_for_location_by_parts(
5743.2.27 by Vincent Ladeuil
Merge the use of _filter_for_location_by_parts, uglier, but better for
2432
            [s.id for s in sections], self.location)
2433
        iter_sections = iter(sections)
2434
        matching_sections = []
5743.6.15 by Vincent Ladeuil
Don't pollute _iter_for_location_by_parts.
2435
        if no_name_section is not None:
2436
            matching_sections.append(
2437
                LocationSection(no_name_section, 0, self.location))
5743.2.31 by Vincent Ladeuil
Both the length and the section id should be used to sort.
2438
        for section_id, extra_path, length in filtered_sections:
5743.2.27 by Vincent Ladeuil
Merge the use of _filter_for_location_by_parts, uglier, but better for
2439
            # a section id is unique for a given store so it's safe to iterate
2440
            # again
2441
            section = iter_sections.next()
2442
            if section_id == section.id:
2443
                matching_sections.append(
2444
                    LocationSection(section, length, extra_path))
5743.6.15 by Vincent Ladeuil
Don't pollute _iter_for_location_by_parts.
2445
        return matching_sections
2446
2447
    def get_sections(self):
2448
        # Override the default implementation as we want to change the order
2449
        matching_sections = self._get_matching_sections()
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2450
        # We want the longest (aka more specific) locations first
5743.2.31 by Vincent Ladeuil
Both the length and the section id should be used to sort.
2451
        sections = sorted(matching_sections,
2452
                          key=lambda section: (section.length, section.id),
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2453
                          reverse=True)
2454
        # Sections mentioning 'ignore_parents' restrict the selection
2455
        for section in sections:
2456
            # FIXME: We really want to use as_bool below -- vila 2011-04-07
2457
            ignore = section.get('ignore_parents', None)
2458
            if ignore is not None:
2459
                ignore = ui.bool_from_string(ignore)
2460
            if ignore:
2461
                break
2462
            # Finally, we have a valid section
2463
            yield section
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2464
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2465
5743.1.34 by Vincent Ladeuil
Merge config-section-matchers into config-stack resolving conflicts
2466
class Stack(object):
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2467
    """A stack of configurations where an option can be defined"""
2468
5743.1.37 by Vincent Ladeuil
Change the way Stacks are built: requires a Store and a mutable section name instead of a callable returning the mutable section.
2469
    def __init__(self, sections_def, store=None, mutable_section_name=None):
5743.1.11 by Vincent Ladeuil
Properly use MutableSection for write operations.
2470
        """Creates a stack of sections with an optional store for changes.
2471
5743.1.36 by Vincent Ladeuil
Clarify that constructor parameters are here to *define* the stack and as such can be re-used. Also make the sections_def parameter mandatory.
2472
        :param sections_def: A list of Section or callables that returns an
2473
            iterable of Section. This defines the Sections for the Stack and
2474
            can be called repeatedly if needed.
5743.1.11 by Vincent Ladeuil
Properly use MutableSection for write operations.
2475
5743.1.37 by Vincent Ladeuil
Change the way Stacks are built: requires a Store and a mutable section name instead of a callable returning the mutable section.
2476
        :param store: The optional Store where modifications will be
2477
            recorded. If none is specified, no modifications can be done.
2478
2479
        :param mutable_section_name: The name of the MutableSection where
2480
            changes are recorded. This requires the ``store`` parameter to be
2481
            specified.
5743.1.11 by Vincent Ladeuil
Properly use MutableSection for write operations.
2482
        """
5743.1.36 by Vincent Ladeuil
Clarify that constructor parameters are here to *define* the stack and as such can be re-used. Also make the sections_def parameter mandatory.
2483
        self.sections_def = sections_def
5743.1.37 by Vincent Ladeuil
Change the way Stacks are built: requires a Store and a mutable section name instead of a callable returning the mutable section.
2484
        self.store = store
2485
        self.mutable_section_name = mutable_section_name
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2486
2487
    def get(self, name):
5743.1.16 by Vincent Ladeuil
Allows empty sections and empty section callables.
2488
        """Return the *first* option value found in the sections.
5743.1.13 by Vincent Ladeuil
Better explain lazy loading and make sure the mutable section respect the design.
2489
5743.1.16 by Vincent Ladeuil
Allows empty sections and empty section callables.
2490
        This is where we guarantee that sections coming from Store are loaded
2491
        lazily: the loading is delayed until we need to either check that an
5743.1.13 by Vincent Ladeuil
Better explain lazy loading and make sure the mutable section respect the design.
2492
        option exists or get its value, which in turn may require to discover
2493
        in which sections it can be defined. Both of these (section and option
2494
        existence) require loading the store (even partially).
2495
        """
5743.1.36 by Vincent Ladeuil
Clarify that constructor parameters are here to *define* the stack and as such can be re-used. Also make the sections_def parameter mandatory.
2496
        # FIXME: No caching of options nor sections yet -- vila 20110503
2497
5743.6.17 by Vincent Ladeuil
Clarify comment.
2498
        # Ensuring lazy loading is achieved by delaying section matching (which
2499
        # implies querying the persistent storage) until it can't be avoided
2500
        # anymore by using callables to describe (possibly empty) section
2501
        # lists.
5743.1.36 by Vincent Ladeuil
Clarify that constructor parameters are here to *define* the stack and as such can be re-used. Also make the sections_def parameter mandatory.
2502
        for section_or_callable in self.sections_def:
5743.1.12 by Vincent Ladeuil
Clarify ConfigStack.get() about lazy evaluation of sections.
2503
            # Each section can expand to multiple ones when a callable is used
2504
            if callable(section_or_callable):
2505
                sections = section_or_callable()
5743.1.9 by Vincent Ladeuil
Fix the issue by allowing delayed section acquisition.
2506
            else:
5743.1.12 by Vincent Ladeuil
Clarify ConfigStack.get() about lazy evaluation of sections.
2507
                sections = [section_or_callable]
2508
            for section in sections:
2509
                value = section.get(name)
2510
                if value is not None:
2511
                    return value
2512
        # No definition was found
2513
        return None
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2514
5743.1.37 by Vincent Ladeuil
Change the way Stacks are built: requires a Store and a mutable section name instead of a callable returning the mutable section.
2515
    def _get_mutable_section(self):
2516
        """Get the MutableSection for the Stack.
5743.1.13 by Vincent Ladeuil
Better explain lazy loading and make sure the mutable section respect the design.
2517
5743.1.15 by Vincent Ladeuil
Test and implement ConfigStack.remove.
2518
        This is where we guarantee that the mutable section is lazily loaded:
5743.1.37 by Vincent Ladeuil
Change the way Stacks are built: requires a Store and a mutable section name instead of a callable returning the mutable section.
2519
        this means we won't load the corresponding store before setting a value
2520
        or deleting an option. In practice the store will often be loaded but
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2521
        this allows helps catching some programming errors.
5743.1.13 by Vincent Ladeuil
Better explain lazy loading and make sure the mutable section respect the design.
2522
        """
5743.1.37 by Vincent Ladeuil
Change the way Stacks are built: requires a Store and a mutable section name instead of a callable returning the mutable section.
2523
        section = self.store.get_mutable_section(self.mutable_section_name)
2524
        return section
2525
2526
    def set(self, name, value):
2527
        """Set a new value for the option."""
2528
        section = self._get_mutable_section()
5743.1.13 by Vincent Ladeuil
Better explain lazy loading and make sure the mutable section respect the design.
2529
        section.set(name, value)
5743.1.7 by Vincent Ladeuil
Simple set implementation.
2530
5743.1.15 by Vincent Ladeuil
Test and implement ConfigStack.remove.
2531
    def remove(self, name):
5743.1.37 by Vincent Ladeuil
Change the way Stacks are built: requires a Store and a mutable section name instead of a callable returning the mutable section.
2532
        """Remove an existing option."""
2533
        section = self._get_mutable_section()
5743.1.15 by Vincent Ladeuil
Test and implement ConfigStack.remove.
2534
        section.remove(name)
2535
5743.1.34 by Vincent Ladeuil
Merge config-section-matchers into config-stack resolving conflicts
2536
    def __repr__(self):
2537
        # Mostly for debugging use
2538
        return "<config.%s(%s)>" % (self.__class__.__name__, id(self))
2539
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2540
5743.6.11 by Vincent Ladeuil
Start parameterizing stack tests.
2541
class GlobalStack(Stack):
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2542
2543
    def __init__(self):
2544
        # Get a GlobalStore
2545
        gstore = GlobalStore()
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2546
        super(GlobalStack, self).__init__([gstore.get_sections], gstore)
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2547
2548
5743.6.11 by Vincent Ladeuil
Start parameterizing stack tests.
2549
class LocationStack(Stack):
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2550
2551
    def __init__(self, location):
2552
        lstore = LocationStore()
2553
        matcher = LocationMatcher(lstore, location)
2554
        gstore = GlobalStore()
2555
        super(LocationStack, self).__init__(
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2556
            [matcher.get_sections, gstore.get_sections], lstore)
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2557
2558
5743.6.11 by Vincent Ladeuil
Start parameterizing stack tests.
2559
class BranchStack(Stack):
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2560
2561
    def __init__(self, branch):
2562
        bstore = BranchStore(branch)
2563
        lstore = LocationStore()
2564
        matcher = LocationMatcher(lstore, branch.base)
2565
        gstore = GlobalStore()
2566
        super(BranchStack, self).__init__(
2567
            [matcher.get_sections, bstore.get_sections, gstore.get_sections],
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2568
            bstore)
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2569
2570
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2571
class cmd_config(commands.Command):
5447.4.19 by Vincent Ladeuil
Add some more documentation.
2572
    __doc__ = """Display, set or remove a configuration option.
2573
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2574
    Display the active value for a given option.
2575
2576
    If --all is specified, NAME is interpreted as a regular expression and all
2577
    matching options are displayed mentioning their scope. The active value
2578
    that bzr will take into account is the first one displayed for each option.
2579
2580
    If no NAME is given, --all .* is implied.
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2581
5447.4.19 by Vincent Ladeuil
Add some more documentation.
2582
    Setting a value is achieved by using name=value without spaces. The value
2583
    is set in the most relevant scope and can be checked by displaying the
2584
    option again.
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2585
    """
2586
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2587
    takes_args = ['name?']
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2588
2589
    takes_options = [
2590
        'directory',
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2591
        # FIXME: This should be a registry option so that plugins can register
2592
        # their own config files (or not) -- vila 20101002
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2593
        commands.Option('scope', help='Reduce the scope to the specified'
2594
                        ' configuration file',
2595
                        type=unicode),
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2596
        commands.Option('all',
2597
            help='Display all the defined values for the matching options.',
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2598
            ),
5447.4.8 by Vincent Ladeuil
Make the test properly fail and provide a fake implementation for ``bzr config --remove opt_name``.
2599
        commands.Option('remove', help='Remove the option from'
2600
                        ' the configuration file'),
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2601
        ]
2602
2603
    @commands.display_command
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2604
    def run(self, name=None, all=False, directory=None, scope=None,
2605
            remove=False):
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2606
        if directory is None:
2607
            directory = '.'
2608
        directory = urlutils.normalize_url(directory)
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2609
        if remove and all:
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2610
            raise errors.BzrError(
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2611
                '--all and --remove are mutually exclusive.')
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2612
        elif remove:
2613
            # Delete the option in the given scope
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2614
            self._remove_config_option(name, directory, scope)
2615
        elif name is None:
2616
            # Defaults to all options
2617
            self._show_matching_options('.*', directory, scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2618
        else:
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2619
            try:
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2620
                name, value = name.split('=', 1)
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2621
            except ValueError:
2622
                # Display the option(s) value(s)
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2623
                if all:
2624
                    self._show_matching_options(name, directory, scope)
2625
                else:
2626
                    self._show_value(name, directory, scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2627
            else:
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2628
                if all:
2629
                    raise errors.BzrError(
2630
                        'Only one option can be set.')
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2631
                # Set the option value
2632
                self._set_config_option(name, value, directory, scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2633
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2634
    def _get_configs(self, directory, scope=None):
2635
        """Iterate the configurations specified by ``directory`` and ``scope``.
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2636
2637
        :param directory: Where the configurations are derived from.
2638
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2639
        :param scope: A specific config to start from.
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2640
        """
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2641
        if scope is not None:
2642
            if scope == 'bazaar':
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2643
                yield GlobalConfig()
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2644
            elif scope == 'locations':
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2645
                yield LocationConfig(directory)
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2646
            elif scope == 'branch':
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2647
                (_, br, _) = bzrdir.BzrDir.open_containing_tree_or_branch(
2648
                    directory)
2649
                yield br.get_config()
2650
        else:
2651
            try:
2652
                (_, br, _) = bzrdir.BzrDir.open_containing_tree_or_branch(
2653
                    directory)
2654
                yield br.get_config()
2655
            except errors.NotBranchError:
2656
                yield LocationConfig(directory)
2657
                yield GlobalConfig()
2658
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2659
    def _show_value(self, name, directory, scope):
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2660
        displayed = False
2661
        for c in self._get_configs(directory, scope):
2662
            if displayed:
2663
                break
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2664
            for (oname, value, section, conf_id, parser) in c._get_options():
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2665
                if name == oname:
5533.1.3 by Vincent Ladeuil
Tweak comment as per poolie's suggestion.
2666
                    # Display only the first value and exit
5533.2.3 by Vincent Ladeuil
Merge 671050-config-policy into 672382-list-values 672382-list-values resolving conflicts
2667
5533.1.3 by Vincent Ladeuil
Tweak comment as per poolie's suggestion.
2668
                    # FIXME: We need to use get_user_option to take policies
2669
                    # into account and we need to make sure the option exists
5533.2.3 by Vincent Ladeuil
Merge 671050-config-policy into 672382-list-values 672382-list-values resolving conflicts
2670
                    # too (hence the two for loops), this needs a better API
2671
                    # -- vila 20101117
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2672
                    value = c.get_user_option(name)
2673
                    # Quote the value appropriately
2674
                    value = parser._quote(value)
2675
                    self.outf.write('%s\n' % (value,))
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2676
                    displayed = True
2677
                    break
2678
        if not displayed:
2679
            raise errors.NoSuchConfigOption(name)
2680
2681
    def _show_matching_options(self, name, directory, scope):
2682
        name = re.compile(name)
2683
        # We want any error in the regexp to be raised *now* so we need to
2684
        # avoid the delay introduced by the lazy regexp.
2685
        name._compile_and_collapse()
2686
        cur_conf_id = None
5533.1.1 by Vincent Ladeuil
Fix ``bzr config`` to respect policies when displaying values and also display sections when appropriate.
2687
        cur_section = None
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2688
        for c in self._get_configs(directory, scope):
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2689
            for (oname, value, section, conf_id, parser) in c._get_options():
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2690
                if name.search(oname):
2691
                    if cur_conf_id != conf_id:
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2692
                        # Explain where the options are defined
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
2693
                        self.outf.write('%s:\n' % (conf_id,))
2694
                        cur_conf_id = conf_id
5533.1.1 by Vincent Ladeuil
Fix ``bzr config`` to respect policies when displaying values and also display sections when appropriate.
2695
                        cur_section = None
2696
                    if (section not in (None, 'DEFAULT')
2697
                        and cur_section != section):
2698
                        # Display the section if it's not the default (or only)
2699
                        # one.
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2700
                        self.outf.write('  [%s]\n' % (section,))
5533.1.1 by Vincent Ladeuil
Fix ``bzr config`` to respect policies when displaying values and also display sections when appropriate.
2701
                        cur_section = section
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2702
                    self.outf.write('  %s = %s\n' % (oname, value))
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2703
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2704
    def _set_config_option(self, name, value, directory, scope):
2705
        for conf in self._get_configs(directory, scope):
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2706
            conf.set_user_option(name, value)
2707
            break
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2708
        else:
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2709
            raise errors.NoSuchConfig(scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2710
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2711
    def _remove_config_option(self, name, directory, scope):
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2712
        if name is None:
2713
            raise errors.BzrCommandError(
2714
                '--remove expects an option to remove.')
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2715
        removed = False
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2716
        for conf in self._get_configs(directory, scope):
5447.4.12 by Vincent Ladeuil
Turn get_options() and get_sections() into private methods because section handling is too messy and needs to be discussed and settled.
2717
            for (section_name, section, conf_id) in conf._get_sections():
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2718
                if scope is not None and conf_id != scope:
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2719
                    # Not the right configuration file
2720
                    continue
2721
                if name in section:
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
2722
                    if conf_id != conf.config_id():
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2723
                        conf = self._get_configs(directory, conf_id).next()
2724
                    # We use the first section in the first config where the
2725
                    # option is defined to remove it
2726
                    conf.remove_user_option(name, section_name)
2727
                    removed = True
2728
                    break
2729
            break
2730
        else:
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2731
            raise errors.NoSuchConfig(scope)
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2732
        if not removed:
2733
            raise errors.NoSuchConfigOption(name)