~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-05-29 15:57:16 UTC
  • mfrom: (3427.5.9 dep_warnings)
  • Revision ID: pqm@pqm.ubuntu.com-20080529155716-0w3kic8lioa63231
(jam) Enable Deprecation Warnings when running -Werror and when
        running selftest

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 by Canonical Ltd
 
1
# Copyright (C) 2005, 2007 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
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
49
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
    symbol_versioning,
 
82
    trace,
 
83
    ui,
 
84
    urlutils,
 
85
    win32utils,
 
86
    )
 
87
from bzrlib.util.configobj import configobj
 
88
""")
 
89
 
78
90
 
79
91
CHECK_IF_POSSIBLE=0
80
92
CHECK_ALWAYS=1
81
93
CHECK_NEVER=2
82
94
 
83
95
 
 
96
SIGN_WHEN_REQUIRED=0
 
97
SIGN_ALWAYS=1
 
98
SIGN_NEVER=2
 
99
 
 
100
 
 
101
POLICY_NONE = 0
 
102
POLICY_NORECURSE = 1
 
103
POLICY_APPENDPATH = 2
 
104
 
 
105
_policy_name = {
 
106
    POLICY_NONE: None,
 
107
    POLICY_NORECURSE: 'norecurse',
 
108
    POLICY_APPENDPATH: 'appendpath',
 
109
    }
 
110
_policy_value = {
 
111
    None: POLICY_NONE,
 
112
    'none': POLICY_NONE,
 
113
    'norecurse': POLICY_NORECURSE,
 
114
    'appendpath': POLICY_APPENDPATH,
 
115
    }
 
116
 
 
117
 
 
118
STORE_LOCATION = POLICY_NONE
 
119
STORE_LOCATION_NORECURSE = POLICY_NORECURSE
 
120
STORE_LOCATION_APPENDPATH = POLICY_APPENDPATH
 
121
STORE_BRANCH = 3
 
122
STORE_GLOBAL = 4
 
123
 
 
124
 
84
125
class ConfigObj(configobj.ConfigObj):
85
126
 
86
127
    def get_bool(self, section, key):
103
144
        """Get the users pop up editor."""
104
145
        raise NotImplementedError
105
146
 
 
147
    def get_mail_client(self):
 
148
        """Get a mail client to use"""
 
149
        selected_client = self.get_user_option('mail_client')
 
150
        try:
 
151
            mail_client_class = {
 
152
                None: mail_client.DefaultMail,
 
153
                # Specific clients
 
154
                'emacsclient': mail_client.EmacsMail,
 
155
                'evolution': mail_client.Evolution,
 
156
                'kmail': mail_client.KMail,
 
157
                'mutt': mail_client.Mutt,
 
158
                'thunderbird': mail_client.Thunderbird,
 
159
                # Generic options
 
160
                'default': mail_client.DefaultMail,
 
161
                'editor': mail_client.Editor,
 
162
                'mapi': mail_client.MAPIClient,
 
163
                'xdg-email': mail_client.XDGEmail,
 
164
            }[selected_client]
 
165
        except KeyError:
 
166
            raise errors.UnknownMailClient(selected_client)
 
167
        return mail_client_class(self)
 
168
 
106
169
    def _get_signature_checking(self):
107
170
        """Template method to override signature checking policy."""
108
171
 
 
172
    def _get_signing_policy(self):
 
173
        """Template method to override signature creation policy."""
 
174
 
109
175
    def _get_user_option(self, option_name):
110
176
        """Template method to provide a user option."""
111
177
        return None
159
225
    
160
226
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
161
227
        
162
 
        $BZREMAIL can be set to override this, then
 
228
        $BZR_EMAIL can be set to override this (as well as the
 
229
        deprecated $BZREMAIL), then
163
230
        the concrete policy type is checked, and finally
164
231
        $EMAIL is examined.
165
232
        If none is found, a reasonable default is (hopefully)
167
234
    
168
235
        TODO: Check it's reasonably well-formed.
169
236
        """
170
 
        v = os.environ.get('BZREMAIL')
 
237
        v = os.environ.get('BZR_EMAIL')
171
238
        if v:
172
239
            return v.decode(bzrlib.user_encoding)
173
 
    
 
240
 
174
241
        v = self._get_user_id()
175
242
        if v:
176
243
            return v
177
 
        
 
244
 
178
245
        v = os.environ.get('EMAIL')
179
246
        if v:
180
247
            return v.decode(bzrlib.user_encoding)
192
259
            return policy
193
260
        return CHECK_IF_POSSIBLE
194
261
 
 
262
    def signing_policy(self):
 
263
        """What is the current policy for signature checking?."""
 
264
        policy = self._get_signing_policy()
 
265
        if policy is not None:
 
266
            return policy
 
267
        return SIGN_WHEN_REQUIRED
 
268
 
195
269
    def signature_needed(self):
