/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Keir Mierle
  • Date: 2007-09-03 23:54:22 UTC
  • mto: This revision was merged to the branch mainline in revision 2824.
  • Revision ID: keir@cs.utoronto.ca-20070903235422-zfj6mk7nts9nmzw2
Add Mutt as a supported client email program. Also rearranges various listings
of the mail clients to be alphabetical; the number of supported email clients
is only going to grow, and this way it is easier to scan for your favorite
email client.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005, 2007 Canonical Ltd
 
2
#   Authors: Robert Collins <robert.collins@canonical.com>
 
3
#            and others
 
4
#
 
5
# This program is free software; you can redistribute it and/or modify
 
6
# it under the terms of the GNU General Public License as published by
 
7
# the Free Software Foundation; either version 2 of the License, or
 
8
# (at your option) any later version.
 
9
#
 
10
# This program is distributed in the hope that it will be useful,
 
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
# GNU General Public License for more details.
 
14
#
 
15
# You should have received a copy of the GNU General Public License
 
16
# along with this program; if not, write to the Free Software
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
18
 
 
19
"""Configuration that affects the behaviour of Bazaar.
 
20
 
 
21
Currently this configuration resides in ~/.bazaar/bazaar.conf
 
22
and ~/.bazaar/locations.conf, which is written to by bzr.
 
23
 
 
24
In bazaar.conf the following options may be set:
 
25
[DEFAULT]
 
26
editor=name-of-program
 
27
email=Your Name <your@email.address>
 
28
check_signatures=require|ignore|check-available(default)
 
29
create_signatures=always|never|when-required(default)
 
30
gpg_signing_command=name-of-program
 
31
log_format=name-of-format
 
32
 
 
33
in locations.conf, you specify the url of a branch and options for it.
 
34
Wildcards may be used - * and ? as normal in shell completion. Options
 
35
set in both bazaar.conf and locations.conf are overridden by the locations.conf
 
36
setting.
 
37
[/home/robertc/source]
 
38
recurse=False|True(default)
 
39
email= as above
 
40
check_signatures= as above 
 
41
create_signatures= as above.
 
42
 
 
43
explanation of options
 
44
----------------------
 
45
editor - this option sets the pop up editor to use during commits.
 
46
email - this option sets the user id bzr will use when committing.
 
47
check_signatures - this option controls whether bzr will require good gpg
 
48
                   signatures, ignore them, or check them if they are 
 
49
                   present.
 
50
create_signatures - this option controls whether bzr will always create 
 
51
                    gpg signatures, never create them, or create them if the
 
52
                    branch is configured to require them.
 
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
 
 
56
In bazaar.conf you can also define aliases in the ALIASES sections, example
 
57
 
 
58
[ALIASES]
 
59
lastlog=log --line -r-10..-1
 
60
ll=log --line -r-10..-1
 
61
h=help
 
62
up=pull
 
63
"""
 
64
 
 
65
import os
 
66
import sys
 
67
 
 
68
from bzrlib.lazy_import import lazy_import
 
69
lazy_import(globals(), """
 
70
import errno
 
71
from fnmatch import fnmatch
 
72
import re
 
73
from StringIO import StringIO
 
74
 
 
75
import bzrlib
 
76
from bzrlib import (
 
77
    errors,
 
78
    mail_client,
 
79
    osutils,
 
80
    symbol_versioning,
 
81
    trace,
 
82
    urlutils,
 
83
    win32utils,
 
84
    )
 
85
import bzrlib.util.configobj.configobj as configobj
 
86
""")
 
87
 
 
88
from bzrlib.trace import mutter, warning
 
89
 
 
90
 
 
91
CHECK_IF_POSSIBLE=0
 
92
CHECK_ALWAYS=1
 
93
CHECK_NEVER=2
 
94
 
 
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
 
 
125
class ConfigObj(configobj.ConfigObj):
 
126
 
 
127
    def get_bool(self, section, key):
 
128
        return self[section].as_bool(key)
 
129
 
 
130
    def get_value(self, section, name):
 
131
        # Try [] for the old DEFAULT section.
 
132
        if section == "DEFAULT":
 
133
            try:
 
134
                return self[name]
 
135
            except KeyError:
 
136
                pass
 
137
        return self[section][name]
 
138
 
 
139
 
 
140
class Config(object):
 
141
    """A configuration policy - what username, editor, gpg needs etc."""
 
