~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Aaron Bentley
  • Date: 2009-03-24 15:47:32 UTC
  • mto: This revision was merged to the branch mainline in revision 4241.
  • Revision ID: aaron@aaronbentley.com-20090324154732-bwkvi4dx3o90a7dq
Add output, emit minimal inventory delta.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 by Canonical Ltd
 
1
# Copyright (C) 2005, 2007, 2008 Canonical Ltd
2
2
#   Authors: Robert Collins <robert.collins@canonical.com>
 
3
#            and others
3
4
#
4
5
# This program is free software; you can redistribute it and/or modify
5
6
# it under the terms of the GNU General Public License as published by
13
14
#
14
15
# You should have received a copy of the GNU General Public License
15
16
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
17
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
18
 
18
19
"""Configuration that affects the behaviour of Bazaar.
19
20
 
20
21
Currently this configuration resides in ~/.bazaar/bazaar.conf
21
 
and ~/.bazaar/branches.conf, which is written to by bzr.
 
22
and ~/.bazaar/locations.conf, which is written to by bzr.
22
23
 
23
24
In bazaar.conf the following options may be set:
24
25
[DEFAULT]
29
30
gpg_signing_command=name-of-program
30
31
log_format=name-of-format
31
32
 
32
 
in branches.conf, you specify the url of a branch and options for it.
 
33
in locations.conf, you specify the url of a branch and options for it.
33
34
Wildcards may be used - * and ? as normal in shell completion. Options
34
 
set in both bazaar.conf and branches.conf are overriden by the branches.conf
 
35
set in both bazaar.conf and locations.conf are overridden by the locations.conf
35
36
setting.
36
37
[/home/robertc/source]
37
38
recurse=False|True(default)
38
39
email= as above
39
 
check_signatures= as abive 
 
40
check_signatures= as above
40
41
create_signatures= as above.
41
42
 
42
43
explanation of options
44
45
editor - this option sets the pop up editor to use during commits.
45
46
email - this option sets the user id bzr will use when committing.
46
47
check_signatures - this option controls whether bzr will require good gpg
47
 
                   signatures, ignore them, or check them if they are 
 
48
                   signatures, ignore them, or check them if they are
48
49
                   present.
49
 
create_signatures - this option controls whether bzr will always create 
 
50
create_signatures - this option controls whether bzr will always create
50
51
                    gpg signatures, never create them, or create them if the
51
52
                    branch is configured to require them.
52
 
                    NB: This option is planned, but not implemented yet.
53
 
log_format - This options set the default log format.  Options are long, 
54
 
             short, line, or a plugin can register new formats
 
53
log_format - this option sets the default log format.  Possible values are
 
54
             long, short, line, or a plugin can register new formats.
55
55
 
56
56
In bazaar.conf you can also define aliases in the ALIASES sections, example
57
57
 
62
62
up=pull
63
63
"""
64
64
 
65
 
 
66
 
import errno
67
65
import os
68
66
import sys
 
67
 
 
68
from bzrlib.lazy_import import lazy_import
 
69
lazy_import(globals(), """
 
70
import errno
69
71
from fnmatch import fnmatch
70
72
import re
 
73
from cStringIO import StringIO
71
74
 
72
75
import bzrlib
73
 
import bzrlib.errors as errors
74
 
from bzrlib.osutils import pathjoin
75
 
from bzrlib.trace import mutter
76
 
import bzrlib.util.configobj.configobj as configobj
77
 
from StringIO import StringIO
 
76
from bzrlib import (
 
77
    debug,
 
78
    errors,
 
79
    mail_client,
 
80
    osutils,
 
81
    registry,
 
82
    symbol_versioning,
 
83
    trace,
 
84
    ui,
 
85
    urlutils,
 
86
    win32utils,
 
87
    )
 
88
from bzrlib.util.configobj import configobj
 
89
""")
 
90
 
78
91
 
79
92
CHECK_IF_POSSIBLE=0
80
93
CHECK_ALWAYS=1
81
94
CHECK_NEVER=2
82
95
 
83
96
 
84
 
class ConfigObj(configobj.ConfigObj):
85
 
 
86
 
    def get_bool(self, section, key):
87
 
        return self[section].as_bool(key)
88
 
 
89
 
    def get_value(self, section, name):
90
 
        # Try [] for the old DEFAULT section.
91
 
        if section == "DEFAULT":
92
 
            try:
93
 
                return self[name]
94
 
            except KeyError:
95
 
                pass
96
 
        return self[section][name]
 
97
SIGN_WHEN_REQUIRED=0
 
98
SIGN_ALWAYS=1
 
99
SIGN_NEVER=2
 
100
 
 
101
 
 
102
POLICY_NONE = 0
 
103
POLICY_NORECURSE = 1
 
104
POLICY_APPENDPATH = 2
 
105
 
 
106
_policy_name = {
 
107
    POLICY_NONE: None,
 
108
    POLICY_NORECURSE: 'norecurse',
 
109
    POLICY_APPENDPATH: 'appendpath',
 
110
    }
 
111
_policy_value = {
 
112
    None: POLICY_NONE,
 
113
    'none': POLICY_NONE,
 
114
    'norecurse': POLICY_NORECURSE,
 
115
    'appendpath': POLICY_APPENDPATH,
 
116
    }
 
117
 
 
118
 
 
119
STORE_LOCATION = POLICY_NONE
 
120
STORE_LOCATION_NORECURSE = POLICY_NORECURSE
 
121
STORE_LOCATION_APPENDPATH = POLICY_APPENDPATH
 
122
STORE_BRANCH = 3
 
123
STORE_GLOBAL = 4
 
124
 
 
125
_ConfigObj = None
 
126
def ConfigObj(*args, **kwargs):
 
127
    global _ConfigObj
 
128
    if _ConfigObj is None:
 
129
        class ConfigObj(configobj.ConfigObj):
 
130
 
 
131
            def get_bool(self, section, key):
 
132
                return self[section].as_bool(key)
 
133
 
 
134
            def get_value(self, section, name):
 
135
                # Try [] for the old DEFAULT section.
 
136
                if section == "DEFAULT":
 
137
                    try:
 
138
                        return self[name]
 
139
                    except KeyError:
 
140
                        pass
 
141
                return self[section][name]
 
142
        _ConfigObj = ConfigObj
 
143
    return _ConfigObj(*args, **kwargs)
97
144
 
98
145
 
99
146
class Config(object):
103
150
        """Get the users pop up editor."""
104
151
        raise NotImplementedError
105
152
 
 
153
    def get_mail_client(self):
 
154
        """Get a mail client to use"""
 
155
        selected_client = self.get_user_option('mail_client')
 
156
        _registry = mail_client.mail_client_registry
 
157
        try:
 
158
            mail_client_class = _registry.get(selected_client)
 
159
        except KeyError:
 
160
            raise errors.UnknownMailClient(selected_client)
 
161
        return mail_client_class(self)
 
162
 
106
163
    def _get_signature_checking(self):