196
270
        """Is a signature needed when committing ?."""
197
 
        policy = self._get_signature_checking()
198
 
        if policy == CHECK_ALWAYS:
 
271
        policy = self._get_signing_policy()
 
272
        if policy is None:
 
273
            policy = self._get_signature_checking()
 
274
            if policy is not None:
 
275
                trace.warning("Please use create_signatures,"
 
276
                              " not check_signatures to set signing policy.")
 
277
            if policy == CHECK_ALWAYS:
 
278
                return True
 
279
        elif policy == SIGN_ALWAYS:
199
280
            return True
200
281
        return False
201
282
 
205
286
    def _get_alias(self, value):
206
287
        pass
207
288
 
 
289
    def get_nickname(self):
 
290
        return self._get_nickname()
 
291
 
 
292
    def _get_nickname(self):
 
293
        return None
 
294
 
 
295
    def get_bzr_remote_path(self):
 
296
        try:
 
297
            return os.environ['BZR_REMOTE_PATH']
 
298
        except KeyError:
 
299
            path = self.get_user_option("bzr_remote_path")
 
300
            if path is None:
 
301
                path = 'bzr'
 
302
            return path
 
303
 
208
304
 
209
305
class IniBasedConfig(Config):
210
306
    """A configuration policy that draws from ini files."""
222
318
            raise errors.ParseConfigError(e.errors, e.config.filename)
223
319
        return self._parser
224
320
 
 
321
    def _get_matching_sections(self):
 
322
        """Return an ordered list of (section_name, extra_path) pairs.
 
323
 
 
324
        If the section contains inherited configuration, extra_path is
 
325
        a string containing the additional path components.
 
326
        """
 
327
        section = self._get_section()
 
328
        if section is not None:
 
329
            return [(section, '')]
 
330
        else:
 
331
            return []
 
332
 
225
333
    def _get_section(self):
226
334
        """Override this to define the section used by the config."""
227
335
        return "DEFAULT"
228
336
 
 
337
    def _get_option_policy(self, section, option_name):
 
338
        """Return the policy for the given (section, option_name) pair."""
 
339
        return POLICY_NONE
 
340
 
229
341
    def _get_signature_checking(self):
230
342
        """See Config._get_signature_checking."""
231
343
        policy = self._get_user_option('check_signatures')
232
344
        if policy:
233
345
            return self._string_to_signature_policy(policy)
234
346
 
 
347
    def _get_signing_policy(self):
 
348
        """See Config._get_signing_policy"""
 
349
        policy = self._get_user_option('create_signatures')
 
350
        if policy:
 
351
            return self._string_to_signing_policy(policy)
 
352
 
235
353
    def _get_user_id(self):
236
354
        """Get the user id from the 'email' key in the current section."""
237
355
        return self._get_user_option('email')
238
356
 
239
357
    def _get_user_option(self, option_name):