142
 
 
143
    def get_editor(self):
 
144
        """Get the users pop up editor."""
 
145
        raise NotImplementedError
 
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
                'default': mail_client.DefaultMail,
 
154
                'editor': mail_client.Editor,
 
155
                'evolution': mail_client.Evolution,
 
156
                'kmail': mail_client.KMail,
 
157
                'mapi': mail_client.MAPIClient,
 
158
                'mutt': mail_client.Mutt,
 
159
                'thunderbird': mail_client.Thunderbird,
 
160
                'xdg-email': mail_client.XDGEmail,
 
161
            }[selected_client]
 
162
        except KeyError:
 
163
            raise errors.UnknownMailClient(selected_client)
 
164
        return mail_client_class(self)
 
165
 
 
166
    def _get_signature_checking(self):
 
167
        """Template method to override signature checking policy."""
 
168
 
 
169
    def _get_signing_policy(self):
 
170
        """Template method to override signature creation policy."""
 
171
 
 
172
    def _get_user_option(self, option_name):
 
173
        """Template method to provide a user option."""
 
174
        return None
 
175
 
 
176
    def get_user_option(self, option_name):
 
177
        """Get a generic option - no special process, no default."""
 
178
        return self._get_user_option(option_name)
 
179
 
 
180
    def gpg_signing_command(self):
 
181
        """What program should be used to sign signatures?"""
 
182
        result = self._gpg_signing_command()
 
183
        if result is None:
 
184
            result = "gpg"
 
185
        return result
 
186
 
 
187
    def _gpg_signing_command(self):
 
188
        """See gpg_signing_command()."""
 
189
        return None
 
190
 
 
191
    def log_format(self):
 
192
        """What log format should be used"""
 
193
        result = self._log_format()
 
194
        if result is None:
 
195
            result = "long"
 
196
        return result
 
197
 
 
198
    def _log_format(self):
 
199
        """See log_format()."""
 
200
        return None
 
201
 
 
202
    def __init__(self):
 
203
        super(Config, self).__init__()
 
204
 
 
205
    def post_commit(self):
 
206
        """An ordered list of python functions to call.
 
207
 
 
208
        Each function takes branch, rev_id as parameters.
 
209
        """
 
210
        return self._post_commit()
 
211
 
 
212
    def _post_commit(self):
 
213
        """See Config.post_commit."""
 
214
        return None
 
215
 
 
216
    def user_email(self):
 
217
        """Return just the email component of a username."""
 
218
        return extract_email_address(self.username())
 
219
 
 
220
    def username(self):
 
221
        """Return email-style username.
 
222
    
 
223
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
 
224
        
 
225
        $BZR_EMAIL can be set to override this (as well as the
 
226
        deprecated $BZREMAIL), then
 
227
        the concrete policy type is checked, and finally
 
228
        $EMAIL is examined.
 
229
        If none is found, a reasonable default is (hopefully)
 
230
        created.
 
231
    
 
232
        TODO: Check it's reasonably well-formed.
 
233
        """
 
234
        v = os.environ.get('BZR_EMAIL')
 
235
        if v:
 
236
            return v.decode(bzrlib.user_encoding)
 
237
        v = os.environ.get('BZREMAIL')
 
238
        if v:
 
239
            warning('BZREMAIL is deprecated in favor of BZR_EMAIL. Please update your configuration.')
 
240
            return v.decode(bzrlib.user_encoding)
 
241
    
 
242
        v = self._get_user_id()
 
243
        if v:
 
244
            return v
 
245
        
 
246
        v = os.environ.get('EMAIL')
 
247
        if v:
 
248
            return v.decode(bzrlib.user_encoding)
 
249
 
 
250
        name, email = _auto_user_id()
 
251
        if name:
 
252
            return '%s <%s>' % (name, email)
 
253
        else:
 
254
            return email
 
255
 
 
256
    def signature_checking(self):
 
257
        """What is the current policy for signature checking?."""
 
258
        policy = self._get_signature_checking()
 
259
        if policy is not None:
 
260
            return policy
 
261
        return CHECK_IF_POSSIBLE
 
262
 
 
263
    def signing_policy(self):
 
264
        """What is the current policy for signature checking?."""
 
265
        policy = self._get_signing_policy()
 
266
        if policy is not None:
 
267
            return policy
 
268
        return SIGN_WHEN_REQUIRED
 
