/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.
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.
981
    """
5764.1.2 by Vincent Ladeuil
This put a common processing into the loop to avoid bad inputs. The
982
    location_parts = location.rstrip('/').split('/')
983
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
    for section in sections:
985
        # location is a local path if possible, so we need
986
        # to convert 'file://' urls to local paths if necessary.
987
988
        # FIXME: I don't think the above comment is still up to date,
989
        # LocationConfig is always instantiated with an url -- vila 2011-04-07
990
991
        # This also avoids having file:///path be a more exact
992
        # match than '/path'.
993
994
        # FIXME: Not sure about the above either, but since the path components
995
        # are compared in sync, adding two empty components (//) is likely to
996
        # trick the comparison and also trick the check on the number of
997
        # components, so we *should* take only the relevant part of the url. On
998
        # the other hand, this means 'file://' urls *can't* be used in sections
999
        # so more work is probably needed -- vila 2011-04-07
1000
        if section.startswith('file://'):
1001
            section_path = urlutils.local_path_from_url(section)
1002
        else:
1003
            section_path = section
1004
        section_parts = section_path.rstrip('/').split('/')
1005
1006
        matched = True
1007
        if len(section_parts) > len(location_parts):
1008
            # More path components in the section, they can't match
1009
            matched = False
1010
        else:
1011
            # Rely on zip truncating in length to the length of the shortest
1012
            # argument sequence.
1013
            names = zip(location_parts, section_parts)
1014
            for name in names:
1015
                if not fnmatch.fnmatch(name[0], name[1]):
1016
                    matched = False
1017
                    break
1018
        if not matched:
1019
            continue
5764.1.1 by Vincent Ladeuil
Extract _match_section_by_parts from LocationConfig._get_matching_sections and more comments to explain the behavior.
1020
        # 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.
1021
        extra_path = '/'.join(location_parts[len(section_parts):])
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
1022
        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.
1023
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1024
5345.5.7 by Vincent Ladeuil
Make LocationConfig use a lock too.
1025
class LocationConfig(LockableConfig):
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1026
    """A configuration object that gives the policy for a location."""
1027
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1028
    def __init__(self, location):
5345.1.2 by Vincent Ladeuil
Get rid of 'branches.conf' references.
1029
        super(LocationConfig, self).__init__(
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1030
            file_name=locations_config_filename())
1878.1.1 by John Arbash Meinel
Entries in locations.conf should prefer local paths if available (bug #53653)
1031
        # local file locations are looked up by local path, rather than
1032
        # by file url. This is because the config file is a user
1033
        # file, and we would rather not expose the user to file urls.
1034
        if location.startswith('file://'):
1035
            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
1036
        self.location = location
1037
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1038
    def config_id(self):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1039
        return 'locations'
1040
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1041
    @classmethod
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
1042
    def from_string(cls, str_or_unicode, location, save=False):
1043
        """Create a config object from a string.
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1044
5345.2.9 by Vincent Ladeuil
Rename IniBaseConfig.from_bytes to from_string.
1045
        :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.
1046
            be utf-8 encoded.
1047
1048
        :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.
1049
1050
        :param save: Whether the file should be saved upon creation.
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1051
        """
1052
        conf = cls(location)
5345.1.26 by Vincent Ladeuil
Merge lockable-config-files into remove-gratuitous-ensure-config-dir-exist-calls resolving conflicts
1053
        conf._create_from_string(str_or_unicode, save)
5345.2.8 by Vincent Ladeuil
Introduce a 'from_bytes' constructor for config objects.
1054
        return conf
1055
1993.3.1 by James Henstridge
first go at making location config lookup recursive
1056
    def _get_matching_sections(self):
1057
        """Return an ordered list of section names matching this location."""
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
1058
        matches = list(_iter_for_location_by_parts(self._get_parser(),
1059
                                                   self.location))
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
1060
        # put the longest (aka more specific) locations first
5764.1.4 by Vincent Ladeuil
Using iterators is even clearer.
1061
        matches.sort(
1062
            key=lambda (section, extra_path, length): (length, section),
1063
            reverse=True)
1064
        for (section, extra_path, length) in matches:
1065
            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
1066
            # should we stop looking for parent configs here?
1993.3.1 by James Henstridge
first go at making location config lookup recursive
1067
            try:
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
1068
                if self._get_parser()[section].as_bool('ignore_parents'):
1069
                    break
1993.3.1 by James Henstridge
first go at making location config lookup recursive
1070
            except KeyError:
1071
                pass
1442.1.9 by Robert Collins
exact section test passes
1072
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.
1073
    def _get_sections(self, name=None):
1074
        """See IniBasedConfig._get_sections()."""
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1075
        # We ignore the name here as the only sections handled are named with
1076
        # the location path and we don't expose embedded sections either.
1077
        parser = self._get_parser()
1078
        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.
1079
            yield (name, parser[name], self.config_id())
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1080
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
1081
    def _get_option_policy(self, section, option_name):
1082
        """Return the policy for the given (section, option_name) pair."""
1083
        # check for the old 'recurse=False' flag
1084
        try:
1085
            recurse = self._get_parser()[section].as_bool('recurse')
1086
        except KeyError:
1087
            recurse = True
1088
        if not recurse:
1089
            return POLICY_NORECURSE
1090
2120.6.10 by James Henstridge
Catch another deprecation warning, and more cleanup
1091
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
1092
        try:
1093
            policy_name = self._get_parser()[section][policy_key]
1094
        except KeyError:
1095
            policy_name = None
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
1096
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
1097
        return _policy_value[policy_name]
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
1098
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1099
    def _set_option_policy(self, section, option_name, option_policy):
1100
        """Set the policy for the given option name in the given section."""
1101
        # The old recurse=False option affects all options in the
1102
        # section.  To handle multiple policies in the section, we
1103
        # need to convert it to a policy_norecurse key.
1104
        try:
1105
            recurse = self._get_parser()[section].as_bool('recurse')
1106
        except KeyError:
1107
            pass
1108
        else:
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1109
            symbol_versioning.warn(
2120.6.11 by James Henstridge
s/0.13/0.14/ in deprecation warning
1110
                'The recurse option is deprecated as of 0.14.  '
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1111
                'The section "%s" has been converted to use policies.'
1112
                % section,
1113
                DeprecationWarning)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1114
            del self._get_parser()[section]['recurse']
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1115
            if not recurse:
1116
                for key in self._get_parser()[section].keys():
1117
                    if not key.endswith(':policy'):
1118
                        self._get_parser()[section][key +
1119
                                                    ':policy'] = 'norecurse'
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1120
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
1121
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
1122
        policy_name = _policy_name[option_policy]
1123
        if policy_name is not None:
1124
            self._get_parser()[section][policy_key] = policy_name
1125
        else:
1126
            if policy_key in self._get_parser()[section]:
1127
                del self._get_parser()[section][policy_key]
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1128
5345.5.7 by Vincent Ladeuil
Make LocationConfig use a lock too.
1129
    @needs_write_lock
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1130
    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.
1131
        """Save option and its value in the configuration."""
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
1132
        if store not in [STORE_LOCATION,
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1133
                         STORE_LOCATION_NORECURSE,
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
1134
                         STORE_LOCATION_APPENDPATH]:
1135
            raise ValueError('bad storage policy %r for %r' %
1136
                (store, option))
5345.5.1 by Vincent Ladeuil
Implement config.reload and make sure we have a file name when using it.
1137
        self.reload()
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
1138
        location = self.location
1139
        if location.endswith('/'):
1140
            location = location[:-1]
5345.1.24 by Vincent Ladeuil
Implement _save for LockableConfig too.
1141
        parser = self._get_parser()
5345.1.21 by Vincent Ladeuil
Slight rewrite to make the method more readable.
1142
        if not location in parser and not location + '/' in parser:
1143
            parser[location] = {}
1144
        elif location + '/' in parser:
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
1145
            location = location + '/'
5345.1.21 by Vincent Ladeuil
Slight rewrite to make the method more readable.
1146
        parser[location][option]=value
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1147
        # the allowed values of store match the config policies
1148
        self._set_option_policy(location, option, store)
4708.2.1 by Martin
Ensure all files opened by bazaar proper are explicitly closed
1149
        self._write_config_file()
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
1150
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1151
1152
class BranchConfig(Config):
1153
    """A configuration object giving the policy for a branch."""
1154
5345.1.3 by Vincent Ladeuil
Make __init__ the first method in the BranchConfig class.
1155
    def __init__(self, branch):
1156
        super(BranchConfig, self).__init__()
1157
        self._location_config = None
1158
        self._branch_data_config = None
1159
        self._global_config = None
1160
        self.branch = branch
1161
        self.option_sources = (self._get_location_config,
1162
                               self._get_branch_data_config,
1163
                               self._get_global_config)
1164
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1165
    def config_id(self):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1166
        return 'branch'
1167
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1168
    def _get_branch_data_config(self):
1169
        if self._branch_data_config is None:
1170
            self._branch_data_config = TreeConfig(self.branch)
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
1171
            self._branch_data_config.config_id = self.config_id
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1172
        return self._branch_data_config
1173
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1174
    def _get_location_config(self):
1175
        if self._location_config is None:
1176
            self._location_config = LocationConfig(self.branch.base)
1177
        return self._location_config
1178
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1179
    def _get_global_config(self):
1180
        if self._global_config is None:
1181
            self._global_config = GlobalConfig()
1182
        return self._global_config
1183
1184
    def _get_best_value(self, option_name):
1185
        """This returns a user option from local, tree or global config.
1186
1187
        They are tried in that order.  Use get_safe_value if trusted values
1188
        are necessary.