240
358
        """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
 
359
        for (section, extra_path) in self._get_matching_sections():
 
360
            try:
 
361
                value = self._get_parser().get_value(section, option_name)
 
362
            except KeyError:
 
363
                continue
 
364
            policy = self._get_option_policy(section, option_name)
 
365
            if policy == POLICY_NONE:
 
366
                return value
 
367
            elif policy == POLICY_NORECURSE:
 
368
                # norecurse items only apply to the exact path
 
369
                if extra_path:
 
370
                    continue
 
371
                else:
 
372
                    return value
 
373
            elif policy == POLICY_APPENDPATH:
 
374
                if extra_path:
 
375
                    value = urlutils.join(value, extra_path)
 
376
                return value
 
377
            else:
 
378
                raise AssertionError('Unexpected config policy %r' % policy)
 
379
        else:
 
380
            return None
246
381
 
247
382
    def _gpg_signing_command(self):
248
383
        """See Config.gpg_signing_command."""
272
407
        raise errors.BzrError("Invalid signatures policy '%s'"
273
408
                              % signature_string)
274
409
 
 
410
    def _string_to_signing_policy(self, signature_string):
 
411
        """Convert a string to a signing policy."""
 
412
        if signature_string.lower() == 'when-required':
 
413
            return SIGN_WHEN_REQUIRED
 
414
        if signature_string.lower() == 'never':
 
415
            return SIGN_NEVER
 
416
        if signature_string.lower() == 'always':
 
417
            return SIGN_ALWAYS
 
418
        raise errors.BzrError("Invalid signing policy '%s'"
 
419
                              % signature_string)
 
420
 
275
421
    def _get_alias(self, value):
276
422
        try:
277
423
            return self._get_parser().get_value("ALIASES", 
279
425
        except KeyError:
280
426
            pass
281
427
 
 
428
    def _get_nickname(self):
 
429
        return self.get_user_option('nickname')
 
430
 
282
431
 
283
432
class GlobalConfig(IniBasedConfig):
284
433
    """The configuration that should be used for a specific location."""
289
438
    def __init__(self):
290
439
        super(GlobalConfig, self).__init__(config_filename)
291
440
 
 
441
    def set_user_option(self, option, value):
 
442
        """Save option and its value in the configuration."""
 
443
        # FIXME: RBC 20051029 This should refresh the parser and also take a
 
444
        # file lock on bazaar.conf.
 
445
        conf_dir = os.path.dirname(self._get_filename())
 
446
        ensure_config_dir_exists(conf_dir)
 
447
        if 'DEFAULT' not in self._get_parser():
 
448
            self._get_parser()['DEFAULT'] = {}
 
449
        self._get_parser()['DEFAULT'][option] = value
 
450
        f = open(self._get_filename(), 'wb')
 
451
        self._get_parser().write(f)
 
452
        f.close()
 
453
 
292
454
 
293
455
class LocationConfig(IniBasedConfig):
294
456
    """A configuration object that gives the policy for a location."""
295
457
 
296
458
    def __init__(self, location):
297
 
        super(LocationConfig, self).__init__(branches_config_filename)
298
 
        self._global_config = None
 
459
        name_generator = locations_config_filename
 
460
        if (not os.path.exists(name_generator()) and
 
461
                os.path.exists(branches_config_filename())):
 
462
            if sys.platform == 'win32':
 
463
                trace.warning('Please rename %s to %s'
 
464
                              % (branches_config_filename(),
 
465
                                 locations_config_filename()))
 
466
            else:
 
467
                trace.warning('Please rename ~/.bazaar/branches.conf'
 
468
                              ' to ~/.bazaar/locations.conf')
 
469
            name_generator = branches_config_filename
 
470
        super(LocationConfig, self).__init__(name_generator)
 
471
        # local file locations are looked up by local path, rather than
 
472
        # by file url. This is because the config file is a user
 
473
        # file, and we would rather not expose the user to file urls.
 
474
        if location.startswith('file://'):
 
475
            location = urlutils.local_path_from_url(location)
299
476
        self.location = location
300
477
 
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
 
        """
 
478
    def _get_matching_sections(self):
 
479
        """Return an ordered list of section names matching this location."""
313
480
        sections = self._get_parser()
314
481
        location_names = self.location.split('/')
315
482
        if self.location.endswith('/'):
316
483
            del location_names[-1]
317
484
        matches=[]
318
485
        for section in sections:
319
 
            section_names = section.split('/')
 
486
            # location is a local path if possible, so we need
 
487
            # to convert 'file://' urls to local paths if necessary.
 
488
            # This also avoids having file:///path be a more exact
 
489
            # match than '/path'.
 
490
            if section.startswith('file://'):
 
491
                section_path = urlutils.local_path_from_url(section)
 
492
            else:
 
493
                section_path = section
 
494
            section_names = section_path.split('/')
320
495
            if section.endswith('/'):
321
496
                del section_names[-1]
322
497
            names = zip(location_names, section_names)
331
506
            # if section is longer, no match.
332
507
            if len(section_names) > len(location_names):
333
508
                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
 
509
            matches.append((len(section_names), section,
 
510
                            '/'.join(location_names[len(section_names):])))
344
511
        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):
 
512
        sections = []
 
513
        for (length, section, extra_path) in matches:
 
514
            sections.append((section, extra_path))
 
515
            # should we stop looking for parent configs here?
 
516
            try:
 
517
                if self._get_parser()[section].as_bool('ignore_parents'):
 
518
                    break
 
519
            except KeyError:
 
520
                pass
 
521
        return sections
 
522
 
 
523
    def _get_option_policy(self, section, option_name):
 
524
        """Return the policy for the given (section, option_name) pair."""
 
525
        # check for the old 'recurse=False' flag
 
526
        try:
 
527
            recurse = self._get_parser()[section].as_bool('recurse')
 
528
        except KeyError:
 
529
            recurse = True
 
530
        if not recurse:
 
531
            return POLICY_NORECURSE
 
532
 
 
533
        policy_key = option_name + ':policy'
 
534
        try:
 
535
            policy_name = self._get_parser()[section][policy_key]
 
536
        except KeyError:
 
537
            policy_name = None
 
538
 
 
539
        return _policy_value[policy_name]
 
540
 
 
541
    def _set_option_policy(self, section, option_name, option_policy):
 
542
        """Set the policy for the given option name in the given section."""
 
543
        # The old recurse=False option affects all options in the
 
544
        # section.  To handle multiple policies in the section, we
 
545
        # need to convert it to a policy_norecurse key.
 
546
        try:
 
547
            recurse = self._get_parser()[section].as_bool('recurse')
 
548
        except KeyError:
 
549
            pass
 
550
        else:
 
551
            symbol_versioning.warn(
 
552
                'The recurse option is deprecated as of 0.14.  '
 
553
                'The section "%s" has been converted to use policies.'
 
554
                % section,
 
555
                DeprecationWarning)
 