269
 
 
270
    def signature_needed(self):
 
271
        """Is a signature needed when committing ?."""
 
272
        policy = self._get_signing_policy()
 
273
        if policy is None:
 
274
            policy = self._get_signature_checking()
 
275
            if policy is not None:
 
276
                warning("Please use create_signatures, not check_signatures "
 
277
                        "to set signing policy.")
 
278
            if policy == CHECK_ALWAYS:
 
279
                return True
 
280
        elif policy == SIGN_ALWAYS:
 
281
            return True
 
282
        return False
 
283
 
 
284
    def get_alias(self, value):
 
285
        return self._get_alias(value)
 
286
 
 
287
    def _get_alias(self, value):
 
288
        pass
 
289
 
 
290
    def get_nickname(self):
 
291
        return self._get_nickname()
 
292
 
 
293
    def _get_nickname(self):
 
294
        return None
 
295
 
 
296
 
 
297
class IniBasedConfig(Config):
 
298
    """A configuration policy that draws from ini files."""
 
299
 
 
300
    def _get_parser(self, file=None):
 
301
        if self._parser is not None:
 
302
            return self._parser
 
303
        if file is None:
 
304
            input = self._get_filename()
 
305
        else:
 
306
            input = file
 
307
        try:
 
308
            self._parser = ConfigObj(input, encoding='utf-8')
 
309
        except configobj.ConfigObjError, e:
 
310
            raise errors.ParseConfigError(e.errors, e.config.filename)
 
311
        return self._parser
 
312
 
 
313
    def _get_matching_sections(self):
 
314
        """Return an ordered list of (section_name, extra_path) pairs.
 
315
 
 
316
        If the section contains inherited configuration, extra_path is
 
317
        a string containing the additional path components.
 
318
        """
 
319
        section = self._get_section()
 
320
        if section is not None:
 
321
            return [(section, '')]
 
322
        else:
 
323
            return []
 
324
 
 
325
    def _get_section(self):
 
326
        """Override this to define the section used by the config."""
 
327
        return "DEFAULT"
 
328
 
 
329
    def _get_option_policy(self, section, option_name):
 
330
        """Return the policy for the given (section, option_name) pair."""
 
331
        return POLICY_NONE
 
332
 
 
333
    def _get_signature_checking(self):
 
334
        """See Config._get_signature_checking."""
 
335
        policy = self._get_user_option('check_signatures')
 
336
        if policy:
 
337
            return self._string_to_signature_policy(policy)
 
338
 
 
339
    def _get_signing_policy(self):
 
340
        """See Config._get_signing_policy"""
 
341
        policy = self._get_user_option('create_signatures')
 
342
        if policy:
 
343
            return self._string_to_signing_policy(policy)
 
344
 
 
345
    def _get_user_id(self):
 
346
        """Get the user id from the 'email' key in the current section."""
 
347
        return self._get_user_option('email')
 
348
 
 
349
    def _get_user_option(self, option_name):
 
350
        """See Config._get_user_option."""
 
351
        for (section, extra_path) in self._get_matching_sections():
 
352
            try:
 
353
                value = self._get_parser().get_value(section, option_name)
 
354
            except KeyError:
 
355
                continue
 
356
            policy = self._get_option_policy(section, option_name)
 
357
            if policy == POLICY_NONE:
 
358
                return value
 
359
            elif policy == POLICY_NORECURSE:
 
360
                # norecurse items only apply to the exact path
 
361
                if extra_path:
 
362
                    continue
 
363
                else:
 
364
                    return value
 
365
            elif policy == POLICY_APPENDPATH:
 
366
                if extra_path:
 
367
                    value = urlutils.join(value, extra_path)
 
368
                return value
 
369
            else:
 
370
                raise AssertionError('Unexpected config policy %r' % policy)
 
371
        else:
 
372
            return None
 
373
 
 
374
    def _gpg_signing_command(self):
 
375
        """See Config.gpg_signing_command."""
 
376
        return self._get_user_option('gpg_signing_command')
 
377
 
 
378
    def _log_format(self):
 
379
        """See Config.log_format."""
 
380
        return self._get_user_option('log_format')
 
381
 
 
382
    def __init__(self, get_filename):
 
383
        super(IniBasedConfig, self).__init__()
 
384
        self._get_filename = get_filename
 
385
        self._parser = None
 
386
        
 
387
    def _post_commit(self):
 