107
164
        """Template method to override signature checking policy."""
108
165
 
 
166
    def _get_signing_policy(self):
 
167
        """Template method to override signature creation policy."""
 
168
 
109
169
    def _get_user_option(self, option_name):
110
170
        """Template method to provide a user option."""
111
171
        return None
156
216
 
157
217
    def username(self):
158
218
        """Return email-style username.
159
 
    
 
219
 
160
220
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
161
 
        
162
 
        $BZREMAIL can be set to override this, then
 
221
 
 
222
        $BZR_EMAIL can be set to override this (as well as the
 
223
        deprecated $BZREMAIL), then
163
224
        the concrete policy type is checked, and finally
164
225
        $EMAIL is examined.
165
226
        If none is found, a reasonable default is (hopefully)
166
227
        created.
167
 
    
 
228
 
168
229
        TODO: Check it's reasonably well-formed.
169
230
        """
170
 
        v = os.environ.get('BZREMAIL')
 
231
        v = os.environ.get('BZR_EMAIL')
171
232
        if v:
172
 
            return v.decode(bzrlib.user_encoding)
173
 
    
 
233
            return v.decode(osutils.get_user_encoding())
 
234
 
174
235
        v = self._get_user_id()
175
236
        if v:
176
237
            return v
177
 
        
 
238
 
178
239
        v = os.environ.get('EMAIL')
179
240
        if v:
180
 
            return v.decode(bzrlib.user_encoding)
 
241
            return v.decode(osutils.get_user_encoding())
181
242
 
182
243
        name, email = _auto_user_id()
183
244
        if name:
192
253
            return policy
193
254
        return CHECK_IF_POSSIBLE
194
255
 
 
256
    def signing_policy(self):
 
257
        """What is the current policy for signature checking?."""
 
258
        policy = self._get_signing_policy()
 
259
        if policy is not None:
 
260
            return policy
 
261
        return SIGN_WHEN_REQUIRED
 
262
 
195
263
    def signature_needed(self):
196
264
        """Is a signature needed when committing ?."""
197
 
        policy = self._get_signature_checking()
198
 
        if policy == CHECK_ALWAYS:
 
265
        policy = self._get_signing_policy()
 
266
        if policy is None:
 
267
            policy = self._get_signature_checking()
 
268
            if policy is not None:
 
269
                trace.warning("Please use create_signatures,"
 
270
                              " not check_signatures to set signing policy.")
 
271
            if policy == CHECK_ALWAYS:
 
272
                return True
 
273
        elif policy == SIGN_ALWAYS:
199
274
            return True
200
275
        return False
201
276
 
205
280
    def _get_alias(self, value):
206
281
        pass
207
282
 
 
283
    def get_nickname(self):
 
284
        return self._get_nickname()
 
285
 
 
286
    def _get_nickname(self):
 
287
        return None
 
288
 
 
289
    def get_bzr_remote_path(self):
 
290
        try:
 
291
            return os.environ['BZR_REMOTE_PATH']
 
292
        except KeyError:
 
293
            path = self.get_user_option("bzr_remote_path")
 
294
            if path is None:
 
295
                path = 'bzr'
 
296
            return path
 
297
 
208
298
 
209
299
class IniBasedConfig(Config):
210
300
    """A configuration policy that draws from ini files."""
222
312
            raise errors.ParseConfigError(e.errors, e.config.filename)
223
313
        return self._parser
224
314
 
 
315
    def _get_matching_sections(self):
 
316
        """Return an ordered list of (section_name, extra_path) pairs.
 
317
 
 
318
        If the section contains inherited configuration, extra_path is
 
319
        a string containing the additional path components.
 
320
        """
 
321
        section = self._get_section()
 
322
        if section is not None:
 
323
            return [(section, '')]
 
324
        else:
 
325
            return []
 
326
 
225
327
    def _get_section(self):
226
328
        """Override this to define the section used by the config."""
227
329
        return "DEFAULT"
228
330
 
 
331
    def _get_option_policy(self, section, option_name):
 
332
        """Return the policy for the given (section, option_name) pair."""
 
333
        return POLICY_NONE
 
334
 
229
335
    def _get_signature_checking(self):
230
336
        """See Config._get_signature_checking."""
231
337
        policy = self._get_user_option('check_signatures')
232
338
        if policy:
233
339
            return self._string_to_signature_policy(policy)
234
340
 
 
341
    def _get_signing_policy(self):
 
342
        """See Config._get_signing_policy"""
 
343
        policy = self._get_user_option('create_signatures')
 
344
        if policy:
 
345
            return self._string_to_signing_policy(policy)
 
346
 
235
347
    def _get_user_id(self):
236
348
        """Get the user id from the 'email' key in the current section."""
237
349
        return self._get_user_option('email')
238
350
 
239
351
    def _get_user_option(self, option_name):
240
352
        """See Config._get_user_option."""
241
 
        try:
242
 
            return self._get_parser().get_value(self._get_section(),
243
 
                                                option_name)
244
 
        except KeyError:
245
 
            pass
 
353
        for (section, extra_path) in self._get_matching_sections():
 
354
            try:
 
355
                value = self._get_parser().get_value(section, option_name)
 
356
            except KeyError:
 
357
                continue
 
358
            policy = self._get_option_policy(section, option_name)
 
359
            if policy == POLICY_NONE:
 
360
                return value
 
361
            elif policy == POLICY_NORECURSE:
 
362
                # norecurse items only apply to the exact path
 
363
                if extra_path:
 
364
                    continue
 
365
                else:
 
366
                    return value
 
367
            elif policy == POLICY_APPENDPATH:
 
368
                if extra_path:
 
369
                    value = urlutils.join(value, extra_path)
 
370
                return value
 
371
            else:
 
372
                raise AssertionError('Unexpected config policy %r' % policy)
 
373
        else:
 
374
            return None
246
375
 
247
376
    def _gpg_signing_command(self):
248
377
        """See Config.gpg_signing_command."""
256
385
        super(IniBasedConfig, self).__init__()
257
386
        self._get_filename = get_filename
258
387
        self._parser = None
259
 
        
 
388
 
260
389
    def _post_commit(self):
261
390
        """See Config.post_commit."""
262
391
        return self._get_user_option('post_commit')
272
401
        raise errors.BzrError("Invalid signatures policy '%s'"
273
402
                              % signature_string)
274
403
 
 
404
    def _string_to_signing_policy(self, signature_string):
 
405
        """Convert a string to a signing policy."""
 
406
        if signature_string.lower() == 'when-required':
 
407
            return SIGN_WHEN_REQUIRED
 
408
        if signature_string.lower() == 'never':
 
409
            return SIGN_NEVER
 
410
        if signature_string.lower() == 'always':
 
411
            return SIGN_ALWAYS
 
412
        raise errors.BzrError("Invalid signing policy '%s'"
 
413
                              % signature_string)
 
414
 
275
415
    def _get_alias(self, value):
276
416
        try:
277
 
            return self._get_parser().get_value("ALIASES", 
 
417
            return self._get_parser().get_value("ALIASES",
278
418
                                                value)
279
419
        except KeyError:
280
420
            pass
281
421
 
 
422
    def _get_nickname(self):
 
423
        return self.get_user_option('nickname')
 
424
 
282
425
 
283
426
class GlobalConfig(IniBasedConfig):
284
427
    """The configuration that should be used for a specific location."""
289
432
    def __init__(self):
290
433
        super(GlobalConfig, self).__init__(config_filename)
291
434
 
 
435
    def set_user_option(self, option, value):
 
436
        """Save option and its value in the configuration."""
 
437
        self._set_option(option, value, 'DEFAULT')
 
438
 
 
439
    def get_aliases(self):
 
440
        """Return the aliases section."""
 
441
        if 'ALIASES' in self._get_parser():
 
442
            return self._get_parser()['ALIASES']
 
443
        else:
 
444
            return {}
 
445
 
 
446
    def set_alias(self, alias_name, alias_command):
 
447
        """Save the alias in the configuration."""
 
448
        self._set_option(alias_name, alias_command, 'ALIASES')
 
449
 
 
450
    def unset_alias(self, alias_name):
 
451
        """Unset an existing alias."""
 
452
        aliases = self._get_parser().get('ALIASES')
 
453
        if not aliases or alias_name not in aliases:
 
454
            raise errors.NoSuchAlias(alias_name)
 
455
        del aliases[alias_name]
 
456
        self._write_config_file()
 
457
 
 
458
    def _set_option(self, option, value, section):
 
459
        # FIXME: RBC 20051029 This should refresh the parser and also take a
 
460
        # file lock on bazaar.conf.
 
461
        conf_dir = os.path.dirname(self._get_filename())
 
462
        ensure_config_dir_exists(conf_dir)
 
463
        self._get_parser().setdefault(section, {})[option] = value
 
464
        self._write_config_file()
 
465
 
 
466
    def _write_config_file(self):
 
467
        f = open(self._get_filename(), 'wb')
 
468
        self._get_parser().write(f)
 
469
        f.close()
 
470
 
292
471
 
293
472
class LocationConfig(IniBasedConfig):
294
473
    """A configuration object that gives the policy for a location."""
295
474
 
296
475
    def __init__(self, location):
297
 
        super(LocationConfig, self).__init__(branches_config_filename)
298
 
        self._global_config = None
 
476
        name_generator = locations_config_filename
 
477
        if (not os.path.exists(name_generator()) and
 
478
                os.path.exists(branches_config_filename())):
 
479
            if sys.platform == 'win32':
 
480
                trace.warning('Please rename %s to %s'
 
481
                              % (branches_config_filename(),
 
482
                                 locations_config_filename()))
 
483
            else:
 
484
                trace.warning('Please rename ~/.bazaar/branches.conf'
 
485
                              ' to ~/.bazaar/locations.conf')
 
486
            name_generator = branches_config_filename
 
487
        super(LocationConfig, self).__init__(name_generator)
 
488
        # local file locations are looked up by local path, rather than
 
489
        # by file url. This is because the config file is a user
 
490
        # file, and we would rather not expose the user to file urls.
 
491
        if location.startswith('file://'):
 
492
            location = urlutils.local_path_from_url(location)
299
493
        self.location = location
300
494
 
301
 
    def _get_global_config(self):
302
 
        if self._global_config is None:
303
 
            self._global_config = GlobalConfig()
304
 
        return self._global_config
305
 
 
306
 
    def _get_section(self):
307
 
        """Get the section we should look in for config items.
308
 
 
309
 
        Returns None if none exists. 
310
 
        TODO: perhaps return a NullSection that thunks through to the 
311
 
              global config.
312
 
        """
 
495
    def _get_matching_sections(self):
 
496
        """Return an ordered list of section names matching this location."""
313
497
        sections = self._get_parser()
314
498
        location_names = self.location.split('/')
315
499
        if self.location.endswith('/'):
316
500
            del location_names[-1]
317
501
        matches=[]
318
502
        for section in sections:
319
 
            section_names = section.split('/')
 
503
            # location is a local path if possible, so we need
 
504
            # to convert 'file://' urls to local paths if necessary.
 
505
            # This also avoids having file:///path be a more exact
 
506
            # match than '/path'.
 
507
            if section.startswith('file://'):
 
508
                section_path = urlutils.local_path_from_url(section)
 
509
            else:
 
510
                section_path = section
 
511
            section_names = section_path.split('/')
320
512
            if section.endswith('/'):
321
513
                del section_names[-1]
322
514
            names = zip(location_names, section_names)
331
523
            # if section is longer, no match.
332
524
            if len(section_names) > len(location_names):
333
525
                continue
334
 
            # if path is longer, and recurse is not true, no match
335
 
            if len(section_names) < len(location_names):
336
 
                try:
337
 
                    if not self._get_parser()[section].as_bool('recurse'):
338
 
                        continue
339
 
                except KeyError:
340
 
                    pass
341
 
            matches.append((len(section_names), section))
342
 
        if not len(matches):
343
 
            return None
 
526
            matches.append((len(section_names), section,
 
527
                            '/'.join(location_names[len(section_names):])))
344
528
        matches.sort(reverse=True)
345
 
        return matches[0][1]
346
 
 
347
 
    def _gpg_signing_command(self):
348
 
        """See Config.gpg_signing_command."""
349
 
        command = super(LocationConfig, self)._gpg_signing_command()
350
 
        if command is not None:
351
 
            return command
352
 
        return self._get_global_config()._gpg_signing_command()
353
 
 
354
 
    def _log_format(self):
355
 
        """See Config.log_format."""
356
 
        command = super(LocationConfig, self)._log_format()
357
 
        if command is not None:
358
 
            return command
359
 
        return self._get_global_config()._log_format()
360
 
 
361
 
    def _get_user_id(self):
362
 
        user_id = super(LocationConfig, self)._get_user_id()
363
 
        if user_id is not None:
364
 
            return user_id
365
 
        return self._get_global_config()._get_user_id()
366
 
 
367
 
    def _get_user_option(self, option_name):
368
 
        """See Config._get_user_option."""
369
 
        option_value = super(LocationConfig, 
370
 
                             self)._get_user_option(option_name)
371
 
        if option_value is not None:
372
 
            return option_value
373
 
        return self._get_global_config()._get_user_option(option_name)
374
 
 
375
 
    def _get_signature_checking(self):
376
 
        """See Config._get_signature_checking."""
377
 
        check = super(LocationConfig, self)._get_signature_checking()
378
 
        if check is not None:
379
 
            return check
380
 
        return self._get_global_config()._get_signature_checking()
381
 
 
382
 
    def _post_commit(self):
383
 
        """See Config.post_commit."""
384
 
        hook = self._get_user_option('post_commit')
385
 
        if hook is not None:
386
 
            return hook
387
 
        return self._get_global_config()._post_commit()
388
 
 
389
 
    def set_user_option(self, option, value):
 
529
        sections = []
 
530
        for (length, section, extra_path) in matches:
 
531
            sections.append((section, extra_path))
 
532
            # should we stop looking for parent configs here?
 
533
            try:
 
534
                if self._get_parser()[section].as_bool('ignore_parents'):
 
535
                    break
 
536
            except KeyError:
 
537
                pass
 
538
        return sections
 
539
 
 
540
    def _get_option_policy(self, section, option_name):
 
541
        """Return the policy for the given (section, option_name) pair."""
 
542
        # check for the old 'recurse=False' flag
 
543
        try:
 
544
            recurse = self._get_parser()[section].as_bool('recurse')
 
545
        except KeyError:
 
546
            recurse = True
 
547
        if not recurse:
 
548
            return POLICY_NORECURSE
 
549
 
 
550
        policy_key = option_name + ':policy'
 
551
        try:
 
552
            policy_name = self._get_parser()[section][policy_key]
 
553
        except KeyError:
 
554
            policy_name = None
 
555
 
 
556
        return _policy_value[policy_name]
 
557
 
 
558
    def _set_option_policy(self, section, option_name, option_policy):
 
559
        """Set the policy for the given option name in the given section."""
 
560
        # The old recurse=False option affects all options in the
 
561
        # section.  To handle multiple policies in the section, we
 
562
        # need to convert it to a policy_norecurse key.
 
563
        try:
 
564
            recurse = self._get_parser()[section].as_bool('recurse')
 
565
        except KeyError:
 
566
            pass
 
567
        else:
 
568
            symbol_versioning.warn(
 
569
                'The recurse option is deprecated as of 0.14.  '
 
570
                'The section "%s" has been converted to use policies.'
 
571
                % section,
 
572
                DeprecationWarning)
 
573
            del self._get_parser()[section]['recurse']
 
574
            if not recurse:
 
575
                for key in self._get_parser()[section].keys():
 
576
                    if not key.endswith(':policy'):
 
577
                        self._get_parser()[section][key +
 
578
                                                    ':policy'] = 'norecurse'
 
579
 
 
580
        policy_key = option_name + ':policy'
 
581
        policy_name = _policy_name[option_policy]
 
582
        if policy_name is not None:
 
583
            self._get_parser()[section][policy_key] = policy_name
 
584
        else:
 
585
            if policy_key in self._get_parser()[section]:
 
586
                del self._get_parser()[section][policy_key]
 
587
 
 
588
    def set_user_option(self, option, value, store=STORE_LOCATION):
390
589
        """Save option and its value in the configuration."""
 
590
        if store not in [STORE_LOCATION,
 
591
                         STORE_LOCATION_NORECURSE,
 
592
                         STORE_LOCATION_APPENDPATH]:
 
593
            raise ValueError('bad storage policy %r for %r' %
 
594
                (store, option))
391
595
        # FIXME: RBC 20051029 This should refresh the parser and also take a
392
 
        # file lock on branches.conf.
 
596
        # file lock on locations.conf.
393
597
        conf_dir = os.path.dirname(self._get_filename())
394
598
        ensure_config_dir_exists(conf_dir)
395
599
        location = self.location
401
605
        elif location + '/' in self._get_parser():
402
606
            location = location + '/'
403
607
        self._get_parser()[location][option]=value
 
608
        # the allowed values of store match the config policies
 
609
        self._set_option_policy(location, option, store)
404
610
        self._get_parser().write(file(self._get_filename(), 'wb'))
405
611
 
406
612
 
407
613
class BranchConfig(Config):
408
614
    """A configuration object giving the policy for a branch."""
409
615
 
 
616
    def _get_branch_data_config(self):
 
617
        if self._branch_data_config is None:
 
618
            self._branch_data_config = TreeConfig(self.branch)
 
619
        return self._branch_data_config
 
620
 
410
621
    def _get_location_config(self):
411
622
        if self._location_config is None:
412
623
            self._location_config = LocationConfig(self.branch.base)
413
624
        return self._location_config
414
625
 
 
626
    def _get_global_config(self):
 
627
        if self._global_config is None:
 
628
            self._global_config = GlobalConfig()
 
629
        return self._global_config
 
630
 
 
631
    def _get_best_value(self, option_name):
 
632
        """This returns a user option from local, tree or global config.
 
633
 
 
634
        They are tried in that order.  Use get_safe_value if trusted values
 
635
        are necessary.
 
636
        """
 
637
        for source in self.option_sources:
 
638
            value = getattr(source(), option_name)()
 
639
            if value is not None:
 
640
                return value
 
641
        return None
 
642
 
 
643
    def _get_safe_value(self, option_name):
 
644
        """This variant of get_best_value never returns untrusted values.
 
645
 
 
646
        It does not return values from the branch data, because the branch may
 
647
        not be controlled by the user.
 
648
 
 
649
        We may wish to allow locations.conf to control whether branches are
 
650
        trusted in the future.
 
651
        """
 
652
        for source in (self._get_location_config, self._get_global_config):
 
653
            value = getattr(source(), option_name)()
 
654
            if value is not None:
 
655
                return value
 
656
        return None
 
657
 
415
658
    def _get_user_id(self):
416
659
        """Return the full user id for the branch.
417
 
    
418
 
        e.g. "John Hacker <jhacker@foo.org>"
 
660
 
 
661
        e.g. "John Hacker <jhacker@example.com>"
419
662
        This is looked up in the email controlfile for the branch.
420
663
        """
421
664
        try:
422
 
            return (self.branch.control_files.get_utf8("email") 
423
 
                    .read()
424
 
                    .decode(bzrlib.user_encoding)
 
665
            return (self.branch._transport.get_bytes("email")
 
666
                    .decode(osutils.get_user_encoding())
425
667
                    .rstrip("\r\n"))
426
668
        except errors.NoSuchFile, e:
427
669
            pass
428
 
        
429
 
        return self._get_location_config()._get_user_id()
 
670
 
 
671
        return self._get_best_value('_get_user_id')
430
672
 
431
673
    def _get_signature_checking(self):
432
674
        """See Config._get_signature_checking."""
433
 
        return self._get_location_config()._get_signature_checking()
 
675
        return self._get_best_value('_get_signature_checking')
 
676
 
 
677
    def _get_signing_policy(self):
 
678
        """See Config._get_signing_policy."""
 
679
        return self._get_best_value('_get_signing_policy')
434
680
 
435
681
    def _get_user_option(self, option_name):
436
682
        """See Config._get_user_option."""
437
 
        return self._get_location_config()._get_user_option(option_name)
 
683
        for source in self.option_sources:
 
684
            value = source()._get_user_option(option_name)
 
685
            if value is not None:
 
686
                return value
 
687
        return None
 
688
 
 
689
    def set_user_option(self, name, value, store=STORE_BRANCH,
 
690
        warn_masked=False):
 
691
        if store == STORE_BRANCH:
 
692
            self._get_branch_data_config().set_option(value, name)
 
693
        elif store == STORE_GLOBAL:
 
694
            self._get_global_config().set_user_option(name, value)
 
695
        else:
 
696
            self._get_location_config().set_user_option(name, value, store)
 
697
        if not warn_masked:
 
698
            return
 
699
        if store in (STORE_GLOBAL, STORE_BRANCH):
 
700
            mask_value = self._get_location_config().get_user_option(name)
 
701
            if mask_value is not None:
 
702
                trace.warning('Value "%s" is masked by "%s" from'
 
703
                              ' locations.conf', value, mask_value)
 
704
            else:
 
705
                if store == STORE_GLOBAL:
 
706
                    branch_config = self._get_branch_data_config()
 
707
                    mask_value = branch_config.get_user_option(name)
 
708
                    if mask_value is not None:
 
709
                        trace.warning('Value "%s" is masked by "%s" from'
 
710
                                      ' branch.conf', value, mask_value)
 
711
 
438
712
 
439
713
    def _gpg_signing_command(self):
440
714
        """See Config.gpg_signing_command."""
441
 
        return self._get_location_config()._gpg_signing_command()
442
 
        
 
715
        return self._get_safe_value('_gpg_signing_command')
 
716
 
443
717
    def __init__(self, branch):
444
718
        super(BranchConfig, self).__init__()
445
719
        self._location_config = None
 
720
        self._branch_data_config = None
 
721
        self._global_config = None
446
722
        self.branch = branch
 
723
        self.option_sources = (self._get_location_config,
 
724
                               self._get_branch_data_config,
 
725
                               self._get_global_config)
447
726
 
448
727
    def _post_commit(self):
449
728
        """See Config.post_commit."""
450
 
        return self._get_location_config()._post_commit()
 
729
        return self._get_safe_value('_post_commit')
 
730
 
 
731
    def _get_nickname(self):
 
732
        value = self._get_explicit_nickname()
 
733
        if value is not None:
 
734
            return value
 
735
        return urlutils.unescape(self.branch.base.split('/')[-2])
 
736
 
 
737
    def has_explicit_nickname(self):
 
738
        """Return true if a nickname has been explicitly assigned."""
 
739
        return self._get_explicit_nickname() is not None
 
740
 
 
741
    def _get_explicit_nickname(self):
 
742
        return self._get_best_value('_get_nickname')
451
743
 
452
744
    def _log_format(self):
453
745
        """See Config.log_format."""
454
 
        return self._get_location_config()._log_format()
 
746
        return self._get_best_value('_log_format')
455
747
 
456
748
 
457
749
def ensure_config_dir_exists(path=None):
466
758
        if sys.platform == 'win32':
467
759
            parent_dir = os.path.dirname(path)
468
760
            if not os.path.isdir(parent_dir):
469
 
                mutter('creating config parent directory: %r', parent_dir)
 
761
                trace.mutter('creating config parent directory: %r', parent_dir)
470
762
            os.mkdir(parent_dir)
471
 
        mutter('creating config directory: %r', path)
 
763
        trace.mutter('creating config directory: %r', path)
472
764
        os.mkdir(path)
473
765
 
474
766
 
476
768
    """Return per-user configuration directory.
477
769
 
478
770
    By default this is ~/.bazaar/
479
 
    
 
771
 
480
772
    TODO: Global option --config-dir to override this.
481
773
    """
482
774
    base = os.environ.get('BZR_HOME', None)
483
775
    if sys.platform == 'win32':
484
776
        if base is None:
485
 
            base = os.environ.get('APPDATA', None)
 
777
            base = win32utils.get_appdata_location_unicode()
486
778
        if base is None:
487
779
            base = os.environ.get('HOME', None)
488
780
        if base is None:
489
 
            raise BzrError('You must have one of BZR_HOME, APPDATA, or HOME set')
490
 
        return pathjoin(base, 'bazaar', '2.0')
 
781
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
 
782
                                  ' or HOME set')
 
783
        return osutils.pathjoin(base, 'bazaar', '2.0')
491
784
    else:
492
785
        # cygwin, linux, and darwin all have a $HOME directory
493
786
        if base is None:
494
787
            base = os.path.expanduser("~")
495
 
        return pathjoin(base, ".bazaar")
 
788
        return osutils.pathjoin(base, ".bazaar")
496
789
 
497
790
 
498
791
def config_filename():
499
792
    """Return per-user configuration ini file filename."""
500
 
    return pathjoin(config_dir(), 'bazaar.conf')
 
793
    return osutils.pathjoin(config_dir(), 'bazaar.conf')
501
794
 
502
795
 
503
796
def branches_config_filename():
504
797
    """Return per-user configuration ini file filename."""
505
 
    return pathjoin(config_dir(), 'branches.conf')
 
798
    return osutils.pathjoin(config_dir(), 'branches.conf')
 
799
 
 
800
 
 
801
def locations_config_filename():
 
802
    """Return per-user configuration ini file filename."""
 
803
    return osutils.pathjoin(config_dir(), 'locations.conf')
 
804
 
 
805
 
 
806
def authentication_config_filename():
 
807
    """Return per-user authentication ini file filename."""
 
808
    return osutils.pathjoin(config_dir(), 'authentication.conf')
 
809
 
 
810
 
 
811
def user_ignore_config_filename():
 
812
    """Return the user default ignore filename"""
 
813
    return osutils.pathjoin(config_dir(), 'ignore')
506
814
 
507
815
 
508
816
def _auto_user_id():
518
826
    """
519
827
    import socket
520
828
 
521
 
    # XXX: Any good way to get real user name on win32?
 
829
    if sys.platform == 'win32':
 
830
        name = win32utils.get_user_name_unicode()
 
831
        if name is None:
 
832
            raise errors.BzrError("Cannot autodetect user name.\n"
 
833
                                  "Please, set your name with command like:\n"
 
834
                                  'bzr whoami "Your Name <name@domain.com>"')
 
835
        host = win32utils.get_host_name_unicode()
 
836
        if host is None:
 
837
            host = socket.gethostname()
 
838
        return name, (name + '@' + host)
522
839
 
523
840
    try:
524
841
        import pwd
525
842
        uid = os.getuid()
526
 
        w = pwd.getpwuid(uid)
 
843
        try:
 
844
            w = pwd.getpwuid(uid)
 
845
        except KeyError:
 
846
            raise errors.BzrCommandError('Unable to determine your name.  '
 
847
                'Please use "bzr whoami" to set it.')
527
848
 
528
 
        try:
529
 
            gecos = w.pw_gecos.decode(bzrlib.user_encoding)
530
 
            username = w.pw_name.decode(bzrlib.user_encoding)
531
 
        except UnicodeDecodeError:
532
 
            # We're using pwd, therefore we're on Unix, so /etc/passwd is ok.
533
 
            raise errors.BzrError("Can't decode username in " \
534
 
                    "/etc/passwd as %s." % bzrlib.user_encoding)
 
849
        # we try utf-8 first, because on many variants (like Linux),
 
850
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
851
        # false positives.  (many users will have their user encoding set to
 
852
        # latin-1, which cannot raise UnicodeError.)
 
853
        try:
 
854
            gecos = w.pw_gecos.decode('utf-8')
 
855
            encoding = 'utf-8'
 
856
        except UnicodeError:
 
857
            try:
 
858
                encoding = osutils.get_user_encoding()
 
859
                gecos = w.pw_gecos.decode(encoding)
 
860
            except UnicodeError:
 
861
                raise errors.BzrCommandError('Unable to determine your name.  '
 
862
                   'Use "bzr whoami" to set it.')
 
863
        try:
 
864
            username = w.pw_name.decode(encoding)
 
865
        except UnicodeError:
 
866
            raise errors.BzrCommandError('Unable to determine your name.  '
 
867
                'Use "bzr whoami" to set it.')
535
868
 
536
869
        comma = gecos.find(',')
537
870
        if comma == -1:
544
877
    except ImportError:
545
878
        import getpass
546
879
        try:
547
 
            realname = username = getpass.getuser().decode(bzrlib.user_encoding)
 
880
            user_encoding = osutils.get_user_encoding()
 
881
            realname = username = getpass.getuser().decode(user_encoding)
548
882
        except UnicodeDecodeError:
549
883
            raise errors.BzrError("Can't decode username as %s." % \
550
 
                    bzrlib.user_encoding)
 
884
                    user_encoding)
551
885
 
552
886
    return realname, (username + '@' + socket.gethostname())
553
887
 
554
888
 
 
889
def parse_username(username):
 
890
    """Parse e-mail username and return a (name, address) tuple."""
 
891
    match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
 
892
    if match is None:
 
893
        return (username, '')
 
894
    else:
 
895
        return (match.group(1), match.group(2))
 
896
 
 
897
 
555
898
def extract_email_address(e):
556
899
    """Return just the address part of an email string.
557
 
    
558
 
    That is just the user@domain part, nothing else. 
 
900
 
 
901
    That is just the user@domain part, nothing else.
559
902
    This part is required to contain only ascii characters.
560
903
    If it can't be extracted, raises an error.
561
 
    
 
904
 
562
905
    >>> extract_email_address('Jane Tester <jane@test.com>')
563
906
    "jane@test.com"
564
907
    """
565
 
    m = re.search(r'[\w+.-]+@[\w+.-]+', e)
566
 
    if not m:
567
 
        raise errors.BzrError("%r doesn't seem to contain "
568
 
                              "a reasonable email address" % e)
569
 
    return m.group(0)
570
 
 
571
 
class TreeConfig(object):
 
908
    name, email = parse_username(e)
 
909
    if not email:
 
910
        raise errors.NoEmailInUsername(e)
 
911
    return email
 
912
 
 
913
 
 
914
class TreeConfig(IniBasedConfig):
572
915
    """Branch configuration data associated with its contents, not location"""
 
916
 
 
917
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
 
918
 
573
919
    def __init__(self, branch):
 
920
        # XXX: Really this should be asking the branch for its configuration
 
921
        # data, rather than relying on a Transport, so that it can work
 
922
        # more cleanly with a RemoteBranch that has no transport.
 
923
        self._config = TransportConfig(branch._transport, 'branch.conf')
574
924
        self.branch = branch
575
925
 
576
 
    def _get_config(self):
577
 
        try:
578
 
            obj = ConfigObj(self.branch.control_files.get('branch.conf'), 
579
 
                            encoding='utf-8')
580
 
        except errors.NoSuchFile:
581
 
            obj = ConfigObj(encoding='utf=8')
582
 
        return obj
 
926
    def _get_parser(self, file=None):
 
927
        if file is not None:
 
928
            return IniBasedConfig._get_parser(file)
 
929
        return self._config._get_configobj()
583
930
 
584
931
    def get_option(self, name, section=None, default=None):
585
932
        self.branch.lock_read()
586
933
        try:
587
 
            obj = self._get_config()
588
 
            try:
589
 
                if section is not None:
590
 
                    obj[section]
591
 
                result = obj[name]
592
 
            except KeyError:
593
 
                result = default
 
934
            return self._config.get_option(name, section, default)
594
935
        finally:
595
936
            self.branch.unlock()
596
937
        return result
599
940
        """Set a per-branch configuration option"""
600
941
        self.branch.lock_write()
601
942
        try:
602
 
            cfg_obj = self._get_config()
603
 
            if section is None:
604
 
                obj = cfg_obj
605
 
            else:
606
 
                try:
607
 
                    obj = cfg_obj[section]
608
 
                except KeyError:
609
 
                    cfg_obj[section] = {}
610
 
                    obj = cfg_obj[section]
611
 
            obj[name] = value
612
 
            out_file = StringIO()
613
 
            cfg_obj.write(out_file)
614
 
            out_file.seek(0)
615
 
            self.branch.control_files.put('branch.conf', out_file)
 
943
            self._config.set_option(value, name, section)
616
944
        finally:
617
945
            self.branch.unlock()
 
946
 
 
947
 
 
948
class AuthenticationConfig(object):
 
949
    """The authentication configuration file based on a ini file.
 
950
 
 
951
    Implements the authentication.conf file described in
 
952
    doc/developers/authentication-ring.txt.
 
953
    """
 
954
 
 
955
    def __init__(self, _file=None):
 
956
        self._config = None # The ConfigObj
 
957
        if _file is None:
 
958
            self._filename = authentication_config_filename()
 
959
            self._input = self._filename = authentication_config_filename()
 
960
        else:
 
961
            # Tests can provide a string as _file
 
962
            self._filename = None
 
963
            self._input = _file
 
964
 
 
965
    def _get_config(self):
 
966
        if self._config is not None:
 
967
            return self._config
 
968
        try:
 
969
            # FIXME: Should we validate something here ? Includes: empty
 
970
            # sections are useless, at least one of
 
971
            # user/password/password_encoding should be defined, etc.
 
972
 
 
973
            # Note: the encoding below declares that the file itself is utf-8
 
974
            # encoded, but the values in the ConfigObj are always Unicode.
 
975
            self._config = ConfigObj(self._input, encoding='utf-8')
 
976
        except configobj.ConfigObjError, e:
 
977
            raise errors.ParseConfigError(e.errors, e.config.filename)
 
978
        return self._config
 
979
 
 
980
    def _save(self):
 
981
        """Save the config file, only tests should use it for now."""
 
982
        conf_dir = os.path.dirname(self._filename)
 
983
        ensure_config_dir_exists(conf_dir)
 
984
        self._get_config().write(file(self._filename, 'wb'))
 
985
 
 
986
    def _set_option(self, section_name, option_name, value):
 
987
        """Set an authentication configuration option"""
 
988
        conf = self._get_config()
 
989
        section = conf.get(section_name)
 
990
        if section is None:
 
991
            conf[section] = {}
 
992
            section = conf[section]
 
993
        section[option_name] = value
 
994
        self._save()
 
995
 
 
996
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
 
997
                        realm=None):
 
998
        """Returns the matching credentials from authentication.conf file.
 
999
 
 
1000
        :param scheme: protocol
 
1001
 
 
1002
        :param host: the server address
 
1003
 
 
1004
        :param port: the associated port (optional)
 
1005
 
 
1006
        :param user: login (optional)
 
1007
 
 
1008
        :param path: the absolute path on the server (optional)
 
1009
        
 
1010
        :param realm: the http authentication realm (optional)
 
1011
 
 
1012
        :return: A dict containing the matching credentials or None.
 
1013
           This includes:
 
1014
           - name: the section name of the credentials in the
 
1015
             authentication.conf file,
 
1016
           - user: can't be different from the provided user if any,
 
1017
           - scheme: the server protocol,
 
1018
           - host: the server address,
 
1019
           - port: the server port (can be None),
 
1020
           - path: the absolute server path (can be None),
 
1021
           - realm: the http specific authentication realm (can be None),
 
1022
           - password: the decoded password, could be None if the credential
 
1023
             defines only the user
 
1024
           - verify_certificates: https specific, True if the server
 
1025
             certificate should be verified, False otherwise.
 
1026
        """
 
1027
        credentials = None
 
1028
        for auth_def_name, auth_def in self._get_config().items():
 
1029
            if type(auth_def) is not configobj.Section:
 
1030
                raise ValueError("%s defined outside a section" % auth_def_name)
 
1031
 
 
1032
            a_scheme, a_host, a_user, a_path = map(
 
1033
                auth_def.get, ['scheme', 'host', 'user', 'path'])
 
1034
 
 
1035
            try:
 
1036
                a_port = auth_def.as_int('port')
 
1037
            except KeyError:
 
1038
                a_port = None
 
1039
            except ValueError:
 
1040
                raise ValueError("'port' not numeric in %s" % auth_def_name)
 
1041
            try:
 
1042
                a_verify_certificates = auth_def.as_bool('verify_certificates')
 
1043
            except KeyError:
 
1044
                a_verify_certificates = True
 
1045
            except ValueError:
 
1046
                raise ValueError(
 
1047
                    "'verify_certificates' not boolean in %s" % auth_def_name)
 
1048
 
 
1049
            # Attempt matching
 
1050
            if a_scheme is not None and scheme != a_scheme:
 
1051
                continue
 
1052
            if a_host is not None:
 
1053
                if not (host == a_host
 
1054
                        or (a_host.startswith('.') and host.endswith(a_host))):
 
1055
                    continue
 
1056
            if a_port is not None and port != a_port:
 
1057
                continue
 
1058
            if (a_path is not None and path is not None
 
1059
                and not path.startswith(a_path)):
 
1060
                continue
 
1061
            if (a_user is not None and user is not None
 
1062
                and a_user != user):
 
1063
                # Never contradict the caller about the user to be used
 
1064
                continue
 
1065
            if a_user is None:
 
1066
                # Can't find a user
 
1067
                continue
 
1068
            # Prepare a credentials dictionary with additional keys
 
1069
            # for the credential providers
 
1070
            credentials = dict(name=auth_def_name,
 
1071
                               user=a_user,
 
1072
                               scheme=a_scheme,
 
1073
                               host=host,
 
1074
                               port=port,
 
1075
                               path=path,
 
1076
                               realm=realm,
 
1077
                               password=auth_def.get('password', None),
 
1078
                               verify_certificates=a_verify_certificates)
 
1079
            # Decode the password in the credentials (or get one)
 
1080
            self.decode_password(credentials,
 
1081
                                 auth_def.get('password_encoding', None))
 
1082
            if 'auth' in debug.debug_flags:
 
1083
                trace.mutter("Using authentication section: %r", auth_def_name)
 
1084
            break
 
1085
 
 
1086
        return credentials
 
1087
 
 
1088
    def set_credentials(self, name, host, user, scheme=None, password=None,
 
1089
                        port=None, path=None, verify_certificates=None,
 
1090
                        realm=None):
 
1091
        """Set authentication credentials for a host.
 
1092
 
 
1093
        Any existing credentials with matching scheme, host, port and path
 
1094
        will be deleted, regardless of name.
 
1095
 
 
1096
        :param name: An arbitrary name to describe this set of credentials.
 
1097
        :param host: Name of the host that accepts these credentials.
 
1098
        :param user: The username portion of these credentials.
 
1099
        :param scheme: The URL scheme (e.g. ssh, http) the credentials apply
 
1100
            to.
 
1101
        :param password: Password portion of these credentials.
 
1102
        :param port: The IP port on the host that these credentials apply to.
 
1103
        :param path: A filesystem path on the host that these credentials
 
1104
            apply to.
 
1105
        :param verify_certificates: On https, verify server certificates if
 
1106
            True.
 
1107
        :param realm: The http authentication realm (optional).
 
1108
        """
 
1109
        values = {'host': host, 'user': user}
 
1110
        if password is not None:
 
1111
            values['password'] = password
 
1112
        if scheme is not None:
 
1113
            values['scheme'] = scheme
 
1114
        if port is not None:
 
1115
            values['port'] = '%d' % port
 
1116
        if path is not None:
 
1117
            values['path'] = path
 
1118
        if verify_certificates is not None:
 
1119
            values['verify_certificates'] = str(verify_certificates)
 
1120
        if realm is not None:
 
1121
            values['realm'] = realm
 
1122
        config = self._get_config()
 
1123
        for_deletion = []
 
1124
        for section, existing_values in config.items():
 
1125
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
 
1126
                if existing_values.get(key) != values.get(key):
 
1127
                    break
 
1128
            else:
 
1129
                del config[section]
 
1130
        config.update({name: values})
 
1131
        self._save()
 
1132
 
 
1133
    def get_user(self, scheme, host, port=None,
 
1134
                 realm=None, path=None, prompt=None):
 
1135
        """Get a user from authentication file.
 
1136
 
 
1137
        :param scheme: protocol
 
1138
 
 
1139
        :param host: the server address
 
1140
 
 
1141
        :param port: the associated port (optional)
 
1142
 
 
1143
        :param realm: the realm sent by the server (optional)
 
1144
 
 
1145
        :param path: the absolute path on the server (optional)
 
1146
 
 
1147
        :return: The found user.
 
1148
        """
 
1149
        credentials = self.get_credentials(scheme, host, port, user=None,
 
1150
                                           path=path, realm=realm)
 
1151
        if credentials is not None:
 
1152
            user = credentials['user']
 
1153
        else:
 
1154
            user = None
 
1155
        return user
 
1156
 
 
1157
    def get_password(self, scheme, host, user, port=None,
 
1158
                     realm=None, path=None, prompt=None):
 
1159
        """Get a password from authentication file or prompt the user for one.
 
1160
 
 
1161
        :param scheme: protocol
 
1162
 
 
1163
        :param host: the server address
 
1164
 
 
1165
        :param port: the associated port (optional)
 
1166
 
 
1167
        :param user: login
 
1168
 
 
1169
        :param realm: the realm sent by the server (optional)
 
1170
 
 
1171
        :param path: the absolute path on the server (optional)
 
1172
 
 
1173
        :return: The found password or the one entered by the user.
 
1174
        """
 
1175
        credentials = self.get_credentials(scheme, host, port, user, path,
 
1176
                                           realm)
 
1177
        if credentials is not None:
 
1178
            password = credentials['password']
 
1179
            if password is not None and scheme is 'ssh':
 
1180
                trace.warning('password ignored in section [%s],'
 
1181
                              ' use an ssh agent instead'
 
1182
                              % credentials['name'])
 
1183
                password = None
 
1184
        else:
 
1185
            password = None
 
1186
        # Prompt user only if we could't find a password
 
1187
        if password is None:
 
1188
            if prompt is None:
 
1189
                # Create a default prompt suitable for most cases
 
1190
                prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
 
1191
            # Special handling for optional fields in the prompt
 
1192
            if port is not None:
 
1193
                prompt_host = '%s:%d' % (host, port)
 
1194
            else:
 
1195
                prompt_host = host
 
1196
            password = ui.ui_factory.get_password(prompt,
 
1197
                                                  host=prompt_host, user=user)
 
1198
        return password
 
1199
 
 
1200
    def decode_password(self, credentials, encoding):
 
1201
        try:
 
1202
            cs = credential_store_registry.get_credential_store(encoding)
 
1203
        except KeyError:
 
1204
            raise ValueError('%r is not a known password_encoding' % encoding)
 
1205
        credentials['password'] = cs.decode_password(credentials)
 
1206
        return credentials
 
1207
 
 
1208
 
 
1209
class CredentialStoreRegistry(registry.Registry):
 
1210
    """A class that registers credential stores.
 
1211
 
 
1212
    A credential store provides access to credentials via the password_encoding
 
1213
    field in authentication.conf sections.
 
1214
 
 
1215
    Except for stores provided by bzr itself,most stores are expected to be
 
1216
    provided by plugins that will therefore use
 
1217
    register_lazy(password_encoding, module_name, member_name, help=help,
 
1218
    info=info) to install themselves.
 
1219
    """
 
1220
 
 
1221
    def get_credential_store(self, encoding=None):
 
1222
        cs = self.get(encoding)
 
1223
        if callable(cs):
 
1224
            cs = cs()
 
1225
        return cs
 
1226
 
 
1227
 
 
1228
credential_store_registry = CredentialStoreRegistry()
 
1229
 
 
1230
 
 
1231
class CredentialStore(object):
 
1232
    """An abstract class to implement storage for credentials"""
 
1233
 
 
1234
    def decode_password(self, credentials):
 
1235
        """Returns a password for the provided credentials in clear text."""
 
1236
        raise NotImplementedError(self.decode_password)
 
1237
 
 
1238
 
 
1239
class PlainTextCredentialStore(CredentialStore):
 
1240
    """Plain text credential store for the authentication.conf file."""
 
1241
 
 
1242
    def decode_password(self, credentials):
 
1243
        """See CredentialStore.decode_password."""
 
1244
        return credentials['password']
 
1245
 
 
1246
 
 
1247
credential_store_registry.register('plain', PlainTextCredentialStore,
 
1248
                                   help=PlainTextCredentialStore.__doc__)
 
1249
credential_store_registry.default_key = 'plain'
 
1250
 
 
1251
 
 
1252
class BzrDirConfig(object):
 
1253
 
 
1254
    def __init__(self, transport):
 
1255
        self._config = TransportConfig(transport, 'control.conf')
 
1256
 
 
1257
    def set_default_stack_on(self, value):
 
1258
        """Set the default stacking location.
 
1259
 
 
1260
        It may be set to a location, or None.
 
1261
 
 
1262
        This policy affects all branches contained by this bzrdir, except for
 
1263
        those under repositories.
 
1264
        """
 
1265
        if value is None:
 
1266
            self._config.set_option('', 'default_stack_on')
 
1267
        else:
 
1268
            self._config.set_option(value, 'default_stack_on')
 
1269
 
 
1270
    def get_default_stack_on(self):
 
1271
        """Return the default stacking location.
 
1272
 
 
1273
        This will either be a location, or None.
 
1274
 
 
1275
        This policy affects all branches contained by this bzrdir, except for
 
1276
        those under repositories.
 
1277
        """
 
1278
        value = self._config.get_option('default_stack_on')
 
1279
        if value == '':
 
1280
            value = None
 
1281
        return value
 
1282
 
 
1283
 
 
1284
class TransportConfig(object):
 
1285
    """A Config that reads/writes a config file on a Transport.
 
1286
 
 
1287
    It is a low-level object that considers config data to be name/value pairs
 
1288
    that may be associated with a section.  Assigning meaning to the these
 
1289
    values is done at higher levels like TreeConfig.
 
1290
    """
 
1291
 
 
1292
    def __init__(self, transport, filename):
 
1293
        self._transport = transport
 
1294
        self._filename = filename
 
1295
 
 
1296
    def get_option(self, name, section=None, default=None):
 
1297
        """Return the value associated with a named option.
 
1298
 
 
1299
        :param name: The name of the value
 
1300
        :param section: The section the option is in (if any)
 
1301
        :param default: The value to return if the value is not set
 
1302
        :return: The value or default value
 
1303
        """
 
1304
        configobj = self._get_configobj()
 
1305
        if section is None:
 
1306
            section_obj = configobj
 
1307
        else:
 
1308
            try:
 
1309
                section_obj = configobj[section]
 
1310
            except KeyError:
 
1311
                return default
 
1312
        return section_obj.get(name, default)
 
1313
 
 
1314
    def set_option(self, value, name, section=None):
 
1315
        """Set the value associated with a named option.
 
1316
 
 
1317
        :param value: The value to set
 
1318
        :param name: The name of the value to set
 
1319
        :param section: The section the option is in (if any)
 
1320
        """
 
1321
        configobj = self._get_configobj()
 
1322
        if section is None:
 
1323
            configobj[name] = value
 
1324
        else:
 
1325
            configobj.setdefault(section, {})[name] = value
 
1326
        self._set_configobj(configobj)
 
1327
 
 
1328
    def _get_configobj(self):
 
1329
        try:
 
1330
            return ConfigObj(self._transport.get(self._filename),
 
1331
                             encoding='utf-8')
 
1332
        except errors.NoSuchFile:
 
1333
            return ConfigObj(encoding='utf-8')
 
1334
 
 
1335
    def _set_configobj(self, configobj):
 
1336
        out_file = StringIO()
 
1337
        configobj.write(out_file)
 
1338
        out_file.seek(0)
 
1339
        self._transport.put_file(self._filename, out_file)