556
            del self._get_parser()[section]['recurse']
 
557
            if not recurse:
 
558
                for key in self._get_parser()[section].keys():
 
559
                    if not key.endswith(':policy'):
 
560
                        self._get_parser()[section][key +
 
561
                                                    ':policy'] = 'norecurse'
 
562
 
 
563
        policy_key = option_name + ':policy'
 
564
        policy_name = _policy_name[option_policy]
 
565
        if policy_name is not None:
 
566
            self._get_parser()[section][policy_key] = policy_name
 
567
        else:
 
568
            if policy_key in self._get_parser()[section]:
 
569
                del self._get_parser()[section][policy_key]
 
570
 
 
571
    def set_user_option(self, option, value, store=STORE_LOCATION):
390
572
        """Save option and its value in the configuration."""
 
573
        if store not in [STORE_LOCATION,
 
574
                         STORE_LOCATION_NORECURSE,
 
575
                         STORE_LOCATION_APPENDPATH]:
 
576
            raise ValueError('bad storage policy %r for %r' %
 
577
                (store, option))
391
578
        # FIXME: RBC 20051029 This should refresh the parser and also take a
392
 
        # file lock on branches.conf.
 
579
        # file lock on locations.conf.
393
580
        conf_dir = os.path.dirname(self._get_filename())
394
581
        ensure_config_dir_exists(conf_dir)
395
582
        location = self.location
401
588
        elif location + '/' in self._get_parser():
402
589
            location = location + '/'
403
590
        self._get_parser()[location][option]=value
 
591
        # the allowed values of store match the config policies
 
592
        self._set_option_policy(location, option, store)
404
593
        self._get_parser().write(file(self._get_filename(), 'wb'))
405
594
 
406
595
 
407
596
class BranchConfig(Config):
408
597
    """A configuration object giving the policy for a branch."""
409
598
 
 
599
    def _get_branch_data_config(self):
 
600
        if self._branch_data_config is None:
 
601
            self._branch_data_config = TreeConfig(self.branch)
 
602
        return self._branch_data_config
 
603
 
410
604
    def _get_location_config(self):
411
605
        if self._location_config is None:
412
606
            self._location_config = LocationConfig(self.branch.base)
413
607
        return self._location_config
414
608
 
 
609
    def _get_global_config(self):
 
610
        if self._global_config is None:
 
611
            self._global_config = GlobalConfig()
 
612
        return self._global_config
 
613
 
 
614
    def _get_best_value(self, option_name):
 
615
        """This returns a user option from local, tree or global config.
 
616
 
 
617
        They are tried in that order.  Use get_safe_value if trusted values
 
618
        are necessary.
 
619
        """
 
620
        for source in self.option_sources:
 
621
            value = getattr(source(), option_name)()
 
622
            if value is not None:
 
623
                return value
 
624
        return None
 
625
 
 
626
    def _get_safe_value(self, option_name):
 
627
        """This variant of get_best_value never returns untrusted values.
 
628
        
 
629
        It does not return values from the branch data, because the branch may
 
630
        not be controlled by the user.
 
631
 
 
632
        We may wish to allow locations.conf to control whether branches are
 
633
        trusted in the future.
 
634
        """
 
635
        for source in (self._get_location_config, self._get_global_config):
 
636
            value = getattr(source(), option_name)()
 
637
            if value is not None:
 
638
                return value
 
639
        return None
 
640
 
415
641
    def _get_user_id(self):
416
642
        """Return the full user id for the branch.
417
643
    
418
 
        e.g. "John Hacker <jhacker@foo.org>"
 
644
        e.g. "John Hacker <jhacker@example.com>"
419
645
        This is looked up in the email controlfile for the branch.
420
646
        """
421
647
        try:
422
 
            return (self.branch.control_files.get_utf8("email") 
423
 
                    .read()
 
648
            return (self.branch._transport.get_bytes("email")
424
649
                    .decode(bzrlib.user_encoding)
425
650
                    .rstrip("\r\n"))
426
651
        except errors.NoSuchFile, e:
427
652
            pass
428
653
        
429
 
        return self._get_location_config()._get_user_id()
 
654
        return self._get_best_value('_get_user_id')
430
655
 
431
656
    def _get_signature_checking(self):
432
657
        """See Config._get_signature_checking."""
433
 
        return self._get_location_config()._get_signature_checking()
 
658
        return self._get_best_value('_get_signature_checking')
 
659
 
 
660
    def _get_signing_policy(self):
 
661
        """See Config._get_signing_policy."""
 
662
        return self._get_best_value('_get_signing_policy')
434
663
 
435
664
    def _get_user_option(self, option_name):
436
665
        """See Config._get_user_option."""
437
 
        return self._get_location_config()._get_user_option(option_name)
 
666
        for source in self.option_sources:
 
667
            value = source()._get_user_option(option_name)
 
668
            if value is not None:
 
669
                return value
 
670
        return None
 
671
 
 
672
    def set_user_option(self, name, value, store=STORE_BRANCH,
 
673
        warn_masked=False):
 
674
        if store == STORE_BRANCH:
 
675
            self._get_branch_data_config().set_option(value, name)
 
676
        elif store == STORE_GLOBAL:
 
677
            self._get_global_config().set_user_option(name, value)
 
678
        else:
 
679
            self._get_location_config().set_user_option(name, value, store)
 
680
        if not warn_masked:
 
681
            return
 
682
        if store in (STORE_GLOBAL, STORE_BRANCH):
 
683
            mask_value = self._get_location_config().get_user_option(name)
 
684
            if mask_value is not None:
 
685
                trace.warning('Value "%s" is masked by "%s" from'
 
686
                              ' locations.conf', value, mask_value)
 
687
            else:
 
688
                if store == STORE_GLOBAL:
 
689
                    branch_config = self._get_branch_data_config()
 
690
                    mask_value = branch_config.get_user_option(name)
 
691
                    if mask_value is not None:
 
692
                        trace.warning('Value "%s" is masked by "%s" from'
 
693
                                      ' branch.conf', value, mask_value)
 
694
 
438
695
 
439
696
    def _gpg_signing_command(self):
440
697
        """See Config.gpg_signing_command."""
441
 
        return self._get_location_config()._gpg_signing_command()
 
698
        return self._get_safe_value('_gpg_signing_command')
442
699
        
443
700
    def __init__(self, branch):
444
701
        super(BranchConfig, self).__init__()
445
702
        self._location_config = None
 
703
        self._branch_data_config = None
 
704
        self._global_config = None
446
705
        self.branch = branch
 
706
        self.option_sources = (self._get_location_config, 
 
707
                               self._get_branch_data_config,
 
708
                               self._get_global_config)
447
709
 
448
710
    def _post_commit(self):
449
711
        """See Config.post_commit."""
450
 
        return self._get_location_config()._post_commit()
 
712
        return self._get_safe_value('_post_commit')
 
713
 
 
714
    def _get_nickname(self):
 
715
        value = self._get_explicit_nickname()
 
716
        if value is not None:
 
717
            return value
 
718
        return urlutils.unescape(self.branch.base.split('/')[-2])
 
719
 
 
720
    def has_explicit_nickname(self):
 
721
        """Return true if a nickname has been explicitly assigned."""
 
722
        return self._get_explicit_nickname() is not None
 
723
 
 
724
    def _get_explicit_nickname(self):
 
725
        return self._get_best_value('_get_nickname')
451
726
 
452
727
    def _log_format(self):
453
728
        """See Config.log_format."""
454
 
        return self._get_location_config()._log_format()
 
729
        return self._get_best_value('_log_format')
455
730
 
456
731
 
457
732
def ensure_config_dir_exists(path=None):
466
741
        if sys.platform == 'win32':
467
742
            parent_dir = os.path.dirname(path)
468
743
            if not os.path.isdir(parent_dir):
469
 
                mutter('creating config parent directory: %r', parent_dir)
 
744
                trace.mutter('creating config parent directory: %r', parent_dir)
470
745
            os.mkdir(parent_dir)
471
 
        mutter('creating config directory: %r', path)
 
746
        trace.mutter('creating config directory: %r', path)
472
747
        os.mkdir(path)
473
748
 
474
749
 
482
757
    base = os.environ.get('BZR_HOME', None)
483
758
    if sys.platform == 'win32':
484
759
        if base is None:
485
 
            base = os.environ.get('APPDATA', None)
 
760
            base = win32utils.get_appdata_location_unicode()
486
761
        if base is None:
487
762
            base = os.environ.get('HOME', None)
488
763
        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')
 
764
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
 
765
                                  ' or HOME set')
 
766
        return osutils.pathjoin(base, 'bazaar', '2.0')
491
767
    else:
492
768
        # cygwin, linux, and darwin all have a $HOME directory
493
769
        if base is None:
494
770
            base = os.path.expanduser("~")
495
 
        return pathjoin(base, ".bazaar")
 
771
        return osutils.pathjoin(base, ".bazaar")
496
772
 
497
773
 
498
774
def config_filename():
499
775
    """Return per-user configuration ini file filename."""
500
 
    return pathjoin(config_dir(), 'bazaar.conf')
 
776
    return osutils.pathjoin(config_dir(), 'bazaar.conf')
501
777
 
502
778
 
503
779
def branches_config_filename():
504
780
    """Return per-user configuration ini file filename."""
505
 
    return pathjoin(config_dir(), 'branches.conf')
 
781
    return osutils.pathjoin(config_dir(), 'branches.conf')
 
782
 
 
783
 
 
784
def locations_config_filename():
 