388
        """See Config.post_commit."""
 
389
        return self._get_user_option('post_commit')
 
390
 
 
391
    def _string_to_signature_policy(self, signature_string):
 
392
        """Convert a string to a signing policy."""
 
393
        if signature_string.lower() == 'check-available':
 
394
            return CHECK_IF_POSSIBLE
 
395
        if signature_string.lower() == 'ignore':
 
396
            return CHECK_NEVER
 
397
        if signature_string.lower() == 'require':
 
398
            return CHECK_ALWAYS
 
399
        raise errors.BzrError("Invalid signatures policy '%s'"
 
400
                              % signature_string)
 
401
 
 
402
    def _string_to_signing_policy(self, signature_string):
 
403
        """Convert a string to a signing policy."""
 
404
        if signature_string.lower() == 'when-required':
 
405
            return SIGN_WHEN_REQUIRED
 
406
        if signature_string.lower() == 'never':
 
407
            return SIGN_NEVER
 
408
        if signature_string.lower() == 'always':
 
409
            return SIGN_ALWAYS
 
410
        raise errors.BzrError("Invalid signing policy '%s'"
 
411
                              % signature_string)
 
412
 
 
413
    def _get_alias(self, value):
 
414
        try:
 
415
            return self._get_parser().get_value("ALIASES", 
 
416
                                                value)
 
417
        except KeyError:
 
418
            pass
 
419
 
 
420
    def _get_nickname(self):
 
421
        return self.get_user_option('nickname')
 
422
 
 
423
 
 
424
class GlobalConfig(IniBasedConfig):
 
425
    """The configuration that should be used for a specific location."""
 
426
 
 
427
    def get_editor(self):
 
428
        return self._get_user_option('editor')
 
429
 
 
430
    def __init__(self):
 
431
        super(GlobalConfig, self).__init__(config_filename)
 
432
 
 
433
    def set_user_option(self, option, value):
 
434
        """Save option and its value in the configuration."""
 
435
        # FIXME: RBC 20051029 This should refresh the parser and also take a
 
436
        # file lock on bazaar.conf.
 
437
        conf_dir = os.path.dirname(self._get_filename())
 
438
        ensure_config_dir_exists(conf_dir)
 
439
        if 'DEFAULT' not in self._get_parser():
 
440
            self._get_parser()['DEFAULT'] = {}
 
441
        self._get_parser()['DEFAULT'][option] = value
 
442
        f = open(self._get_filename(), 'wb')
 
443
        self._get_parser().write(f)
 
444
        f.close()
 
445
 
 
446
 
 
447
class LocationConfig(IniBasedConfig):
 
448
    """A configuration object that gives the policy for a location."""
 
449
 
 
450
    def __init__(self, location):
 
451
        name_generator = locations_config_filename
 
452
        if (not os.path.exists(name_generator()) and 
 
453
                os.path.exists(branches_config_filename())):
 
454
            if sys.platform == 'win32':
 
455
                warning('Please rename %s to %s' 
 
456
                         % (branches_config_filename(),
 
457
                            locations_config_filename()))
 
458
            else:
 
459
                warning('Please rename ~/.bazaar/branches.conf'
 
460
                        ' to ~/.bazaar/locations.conf')
 
461
            name_generator = branches_config_filename
 
462
        super(LocationConfig, self).__init__(name_generator)
 
463
        # local file locations are looked up by local path, rather than
 
464
        # by file url. This is because the config file is a user
 
465
        # file, and we would rather not expose the user to file urls.
 
466
        if location.startswith('file://'):
 
467
            location = urlutils.local_path_from_url(location)
 
468
        self.location = location
 
469
 
 
470
    def _get_matching_sections(self):
 
471
        """Return an ordered list of section names matching this location."""
 
472
        sections = self._get_parser()
 
473
        location_names = self.location.split('/')
 
474
        if self.location.endswith('/'):
 
475
            del location_names[-1]
 
476
        matches=[]
 
477
        for section in sections:
 
478
            # location is a local path if possible, so we need
 
479
            # to convert 'file://' urls to local paths if necessary.
 
480
            # This also avoids having file:///path be a more exact
 
481
            # match than '/path'.
 
482
            if section.startswith('file://'):
 
483
                section_path = urlutils.local_path_from_url(section)
 
484
            else:
 
485
                section_path = section
 
486
            section_names = section_path.split('/')
 