1189
        """
1190
        for source in self.option_sources:
1191
            value = getattr(source(), option_name)()
1192
            if value is not None:
1193
                return value
1194
        return None
1195
1196
    def _get_safe_value(self, option_name):
1197
        """This variant of get_best_value never returns untrusted values.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1198
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1199
        It does not return values from the branch data, because the branch may
1200
        not be controlled by the user.
1201
1202
        We may wish to allow locations.conf to control whether branches are
1203
        trusted in the future.
1204
        """
1205
        for source in (self._get_location_config, self._get_global_config):
1206
            value = getattr(source(), option_name)()
1207
            if value is not None:
1208
                return value
1209
        return None
1210
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1211
    def _get_user_id(self):
1212
        """Return the full user id for the branch.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1213
3407.2.14 by Martin Pool
Remove more cases of getting transport via control_files
1214
        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
1215
        This is looked up in the email controlfile for the branch.
1216
        """
1217
        try:
3407.2.16 by Martin Pool
Remove RemoteBranch reliance on control_files._transport
1218
            return (self.branch._transport.get_bytes("email")
3224.5.4 by Andrew Bennetts
Fix test suite, mainly weeding out uses of bzrlib.user_encoding.
1219
                    .decode(osutils.get_user_encoding())
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
1220
                    .rstrip("\r\n"))
1221
        except errors.NoSuchFile, e:
1222
            pass
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1223
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1224
        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
1225
4603.1.10 by Aaron Bentley
Provide change editor via config.
1226
    def _get_change_editor(self):
1227
        return self._get_best_value('_get_change_editor')
1228
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
1229
    def _get_signature_checking(self):
1230
        """See Config._get_signature_checking."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1231
        return self._get_best_value('_get_signature_checking')
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
1232
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
1233
    def _get_signing_policy(self):
1234
        """See Config._get_signing_policy."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1235
        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
1236
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
1237
    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.
1238
        """See Config._get_user_option."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1239
        for source in self.option_sources:
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
1240
            value = source()._get_user_option(option_name)
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1241
            if value is not None:
1242
                return value
1243
        return None
1244
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
    def _get_sections(self, name=None):
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1246
        """See IniBasedConfig.get_sections()."""
1247
        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.
1248
            for section in source()._get_sections(name):
5447.4.4 by Vincent Ladeuil
Implement config.get_sections() to clarify how sections can be used.
1249
                yield section
1250
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
    def _get_options(self, sections=None):
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1252
        opts = []
1253
        # 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.
1254
        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.
1255
            yield option
1256
        # Then the branch options
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
1257
        branch_config = self._get_branch_data_config()
1258
        if sections is None:
1259
            sections = [('DEFAULT', branch_config._get_parser())]
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1260
        # FIXME: We shouldn't have to duplicate the code in IniBasedConfig but
1261
        # 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.
1262
        config_id = self.config_id()
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1263
        for (section_name, section) in sections:
1264
            for (name, value) in section.iteritems():
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
1265
                yield (name, value, section_name,
1266
                       config_id, branch_config._get_parser())
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
1267
        # 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.
1268
        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.
1269
            yield option
5447.4.1 by Vincent Ladeuil
Implement config.get_options_matching_regexp.
1270
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
1271
    def set_user_option(self, name, value, store=STORE_BRANCH,
1272
        warn_masked=False):
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1273
        if store == STORE_BRANCH:
1770.2.6 by Aaron Bentley
Ensure branch.conf works properly
1274
            self._get_branch_data_config().set_option(value, name)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1275
        elif store == STORE_GLOBAL:
2120.6.7 by James Henstridge
Fix GlobalConfig.set_user_option() call
1276
            self._get_global_config().set_user_option(name, value)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
1277
        else:
1278
            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)
1279
        if not warn_masked:
1280
            return
1281
        if store in (STORE_GLOBAL, STORE_BRANCH):
1282
            mask_value = self._get_location_config().get_user_option(name)
1283
            if mask_value is not None:
1284
                trace.warning('Value "%s" is masked by "%s" from'
1285
                              ' locations.conf', value, mask_value)
1286
            else:
1287
                if store == STORE_GLOBAL:
1288
                    branch_config = self._get_branch_data_config()
1289
                    mask_value = branch_config.get_user_option(name)
1290
                    if mask_value is not None:
1291
                        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
1292
                                      ' branch.conf', value, mask_value)
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
1293
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
1294
    def remove_user_option(self, option_name, section_name=None):
1295
        self._get_branch_data_config().remove_option(option_name, section_name)
1296
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
1297
    def _gpg_signing_command(self):
1442.1.56 by Robert Collins
gpg_signing_command configuration item
1298
        """See Config.gpg_signing_command."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1299
        return self._get_safe_value('_gpg_signing_command')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1300
1472 by Robert Collins
post commit hook, first pass implementation
1301
    def _post_commit(self):
1302
        """See Config.post_commit."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1303
        return self._get_safe_value('_post_commit')
1472 by Robert Collins
post commit hook, first pass implementation
1304
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
1305
    def _get_nickname(self):
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
1306
        value = self._get_explicit_nickname()
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
1307
        if value is not None:
1308
            return value
2120.5.2 by Alexander Belchenko
(jam) Fix for bug #66857
1309
        return urlutils.unescape(self.branch.base.split('/')[-2])
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
1310
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
1311
    def has_explicit_nickname(self):
1312
        """Return true if a nickname has been explicitly assigned."""
1313
        return self._get_explicit_nickname() is not None
1314
1315
    def _get_explicit_nickname(self):
1316
        return self._get_best_value('_get_nickname')
1317
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
1318
    def _log_format(self):
1319
        """See Config.log_format."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1320
        return self._get_best_value('_log_format')
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
1321
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
1322
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
1323
def ensure_config_dir_exists(path=None):
5519.4.4 by Neil Martinsen-Burrell
restore ensure_config_dir since XDG_CONFIG_HOME is optional
1324
    """Make sure a configuration directory exists.
1325
    This makes sure that the directory exists.
1326
    On windows, since configuration directories are 2 levels deep,
1327
    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
1328
    """
1329
    if path is None:
1330
        path = config_dir()
1331
    if not os.path.isdir(path):
5519.4.4 by Neil Martinsen-Burrell
restore ensure_config_dir since XDG_CONFIG_HOME is optional
1332
        if sys.platform == 'win32':
1333
            parent_dir = os.path.dirname(path)
1334
            if not os.path.isdir(parent_dir):
1335
                trace.mutter('creating config parent directory: %r', parent_dir)
1336
                os.mkdir(parent_dir)
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1337
        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
1338
        os.mkdir(path)
5116.2.6 by Parth Malwankar
renamed copy_ownership to copy_ownership_from_path.
1339
        osutils.copy_ownership_from_path(path)
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
1340
1532 by Robert Collins
Merge in John Meinels integration branch.
1341
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
1342
def config_dir():
1343
    """Return per-user configuration directory.
1344
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1345
    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
1346
    and Linux.  On Linux, if there is a $XDG_CONFIG_HOME/bazaar directory,
1347
    that will be used instead.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1348
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
1349
    TODO: Global option --config-dir to override this.
1350
    """
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1351
    base = os.environ.get('BZR_HOME', None)
1352
    if sys.platform == 'win32':
5598.2.2 by John Arbash Meinel
Change the comment slightly
1353
        # environ variables on Windows are in user encoding/mbcs. So decode
1354
        # 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.
1355
        if base is not None:
1356
            base = base.decode('mbcs')
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1357
        if base is None:
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
1358
            base = win32utils.get_appdata_location_unicode()
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1359
        if base is None:
1360
            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.
1361
            if base is not None:
1362
                base = base.decode('mbcs')
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1363
        if base is None:
2991.2.2 by Vincent Ladeuil
No tests worth adding after upgrading to configobj-4.4.0.
1364
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
1365
                                  ' or HOME set')
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1366
        return osutils.pathjoin(base, 'bazaar', '2.0')
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1367
    elif sys.platform == 'darwin':
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1368
        if base is None:
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1369
            # this takes into account $HOME
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
1370
            base = os.path.expanduser("~")
5519.4.1 by Neil Martinsen-Burrell
spec and first implementation, next tests
1371
        return osutils.pathjoin(base, '.bazaar')
1372
    else:
1373
        if base is None:
5519.4.3 by Neil Martinsen-Burrell
be permissive about using $XDG_CONFIG_HOME/bazaar, but dont complain
1374
1375
            xdg_dir = os.environ.get('XDG_CONFIG_HOME', None)
1376
            if xdg_dir is None:
1377
                xdg_dir = osutils.pathjoin(os.path.expanduser("~"), ".config")
1378
            xdg_dir = osutils.pathjoin(xdg_dir, 'bazaar')
1379
            if osutils.isdir(xdg_dir):
1380
                trace.mutter(
1381
                    "Using configuration in XDG directory %s." % xdg_dir)
1382
                return xdg_dir
1383
1384
            base = os.path.expanduser("~")
5519.4.4 by Neil Martinsen-Burrell
restore ensure_config_dir since XDG_CONFIG_HOME is optional
1385
        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 \
1386
1387
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
def config_filename():
1389
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1390
    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.
1391
1392
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
1393
def locations_config_filename():
1394
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1395
    return osutils.pathjoin(config_dir(), 'locations.conf')
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
1396
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
1397
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1398
def authentication_config_filename():
1399
    """Return per-user authentication ini file filename."""
1400
    return osutils.pathjoin(config_dir(), 'authentication.conf')
1401
1402
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
1403
def user_ignore_config_filename():
1404
    """Return the user default ignore filename"""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
1405
    return osutils.pathjoin(config_dir(), 'ignore')
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
1406
1407
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1408
def crash_dir():
1409
    """Return the directory name to store crash files.
1410
1411
    This doesn't implicitly create it.