785
    """Return per-user configuration ini file filename."""
 
786
    return osutils.pathjoin(config_dir(), 'locations.conf')
 
787
 
 
788
 
 
789
def authentication_config_filename():
 
790
    """Return per-user authentication ini file filename."""
 
791
    return osutils.pathjoin(config_dir(), 'authentication.conf')
 
792
 
 
793
 
 
794
def user_ignore_config_filename():
 
795
    """Return the user default ignore filename"""
 
796
    return osutils.pathjoin(config_dir(), 'ignore')
506
797
 
507
798
 
508
799
def _auto_user_id():
518
809
    """
519
810
    import socket
520
811
 
521
 
    # XXX: Any good way to get real user name on win32?
 
812
    if sys.platform == 'win32':
 
813
        name = win32utils.get_user_name_unicode()
 
814
        if name is None:
 
815
            raise errors.BzrError("Cannot autodetect user name.\n"
 
816
                                  "Please, set your name with command like:\n"
 
817
                                  'bzr whoami "Your Name <name@domain.com>"')
 
818
        host = win32utils.get_host_name_unicode()
 
819
        if host is None:
 
820
            host = socket.gethostname()
 
821
        return name, (name + '@' + host)
522
822
 
523
823
    try:
524
824
        import pwd
525
825
        uid = os.getuid()
526
826
        w = pwd.getpwuid(uid)
527
827
 
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)
 
828
        # we try utf-8 first, because on many variants (like Linux),
 
829
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
830
        # false positives.  (many users will have their user encoding set to
 
831
        # latin-1, which cannot raise UnicodeError.)
 
832
        try:
 
833
            gecos = w.pw_gecos.decode('utf-8')
 
834
            encoding = 'utf-8'
 
835
        except UnicodeError:
 
836
            try:
 
837
                gecos = w.pw_gecos.decode(bzrlib.user_encoding)
 
838
                encoding = bzrlib.user_encoding
 
839
            except UnicodeError:
 
840
                raise errors.BzrCommandError('Unable to determine your name.  '
 
841
                   'Use "bzr whoami" to set it.')
 
842
        try:
 
843
            username = w.pw_name.decode(encoding)
 
844
        except UnicodeError:
 
845
            raise errors.BzrCommandError('Unable to determine your name.  '
 
846
                'Use "bzr whoami" to set it.')
535
847
 
536
848
        comma = gecos.find(',')
537
849
        if comma == -1:
552
864
    return realname, (username + '@' + socket.gethostname())
553
865
 
554
866
 
 
867
def parse_username(username):
 
868
    """Parse e-mail username and return a (name, address) tuple."""
 
869
    match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
 
870
    if match is None:
 
871
        return (username, '')
 
872
    else:
 
873
        return (match.group(1), match.group(2))
 
874
 
 
875
 
555
876
def extract_email_address(e):
556
877
    """Return just the address part of an email string.
557
 
    
 
878
 
558
879
    That is just the user@domain part, nothing else. 
559
880
    This part is required to contain only ascii characters.
560
881
    If it can't be extracted, raises an error.
561
 
    
 
882
 
562
883
    >>> extract_email_address('Jane Tester <jane@test.com>')
563
884
    "jane@test.com"
564
885
    """
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):
 
886
    name, email = parse_username(e)
 
887
    if not email:
 
888
        raise errors.NoEmailInUsername(e)
 
889
    return email
 
890
 
 
891
 
 
892
class TreeConfig(IniBasedConfig):
572
893
    """Branch configuration data associated with its contents, not location"""
 
894
 
 
895
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
 
896
 
573
897
    def __init__(self, branch):
 
898
        # XXX: Really this should be asking the branch for its configuration
 
899
        # data, rather than relying on a Transport, so that it can work 
 
900
        # more cleanly with a RemoteBranch that has no transport.
 
901
        self._config = TransportConfig(branch._transport, 'branch.conf')
574
902
        self.branch = branch
575
903
 
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
 
904
    def _get_parser(self, file=None):
 
905
        if file is not None:
 
906
            return IniBasedConfig._get_parser(file)
 
907
        return self._config._get_configobj()
583
908
 
584
909
    def get_option(self, name, section=None, default=None):
585
910
        self.branch.lock_read()
586
911
        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
 
912
            return self._config.get_option(name, section, default)
594
913
        finally:
595
914
            self.branch.unlock()
596
915
        return result
599
918
        """Set a per-branch configuration option"""
600
919
        self.branch.lock_write()
601
920
        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)
 
921
            self._config.set_option(value, name, section)
616
922
        finally:
617
923
            self.branch.unlock()
 
924
 
 
925
 
 
926
class AuthenticationConfig(object):
 
927
    """The authentication configuration file based on a ini file.
 
928
 
 
929
    Implements the authentication.conf file described in
 
930
    doc/developers/authentication-ring.txt.
 