487
            if section.endswith('/'):
 
488
                del section_names[-1]
 
489
            names = zip(location_names, section_names)
 
490
            matched = True
 
491
            for name in names:
 
492
                if not fnmatch(name[0], name[1]):
 
493
                    matched = False
 
494
                    break
 
495
            if not matched:
 
496
                continue
 
497
            # so, for the common prefix they matched.
 
498
            # if section is longer, no match.
 
499
            if len(section_names) > len(location_names):
 
500
                continue
 
501
            matches.append((len(section_names), section,
 
502
                            '/'.join(location_names[len(section_names):])))
 
503
        matches.sort(reverse=True)
 
504
        sections = []
 
505
        for (length, section, extra_path) in matches:
 
506
            sections.append((section, extra_path))
 
507
            # should we stop looking for parent configs here?
 
508
            try:
 
509
                if self._get_parser()[section].as_bool('ignore_parents'):
 
510
                    break
 
511
            except KeyError:
 
512
                pass
 
513
        return sections
 
514
 
 
515
    def _get_option_policy(self, section, option_name):
 
516
        """Return the policy for the given (section, option_name) pair."""
 
517
        # check for the old 'recurse=False' flag
 
518
        try:
 
519
            recurse = self._get_parser()[section].as_bool('recurse')
 
520
        except KeyError:
 
521
            recurse = True
 
522
        if not recurse:
 
523
            return POLICY_NORECURSE
 
524
 
 
525
        policy_key = option_name + ':policy'
 
526
        try:
 
527
            policy_name = self._get_parser()[section][policy_key]
 
528
        except KeyError:
 
529
            policy_name = None
 
530
 
 
531
        return _policy_value[policy_name]
 
532
 
 
533
    def _set_option_policy(self, section, option_name, option_policy):
 
534
        """Set the policy for the given option name in the given section."""
 
535
        # The old recurse=False option affects all options in the
 
536
        # section.  To handle multiple policies in the section, we
 
537
        # need to convert it to a policy_norecurse key.
 
538
        try:
 
539
            recurse = self._get_parser()[section].as_bool('recurse')
 
540
        except KeyError:
 
541
            pass
 
542
        else:
 
543
            symbol_versioning.warn(
 
544
                'The recurse option is deprecated as of 0.14.  '
 
545
                'The section "%s" has been converted to use policies.'
 
546
                % section,
 
547
                DeprecationWarning)
 
548
            del self._get_parser()[section]['recurse']
 
549
            if not recurse:
 
550
                for key in self._get_parser()[section].keys():
 
551
                    if not key.endswith(':policy'):
 
552
                        self._get_parser()[section][key +
 
553
                                                    ':policy'] = 'norecurse'
 
554
 
 
555
        policy_key = option_name + ':policy'
 
556
        policy_name = _policy_name[option_policy]
 
557
        if policy_name is not None:
 
558
            self._get_parser()[section][policy_key] = policy_name
 
559
        else:
 
560
            if policy_key in self._get_parser()[section]:
 
561
                del self._get_parser()[section][policy_key]
 
562
 
 
563
    def set_user_option(self, option, value, store=STORE_LOCATION):
 
564
        """Save option and its value in the configuration."""
 
565
        assert store in [STORE_LOCATION,
 
566
                         STORE_LOCATION_NORECURSE,
 
567
                         STORE_LOCATION_APPENDPATH], 'bad storage policy'
 
568
        # FIXME: RBC 20051029 This should refresh the parser and also take a
 
569
        # file lock on locations.conf.
 
570
        conf_dir = os.path.dirname(self._get_filename())
 
571
        ensure_config_dir_exists(conf_dir)
 
572
        location = self.location
 
573
        if location.endswith('/'):
 
574
            location = location[:-1]
 
575
        if (not location in self._get_parser() and
 
576
            not location + '/' in self._get_parser()):
 
577
            self._get_parser()[location]={}
 
578
        elif location + '/' in self._get_parser():
 
579
            location = location + '/'
 
580
        self._get_parser()[location][option]=value
 
581
        # the allowed values of store match the config policies
 
582
        self._set_option_policy(location, option, store)
 
583
        self._get_parser().write(file(self._get_filename(), 'wb'))
 
584
 
 
585
 
 
586
class BranchConfig(Config):
 
587
    """A configuration object giving the policy for a branch."""
 