1412
4634.128.2 by Martin Pool
Write crash files into /var/crash where apport can see them.
1413
    On Windows it's in the config directory; elsewhere it's /var/crash
4634.128.18 by Martin Pool
Update apport crash tests
1414
    which may be monitored by apport.  It can be overridden by
1415
    $APPORT_CRASH_DIR.
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1416
    """
1417
    if sys.platform == 'win32':
1418
        return osutils.pathjoin(config_dir(), 'Crash')
1419
    else:
4634.128.2 by Martin Pool
Write crash files into /var/crash where apport can see them.
1420
        # XXX: hardcoded in apport_python_hook.py; therefore here too -- mbp
1421
        # 2010-01-31
4634.128.18 by Martin Pool
Update apport crash tests
1422
        return os.environ.get('APPORT_CRASH_DIR', '/var/crash')
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1423
1424
1425
def xdg_cache_dir():
4584.3.23 by Martin Pool
Correction to xdg_cache_dir and add a simple test
1426
    # See http://standards.freedesktop.org/basedir-spec/latest/ar01s03.html
1427
    # Possibly this should be different on Windows?
1428
    e = os.environ.get('XDG_CACHE_DIR', None)
1429
    if e:
1430
        return e
1431
    else:
1432
        return os.path.expanduser('~/.cache')
4584.3.4 by Martin Pool
Add crash_dir and xdg_cache_dir functions
1433
1434
5050.72.1 by Martin Pool
Set email address from /etc/mailname if possible
1435
def _get_default_mail_domain():
1436
    """If possible, return the assumed default email domain.
1437
1438
    :returns: string mail domain, or None.
1439
    """
1440
    if sys.platform == 'win32':
1441
        # No implementation yet; patches welcome
1442
        return None
1443
    try:
1444
        f = open('/etc/mailname')
1445
    except (IOError, OSError), e:
1446
        return None
1447
    try:
1448
        domain = f.read().strip()
1449
        return domain
1450
    finally:
1451
        f.close()
1452
1453
1454
def _auto_user_id():
1455
    """Calculate automatic user identification.
1456
1457
    :returns: (realname, email), either of which may be None if they can't be
1458
    determined.
1459
1460
    Only used when none is set in the environment or the id file.
1461
1462
    This only returns an email address if we can be fairly sure the 
1463
    address is reasonable, ie if /etc/mailname is set on unix.
1464
1465
    This doesn't use the FQDN as the default domain because that may be 
1466
    slow, and it doesn't use the hostname alone because that's not normally 
1467
    a reasonable address.
1468
    """
1469
    if sys.platform == 'win32':
1470
        # No implementation to reliably determine Windows default mail
1471
        # address; please add one.
1472
        return None, None
1473
1474
    default_mail_domain = _get_default_mail_domain()
1475
    if not default_mail_domain:
1476
        return None, None
1477
1478
    import pwd
1479
    uid = os.getuid()
1480
    try:
1481
        w = pwd.getpwuid(uid)
1482
    except KeyError:
1483
        mutter('no passwd entry for uid %d?' % uid)
1484
        return None, None
1485
1486
    # we try utf-8 first, because on many variants (like Linux),
1487
    # /etc/passwd "should" be in utf-8, and because it's unlikely to give
1488
    # false positives.  (many users will have their user encoding set to
1489
    # latin-1, which cannot raise UnicodeError.)
1490
    try:
1491
        gecos = w.pw_gecos.decode('utf-8')
1492
        encoding = 'utf-8'
1493
    except UnicodeError:
1494
        try:
1495
            encoding = osutils.get_user_encoding()
1496
            gecos = w.pw_gecos.decode(encoding)
1497
        except UnicodeError, e:
1498
            mutter("cannot decode passwd entry %s" % w)
1499
            return None, None
1500
    try:
1501
        username = w.pw_name.decode(encoding)
1502
    except UnicodeError, e:
1503
        mutter("cannot decode passwd entry %s" % w)
1504
        return None, None
1505
1506
    comma = gecos.find(',')
1507
    if comma == -1:
1508
        realname = gecos
1509
    else:
1510
        realname = gecos[:comma]
1511
1512
    return realname, (username + '@' + default_mail_domain)
1513
1514
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
1515
def parse_username(username):
1516
    """Parse e-mail username and return a (name, address) tuple."""
1517
    match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
1518
    if match is None:
1519
        return (username, '')
1520
    else:
1521
        return (match.group(1), match.group(2))
1522
1523
1185.16.52 by Martin Pool
- add extract_email_address
1524
def extract_email_address(e):
1525
    """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.
1526
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1527
    That is just the user@domain part, nothing else.
1185.16.52 by Martin Pool
- add extract_email_address
1528
    This part is required to contain only ascii characters.
1529
    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.
1530
1185.16.52 by Martin Pool
- add extract_email_address
1531
    >>> extract_email_address('Jane Tester <jane@test.com>')
1532
    "jane@test.com"
1533
    """
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
1534
    name, email = parse_username(e)
1535
    if not email:
2055.2.2 by John Arbash Meinel
Switch extract_email_address() to use a more specific exception
1536
        raise errors.NoEmailInUsername(e)
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
1537
    return email
1185.35.11 by Aaron Bentley
Added support for branch nicks
1538
1185.85.30 by John Arbash Meinel
Fixing 'bzr push' exposed that IniBasedConfig didn't handle unicode.
1539
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1540
class TreeConfig(IniBasedConfig):
1185.35.11 by Aaron Bentley
Added support for branch nicks
1541
    """Branch configuration data associated with its contents, not location"""
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1542
3408.3.1 by Martin Pool
Remove erroneous handling of branch.conf for RemoteBranch
1543
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
1544
1185.35.11 by Aaron Bentley
Added support for branch nicks
1545
    def __init__(self, branch):
4226.1.5 by Robert Collins
Reinstate the use of the Branch.get_config_file verb.
1546
        self._config = branch._get_config()
1185.35.11 by Aaron Bentley
Added support for branch nicks
1547
        self.branch = branch
1548
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
1549
    def _get_parser(self, file=None):
1550
        if file is not None:
1551
            return IniBasedConfig._get_parser(file)
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1552
        return self._config._get_configobj()
1185.35.11 by Aaron Bentley
Added support for branch nicks
1553
1554
    def get_option(self, name, section=None, default=None):
1555
        self.branch.lock_read()
1556
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1557
            return self._config.get_option(name, section, default)
1185.35.11 by Aaron Bentley
Added support for branch nicks
1558
        finally:
1559
            self.branch.unlock()
1560
1561
    def set_option(self, value, name, section=None):
1562
        """Set a per-branch configuration option"""
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
1563
        # FIXME: We shouldn't need to lock explicitly here but rather rely on
1564
        # higher levels providing the right lock -- vila 20101004
1185.35.11 by Aaron Bentley
Added support for branch nicks
1565
        self.branch.lock_write()
1566
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1567
            self._config.set_option(value, name, section)
1185.35.11 by Aaron Bentley
Added support for branch nicks
1568
        finally:
1569
            self.branch.unlock()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1570
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
1571
    def remove_option(self, option_name, section_name=None):
1572
        # FIXME: We shouldn't need to lock explicitly here but rather rely on
1573
        # higher levels providing the right lock -- vila 20101004
1574
        self.branch.lock_write()
1575
        try:
1576
            self._config.remove_option(option_name, section_name)
1577
        finally:
1578
            self.branch.unlock()
1579
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1580
1581
class AuthenticationConfig(object):
1582
    """The authentication configuration file based on a ini file.
1583
1584
    Implements the authentication.conf file described in
1585
    doc/developers/authentication-ring.txt.
1586
    """
1587
1588
    def __init__(self, _file=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1589
        self._config = None # The ConfigObj
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1590
        if _file is None:
2900.2.24 by Vincent Ladeuil
Review feedback.
1591
            self._filename = authentication_config_filename()
1592
            self._input = self._filename = authentication_config_filename()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1593
        else:
2900.2.24 by Vincent Ladeuil
Review feedback.
1594
            # Tests can provide a string as _file
1595
            self._filename = None
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1596
            self._input = _file
1597
1598
    def _get_config(self):
1599
        if self._config is not None:
1600
            return self._config
1601
        try:
2900.2.22 by Vincent Ladeuil
Polishing.
1602
            # FIXME: Should we validate something here ? Includes: empty
1603
            # sections are useless, at least one of
1604
            # user/password/password_encoding should be defined, etc.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1605
1606
            # Note: the encoding below declares that the file itself is utf-8
1607
            # encoded, but the values in the ConfigObj are always Unicode.
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1608
            self._config = ConfigObj(self._input, encoding='utf-8')
1609
        except configobj.ConfigObjError, e:
1610
            raise errors.ParseConfigError(e.errors, e.config.filename)
1611
        return self._config
1612
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1613
    def _save(self):
1614
        """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.
1615
        conf_dir = os.path.dirname(self._filename)
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1616
        ensure_config_dir_exists(conf_dir)
4708.2.2 by Martin
Workingtree changes sitting around since November, more explict closing of files in bzrlib
1617
        f = file(self._filename, 'wb')
1618
        try:
1619
            self._get_config().write(f)
1620
        finally:
1621
            f.close()
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1622
1623
    def _set_option(self, section_name, option_name, value):
1624
        """Set an authentication configuration option"""
1625
        conf = self._get_config()
1626
        section = conf.get(section_name)
1627
        if section is None:
1628
            conf[section] = {}
1629
            section = conf[section]
1630
        section[option_name] = value
1631
        self._save()