931
    """
 
932
 
 
933
    def __init__(self, _file=None):
 
934
        self._config = None # The ConfigObj
 
935
        if _file is None:
 
936
            self._filename = authentication_config_filename()
 
937
            self._input = self._filename = authentication_config_filename()
 
938
        else:
 
939
            # Tests can provide a string as _file
 
940
            self._filename = None
 
941
            self._input = _file
 
942
 
 
943
    def _get_config(self):
 
944
        if self._config is not None:
 
945
            return self._config
 
946
        try:
 
947
            # FIXME: Should we validate something here ? Includes: empty
 
948
            # sections are useless, at least one of
 
949
            # user/password/password_encoding should be defined, etc.
 
950
 
 
951
            # Note: the encoding below declares that the file itself is utf-8
 
952
            # encoded, but the values in the ConfigObj are always Unicode.
 
953
            self._config = ConfigObj(self._input, encoding='utf-8')
 
954
        except configobj.ConfigObjError, e:
 
955
            raise errors.ParseConfigError(e.errors, e.config.filename)
 
956
        return self._config
 
957
 
 
958
    def _save(self):
 
959
        """Save the config file, only tests should use it for now."""
 
960
        conf_dir = os.path.dirname(self._filename)
 
961
        ensure_config_dir_exists(conf_dir)
 
962
        self._get_config().write(file(self._filename, 'wb'))
 
963
 
 
964
    def _set_option(self, section_name, option_name, value):
 
965
        """Set an authentication configuration option"""
 
966
        conf = self._get_config()
 
967
        section = conf.get(section_name)
 
968
        if section is None:
 
969
            conf[section] = {}
 
970
            section = conf[section]
 
971
        section[option_name] = value
 
972
        self._save()
 
973
 
 
974
    def get_credentials(self, scheme, host, port=None, user=None, path=None):
 
975
        """Returns the matching credentials from authentication.conf file.
 
976
 
 
977
        :param scheme: protocol
 
978
 
 
979
        :param host: the server address
 
980
 
 
981
        :param port: the associated port (optional)
 
982
 
 
983
        :param user: login (optional)
 
984
 
 
985
        :param path: the absolute path on the server (optional)
 
986
 
 
987
        :return: A dict containing the matching credentials or None.
 
988
           This includes:
 
989
           - name: the section name of the credentials in the
 
990
             authentication.conf file,
 
991
           - user: can't de different from the provided user if any,
 
992
           - password: the decoded password, could be None if the credential
 
993
             defines only the user
 
994
           - verify_certificates: https specific, True if the server
 
995
             certificate should be verified, False otherwise.
 
996
        """
 
997
        credentials = None
 
998
        for auth_def_name, auth_def in self._get_config().items():
 
999
            if type(auth_def) is not configobj.Section:
 
1000
                raise ValueError("%s defined outside a section" % auth_def_name)
 
1001
 
 
1002
            a_scheme, a_host, a_user, a_path = map(
 
1003
                auth_def.get, ['scheme', 'host', 'user', 'path'])
 
1004
 
 
1005
            try:
 
1006
                a_port = auth_def.as_int('port')
 
1007
            except KeyError:
 
1008
                a_port = None
 
1009
            except ValueError:
 
1010
                raise ValueError("'port' not numeric in %s" % auth_def_name)
 
1011
            try:
 
1012
                a_verify_certificates = auth_def.as_bool('verify_certificates')
 
1013
            except KeyError:
 
1014
                a_verify_certificates = True
 
1015
            except ValueError:
 
1016
                raise ValueError(
 
1017
                    "'verify_certificates' not boolean in %s" % auth_def_name)
 
1018
 
 
1019
            # Attempt matching
 
1020
            if a_scheme is not None and scheme != a_scheme:
 
1021
                continue
 
1022
            if a_host is not None:
 
1023
                if not (host == a_host
 
1024
                        or (a_host.startswith('.') and host.endswith(a_host))):
 
1025
                    continue
 
1026
            if a_port is not None and port != a_port:
 
1027
                continue
 
1028
            if (a_path is not None and path is not None
 
1029
                and not path.startswith(a_path)):
 
1030
                continue
 
1031
            if (a_user is not None and user is not None
 
1032
                and a_user != user):
 
1033
                # Never contradict the caller about the user to be used
 
1034
                continue
 
1035
            if a_user is None:
 
1036
                # Can't find a user
 
1037
                continue
 
1038
            credentials = dict(name=auth_def_name,
 
1039
                               user=a_user,
 
1040
                               password=auth_def.get('password', None),
 
1041
                               verify_certificates=a_verify_certificates)
 
1042
            self.decode_password(credentials,
 
1043
                                 auth_def.get('password_encoding', None))
 
1044
            if 'auth' in debug.debug_flags:
 
1045
                trace.mutter("Using authentication section: %r", auth_def_name)
 
1046
            break
 
1047
 
 
1048
        return credentials
 
1049
 
 
1050
    def get_user(self, scheme, host, port=None,
 
1051
                 realm=None, path=None, prompt=None):
 
1052
        """Get a user from authentication file.
 