588
 
 
589
    def _get_branch_data_config(self):
 
590
        if self._branch_data_config is None:
 
591
            self._branch_data_config = TreeConfig(self.branch)
 
592
        return self._branch_data_config
 
593
 
 
594
    def _get_location_config(self):
 
595
        if self._location_config is None:
 
596
            self._location_config = LocationConfig(self.branch.base)
 
597
        return self._location_config
 
598
 
 
599
    def _get_global_config(self):
 
600
        if self._global_config is None:
 
601
            self._global_config = GlobalConfig()
 
602
        return self._global_config
 
603
 
 
604
    def _get_best_value(self, option_name):
 
605
        """This returns a user option from local, tree or global config.
 
606
 
 
607
        They are tried in that order.  Use get_safe_value if trusted values
 
608
        are necessary.
 
609
        """
 
610
        for source in self.option_sources:
 
611
            value = getattr(source(), option_name)()
 
612
            if value is not None:
 
613
                return value
 
614
        return None
 
615
 
 
616
    def _get_safe_value(self, option_name):
 
617
        """This variant of get_best_value never returns untrusted values.
 
618
        
 
619
        It does not return values from the branch data, because the branch may
 
620
        not be controlled by the user.
 
621
 
 
622
        We may wish to allow locations.conf to control whether branches are
 
623
        trusted in the future.
 
624
        """
 
625
        for source in (self._get_location_config, self._get_global_config):
 
626
            value = getattr(source(), option_name)()
 
627
            if value is not None:
 
628
                return value
 
629
        return None
 
630
 
 
631
    def _get_user_id(self):
 
632
        """Return the full user id for the branch.
 
633
    
 
634
        e.g. "John Hacker <jhacker@foo.org>"
 
635
        This is looked up in the email controlfile for the branch.
 
636
        """
 
637
        try:
 
638
            return (self.branch.control_files.get_utf8("email") 
 
639
                    .read()
 
640
                    .decode(bzrlib.user_encoding)
 
641
                    .rstrip("\r\n"))
 
642
        except errors.NoSuchFile, e:
 
643
            pass
 
644
        
 
645
        return self._get_best_value('_get_user_id')
 
646
 
 
647
    def _get_signature_checking(self):
 
648
        """See Config._get_signature_checking."""
 
649
        return self._get_best_value('_get_signature_checking')
 
650
 
 
651
    def _get_signing_policy(self):
 
652
        """See Config._get_signing_policy."""
 
653
        return self._get_best_value('_get_signing_policy')
 
654
 
 
655
    def _get_user_option(self, option_name):
 
656
        """See Config._get_user_option."""
 
657
        for source in self.option_sources:
 
658
            value = source()._get_user_option(option_name)
 
659
            if value is not None:
 
660
                return value
 
661
        return None
 
662
 
 
663
    def set_user_option(self, name, value, store=STORE_BRANCH,
 
664
        warn_masked=False):
 
665
        if store == STORE_BRANCH:
 
666
            self._get_branch_data_config().set_option(value, name)
 
667
        elif store == STORE_GLOBAL:
 
668
            self._get_global_config().set_user_option(name, value)
 
669
        else:
 
670
            self._get_location_config().set_user_option(name, value, store)
 
671
        if not warn_masked:
 
672
            return
 
673
        if store in (STORE_GLOBAL, STORE_BRANCH):
 
674
            mask_value = self._get_location_config().get_user_option(name)
 
675
            if mask_value is not None:
 
676
                trace.warning('Value "%s" is masked by "%s" from'
 
677
                              ' locations.conf', value, mask_value)
 
678
            else:
 
679
                if store == STORE_GLOBAL:
 
680
                    branch_config = self._get_branch_data_config()
 
681
                    mask_value = branch_config.get_user_option(name)
 
682
                    if mask_value is not None:
 
683
                        trace.warning('Value "%s" is masked by "%s" from'
 
684
                                      ' branch.conf', value, mask_value)
 
685
 
 
686
 
 
687
    def _gpg_signing_command(self):
 
688
        """See Config.gpg_signing_command."""
 
689
        return self._get_safe_value('_gpg_signing_command')
 
690
        
 
691
    def __init__(self, branch):
 
692
        super(BranchConfig, self).__init__()
 
693
        self._location_config = None
 
694
        self._branch_data_config = None
 
695
        self._global_config = None
 
696
        self.branch = branch
 