1632
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1633
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
1634
                        realm=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1635
        """Returns the matching credentials from authentication.conf file.
1636
1637
        :param scheme: protocol
1638
1639
        :param host: the server address
1640
1641
        :param port: the associated port (optional)
1642
1643
        :param user: login (optional)
1644
1645
        :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
1646
        
1647
        :param realm: the http authentication realm (optional)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1648
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1649
        :return: A dict containing the matching credentials or None.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1650
           This includes:
1651
           - name: the section name of the credentials in the
1652
             authentication.conf file,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1653
           - 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.
1654
           - scheme: the server protocol,
1655
           - host: the server address,
1656
           - port: the server port (can be None),
1657
           - path: the absolute server path (can be None),
1658
           - realm: the http specific authentication realm (can be None),
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1659
           - password: the decoded password, could be None if the credential
1660
             defines only the user
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1661
           - verify_certificates: https specific, True if the server
1662
             certificate should be verified, False otherwise.
1663
        """
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1664
        credentials = None
1665
        for auth_def_name, auth_def in self._get_config().items():
3418.2.1 by Vincent Ladeuil
Fix #217650 by catching declarations outside sections.
1666
            if type(auth_def) is not configobj.Section:
1667
                raise ValueError("%s defined outside a section" % auth_def_name)
1668
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1669
            a_scheme, a_host, a_user, a_path = map(
1670
                auth_def.get, ['scheme', 'host', 'user', 'path'])
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1671
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1672
            try:
1673
                a_port = auth_def.as_int('port')
1674
            except KeyError:
1675
                a_port = None
2900.2.22 by Vincent Ladeuil
Polishing.
1676
            except ValueError:
1677
                raise ValueError("'port' not numeric in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1678
            try:
1679
                a_verify_certificates = auth_def.as_bool('verify_certificates')
1680
            except KeyError:
1681
                a_verify_certificates = True
2900.2.22 by Vincent Ladeuil
Polishing.
1682
            except ValueError:
1683
                raise ValueError(
1684
                    "'verify_certificates' not boolean in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1685
1686
            # Attempt matching
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1687
            if a_scheme is not None and scheme != a_scheme:
1688
                continue
1689
            if a_host is not None:
1690
                if not (host == a_host
1691
                        or (a_host.startswith('.') and host.endswith(a_host))):
1692
                    continue
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1693
            if a_port is not None and port != a_port:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1694
                continue
1695
            if (a_path is not None and path is not None
1696
                and not path.startswith(a_path)):
1697
                continue
1698
            if (a_user is not None and user is not None
1699
                and a_user != user):
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1700
                # Never contradict the caller about the user to be used
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1701
                continue
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1702
            if a_user is None:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1703
                # Can't find a user
1704
                continue
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1705
            # Prepare a credentials dictionary with additional keys
1706
            # for the credential providers
2900.2.24 by Vincent Ladeuil
Review feedback.
1707
            credentials = dict(name=auth_def_name,
3418.4.2 by Vincent Ladeuil
Fix bug #199440 by taking into account that a section may not
1708
                               user=a_user,
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1709
                               scheme=a_scheme,
1710
                               host=host,
1711
                               port=port,
1712
                               path=path,
1713
                               realm=realm,
3418.4.2 by Vincent Ladeuil
Fix bug #199440 by taking into account that a section may not
1714
                               password=auth_def.get('password', None),
2900.2.24 by Vincent Ladeuil
Review feedback.
1715
                               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
1716
            # Decode the password in the credentials (or get one)
2900.2.22 by Vincent Ladeuil
Polishing.
1717
            self.decode_password(credentials,
1718
                                 auth_def.get('password_encoding', None))
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1719
            if 'auth' in debug.debug_flags:
1720
                trace.mutter("Using authentication section: %r", auth_def_name)
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1721
            break
1722
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1723
        if credentials is None:
1724
            # No credentials were found in authentication.conf, try the fallback
1725
            # credentials stores.
1726
            credentials = credential_store_registry.get_fallback_credentials(
1727
                scheme, host, port, user, path, realm)
1728
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1729
        return credentials
1730
3777.3.2 by Aaron Bentley
Reverse order of scheme and password
1731
    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
1732
                        port=None, path=None, verify_certificates=None,
1733
                        realm=None):
3777.3.1 by Aaron Bentley
Update docs
1734
        """Set authentication credentials for a host.
1735
1736
        Any existing credentials with matching scheme, host, port and path
1737
        will be deleted, regardless of name.
1738
1739
        :param name: An arbitrary name to describe this set of credentials.
1740
        :param host: Name of the host that accepts these credentials.
1741
        :param user: The username portion of these credentials.
1742
        :param scheme: The URL scheme (e.g. ssh, http) the credentials apply
1743
            to.
3777.3.2 by Aaron Bentley
Reverse order of scheme and password
1744
        :param password: Password portion of these credentials.
3777.3.1 by Aaron Bentley
Update docs
1745
        :param port: The IP port on the host that these credentials apply to.
1746
        :param path: A filesystem path on the host that these credentials
1747
            apply to.
1748
        :param verify_certificates: On https, verify server certificates if
1749
            True.
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1750
        :param realm: The http authentication realm (optional).
3777.3.1 by Aaron Bentley
Update docs
1751
        """
3777.1.8 by Aaron Bentley
Commit work-in-progress
1752
        values = {'host': host, 'user': user}
1753
        if password is not None:
1754
            values['password'] = password
1755
        if scheme is not None:
1756
            values['scheme'] = scheme
1757
        if port is not None:
1758
            values['port'] = '%d' % port
1759
        if path is not None:
1760
            values['path'] = path
3777.1.10 by Aaron Bentley
Ensure credentials are stored
1761
        if verify_certificates is not None:
1762
            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
1763
        if realm is not None:
1764
            values['realm'] = realm
3777.1.11 by Aaron Bentley
Ensure changed-name updates clear old values
1765
        config = self._get_config()
1766
        for_deletion = []
1767
        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
1768
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
3777.1.11 by Aaron Bentley
Ensure changed-name updates clear old values
1769
                if existing_values.get(key) != values.get(key):
1770
                    break
1771
            else:
1772
                del config[section]
1773
        config.update({name: values})
3777.1.10 by Aaron Bentley
Ensure credentials are stored
1774
        self._save()
3777.1.8 by Aaron Bentley
Commit work-in-progress
1775
4304.2.1 by Vincent Ladeuil
Fix bug #367726 by reverting some default user handling introduced
1776
    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.
1777
                 prompt=None, ask=False, default=None):
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1778
        """Get a user from authentication file.
1779
1780
        :param scheme: protocol
1781
1782
        :param host: the server address
1783
1784
        :param port: the associated port (optional)
1785
1786
        :param realm: the realm sent by the server (optional)
1787
1788
        :param path: the absolute path on the server (optional)
1789
4222.3.4 by Jelmer Vernooij
Default to getpass.getuser() in AuthenticationConfig.get_user(), but allow
1790
        :param ask: Ask the user if there is no explicitly configured username 
1791
                    (optional)
1792
4304.2.1 by Vincent Ladeuil
Fix bug #367726 by reverting some default user handling introduced
1793
        :param default: The username returned if none is defined (optional).
1794
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1795
        :return: The found user.
1796
        """
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1797
        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
1798
                                           path=path, realm=realm)
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1799
        if credentials is not None:
1800
            user = credentials['user']
1801
        else:
1802
            user = None
4222.3.2 by Jelmer Vernooij
Prompt for user names if they are not in the configuration.
1803
        if user is None:
4222.3.4 by Jelmer Vernooij
Default to getpass.getuser() in AuthenticationConfig.get_user(), but allow
1804
            if ask:
1805
                if prompt is None:
1806
                    # Create a default prompt suitable for most cases
1807
                    prompt = scheme.upper() + ' %(host)s username'
1808
                # Special handling for optional fields in the prompt
1809
                if port is not None:
1810
                    prompt_host = '%s:%d' % (host, port)
1811
                else:
1812
                    prompt_host = host
1813
                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.
1814
            else:
4222.3.10 by Jelmer Vernooij
Avoid using the default username in the case of SMTP.
1815
                user = default
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1816
        return user
1817
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1818
    def get_password(self, scheme, host, user, port=None,
1819
                     realm=None, path=None, prompt=None):
1820
        """Get a password from authentication file or prompt the user for one.
1821
1822
        :param scheme: protocol
1823
1824
        :param host: the server address
1825
1826
        :param port: the associated port (optional)
1827
1828
        :param user: login
1829
1830
        :param realm: the realm sent by the server (optional)
1831
1832
        :param path: the absolute path on the server (optional)
1833
1834
        :return: The found password or the one entered by the user.
1835
        """
4081.1.1 by Jean-Francois Roy
A 'realm' optional argument was added to the get_credentials and set_credentials
1836
        credentials = self.get_credentials(scheme, host, port, user, path,
1837
                                           realm)
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1838
        if credentials is not None:
1839
            password = credentials['password']
3420.1.3 by Vincent Ladeuil
John's review feedback.
1840
            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.
1841
                trace.warning('password ignored in section [%s],'
1842
                              ' use an ssh agent instead'
1843
                              % credentials['name'])
1844
                password = None
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1845
        else:
1846
            password = None
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1847
        # 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).
1848
        if password is None:
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1849
            if prompt is None:
3420.1.2 by Vincent Ladeuil
Fix bug #203186 by ignoring passwords for ssh and warning user.
1850
                # Create a default prompt suitable for most cases
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1851
                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
1852
            # Special handling for optional fields in the prompt
1853
            if port is not None:
1854
                prompt_host = '%s:%d' % (host, port)
1855
            else:
1856
                prompt_host = host
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1857
            password = ui.ui_factory.get_password(prompt,
1858
                                                  host=prompt_host, user=user)
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1859
        return password
1860
2900.2.22 by Vincent Ladeuil
Polishing.
1861
    def decode_password(self, credentials, encoding):
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1862
        try:
1863
            cs = credential_store_registry.get_credential_store(encoding)
1864
        except KeyError:
1865
            raise ValueError('%r is not a known password_encoding' % encoding)
1866
        credentials['password'] = cs.decode_password(credentials)
2900.2.22 by Vincent Ladeuil
Polishing.
1867
        return credentials
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1868
3242.3.17 by Aaron Bentley
Whitespace cleanup
1869
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1870
class CredentialStoreRegistry(registry.Registry):
1871
    """A class that registers credential stores.
1872
1873
    A credential store provides access to credentials via the password_encoding
1874
    field in authentication.conf sections.
1875
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1876
    Except for stores provided by bzr itself, most stores are expected to be
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1877
    provided by plugins that will therefore use
1878
    register_lazy(password_encoding, module_name, member_name, help=help,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1879
    fallback=fallback) to install themselves.
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1880
1881
    A fallback credential store is one that is queried if no credentials can be
1882
    found via authentication.conf.
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1883
    """
1884
1885
    def get_credential_store(self, encoding=None):
1886
        cs = self.get(encoding)
1887
        if callable(cs):
1888
            cs = cs()
1889
        return cs
1890
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1891
    def is_fallback(self, name):
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1892
        """Check if the named credentials store should be used as fallback."""
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1893
        return self.get_info(name)
1894
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1895
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1896
                                 path=None, realm=None):
1897
        """Request credentials from all fallback credentials stores.
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1898
1899
        The first credentials store that can provide credentials wins.
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1900
        """
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1901
        credentials = None
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1902
        for name in self.keys():
4283.1.2 by Jelmer Vernooij
Add tests, NEWS item.
1903
            if not self.is_fallback(name):
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1904
                continue
1905
            cs = self.get_credential_store(name)
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1906
            credentials = cs.get_credentials(scheme, host, port, user,
1907
                                             path, realm)
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1908
            if credentials is not None:
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1909
                # We found some credentials
1910
                break
1911
        return credentials
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1912
1913
    def register(self, key, obj, help=None, override_existing=False,
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1914
                 fallback=False):
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1915
        """Register a new object to a name.
1916
1917
        :param key: This is the key to use to request the object later.
1918
        :param obj: The object to register.
1919
        :param help: Help text for this entry. This may be a string or
1920
                a callable. If it is a callable, it should take two
1921
                parameters (registry, key): this registry and the key that
1922
                the help was registered under.
1923
        :param override_existing: Raise KeyErorr if False and something has
1924
                already been registered for that key. If True, ignore if there
1925
                is an existing key (always register the new value).
1926
        :param fallback: Whether this credential store should be 
1927
                used as fallback.
1928
        """
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1929
        return super(CredentialStoreRegistry,
1930
                     self).register(key, obj, help, info=fallback,
1931
                                    override_existing=override_existing)
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1932
1933
    def register_lazy(self, key, module_name, member_name,
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1934
                      help=None, override_existing=False,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1935
                      fallback=False):
1936
        """Register a new credential store to be loaded on request.
1937
1938
        :param module_name: The python path to the module. Such as 'os.path'.
1939
        :param member_name: The member of the module to return.  If empty or
1940
                None, get() will return the module itself.
1941
        :param help: Help text for this entry. This may be a string or
1942
                a callable.
1943
        :param override_existing: If True, replace the existing object
1944
                with the new one. If False, if there is already something
1945
                registered with the same key, raise a KeyError
1946
        :param fallback: Whether this credential store should be 
1947
                used as fallback.
1948
        """
1949
        return super(CredentialStoreRegistry, self).register_lazy(
1950
            key, module_name, member_name, help,
1951
            info=fallback, override_existing=override_existing)
1952
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1953
1954
credential_store_registry = CredentialStoreRegistry()
1955
1956
1957
class CredentialStore(object):
1958
    """An abstract class to implement storage for credentials"""
1959
1960
    def decode_password(self, credentials):
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1961
        """Returns a clear text password for the provided credentials."""
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1962
        raise NotImplementedError(self.decode_password)
1963
4283.2.1 by Vincent Ladeuil
Add a test and cleanup some PEP8 issues.
1964
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
4283.1.1 by Jelmer Vernooij
Support fallback credential stores.
1965
                        realm=None):
1966
        """Return the matching credentials from this credential store.
1967
1968
        This method is only called on fallback credential stores.
1969
        """
1970
        raise NotImplementedError(self.get_credentials)
1971
1972
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1973
1974
class PlainTextCredentialStore(CredentialStore):
5131.2.1 by Martin
Permit bzrlib to run under python -OO by explictly assigning to __doc__ for user-visible docstrings
1975
    __doc__ = """Plain text credential store for the authentication.conf file"""
3757.3.1 by Vincent Ladeuil
Add credential stores plugging.
1976
1977
    def decode_password(self, credentials):
1978
        """See CredentialStore.decode_password."""
1979
        return credentials['password']
1980
1981
1982
credential_store_registry.register('plain', PlainTextCredentialStore,
1983
                                   help=PlainTextCredentialStore.__doc__)
1984
credential_store_registry.default_key = 'plain'
1985
1986
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
1987
class BzrDirConfig(object):
1988
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
1989
    def __init__(self, bzrdir):
1990
        self._bzrdir = bzrdir
1991
        self._config = bzrdir._get_config()
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1992
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
1993
    def set_default_stack_on(self, value):
1994
        """Set the default stacking location.
1995
1996
        It may be set to a location, or None.
1997
1998
        This policy affects all branches contained by this bzrdir, except for
1999
        those under repositories.
2000
        """
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
2001
        if self._config is None:
2002
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2003
        if value is None:
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2004
            self._config.set_option('', 'default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2005
        else:
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2006
            self._config.set_option(value, 'default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2007
2008
    def get_default_stack_on(self):
2009
        """Return the default stacking location.
2010
2011
        This will either be a location, or None.
2012
2013
        This policy affects all branches contained by this bzrdir, except for
2014
        those under repositories.
2015
        """
4288.1.1 by Robert Collins
Add support for a RemoteBzrDirConfig to support optimising push operations which need to look for default stacking locations.
2016
        if self._config is None:
2017
            return None
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2018
        value = self._config.get_option('default_stack_on')
3242.3.11 by Aaron Bentley
Clean up BzrDirConfig usage
2019
        if value == '':
2020
            value = None
2021
        return value
2022
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2023
2024
class TransportConfig(object):
3242.1.5 by Aaron Bentley
Update per review comments
2025
    """A Config that reads/writes a config file on a Transport.
3242.1.4 by Aaron Bentley
Clean-up
2026
2027
    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.
2028
    that may be associated with a section.  Assigning meaning to these values
2029
    is done at higher levels like TreeConfig.
3242.1.4 by Aaron Bentley
Clean-up
2030
    """
3242.3.14 by Aaron Bentley
Make BzrDirConfig use TransportConfig
2031
2032
    def __init__(self, transport, filename):
2033
        self._transport = transport
2034
        self._filename = filename
2035
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
2036
    def get_option(self, name, section=None, default=None):
2037
        """Return the value associated with a named option.
2038
2039
        :param name: The name of the value
2040
        :param section: The section the option is in (if any)
2041
        :param default: The value to return if the value is not set
2042
        :return: The value or default value
2043
        """
2044
        configobj = self._get_configobj()
2045
        if section is None:
2046
            section_obj = configobj
2047
        else:
2048
            try:
2049
                section_obj = configobj[section]
2050
            except KeyError:
2051
                return default
2052
        return section_obj.get(name, default)
2053
2054
    def set_option(self, value, name, section=None):
2055
        """Set the value associated with a named option.
2056
2057
        :param value: The value to set
2058
        :param name: The name of the value to set
2059
        :param section: The section the option is in (if any)
2060
        """
2061
        configobj = self._get_configobj()
2062
        if section is None:
2063
            configobj[name] = value
2064
        else:
2065
            configobj.setdefault(section, {})[name] = value
2066
        self._set_configobj(configobj)
2067
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2068
    def remove_option(self, option_name, section_name=None):
2069
        configobj = self._get_configobj()
2070
        if section_name is None:
2071
            del configobj[option_name]
2072
        else:
2073
            del configobj[section_name][option_name]
2074
        self._set_configobj(configobj)
2075
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
2076
    def _get_config_file(self):
2077
        try:
4852.1.10 by John Arbash Meinel
Use a StringIO instead, otherwise we get failures with smart server requests.
2078
            return StringIO(self._transport.get_bytes(self._filename))
4288.1.2 by Robert Collins
Create a server verb for doing BzrDir.get_config()
2079
        except errors.NoSuchFile:
2080
            return StringIO()
2081
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
2082
    def _get_configobj(self):
4708.2.1 by Martin
Ensure all files opened by bazaar proper are explicitly closed
2083
        f = self._get_config_file()
2084
        try:
2085
            return ConfigObj(f, encoding='utf-8')
2086
        finally:
2087
            f.close()
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
2088
2089
    def _set_configobj(self, configobj):
2090
        out_file = StringIO()
2091
        configobj.write(out_file)
2092
        out_file.seek(0)
2093
        self._transport.put_file(self._filename, out_file)
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2094
2095
5743.3.11 by Vincent Ladeuil
Config sections only implement read access.
2096
class Section(object):
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2097
    """A section defines a dict of options.
2098
2099
    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.
2100
    options. It is *not* a python dict object though and doesn't try to mimic
2101
    its API.
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2102
    """
2103
2104
    def __init__(self, section_id, options):
2105
        self.id = section_id
2106
        # We re-use the dict-like object received
2107
        self.options = options
2108
2109
    def get(self, name, default=None):
2110
        return self.options.get(name, default)
2111
5743.3.12 by Vincent Ladeuil
Add an ad-hoc __repr__.
2112
    def __repr__(self):
2113
        # Mostly for debugging use
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2114
        return "<config.%s id=%s>" % (self.__class__.__name__, self.id)
5743.3.12 by Vincent Ladeuil
Add an ad-hoc __repr__.
2115
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2116
5743.3.6 by Vincent Ladeuil
Use a name less likely to be reused.
2117
_NewlyCreatedOption = object()
5743.2.2 by Vincent Ladeuil
Add tests for remove.
2118
"""Was the option created during the MutableSection lifetime"""
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2119
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2120
5743.3.11 by Vincent Ladeuil
Config sections only implement read access.
2121
class MutableSection(Section):
5743.3.1 by Vincent Ladeuil
Add a docstring and dates to FIXMEs.
2122
    """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.
2123
2124
    def __init__(self, section_id, options):
2125
        super(MutableSection, self).__init__(section_id, options)
2126
        self.orig = {}
2127
2128
    def set(self, name, value):
5743.2.2 by Vincent Ladeuil
Add tests for remove.
2129
        if name not in self.options:
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2130
            # This is a new option
5743.3.6 by Vincent Ladeuil
Use a name less likely to be reused.
2131
            self.orig[name] = _NewlyCreatedOption
5743.2.3 by Vincent Ladeuil
The option is either new or has an existing value.
2132
        elif name not in self.orig:
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2133
            self.orig[name] = self.get(name, None)
2134
        self.options[name] = value
2135
2136
    def remove(self, name):
2137
        if name not in self.orig:
2138
            self.orig[name] = self.get(name, None)
2139
        del self.options[name]
2140
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2141
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2142
class Store(object):
2143
    """Abstract interface to persistent storage for configuration options."""
2144
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2145
    readonly_section_class = Section
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2146
    mutable_section_class = MutableSection
5743.4.21 by Vincent Ladeuil
All stores should provide _load_from_string to reuse the existing tests.
2147
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2148
    def is_loaded(self):
2149
        """Returns True if the Store has been loaded.
2150
2151
        This is used to implement lazy loading and ensure the persistent
2152
        storage is queried only when needed.
2153
        """
2154
        raise NotImplementedError(self.is_loaded)
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2155
2156
    def load(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2157
        """Loads the Store from persistent storage."""
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2158
        raise NotImplementedError(self.load)
2159
5743.4.21 by Vincent Ladeuil
All stores should provide _load_from_string to reuse the existing tests.
2160
    def _load_from_string(self, str_or_unicode):
2161
        """Create a store from a string in configobj syntax.
2162
2163
        :param str_or_unicode: A string representing the file content. This will
2164
            be encoded to suit store needs internally.
2165
2166
        This is for tests and should not be used in production unless a
2167
        convincing use case can be demonstrated :)
2168
        """
2169
        raise NotImplementedError(self._load_from_string)
2170
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2171
    def save(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2172
        """Saves the Store to persistent storage."""
5743.4.10 by Vincent Ladeuil
Fix copy/paste, bad.
2173
        raise NotImplementedError(self.save)
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2174
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2175
    def external_url(self):
2176
        raise NotImplementedError(self.external_url)
2177
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2178
    def get_sections(self):
2179
        """Returns an ordered iterable of existing sections.
2180
2181
        :returns: An iterable of (name, dict).
2182
        """
2183
        raise NotImplementedError(self.get_sections)
2184
5743.4.2 by Vincent Ladeuil
Stores don't implement set_option, they just provide a mutable section.
2185
    def get_mutable_section(self, section_name=None):
2186
        """Returns the specified mutable section.
2187
2188
        :param section_name: The section identifier
2189
        """
2190
        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).
2191
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2192
    def __repr__(self):
2193
        # Mostly for debugging use
5743.5.17 by Vincent Ladeuil
Use external_url to identify stores.
2194
        return "<config.%s(%s)>" % (self.__class__.__name__,
5743.5.18 by Vincent Ladeuil
Fix typo.
2195
                                    self.external_url())
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2196
2197
2198
class IniFileStore(Store):
2199
    """A config Store using ConfigObj for storage.
2200
2201
    :ivar transport: The transport object where the config file is located.
2202
2203
    :ivar file_name: The config file basename in the transport directory.
2204
2205
    :ivar _config_obj: Private member to hold the ConfigObj instance used to
2206
        serialize/deserialize the config file.
2207
    """
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2208
2209
    def __init__(self, transport, file_name):
2210
        """A config Store using ConfigObj for storage.
2211
2212
        :param transport: The transport object where the config file is located.
2213
2214
        :param file_name: The config file basename in the transport directory.
2215
        """
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2216
        super(IniFileStore, self).__init__()
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2217
        self.transport = transport
2218
        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.
2219
        self._config_obj = None
2220
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2221
    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.
2222
        return self._config_obj != None
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2223
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2224
    def load(self):
2225
        """Load the store from the associated file."""
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2226
        if self.is_loaded():
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2227
            return
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2228
        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.
2229
        self._load_from_string(content)
2230
2231
    def _load_from_string(self, str_or_unicode):
2232
        """Create a config store from a string.
2233
2234
        :param str_or_unicode: A string representing the file content. This will
2235
            be utf-8 encoded internally.
2236
2237
        This is for tests and should not be used in production unless a
2238
        convincing use case can be demonstrated :)
2239
        """
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2240
        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.
2241
            raise AssertionError('Already loaded: %r' % (self._config_obj,))
2242
        co_input = StringIO(str_or_unicode.encode('utf-8'))
5743.4.3 by Vincent Ladeuil
Implement get_mutable_section.
2243
        try:
5743.2.7 by Vincent Ladeuil
Implement loading a config store from a string or a file.
2244
            # The config files are always stored utf8-encoded
2245
            self._config_obj = ConfigObj(co_input, encoding='utf-8')
2246
        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.
2247
            self._config_obj = None
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2248
            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.
2249
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.
2250
    def save(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2251
        if not self.is_loaded():
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2252
            # Nothing to save
2253
            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.
2254
        out = StringIO()
2255
        self._config_obj.write(out)
2256
        self.transport.put_bytes(self.file_name, out.getvalue())
2257
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2258
    def external_url(self):
2259
        # FIXME: external_url should really accepts an optional relpath
2260
        # parameter (bug #750169) :-/ -- vila 2011-04-04
2261
        # The following will do in the interim but maybe we don't want to
2262
        # expose a path here but rather a config ID and its associated
2263
        # object </hand wawe>.
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2264
        return urlutils.join(self.transport.external_url(), self.file_name)
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2265
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2266
    def get_sections(self):
2267
        """Get the configobj section in the file order.
2268
2269
        :returns: An iterable of (name, dict).
2270
        """
2271
        # We need a loaded store
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2272
        try:
2273
            self.load()
2274
        except errors.NoSuchFile:
2275
            # If the file doesn't exist, there is no sections
2276
            return
5743.2.10 by Vincent Ladeuil
Implement store.get_sections() as an iterator and provides the configobj implementation.
2277
        cobj = self._config_obj
2278
        if cobj.scalars:
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2279
            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.
2280
        for section_name in cobj.sections:
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2281
            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.
2282
5743.4.2 by Vincent Ladeuil
Stores don't implement set_option, they just provide a mutable section.
2283
    def get_mutable_section(self, section_name=None):
5743.2.11 by Vincent Ladeuil
Basic store.set implementation.
2284
        # We need a loaded store
5743.4.19 by Vincent Ladeuil
Clarify that only Store.get_mutable_section() can accept an empty file.
2285
        try:
2286
            self.load()
2287
        except errors.NoSuchFile:
2288
            # The file doesn't exist, let's pretend it was empty
2289
            self._load_from_string('')
5743.2.11 by Vincent Ladeuil
Basic store.set implementation.
2290
        if section_name is None:
2291
            section = self._config_obj
2292
        else:
2293
            section = self._config_obj.setdefault(section_name, {})
5743.4.22 by Vincent Ladeuil
Allow daughter classes to use different Section classes if/when needed.
2294
        return self.mutable_section_class(section_name, section)
5743.2.11 by Vincent Ladeuil
Basic store.set implementation.
2295
5743.2.1 by Vincent Ladeuil
Basic tests and implementations for read-only and mutable sections.
2296
5743.4.16 by Vincent Ladeuil
Some doc for the stores.
2297
# Note that LockableConfigObjStore inherits from ConfigObjStore because we need
2298
# unlockable stores for use with objects that can already ensure the locking
2299
# (think branches). If different stores (not based on ConfigObj) are created,
2300
# they may face the same issue.
2301
2302
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2303
class LockableIniFileStore(IniFileStore):
5743.4.9 by Vincent Ladeuil
Implement a LockableConfigObjStore to be able to mimick the actual behaviour.
2304
    """A ConfigObjStore using locks on save to ensure store integrity."""
2305
2306
    def __init__(self, transport, file_name, lock_dir_name=None):
2307
        """A config Store using ConfigObj for storage.
2308
2309
        :param transport: The transport object where the config file is located.
2310
2311
        :param file_name: The config file basename in the transport directory.
2312
        """
2313
        if lock_dir_name is None:
2314
            lock_dir_name = 'lock'
2315
        self.lock_dir_name = lock_dir_name
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2316
        super(LockableIniFileStore, self).__init__(transport, file_name)
5743.4.9 by Vincent Ladeuil
Implement a LockableConfigObjStore to be able to mimick the actual behaviour.
2317
        self._lock = lockdir.LockDir(self.transport, self.lock_dir_name)
2318
2319
    def lock_write(self, token=None):
2320
        """Takes a write lock in the directory containing the config file.
2321
2322
        If the directory doesn't exist it is created.