1053
 
 
1054
        :param scheme: protocol
 
1055
 
 
1056
        :param host: the server address
 
1057
 
 
1058
        :param port: the associated port (optional)
 
1059
 
 
1060
        :param realm: the realm sent by the server (optional)
 
1061
 
 
1062
        :param path: the absolute path on the server (optional)
 
1063
 
 
1064
        :return: The found user.
 
1065
        """
 
1066
        credentials = self.get_credentials(scheme, host, port, user=None,
 
1067
                                           path=path)
 
1068
        if credentials is not None:
 
1069
            user = credentials['user']
 
1070
        else:
 
1071
            user = None
 
1072
        return user
 
1073
 
 
1074
    def get_password(self, scheme, host, user, port=None,
 
1075
                     realm=None, path=None, prompt=None):
 
1076
        """Get a password from authentication file or prompt the user for one.
 
1077
 
 
1078
        :param scheme: protocol
 
1079
 
 
1080
        :param host: the server address
 
1081
 
 
1082
        :param port: the associated port (optional)
 
1083
 
 
1084
        :param user: login
 
1085
 
 
1086
        :param realm: the realm sent by the server (optional)
 
1087
 
 
1088
        :param path: the absolute path on the server (optional)
 
1089
 
 
1090
        :return: The found password or the one entered by the user.
 
1091
        """
 
1092
        credentials = self.get_credentials(scheme, host, port, user, path)
 
1093
        if credentials is not None:
 
1094
            password = credentials['password']
 
1095
            if password is not None and scheme is 'ssh':
 
1096
                trace.warning('password ignored in section [%s],'
 
1097
                              ' use an ssh agent instead'
 
1098
                              % credentials['name'])
 
1099
                password = None
 
1100
        else:
 
1101
            password = None
 
1102
        # Prompt user only if we could't find a password
 
1103
        if password is None:
 
1104
            if prompt is None:
 
1105
                # Create a default prompt suitable for most cases
 
1106
                prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
 
1107
            # Special handling for optional fields in the prompt
 
1108
            if port is not None:
 
1109
                prompt_host = '%s:%d' % (host, port)
 
1110
            else:
 
1111
                prompt_host = host
 
1112
            password = ui.ui_factory.get_password(prompt,
 
1113
                                                  host=prompt_host, user=user)
 
1114
        return password
 
1115
 
 
1116
    def decode_password(self, credentials, encoding):
 
1117
        return credentials
 
1118
 
 
1119
 
 
1120
class TransportConfig(object):
 
1121
    """A Config that reads/writes a config file on a Transport.
 
1122
 
 
1123
    It is a low-level object that considers config data to be name/value pairs
 
1124
    that may be associated with a section.  Assigning meaning to the these
 
1125
    values is done at higher levels like TreeConfig.
 
1126
    """
 
1127
 
 
1128
    def __init__(self, transport, filename):
 
1129
        self._transport = transport
 
1130
        self._filename = filename
 
1131
 
 
1132
    def get_option(self, name, section=None, default=None):
 
1133
        """Return the value associated with a named option.
 
1134
 
 
1135
        :param name: The name of the value
 
1136
        :param section: The section the option is in (if any)
 
1137
        :param default: The value to return if the value is not set
 
1138
        :return: The value or default value
 
1139
        """
 
1140
        configobj = self._get_configobj()
 
1141
        if section is None:
 
1142
            section_obj = configobj
 
1143
        else:
 
1144
            try:
 
1145
                section_obj = configobj[section]
 
1146
            except KeyError:
 
1147
                return default
 
1148
        return section_obj.get(name, default)
 
1149
 
 
1150
    def set_option(self, value, name, section=None):
 
1151
        """Set the value associated with a named option.
 
1152
 
 
1153
        :param value: The value to set
 
1154
        :param name: The name of the value to set
 
1155
        :param section: The section the option is in (if any)
 
1156
        """
 
1157
        configobj = self._get_configobj()
 
1158
        if section is None:
 
1159
            configobj[name] = value
 
1160
        else:
 
1161
            configobj.setdefault(section, {})[name] = value
 
1162
        self._set_configobj(configobj)
 
1163
 
 
1164
    def _get_configobj(self):
 
1165
        try:
 
1166
            return ConfigObj(self._transport.get(self._filename),
 
1167
                             encoding='utf-8')
 
1168
        except errors.NoSuchFile:
 
1169
            return ConfigObj(encoding='utf-8')
 
1170
 
 
1171
    def _set_configobj(self, configobj):
 
1172
        out_file = StringIO()
 
1173
        configobj.write(out_file)
 
1174
        out_file.seek(0)
 
1175
        self._transport.put_file(self._filename, out_file)