697
        self.option_sources = (self._get_location_config, 
 
698
                               self._get_branch_data_config,
 
699
                               self._get_global_config)
 
700
 
 
701
    def _post_commit(self):
 
702
        """See Config.post_commit."""
 
703
        return self._get_safe_value('_post_commit')
 
704
 
 
705
    def _get_nickname(self):
 
706
        value = self._get_explicit_nickname()
 
707
        if value is not None:
 
708
            return value
 
709
        return urlutils.unescape(self.branch.base.split('/')[-2])
 
710
 
 
711
    def has_explicit_nickname(self):
 
712
        """Return true if a nickname has been explicitly assigned."""
 
713
        return self._get_explicit_nickname() is not None
 
714
 
 
715
    def _get_explicit_nickname(self):
 
716
        return self._get_best_value('_get_nickname')
 
717
 
 
718
    def _log_format(self):
 
719
        """See Config.log_format."""
 
720
        return self._get_best_value('_log_format')
 
721
 
 
722
 
 
723
def ensure_config_dir_exists(path=None):
 
724
    """Make sure a configuration directory exists.
 
725
    This makes sure that the directory exists.
 
726
    On windows, since configuration directories are 2 levels deep,
 
727
    it makes sure both the directory and the parent directory exists.
 
728
    """
 
729
    if path is None:
 
730
        path = config_dir()
 
731
    if not os.path.isdir(path):
 
732
        if sys.platform == 'win32':
 
733
            parent_dir = os.path.dirname(path)
 
734
            if not os.path.isdir(parent_dir):
 
735
                mutter('creating config parent directory: %r', parent_dir)
 
736
            os.mkdir(parent_dir)
 
737
        mutter('creating config directory: %r', path)
 
738
        os.mkdir(path)
 
739
 
 
740
 
 
741
def config_dir():
 
742
    """Return per-user configuration directory.
 
743
 
 
744
    By default this is ~/.bazaar/
 
745
    
 
746
    TODO: Global option --config-dir to override this.
 
747
    """
 
748
    base = os.environ.get('BZR_HOME', None)
 
749
    if sys.platform == 'win32':
 
750
        if base is None:
 
751
            base = win32utils.get_appdata_location_unicode()
 
752
        if base is None:
 
753
            base = os.environ.get('HOME', None)
 
754
        if base is None:
 
755
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA, or HOME set')
 
756
        return osutils.pathjoin(base, 'bazaar', '2.0')
 
757
    else:
 
758
        # cygwin, linux, and darwin all have a $HOME directory
 
759
        if base is None:
 
760
            base = os.path.expanduser("~")
 
761
        return osutils.pathjoin(base, ".bazaar")
 
762
 
 
763
 
 
764
def config_filename():
 
765
    """Return per-user configuration ini file filename."""
 
766
    return osutils.pathjoin(config_dir(), 'bazaar.conf')
 
767
 
 
768
 
 
769
def branches_config_filename():
 
770
    """Return per-user configuration ini file filename."""
 
771
    return osutils.pathjoin(config_dir(), 'branches.conf')
 
772
 
 
773
 
 
774
def locations_config_filename():
 
775
    """Return per-user configuration ini file filename."""
 
776
    return osutils.pathjoin(config_dir(), 'locations.conf')
 
777
 
 
778
 
 
779
def user_ignore_config_filename():
 
780
    """Return the user default ignore filename"""
 
781
    return osutils.pathjoin(config_dir(), 'ignore')
 
782
 
 
783
 
 
784
def _auto_user_id():
 
785
    """Calculate automatic user identification.
 
786
 
 
787
    Returns (realname, email).
 
788
 
 
789
    Only used when none is set in the environment or the id file.
 
790
 
 
791
    This previously used the FQDN as the default domain, but that can
 
792
    be very slow on machines where DNS is broken.  So now we simply
 
793
    use the hostname.
 
794
    """
 
795
    import socket
 
796
 
 
797
    if sys.platform == 'win32':
 
798
        name = win32utils.get_user_name_unicode()
 
799
        if name is None:
 
800
            raise errors.BzrError("Cannot autodetect user name.\n"
 
801
                                  "Please, set your name with command like:\n"
 
802
                                  'bzr whoami "Your Name <name@domain.com>"')
 
803
        host = win32utils.get_host_name_unicode()
 
804
        if host is None:
 
805
            host = socket.gethostname()
 