2323
        """
2324
        # FIXME: This doesn't check the ownership of the created directories as
2325
        # ensure_config_dir_exists does. It should if the transport is local
2326
        # -- vila 2011-04-06
2327
        self.transport.create_prefix()
2328
        return self._lock.lock_write(token)
2329
2330
    def unlock(self):
2331
        self._lock.unlock()
2332
2333
    def break_lock(self):
2334
        self._lock.break_lock()
2335
2336
    @needs_write_lock
2337
    def save(self):
5743.4.25 by Vincent Ladeuil
Address review comments by jelmer and poolie.
2338
        super(LockableIniFileStore, self).save()
5743.4.9 by Vincent Ladeuil
Implement a LockableConfigObjStore to be able to mimick the actual behaviour.
2339
2340
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2341
# FIXME: global, bazaar, shouldn't that be 'user' instead or even
2342
# 'user_defaults' as opposed to 'user_overrides', 'system_defaults'
2343
# (/etc/bzr/bazaar.conf) and 'system_overrides' ? -- vila 2011-04-05
5743.5.15 by Vincent Ladeuil
Mention poolie's point about focusing tests.
2344
2345
# FIXME: Moreover, we shouldn't need classes for these stores either, factory
2346
# functions or a registry will make it easier and clearer for tests, focusing
2347
# on the relevant parts of the API that needs testing -- vila 20110503 (based
2348
# on a poolie's remark)
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2349
class GlobalStore(LockableIniFileStore):
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2350
2351
    def __init__(self, possible_transports=None):
2352
        t = transport.get_transport(config_dir(),
2353
                                    possible_transports=possible_transports)
2354
        super(GlobalStore, self).__init__(t, 'bazaar.conf')
2355
2356
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2357
class LocationStore(LockableIniFileStore):
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2358
2359
    def __init__(self, possible_transports=None):
2360
        t = transport.get_transport(config_dir(),
2361
                                    possible_transports=possible_transports)
5743.5.10 by Vincent Ladeuil
Parametrize the generic tests against the concrete stores.
2362
        super(LocationStore, self).__init__(t, 'locations.conf')
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2363
2364
5743.5.13 by Vincent Ladeuil
Merge config-abstract-store into config-concrete-stores resolving conflicts
2365
class BranchStore(IniFileStore):
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2366
2367
    def __init__(self, branch):
2368
        super(BranchStore, self).__init__(branch.control_transport,
2369
                                          'branch.conf')
2370
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2371
class SectionMatcher(object):
2372
    """Select sections into a given Store.
2373
2374
    This intended to be used to postpone getting an iterable of sections from a
2375
    store.
2376
    """
2377
2378
    def __init__(self, store):
2379
        self.store = store
2380
2381
    def get_sections(self):
5743.2.29 by Vincent Ladeuil
Add doc for the section matchers.
2382
        # 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.
2383
        # sections.
2384
        sections = self.store.get_sections()
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2385
        # Walk the revisions in the order provided
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2386
        for s in sections:
2387
            if self.match(s):
2388
                yield s
2389
2390
    def match(self, secion):
2391
        raise NotImplementedError(self.match)
2392
2393
5743.2.37 by Vincent Ladeuil
Merge config-concrete-stores into config-section-matchers resolving conflicts
2394
class LocationSection(Section):
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2395
2396
    def __init__(self, section, length, extra_path):
2397
        super(LocationSection, self).__init__(section.id, section.options)
2398
        self.length = length
2399
        self.extra_path = extra_path
2400
2401
    def get(self, name, default=None):
2402
        value = super(LocationSection, self).get(name, default)
2403
        if value is not None:
2404
            policy_name = self.get(name + ':policy', None)
2405
            policy = _policy_value.get(policy_name, POLICY_NONE)
2406
            if policy == POLICY_APPENDPATH:
2407
                value = urlutils.join(value, self.extra_path)
2408
        return value
2409
2410
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2411
class LocationMatcher(SectionMatcher):
2412
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2413
    def __init__(self, store, location):
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2414
        super(LocationMatcher, self).__init__(store)
2415
        self.location = location
2416
5743.6.15 by Vincent Ladeuil
Don't pollute _iter_for_location_by_parts.
2417
    def _get_matching_sections(self):
2418
        """Get all sections matching ``location``."""
2419
        # We slightly diverge from LocalConfig here by allowing the no-name
2420
        # section as the most generic one and the lower priority.
2421
        no_name_section = None
2422
        sections = []
2423
        # Filter out the no_name_section so _iter_for_location_by_parts can be
2424
        # used (it assumes all sections have a name).
2425
        for section in self.store.get_sections():
2426
            if section.id is None:
2427
                no_name_section = section
2428
            else:
2429
                sections.append(section)
2430
        # Unfortunately _iter_for_location_by_parts deals with section names so
2431
        # we have to resync.
5743.2.31 by Vincent Ladeuil
Both the length and the section id should be used to sort.
2432
        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
2433
            [s.id for s in sections], self.location)
2434
        iter_sections = iter(sections)
2435
        matching_sections = []
5743.6.15 by Vincent Ladeuil
Don't pollute _iter_for_location_by_parts.
2436
        if no_name_section is not None:
2437
            matching_sections.append(
2438
                # FIXME: ``location`` may need to be normalized for appendpath
2439
                # to work correctly ? -- vila 20110504
2440
                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.
2441
        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
2442
            # a section id is unique for a given store so it's safe to iterate
2443
            # again
2444
            section = iter_sections.next()
2445
            if section_id == section.id:
2446
                matching_sections.append(
2447
                    LocationSection(section, length, extra_path))
5743.6.15 by Vincent Ladeuil
Don't pollute _iter_for_location_by_parts.
2448
        return matching_sections
2449
2450
    def get_sections(self):
2451
        # Override the default implementation as we want to change the order
2452
        matching_sections = self._get_matching_sections()
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2453
        # 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.
2454
        sections = sorted(matching_sections,
2455
                          key=lambda section: (section.length, section.id),
5743.2.24 by Vincent Ladeuil
Complete location config helpers with basic tests.
2456
                          reverse=True)
2457
        # Sections mentioning 'ignore_parents' restrict the selection
2458
        for section in sections:
2459
            # FIXME: We really want to use as_bool below -- vila 2011-04-07
2460
            ignore = section.get('ignore_parents', None)
2461
            if ignore is not None:
2462
                ignore = ui.bool_from_string(ignore)
2463
            if ignore:
2464
                break
2465
            # Finally, we have a valid section
2466
            yield section
5743.2.22 by Vincent Ladeuil
Some minimal SectionMatcher implementation to setup the test infrastucture.
2467
5743.2.13 by Vincent Ladeuil
Trivial implementations for stores with smoke tests.
2468
5743.1.34 by Vincent Ladeuil
Merge config-section-matchers into config-stack resolving conflicts
2469
class Stack(object):
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2470
    """A stack of configurations where an option can be defined"""
2471
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.
2472
    def __init__(self, sections_def, store=None, mutable_section_name=None):
5743.1.11 by Vincent Ladeuil
Properly use MutableSection for write operations.
2473
        """Creates a stack of sections with an optional store for changes.
2474
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.
2475
        :param sections_def: A list of Section or callables that returns an
2476
            iterable of Section. This defines the Sections for the Stack and
2477
            can be called repeatedly if needed.
5743.1.11 by Vincent Ladeuil
Properly use MutableSection for write operations.
2478
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.
2479
        :param store: The optional Store where modifications will be
2480
            recorded. If none is specified, no modifications can be done.
2481
2482
        :param mutable_section_name: The name of the MutableSection where
2483
            changes are recorded. This requires the ``store`` parameter to be
2484
            specified.
5743.1.11 by Vincent Ladeuil
Properly use MutableSection for write operations.
2485
        """
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.
2486
        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.
2487
        self.store = store
2488
        self.mutable_section_name = mutable_section_name
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2489
2490
    def get(self, name):
5743.1.16 by Vincent Ladeuil
Allows empty sections and empty section callables.
2491
        """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.
2492
5743.1.16 by Vincent Ladeuil
Allows empty sections and empty section callables.
2493
        This is where we guarantee that sections coming from Store are loaded
2494
        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.
2495
        option exists or get its value, which in turn may require to discover
2496
        in which sections it can be defined. Both of these (section and option
2497
        existence) require loading the store (even partially).
2498
        """
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.
2499
        # FIXME: No caching of options nor sections yet -- vila 20110503
2500
5743.1.16 by Vincent Ladeuil
Allows empty sections and empty section callables.
2501
        # Ensuring lazy loading is achieved by delaying section matching until
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2502
        # it can't be avoided anymore by using callables to describe (possibly
5743.1.16 by Vincent Ladeuil
Allows empty sections and empty section callables.
2503
        # empty) section 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.
2504
        for section_or_callable in self.sections_def:
5743.1.12 by Vincent Ladeuil
Clarify ConfigStack.get() about lazy evaluation of sections.
2505
            # Each section can expand to multiple ones when a callable is used
2506
            if callable(section_or_callable):
2507
                sections = section_or_callable()
5743.1.9 by Vincent Ladeuil
Fix the issue by allowing delayed section acquisition.
2508
            else:
5743.1.12 by Vincent Ladeuil
Clarify ConfigStack.get() about lazy evaluation of sections.
2509
                sections = [section_or_callable]
2510
            for section in sections:
2511
                value = section.get(name)
2512
                if value is not None:
2513
                    return value
2514
        # No definition was found
2515
        return None
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2516
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.
2517
    def _get_mutable_section(self):
2518
        """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.
2519
5743.1.15 by Vincent Ladeuil
Test and implement ConfigStack.remove.
2520
        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.
2521
        this means we won't load the corresponding store before setting a value
2522
        or deleting an option. In practice the store will often be loaded but
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2523
        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.
2524
        """
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.
2525
        section = self.store.get_mutable_section(self.mutable_section_name)
2526
        return section
2527
2528
    def set(self, name, value):
2529
        """Set a new value for the option."""
2530
        section = self._get_mutable_section()
5743.1.13 by Vincent Ladeuil
Better explain lazy loading and make sure the mutable section respect the design.
2531
        section.set(name, value)
5743.1.7 by Vincent Ladeuil
Simple set implementation.
2532
5743.1.15 by Vincent Ladeuil
Test and implement ConfigStack.remove.
2533
    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.
2534
        """Remove an existing option."""
2535
        section = self._get_mutable_section()
5743.1.15 by Vincent Ladeuil
Test and implement ConfigStack.remove.
2536
        section.remove(name)
2537
5743.1.34 by Vincent Ladeuil
Merge config-section-matchers into config-stack resolving conflicts
2538
    def __repr__(self):
2539
        # Mostly for debugging use
2540
        return "<config.%s(%s)>" % (self.__class__.__name__, id(self))
2541
5743.1.1 by Vincent Ladeuil
Start implementing a config stack.
2542
5743.6.11 by Vincent Ladeuil
Start parameterizing stack tests.
2543
class GlobalStack(Stack):
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2544
2545
    def __init__(self):
2546
        # Get a GlobalStore
2547
        gstore = GlobalStore()
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2548
        super(GlobalStack, self).__init__([gstore.get_sections], gstore)
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2549
2550
5743.6.11 by Vincent Ladeuil
Start parameterizing stack tests.
2551
class LocationStack(Stack):
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2552
2553
    def __init__(self, location):
2554
        lstore = LocationStore()
2555
        matcher = LocationMatcher(lstore, location)
2556
        gstore = GlobalStore()
2557
        super(LocationStack, self).__init__(
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2558
            [matcher.get_sections, gstore.get_sections], lstore)
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2559
2560
5743.6.11 by Vincent Ladeuil
Start parameterizing stack tests.
2561
class BranchStack(Stack):
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2562
2563
    def __init__(self, branch):
2564
        bstore = BranchStore(branch)
2565
        lstore = LocationStore()
2566
        matcher = LocationMatcher(lstore, branch.base)
2567
        gstore = GlobalStore()
2568
        super(BranchStack, self).__init__(
2569
            [matcher.get_sections, bstore.get_sections, gstore.get_sections],
5743.6.14 by Vincent Ladeuil
Parametrize the Stack tests.
2570
            bstore)
5743.6.1 by Vincent Ladeuil
Outline concrete stacks and basic smoke tests.
2571
2572
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2573
class cmd_config(commands.Command):
5447.4.19 by Vincent Ladeuil
Add some more documentation.
2574
    __doc__ = """Display, set or remove a configuration option.
2575
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2576
    Display the active value for a given option.
2577
2578
    If --all is specified, NAME is interpreted as a regular expression and all
2579
    matching options are displayed mentioning their scope. The active value
2580
    that bzr will take into account is the first one displayed for each option.
2581
2582
    If no NAME is given, --all .* is implied.
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2583
5447.4.19 by Vincent Ladeuil
Add some more documentation.
2584
    Setting a value is achieved by using name=value without spaces. The value
2585
    is set in the most relevant scope and can be checked by displaying the
2586
    option again.
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2587
    """
2588
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2589
    takes_args = ['name?']
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2590
2591
    takes_options = [
2592
        'directory',
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2593
        # FIXME: This should be a registry option so that plugins can register
2594
        # their own config files (or not) -- vila 20101002
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2595
        commands.Option('scope', help='Reduce the scope to the specified'
2596
                        ' configuration file',
2597
                        type=unicode),
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2598
        commands.Option('all',
2599
            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.
2600
            ),
5447.4.8 by Vincent Ladeuil
Make the test properly fail and provide a fake implementation for ``bzr config --remove opt_name``.
2601
        commands.Option('remove', help='Remove the option from'
2602
                        ' the configuration file'),
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2603
        ]
2604
2605
    @commands.display_command
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2606
    def run(self, name=None, all=False, directory=None, scope=None,
2607
            remove=False):
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2608
        if directory is None:
2609
            directory = '.'
2610
        directory = urlutils.normalize_url(directory)
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2611
        if remove and all:
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2612
            raise errors.BzrError(
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2613
                '--all and --remove are mutually exclusive.')
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2614
        elif remove:
2615
            # Delete the option in the given scope
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2616
            self._remove_config_option(name, directory, scope)
2617
        elif name is None:
2618
            # Defaults to all options
2619
            self._show_matching_options('.*', directory, scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2620
        else:
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2621
            try:
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2622
                name, value = name.split('=', 1)
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2623
            except ValueError:
2624
                # Display the option(s) value(s)
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2625
                if all:
2626
                    self._show_matching_options(name, directory, scope)
2627
                else:
2628
                    self._show_value(name, directory, scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2629
            else:
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2630
                if all:
2631
                    raise errors.BzrError(
2632
                        'Only one option can be set.')
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2633
                # Set the option value
2634
                self._set_config_option(name, value, directory, scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2635
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2636
    def _get_configs(self, directory, scope=None):
2637
        """Iterate the configurations specified by ``directory`` and ``scope``.
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2638
2639
        :param directory: Where the configurations are derived from.
2640
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2641
        :param scope: A specific config to start from.
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2642
        """
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2643
        if scope is not None:
2644
            if scope == 'bazaar':
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2645
                yield GlobalConfig()
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2646
            elif scope == 'locations':
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2647
                yield LocationConfig(directory)
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2648
            elif scope == 'branch':
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2649
                (_, br, _) = bzrdir.BzrDir.open_containing_tree_or_branch(
2650
                    directory)
2651
                yield br.get_config()
2652
        else:
2653
            try:
2654
                (_, br, _) = bzrdir.BzrDir.open_containing_tree_or_branch(
2655
                    directory)
2656
                yield br.get_config()
2657
            except errors.NotBranchError:
2658
                yield LocationConfig(directory)
2659
                yield GlobalConfig()
2660
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2661
    def _show_value(self, name, directory, scope):
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2662
        displayed = False
2663
        for c in self._get_configs(directory, scope):
2664
            if displayed:
2665
                break
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2666
            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
2667
                if name == oname:
5533.1.3 by Vincent Ladeuil
Tweak comment as per poolie's suggestion.
2668
                    # 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
2669
5533.1.3 by Vincent Ladeuil
Tweak comment as per poolie's suggestion.
2670
                    # FIXME: We need to use get_user_option to take policies
2671
                    # 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
2672
                    # too (hence the two for loops), this needs a better API
2673
                    # -- vila 20101117
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2674
                    value = c.get_user_option(name)
2675
                    # Quote the value appropriately
2676
                    value = parser._quote(value)
2677
                    self.outf.write('%s\n' % (value,))
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2678
                    displayed = True
2679
                    break
2680
        if not displayed:
2681
            raise errors.NoSuchConfigOption(name)
2682
2683
    def _show_matching_options(self, name, directory, scope):
2684
        name = re.compile(name)
2685
        # We want any error in the regexp to be raised *now* so we need to
2686
        # avoid the delay introduced by the lazy regexp.
2687
        name._compile_and_collapse()
2688
        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.
2689
        cur_section = None
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2690
        for c in self._get_configs(directory, scope):
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2691
            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
2692
                if name.search(oname):
2693
                    if cur_conf_id != conf_id:
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2694
                        # Explain where the options are defined
5447.4.3 by Vincent Ladeuil
Simplify code and design by only defining get_options() where relevant.
2695
                        self.outf.write('%s:\n' % (conf_id,))
2696
                        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.
2697
                        cur_section = None
2698
                    if (section not in (None, 'DEFAULT')
2699
                        and cur_section != section):
2700
                        # Display the section if it's not the default (or only)
2701
                        # one.
5533.2.1 by Vincent Ladeuil
``bzr config`` properly displays list values
2702
                        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.
2703
                        cur_section = section
5506.2.3 by Vincent Ladeuil
Take review comments into account and drive-by fix bug #670251
2704
                    self.outf.write('  %s = %s\n' % (oname, value))
5447.4.2 by Vincent Ladeuil
Implement the 'brz config' command. Read-only.
2705
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2706
    def _set_config_option(self, name, value, directory, scope):
2707
        for conf in self._get_configs(directory, scope):
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2708
            conf.set_user_option(name, value)
2709
            break
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2710
        else:
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2711
            raise errors.NoSuchConfig(scope)
5447.4.5 by Vincent Ladeuil
Implement ``bzr config option=value``.
2712
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2713
    def _remove_config_option(self, name, directory, scope):
5506.2.1 by Vincent Ladeuil
Implements ``bzr config --active option`` displaying only the value.
2714
        if name is None:
2715
            raise errors.BzrCommandError(
2716
                '--remove expects an option to remove.')
5447.4.9 by Vincent Ladeuil
Refactor under tests umbrella.
2717
        removed = False
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2718
        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.
2719
            for (section_name, section, conf_id) in conf._get_sections():
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2720
                if scope is not None and conf_id != scope:
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2721
                    # Not the right configuration file
2722
                    continue
2723
                if name in section:
5447.4.16 by Vincent Ladeuil
Use config_id instead of id as suggested by poolie.
2724
                    if conf_id != conf.config_id():
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2725
                        conf = self._get_configs(directory, conf_id).next()
2726
                    # We use the first section in the first config where the
2727
                    # option is defined to remove it
2728
                    conf.remove_user_option(name, section_name)
2729
                    removed = True
2730
                    break
2731
            break
2732
        else:
5447.4.17 by Vincent Ladeuil
Rename config --force to config --scope.
2733
            raise errors.NoSuchConfig(scope)
5447.4.11 by Vincent Ladeuil
Implement ``bzr config --remove <option>``.
2734
        if not removed:
2735
            raise errors.NoSuchConfigOption(name)