806
        return name, (name + '@' + host)
 
807
 
 
808
    try:
 
809
        import pwd
 
810
        uid = os.getuid()
 
811
        w = pwd.getpwuid(uid)
 
812
 
 
813
        # we try utf-8 first, because on many variants (like Linux),
 
814
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
815
        # false positives.  (many users will have their user encoding set to
 
816
        # latin-1, which cannot raise UnicodeError.)
 
817
        try:
 
818
            gecos = w.pw_gecos.decode('utf-8')
 
819
            encoding = 'utf-8'
 
820
        except UnicodeError:
 
821
            try:
 
822
                gecos = w.pw_gecos.decode(bzrlib.user_encoding)
 
823
                encoding = bzrlib.user_encoding
 
824
            except UnicodeError:
 
825
                raise errors.BzrCommandError('Unable to determine your name.  '
 
826
                   'Use "bzr whoami" to set it.')
 
827
        try:
 
828
            username = w.pw_name.decode(encoding)
 
829
        except UnicodeError:
 
830
            raise errors.BzrCommandError('Unable to determine your name.  '
 
831
                'Use "bzr whoami" to set it.')
 
832
 
 
833
        comma = gecos.find(',')
 
834
        if comma == -1:
 
835
            realname = gecos
 
836
        else:
 
837
            realname = gecos[:comma]
 
838
        if not realname:
 
839
            realname = username
 
840
 
 
841
    except ImportError:
 
842
        import getpass
 
843
        try:
 
844
            realname = username = getpass.getuser().decode(bzrlib.user_encoding)
 
845
        except UnicodeDecodeError:
 
846
            raise errors.BzrError("Can't decode username as %s." % \
 
847
                    bzrlib.user_encoding)
 
848
 
 
849
    return realname, (username + '@' + socket.gethostname())
 
850
 
 
851
 
 
852
def extract_email_address(e):
 
853
    """Return just the address part of an email string.
 
854
    
 
855
    That is just the user@domain part, nothing else. 
 
856
    This part is required to contain only ascii characters.
 
857
    If it can't be extracted, raises an error.
 
858
    
 
859
    >>> extract_email_address('Jane Tester <jane@test.com>')
 
860
    "jane@test.com"
 
861
    """
 
862
    m = re.search(r'[\w+.-]+@[\w+.-]+', e)
 
863
    if not m:
 
864
        raise errors.NoEmailInUsername(e)
 
865
    return m.group(0)
 
866
 
 
867
 
 
868
class TreeConfig(IniBasedConfig):
 
869
    """Branch configuration data associated with its contents, not location"""
 
870
    def __init__(self, branch):
 
871
        self.branch = branch
 
872
 
 
873
    def _get_parser(self, file=None):
 
874
        if file is not None:
 
875
            return IniBasedConfig._get_parser(file)
 
876
        return self._get_config()
 
877
 
 
878
    def _get_config(self):
 
879
        try:
 
880
            obj = ConfigObj(self.branch.control_files.get('branch.conf'),
 
881
                            encoding='utf-8')
 
882
        except errors.NoSuchFile:
 
883
            obj = ConfigObj(encoding='utf=8')
 
884
        return obj
 
885
 
 
886
    def get_option(self, name, section=None, default=None):
 
887
        self.branch.lock_read()
 
888
        try:
 
889
            obj = self._get_config()
 
890
            try:
 
891
                if section is not None:
 
892
                    obj = obj[section]
 
893
                result = obj[name]
 
894
            except KeyError:
 
895
                result = default
 
896
        finally:
 
897
            self.branch.unlock()
 
898
        return result
 
899
 
 
900
    def set_option(self, value, name, section=None):
 
901
        """Set a per-branch configuration option"""
 
902
        self.branch.lock_write()
 
903
        try:
 
904
            cfg_obj = self._get_config()
 
905
            if section is None:
 
906
                obj = cfg_obj
 
907
            else:
 
908
                try:
 
909
                    obj = cfg_obj[section]
 
910
                except KeyError:
 
911
                    cfg_obj[section] = {}
 
912
                    obj = cfg_obj[section]
 
913
            obj[name] = value
 
914
            out_file = StringIO()
 
915
            cfg_obj.write(out_file)
 
916
            out_file.seek(0)
 
917
            self.branch.control_files.put('branch.conf', out_file)
 
918
        finally:
 
919
            self.branch.unlock()