/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
1
# Copyright (C) 2005, 2007 Canonical Ltd
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
2
#   Authors: Robert Collins <robert.collins@canonical.com>
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
3
#            and others
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
4
#
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
#
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
# GNU General Public License for more details.
14
#
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
1442.1.20 by Robert Collins
add some documentation on options
19
"""Configuration that affects the behaviour of Bazaar.
20
21
Currently this configuration resides in ~/.bazaar/bazaar.conf
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
22
and ~/.bazaar/locations.conf, which is written to by bzr.
1442.1.20 by Robert Collins
add some documentation on options
23
1461 by Robert Collins
Typo in config.py (Thanks Fabbione)
24
In bazaar.conf the following options may be set:
1442.1.20 by Robert Collins
add some documentation on options
25
[DEFAULT]
26
editor=name-of-program
27
email=Your Name <your@email.address>
28
check_signatures=require|ignore|check-available(default)
29
create_signatures=always|never|when-required(default)
1442.1.56 by Robert Collins
gpg_signing_command configuration item
30
gpg_signing_command=name-of-program
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
31
log_format=name-of-format
1442.1.20 by Robert Collins
add some documentation on options
32
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
33
in locations.conf, you specify the url of a branch and options for it.
1442.1.20 by Robert Collins
add some documentation on options
34
Wildcards may be used - * and ? as normal in shell completion. Options
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
35
set in both bazaar.conf and locations.conf are overridden by the locations.conf
1442.1.20 by Robert Collins
add some documentation on options
36
setting.
37
[/home/robertc/source]
38
recurse=False|True(default)
39
email= as above
1759.2.1 by Jelmer Vernooij
Fix some types (found using aspell).
40
check_signatures= as above 
1442.1.20 by Robert Collins
add some documentation on options
41
create_signatures= as above.
42
43
explanation of options
44
----------------------
45
editor - this option sets the pop up editor to use during commits.
46
email - this option sets the user id bzr will use when committing.
47
check_signatures - this option controls whether bzr will require good gpg
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.
1887.2.1 by Adeodato Simó
Fix some typos and grammar issues.
53
log_format - this option sets the default log format.  Possible values are
54
             long, short, line, or a plugin can register new formats.
1553.6.2 by Erik Bågfors
documentation and NEWS
55
56
In bazaar.conf you can also define aliases in the ALIASES sections, example
57
58
[ALIASES]
59
lastlog=log --line -r-10..-1
60
ll=log --line -r-10..-1
61
h=help
62
up=pull
1442.1.20 by Robert Collins
add some documentation on options
63
"""
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
64
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
65
import os
66
import sys
1474 by Robert Collins
Merge from Aaron Bentley.
67
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
68
from bzrlib.lazy_import import lazy_import
69
lazy_import(globals(), """
1474 by Robert Collins
Merge from Aaron Bentley.
70
import errno
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
71
from fnmatch import fnmatch
72
import re
2900.2.22 by Vincent Ladeuil
Polishing.
73
from cStringIO import StringIO
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
74
75
import bzrlib
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
76
from bzrlib import (
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
77
    debug,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
78
    errors,
2681.1.8 by Aaron Bentley
Add Thunderbird support to bzr send
79
    mail_client,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
80
    osutils,
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
81
    symbol_versioning,
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
82
    trace,
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
83
    ui,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
84
    urlutils,
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
85
    win32utils,
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
86
    )
2991.2.4 by Vincent Ladeuil
Various fixes following local testing environment rebuild.
87
from bzrlib.util.configobj import configobj
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
88
""")
89
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
90
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
91
CHECK_IF_POSSIBLE=0
92
CHECK_ALWAYS=1
93
CHECK_NEVER=2
94
95
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
96
SIGN_WHEN_REQUIRED=0
97
SIGN_ALWAYS=1
98
SIGN_NEVER=2
99
100
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
101
POLICY_NONE = 0
102
POLICY_NORECURSE = 1
103
POLICY_APPENDPATH = 2
104
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
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
    }
2120.6.4 by James Henstridge
add support for specifying policy when storing options
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
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
124
1474 by Robert Collins
Merge from Aaron Bentley.
125
class ConfigObj(configobj.ConfigObj):
126
127
    def get_bool(self, section, key):
1556.2.2 by Aaron Bentley
Fixed get_bool
128
        return self[section].as_bool(key)
1474 by Robert Collins
Merge from Aaron Bentley.
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
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
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
2681.1.8 by Aaron Bentley
Add Thunderbird support to bzr send
147
    def get_mail_client(self):
148
        """Get a mail client to use"""
149
        selected_client = self.get_user_option('mail_client')
2681.1.10 by Aaron Bentley
Clean up handling of unknown mail clients
150
        try:
151
            mail_client_class = {
2681.3.4 by Lukáš Lalinsky
- Rename 'windows' to 'mapi'
152
                None: mail_client.DefaultMail,
2790.2.2 by Keir Mierle
Change alphabetic ordering into two categories; one for specific clients the other for generic options.
153
                # Specific clients
3324.4.1 by Xavier Maillard
Replace mail-mode call with compose-mail from GNU Emacs.
154
                'emacsclient': mail_client.EmacsMail,
2681.2.1 by Lukáš Lalinsky
Support for Evolution mail client.
155
                'evolution': mail_client.Evolution,
2790.2.1 by Keir Mierle
Add Mutt as a supported client email program. Also rearranges various listings
156
                'kmail': mail_client.KMail,
157
                'mutt': mail_client.Mutt,
158
                'thunderbird': mail_client.Thunderbird,
2790.2.2 by Keir Mierle
Change alphabetic ordering into two categories; one for specific clients the other for generic options.
159
                # Generic options
160
                'default': mail_client.DefaultMail,
161
                'editor': mail_client.Editor,
162
                'mapi': mail_client.MAPIClient,
2681.1.23 by Aaron Bentley
Add support for xdg-email
163
                'xdg-email': mail_client.XDGEmail,
2681.1.10 by Aaron Bentley
Clean up handling of unknown mail clients
164
            }[selected_client]
165
        except KeyError:
166
            raise errors.UnknownMailClient(selected_client)
167
        return mail_client_class(self)
2681.1.8 by Aaron Bentley
Add Thunderbird support to bzr send
168
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
169
    def _get_signature_checking(self):
170
        """Template method to override signature checking policy."""
171
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
172
    def _get_signing_policy(self):
173
        """Template method to override signature creation policy."""
174
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
175
    def _get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
176
        """Template method to provide a user option."""
177
        return None
178
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
179
    def get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
180
        """Get a generic option - no special process, no default."""
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
181
        return self._get_user_option(option_name)
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
182
1442.1.56 by Robert Collins
gpg_signing_command configuration item
183
    def gpg_signing_command(self):
184
        """What program should be used to sign signatures?"""
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
185
        result = self._gpg_signing_command()
186
        if result is None:
187
            result = "gpg"
188
        return result
189
190
    def _gpg_signing_command(self):
191
        """See gpg_signing_command()."""
192
        return None
1442.1.56 by Robert Collins
gpg_signing_command configuration item
193
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
194
    def log_format(self):
195
        """What log format should be used"""
196
        result = self._log_format()
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
197
        if result is None:
198
            result = "long"
199
        return result
200
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
201
    def _log_format(self):
202
        """See log_format()."""
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
203
        return None
204
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
205
    def __init__(self):
206
        super(Config, self).__init__()
207
1472 by Robert Collins
post commit hook, first pass implementation
208
    def post_commit(self):
209
        """An ordered list of python functions to call.
210
211
        Each function takes branch, rev_id as parameters.
212
        """
213
        return self._post_commit()
214
215
    def _post_commit(self):
216
        """See Config.post_commit."""
217
        return None
218
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
219
    def user_email(self):
220
        """Return just the email component of a username."""
1185.33.31 by Martin Pool
Make annotate cope better with revisions committed without a valid
221
        return extract_email_address(self.username())
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
222
223
    def username(self):
224
        """Return email-style username.
225
    
226
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
227
        
1861.4.1 by Matthieu Moy
BZREMAIL renamed to BZR_EMAIL.
228
        $BZR_EMAIL can be set to override this (as well as the
229
        deprecated $BZREMAIL), then
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
230
        the concrete policy type is checked, and finally
1185.37.2 by Jamie Wilkinson
Fix a typo and grammar in Config.username() docstring.
231
        $EMAIL is examined.
232
        If none is found, a reasonable default is (hopefully)
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
233
        created.
234
    
235
        TODO: Check it's reasonably well-formed.
236
        """
1861.4.1 by Matthieu Moy
BZREMAIL renamed to BZR_EMAIL.
237
        v = os.environ.get('BZR_EMAIL')
238
        if v:
239
            return v.decode(bzrlib.user_encoding)
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
240
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
241
        v = self._get_user_id()
242
        if v:
243
            return v
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
244
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
245
        v = os.environ.get('EMAIL')
246
        if v:
247
            return v.decode(bzrlib.user_encoding)
248
249
        name, email = _auto_user_id()
250
        if name:
251
            return '%s <%s>' % (name, email)
252
        else:
253
            return email
254
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
255
    def signature_checking(self):
256
        """What is the current policy for signature checking?."""
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
257
        policy = self._get_signature_checking()
258
        if policy is not None:
259
            return policy
1442.1.14 by Robert Collins
Create a default signature checking policy of CHECK_IF_POSSIBLE
260
        return CHECK_IF_POSSIBLE
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
261
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
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
1442.1.21 by Robert Collins
create signature_needed() call for commit to trigger creating signatures
269
    def signature_needed(self):
270
        """Is a signature needed when committing ?."""
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
271
        policy = self._get_signing_policy()
272
        if policy is None:
273
            policy = self._get_signature_checking()
274
            if policy is not None:
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
275
                trace.warning("Please use create_signatures,"
276
                              " not check_signatures to set signing policy.")
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
277
            if policy == CHECK_ALWAYS:
278
                return True
279
        elif policy == SIGN_ALWAYS:
1442.1.21 by Robert Collins
create signature_needed() call for commit to trigger creating signatures
280
            return True
281
        return False
282
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
283
    def get_alias(self, value):
284
        return self._get_alias(value)
285
286
    def _get_alias(self, value):
287
        pass
288
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
289
    def get_nickname(self):
290
        return self._get_nickname()
291
292
    def _get_nickname(self):
293
        return None
294
1551.18.17 by Aaron Bentley
Introduce bzr_remote_path configuration variable
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
1442.1.15 by Robert Collins
make getting the signature checking policy a template method
304
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
305
class IniBasedConfig(Config):
306
    """A configuration policy that draws from ini files."""
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
307
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
308
    def _get_parser(self, file=None):
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
309
        if self._parser is not None:
310
            return self._parser
1185.12.49 by Aaron Bentley
Switched to ConfigObj
311
        if file is None:
312
            input = self._get_filename()
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
313
        else:
1185.12.49 by Aaron Bentley
Switched to ConfigObj
314
            input = file
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
315
        try:
1551.2.20 by Aaron Bentley
Treated config files as utf-8
316
            self._parser = ConfigObj(input, encoding='utf-8')
1474 by Robert Collins
Merge from Aaron Bentley.
317
        except configobj.ConfigObjError, e:
1185.12.51 by Aaron Bentley
Allowed second call of _get_parser() to not require a file
318
            raise errors.ParseConfigError(e.errors, e.config.filename)
1185.12.49 by Aaron Bentley
Switched to ConfigObj
319
        return self._parser
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
320
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
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
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
333
    def _get_section(self):
334
        """Override this to define the section used by the config."""
335
        return "DEFAULT"
336
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
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
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
341
    def _get_signature_checking(self):
342
        """See Config._get_signature_checking."""
1474 by Robert Collins
Merge from Aaron Bentley.
343
        policy = self._get_user_option('check_signatures')
344
        if policy:
345
            return self._string_to_signature_policy(policy)
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
346
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
347
    def _get_signing_policy(self):
1773.4.3 by Martin Pool
[merge] bzr.dev
348
        """See Config._get_signing_policy"""
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
349
        policy = self._get_user_option('create_signatures')
350
        if policy:
351
            return self._string_to_signing_policy(policy)
352
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
353
    def _get_user_id(self):
354
        """Get the user id from the 'email' key in the current section."""
1474 by Robert Collins
Merge from Aaron Bentley.
355
        return self._get_user_option('email')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
356
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
357
    def _get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
358
        """See Config._get_user_option."""
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
359
        for (section, extra_path) in self._get_matching_sections():
360
            try:
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
361
                value = self._get_parser().get_value(section, option_name)
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
362
            except KeyError:
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
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:
2120.6.3 by James Henstridge
add some more tests for getting policy options, and behaviour of get_user_option in the presence of config policies
374
                if extra_path:
375
                    value = urlutils.join(value, extra_path)
376
                return value
2120.6.6 by James Henstridge
fix test_set_push_location test
377
            else:
378
                raise AssertionError('Unexpected config policy %r' % policy)
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
379
        else:
1993.3.1 by James Henstridge
first go at making location config lookup recursive
380
            return None
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
381
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
382
    def _gpg_signing_command(self):
1442.1.56 by Robert Collins
gpg_signing_command configuration item
383
        """See Config.gpg_signing_command."""
1472 by Robert Collins
post commit hook, first pass implementation
384
        return self._get_user_option('gpg_signing_command')
1442.1.56 by Robert Collins
gpg_signing_command configuration item
385
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
386
    def _log_format(self):
387
        """See Config.log_format."""
388
        return self._get_user_option('log_format')
1553.2.4 by Erik Bågfors
Support for setting the default log format at a configuration option
389
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
390
    def __init__(self, get_filename):
391
        super(IniBasedConfig, self).__init__()
392
        self._get_filename = get_filename
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
393
        self._parser = None
1472 by Robert Collins
post commit hook, first pass implementation
394
        
395
    def _post_commit(self):
396
        """See Config.post_commit."""
397
        return self._get_user_option('post_commit')
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
398
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
399
    def _string_to_signature_policy(self, signature_string):
400
        """Convert a string to a signing policy."""
1442.1.17 by Robert Collins
allow global overriding of signature policy to force checking, or (pointless but allowed) to set auto checking
401
        if signature_string.lower() == 'check-available':
402
            return CHECK_IF_POSSIBLE
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
403
        if signature_string.lower() == 'ignore':
404
            return CHECK_NEVER
1442.1.17 by Robert Collins
allow global overriding of signature policy to force checking, or (pointless but allowed) to set auto checking
405
        if signature_string.lower() == 'require':
406
            return CHECK_ALWAYS
1442.1.16 by Robert Collins
allow global overriding of signature policy to never check
407
        raise errors.BzrError("Invalid signatures policy '%s'"
408
                              % signature_string)
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
409
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
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
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
421
    def _get_alias(self, value):
422
        try:
423
            return self._get_parser().get_value("ALIASES", 
424
                                                value)
425
        except KeyError:
426
            pass
427
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
428
    def _get_nickname(self):
429
        return self.get_user_option('nickname')
430
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
431
432
class GlobalConfig(IniBasedConfig):
433
    """The configuration that should be used for a specific location."""
434
435
    def get_editor(self):
1474 by Robert Collins
Merge from Aaron Bentley.
436
        return self._get_user_option('editor')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
437
438
    def __init__(self):
439
        super(GlobalConfig, self).__init__(config_filename)
440
1816.2.1 by Robey Pointer
add set_user_option to GlobalConfig, and make /etc/passwd username lookup try harder with encodings
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
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
454
455
class LocationConfig(IniBasedConfig):
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
456
    """A configuration object that gives the policy for a location."""
457
458
    def __init__(self, location):
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
459
        name_generator = locations_config_filename
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
460
        if (not os.path.exists(name_generator()) and
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
461
                os.path.exists(branches_config_filename())):
1830.2.1 by John Arbash Meinel
Make it clearer what config file needs to be renamed.
462
            if sys.platform == 'win32':
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
463
                trace.warning('Please rename %s to %s'
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
464
                              % (branches_config_filename(),
465
                                 locations_config_filename()))
1830.2.1 by John Arbash Meinel
Make it clearer what config file needs to be renamed.
466
            else:
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
467
                trace.warning('Please rename ~/.bazaar/branches.conf'
468
                              ' to ~/.bazaar/locations.conf')
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
469
            name_generator = branches_config_filename
470
        super(LocationConfig, self).__init__(name_generator)
1878.1.1 by John Arbash Meinel
Entries in locations.conf should prefer local paths if available (bug #53653)
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)
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
476
        self.location = location
477
1993.3.1 by James Henstridge
first go at making location config lookup recursive
478
    def _get_matching_sections(self):
479
        """Return an ordered list of section names matching this location."""
1185.12.49 by Aaron Bentley
Switched to ConfigObj
480
        sections = self._get_parser()
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
481
        location_names = self.location.split('/')
1442.1.18 by Robert Collins
permit per branch location overriding of signature checking policy
482
        if self.location.endswith('/'):
483
            del location_names[-1]
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
484
        matches=[]
1442.1.10 by Robert Collins
explicit over glob test passes
485
        for section in sections:
1878.1.1 by John Arbash Meinel
Entries in locations.conf should prefer local paths if available (bug #53653)
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('/')
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
495
            if section.endswith('/'):
496
                del section_names[-1]
497
            names = zip(location_names, section_names)
498
            matched = True
499
            for name in names:
500
                if not fnmatch(name[0], name[1]):
501
                    matched = False
502
                    break
503
            if not matched:
504
                continue
505
            # so, for the common prefix they matched.
506
            # if section is longer, no match.
507
            if len(section_names) > len(location_names):
508
                continue
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
509
            matches.append((len(section_names), section,
510
                            '/'.join(location_names[len(section_names):])))
1442.1.12 by Robert Collins
LocationConfig section retrieval falls into my lap
511
        matches.sort(reverse=True)
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
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?
1993.3.1 by James Henstridge
first go at making location config lookup recursive
516
            try:
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
517
                if self._get_parser()[section].as_bool('ignore_parents'):
518
                    break
1993.3.1 by James Henstridge
first go at making location config lookup recursive
519
            except KeyError:
520
                pass
1993.3.3 by James Henstridge
make _get_matching_sections() return (section, extra_path) tuples, and adjust other code to match
521
        return sections
1442.1.9 by Robert Collins
exact section test passes
522
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
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
2120.6.10 by James Henstridge
Catch another deprecation warning, and more cleanup
533
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
534
        try:
535
            policy_name = self._get_parser()[section][policy_key]
536
        except KeyError:
537
            policy_name = None
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
538
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
539
        return _policy_value[policy_name]
2120.6.1 by James Henstridge
add support for norecurse and appendpath policies when reading configuration files
540
2120.6.4 by James Henstridge
add support for specifying policy when storing options
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:
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
551
            symbol_versioning.warn(
2120.6.11 by James Henstridge
s/0.13/0.14/ in deprecation warning
552
                'The recurse option is deprecated as of 0.14.  '
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
553
                'The section "%s" has been converted to use policies.'
554
                % section,
555
                DeprecationWarning)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
556
            del self._get_parser()[section]['recurse']
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
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'
2120.6.4 by James Henstridge
add support for specifying policy when storing options
562
2120.6.9 by James Henstridge
Fixes for issues brought up in John's review
563
        policy_key = option_name + ':policy'
2120.6.8 by James Henstridge
Change syntax for setting config option policies. Rather than
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]
2120.6.4 by James Henstridge
add support for specifying policy when storing options
570
571
    def set_user_option(self, option, value, store=STORE_LOCATION):
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
572
        """Save option and its value in the configuration."""
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
573
        if store not in [STORE_LOCATION,
2120.6.4 by James Henstridge
add support for specifying policy when storing options
574
                         STORE_LOCATION_NORECURSE,
3376.2.4 by Martin Pool
Remove every assert statement from bzrlib!
575
                         STORE_LOCATION_APPENDPATH]:
576
            raise ValueError('bad storage policy %r for %r' %
577
                (store, option))
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
578
        # FIXME: RBC 20051029 This should refresh the parser and also take a
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
579
        # file lock on locations.conf.
1185.31.39 by John Arbash Meinel
Replacing os.getcwdu() with osutils.getcwd(),
580
        conf_dir = os.path.dirname(self._get_filename())
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
581
        ensure_config_dir_exists(conf_dir)
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
582
        location = self.location
583
        if location.endswith('/'):
584
            location = location[:-1]
585
        if (not location in self._get_parser() and
586
            not location + '/' in self._get_parser()):
587
            self._get_parser()[location]={}
588
        elif location + '/' in self._get_parser():
589
            location = location + '/'
590
        self._get_parser()[location][option]=value
2120.6.4 by James Henstridge
add support for specifying policy when storing options
591
        # the allowed values of store match the config policies
592
        self._set_option_policy(location, option, store)
1551.2.49 by abentley
Made ConfigObj output binary-identical files on win32 and *nix
593
        self._get_parser().write(file(self._get_filename(), 'wb'))
1490 by Robert Collins
Implement a 'bzr push' command, with saved locations; update diff to return 1.
594
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
595
596
class BranchConfig(Config):
597
    """A configuration object giving the policy for a branch."""
598
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
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
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
604
    def _get_location_config(self):
605
        if self._location_config is None:
606
            self._location_config = LocationConfig(self.branch.base)
607
        return self._location_config
608
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
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
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
641
    def _get_user_id(self):
642
        """Return the full user id for the branch.
643
    
644
        e.g. "John Hacker <jhacker@foo.org>"
645
        This is looked up in the email controlfile for the branch.
646
        """
647
        try:
3388.2.3 by Martin Pool
Fix up more uses of LockableFiles.get_utf8 in tests
648
            return (self.branch.control_files._transport.get_bytes("email")
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
649
                    .decode(bzrlib.user_encoding)
650
                    .rstrip("\r\n"))
651
        except errors.NoSuchFile, e:
652
            pass
653
        
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
654
        return self._get_best_value('_get_user_id')
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
655
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
656
    def _get_signature_checking(self):
657
        """See Config._get_signature_checking."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
658
        return self._get_best_value('_get_signature_checking')
1442.1.19 by Robert Collins
BranchConfigs inherit signature_checking policy from their LocationConfig.
659
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
660
    def _get_signing_policy(self):
661
        """See Config._get_signing_policy."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
662
        return self._get_best_value('_get_signing_policy')
1770.2.1 by Aaron Bentley
Use create_signature for signing policy, deprecate check_signatures for this
663
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
664
    def _get_user_option(self, option_name):
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
665
        """See Config._get_user_option."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
666
        for source in self.option_sources:
1993.3.6 by James Henstridge
get rid of the recurse argument to get_user_option()
667
            value = source()._get_user_option(option_name)
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
668
            if value is not None:
669
                return value
670
        return None
671
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
672
    def set_user_option(self, name, value, store=STORE_BRANCH,
673
        warn_masked=False):
2120.6.4 by James Henstridge
add support for specifying policy when storing options
674
        if store == STORE_BRANCH:
1770.2.6 by Aaron Bentley
Ensure branch.conf works properly
675
            self._get_branch_data_config().set_option(value, name)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
676
        elif store == STORE_GLOBAL:
2120.6.7 by James Henstridge
Fix GlobalConfig.set_user_option() call
677
            self._get_global_config().set_user_option(name, value)
2120.6.4 by James Henstridge
add support for specifying policy when storing options
678
        else:
679
            self._get_location_config().set_user_option(name, value, store)
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
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'
1551.15.37 by Aaron Bentley
Don't treat a format string as a normal string
693
                                      ' branch.conf', value, mask_value)
1551.15.35 by Aaron Bentley
Warn when setting config values that will be masked (#122286)
694
1442.1.69 by Robert Collins
config.Config has a 'get_user_option' call that accepts an option name.
695
1442.1.59 by Robert Collins
Add re-sign command to generate a digital signature on a single revision.
696
    def _gpg_signing_command(self):
1442.1.56 by Robert Collins
gpg_signing_command configuration item
697
        """See Config.gpg_signing_command."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
698
        return self._get_safe_value('_gpg_signing_command')
1442.1.56 by Robert Collins
gpg_signing_command configuration item
699
        
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
700
    def __init__(self, branch):
701
        super(BranchConfig, self).__init__()
702
        self._location_config = None
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
703
        self._branch_data_config = None
704
        self._global_config = None
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
705
        self.branch = branch
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
706
        self.option_sources = (self._get_location_config, 
707
                               self._get_branch_data_config,
708
                               self._get_global_config)
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
709
1472 by Robert Collins
post commit hook, first pass implementation
710
    def _post_commit(self):
711
        """See Config.post_commit."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
712
        return self._get_safe_value('_post_commit')
1472 by Robert Collins
post commit hook, first pass implementation
713
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
714
    def _get_nickname(self):
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
715
        value = self._get_explicit_nickname()
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
716
        if value is not None:
717
            return value
2120.5.2 by Alexander Belchenko
(jam) Fix for bug #66857
718
        return urlutils.unescape(self.branch.base.split('/')[-2])
1770.2.7 by Aaron Bentley
Set/get nickname using BranchConfig
719
1824.1.1 by Robert Collins
Add BranchConfig.has_explicit_nickname call.
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')
726
1553.2.9 by Erik Bågfors
log_formatter => log_format for "named" formatters
727
    def _log_format(self):
728
        """See Config.log_format."""
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
729
        return self._get_best_value('_log_format')
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
730
1553.6.12 by Erik Bågfors
remove AliasConfig, based on input from abentley
731
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
732
def ensure_config_dir_exists(path=None):
733
    """Make sure a configuration directory exists.
734
    This makes sure that the directory exists.
735
    On windows, since configuration directories are 2 levels deep,
736
    it makes sure both the directory and the parent directory exists.
737
    """
738
    if path is None:
739
        path = config_dir()
740
    if not os.path.isdir(path):
741
        if sys.platform == 'win32':
742
            parent_dir = os.path.dirname(path)
743
            if not os.path.isdir(parent_dir):
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
744
                trace.mutter('creating config parent directory: %r', parent_dir)
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
745
            os.mkdir(parent_dir)
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
746
        trace.mutter('creating config directory: %r', path)
1185.31.43 by John Arbash Meinel
Reintroduced ensure_config_dir_exists() for sftp
747
        os.mkdir(path)
748
1532 by Robert Collins
Merge in John Meinels integration branch.
749
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
750
def config_dir():
751
    """Return per-user configuration directory.
752
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
753
    By default this is ~/.bazaar/
1442.1.1 by Robert Collins
move config_dir into bzrlib.config
754
    
755
    TODO: Global option --config-dir to override this.
756
    """
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
757
    base = os.environ.get('BZR_HOME', None)
758
    if sys.platform == 'win32':
759
        if base is None:
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
760
            base = win32utils.get_appdata_location_unicode()
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
761
        if base is None:
762
            base = os.environ.get('HOME', None)
763
        if base is None:
2991.2.2 by Vincent Ladeuil
No tests worth adding after upgrading to configobj-4.4.0.
764
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
765
                                  ' or HOME set')
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
766
        return osutils.pathjoin(base, 'bazaar', '2.0')
1185.38.1 by John Arbash Meinel
Adding my win32 patch for moving the home directory.
767
    else:
768
        # cygwin, linux, and darwin all have a $HOME directory
769
        if base is None:
770
            base = os.path.expanduser("~")
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
771
        return osutils.pathjoin(base, ".bazaar")
1185.31.32 by John Arbash Meinel
Updated the bzr sourcecode to use bzrlib.osutils.pathjoin rather than os.path.join to enforce internal use of / instead of \
772
773
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
774
def config_filename():
775
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
776
    return osutils.pathjoin(config_dir(), 'bazaar.conf')
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
777
778
1442.1.6 by Robert Collins
first stage major overhaul of configs, giving use BranchConfigs, LocationConfigs and GlobalConfigs
779
def branches_config_filename():
780
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
781
    return osutils.pathjoin(config_dir(), 'branches.conf')
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
782
1830.2.1 by John Arbash Meinel
Make it clearer what config file needs to be renamed.
783
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
784
def locations_config_filename():
785
    """Return per-user configuration ini file filename."""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
786
    return osutils.pathjoin(config_dir(), 'locations.conf')
1770.2.2 by Aaron Bentley
Rename branches.conf to locations.conf
787
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
788
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
789
def authentication_config_filename():
790
    """Return per-user authentication ini file filename."""
791
    return osutils.pathjoin(config_dir(), 'authentication.conf')
792
793
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
794
def user_ignore_config_filename():
795
    """Return the user default ignore filename"""
1996.3.31 by John Arbash Meinel
Make bzrlib.config use lazy importing
796
    return osutils.pathjoin(config_dir(), 'ignore')
1836.1.6 by John Arbash Meinel
Creating a helper function for getting the user ignore filename
797
798
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
799
def _auto_user_id():
800
    """Calculate automatic user identification.
801
802
    Returns (realname, email).
803
804
    Only used when none is set in the environment or the id file.
805
806
    This previously used the FQDN as the default domain, but that can
807
    be very slow on machines where DNS is broken.  So now we simply
808
    use the hostname.
809
    """
810
    import socket
811
2245.4.3 by Alexander Belchenko
config.py: changing _auto_user_id() and config_dir() to use functions from win32utils
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)
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
822
823
    try:
824
        import pwd
825
        uid = os.getuid()
826
        w = pwd.getpwuid(uid)
1553.4.5 by Michael Ellerman
Produce an intelligible error if the user's /etc/passwd is not encoded in
827
1816.2.1 by Robey Pointer
add set_user_option to GlobalConfig, and make /etc/passwd username lookup try harder with encodings
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.')
1553.4.5 by Michael Ellerman
Produce an intelligible error if the user's /etc/passwd is not encoded in
847
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
848
        comma = gecos.find(',')
849
        if comma == -1:
850
            realname = gecos
851
        else:
852
            realname = gecos[:comma]
853
        if not realname:
854
            realname = username
855
856
    except ImportError:
857
        import getpass
1553.4.5 by Michael Ellerman
Produce an intelligible error if the user's /etc/passwd is not encoded in
858
        try:
859
            realname = username = getpass.getuser().decode(bzrlib.user_encoding)
860
        except UnicodeDecodeError:
861
            raise errors.BzrError("Can't decode username as %s." % \
862
                    bzrlib.user_encoding)
1442.1.2 by Robert Collins
create a config module - there is enough config logic to make this worthwhile, and start testing config processing.
863
864
    return realname, (username + '@' + socket.gethostname())
865
866
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
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
1185.16.52 by Martin Pool
- add extract_email_address
876
def extract_email_address(e):
877
    """Return just the address part of an email string.
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
878
1185.16.52 by Martin Pool
- add extract_email_address
879
    That is just the user@domain part, nothing else. 
880
    This part is required to contain only ascii characters.
881
    If it can't be extracted, raises an error.
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
882
1185.16.52 by Martin Pool
- add extract_email_address
883
    >>> extract_email_address('Jane Tester <jane@test.com>')
884
    "jane@test.com"
885
    """
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
886
    name, email = parse_username(e)
887
    if not email:
2055.2.2 by John Arbash Meinel
Switch extract_email_address() to use a more specific exception
888
        raise errors.NoEmailInUsername(e)
3063.3.2 by Lukáš Lalinský
Move the name and e-mail address extraction logic to config.parse_username.
889
    return email
1185.35.11 by Aaron Bentley
Added support for branch nicks
890
1185.85.30 by John Arbash Meinel
Fixing 'bzr push' exposed that IniBasedConfig didn't handle unicode.
891
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
892
class TreeConfig(IniBasedConfig):
1185.35.11 by Aaron Bentley
Added support for branch nicks
893
    """Branch configuration data associated with its contents, not location"""
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
894
3408.3.1 by Martin Pool
Remove erroneous handling of branch.conf for RemoteBranch
895
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
896
1185.35.11 by Aaron Bentley
Added support for branch nicks
897
    def __init__(self, branch):
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
898
        transport = branch.control_files._transport
899
        self._config = TransportConfig(transport, 'branch.conf')
1185.35.11 by Aaron Bentley
Added support for branch nicks
900
        self.branch = branch
901
1770.2.5 by Aaron Bentley
Integrate branch.conf into BranchConfig
902
    def _get_parser(self, file=None):
903
        if file is not None:
904
            return IniBasedConfig._get_parser(file)
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
905
        return self._config._get_configobj()
1185.35.11 by Aaron Bentley
Added support for branch nicks
906
907
    def get_option(self, name, section=None, default=None):
908
        self.branch.lock_read()
909
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
910
            return self._config.get_option(name, section, default)
1185.35.11 by Aaron Bentley
Added support for branch nicks
911
        finally:
912
            self.branch.unlock()
913
        return result
914
915
    def set_option(self, value, name, section=None):
916
        """Set a per-branch configuration option"""
917
        self.branch.lock_write()
918
        try:
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
919
            self._config.set_option(value, name, section)
1185.35.11 by Aaron Bentley
Added support for branch nicks
920
        finally:
921
            self.branch.unlock()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
922
923
924
class AuthenticationConfig(object):
925
    """The authentication configuration file based on a ini file.
926
927
    Implements the authentication.conf file described in
928
    doc/developers/authentication-ring.txt.
929
    """
930
931
    def __init__(self, _file=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
932
        self._config = None # The ConfigObj
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
933
        if _file is None:
2900.2.24 by Vincent Ladeuil
Review feedback.
934
            self._filename = authentication_config_filename()
935
            self._input = self._filename = authentication_config_filename()
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
936
        else:
2900.2.24 by Vincent Ladeuil
Review feedback.
937
            # Tests can provide a string as _file
938
            self._filename = None
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
939
            self._input = _file
940
941
    def _get_config(self):
942
        if self._config is not None:
943
            return self._config
944
        try:
2900.2.22 by Vincent Ladeuil
Polishing.
945
            # FIXME: Should we validate something here ? Includes: empty
946
            # sections are useless, at least one of
947
            # user/password/password_encoding should be defined, etc.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
948
949
            # Note: the encoding below declares that the file itself is utf-8
950
            # encoded, but the values in the ConfigObj are always Unicode.
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
951
            self._config = ConfigObj(self._input, encoding='utf-8')
952
        except configobj.ConfigObjError, e:
953
            raise errors.ParseConfigError(e.errors, e.config.filename)
954
        return self._config
955
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
956
    def _save(self):
957
        """Save the config file, only tests should use it for now."""
2900.2.26 by Vincent Ladeuil
Fix forgotten reference to _get_filename and duplicated code.
958
        conf_dir = os.path.dirname(self._filename)
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
959
        ensure_config_dir_exists(conf_dir)
2900.2.26 by Vincent Ladeuil
Fix forgotten reference to _get_filename and duplicated code.
960
        self._get_config().write(file(self._filename, 'wb'))
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
961
962
    def _set_option(self, section_name, option_name, value):
963
        """Set an authentication configuration option"""
964
        conf = self._get_config()
965
        section = conf.get(section_name)
966
        if section is None:
967
            conf[section] = {}
968
            section = conf[section]
969
        section[option_name] = value
970
        self._save()
971
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
972
    def get_credentials(self, scheme, host, port=None, user=None, path=None):
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
973
        """Returns the matching credentials from authentication.conf file.
974
975
        :param scheme: protocol
976
977
        :param host: the server address
978
979
        :param port: the associated port (optional)
980
981
        :param user: login (optional)
982
983
        :param path: the absolute path on the server (optional)
984
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
985
        :return: A dict containing the matching credentials or None.
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
986
           This includes:
987
           - name: the section name of the credentials in the
988
             authentication.conf file,
989
           - user: can't de different from the provided user if any,
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
990
           - password: the decoded password, could be None if the credential
991
             defines only the user
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
992
           - verify_certificates: https specific, True if the server
993
             certificate should be verified, False otherwise.
994
        """
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
995
        credentials = None
996
        for auth_def_name, auth_def in self._get_config().items():
3418.2.1 by Vincent Ladeuil
Fix #217650 by catching declarations outside sections.
997
            if type(auth_def) is not configobj.Section:
998
                raise ValueError("%s defined outside a section" % auth_def_name)
999
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1000
            a_scheme, a_host, a_user, a_path = map(
1001
                auth_def.get, ['scheme', 'host', 'user', 'path'])
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1002
2900.2.5 by Vincent Ladeuil
ake ftp aware of authentication config.
1003
            try:
1004
                a_port = auth_def.as_int('port')
1005
            except KeyError:
1006
                a_port = None
2900.2.22 by Vincent Ladeuil
Polishing.
1007
            except ValueError:
1008
                raise ValueError("'port' not numeric in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1009
            try:
1010
                a_verify_certificates = auth_def.as_bool('verify_certificates')
1011
            except KeyError:
1012
                a_verify_certificates = True
2900.2.22 by Vincent Ladeuil
Polishing.
1013
            except ValueError:
1014
                raise ValueError(
1015
                    "'verify_certificates' not boolean in %s" % auth_def_name)
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1016
1017
            # Attempt matching
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1018
            if a_scheme is not None and scheme != a_scheme:
1019
                continue
1020
            if a_host is not None:
1021
                if not (host == a_host
1022
                        or (a_host.startswith('.') and host.endswith(a_host))):
1023
                    continue
2900.2.4 by Vincent Ladeuil
Cosmetic changes.
1024
            if a_port is not None and port != a_port:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1025
                continue
1026
            if (a_path is not None and path is not None
1027
                and not path.startswith(a_path)):
1028
                continue
1029
            if (a_user is not None and user is not None
1030
                and a_user != user):
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1031
                # Never contradict the caller about the user to be used
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1032
                continue
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1033
            if a_user is None:
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1034
                # Can't find a user
1035
                continue
2900.2.24 by Vincent Ladeuil
Review feedback.
1036
            credentials = dict(name=auth_def_name,
3418.4.2 by Vincent Ladeuil
Fix bug #199440 by taking into account that a section may not
1037
                               user=a_user,
1038
                               password=auth_def.get('password', None),
2900.2.24 by Vincent Ladeuil
Review feedback.
1039
                               verify_certificates=a_verify_certificates)
2900.2.22 by Vincent Ladeuil
Polishing.
1040
            self.decode_password(credentials,
1041
                                 auth_def.get('password_encoding', None))
2900.2.10 by Vincent Ladeuil
Add -Dauth handling.
1042
            if 'auth' in debug.debug_flags:
1043
                trace.mutter("Using authentication section: %r", auth_def_name)
2900.2.3 by Vincent Ladeuil
Credentials matching implementation.
1044
            break
1045
1046
        return credentials
1047
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1048
    def get_user(self, scheme, host, port=None,
1049
                 realm=None, path=None, prompt=None):
1050
        """Get a user from authentication file.
1051
1052
        :param scheme: protocol
1053
1054
        :param host: the server address
1055
1056
        :param port: the associated port (optional)
1057
1058
        :param realm: the realm sent by the server (optional)
1059
1060
        :param path: the absolute path on the server (optional)
1061
1062
        :return: The found user.
1063
        """
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1064
        credentials = self.get_credentials(scheme, host, port, user=None,
1065
                                           path=path)
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1066
        if credentials is not None:
1067
            user = credentials['user']
1068
        else:
1069
            user = None
1070
        return user
1071
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1072
    def get_password(self, scheme, host, user, port=None,
1073
                     realm=None, path=None, prompt=None):
1074
        """Get a password from authentication file or prompt the user for one.
1075
1076
        :param scheme: protocol
1077
1078
        :param host: the server address
1079
1080
        :param port: the associated port (optional)
1081
1082
        :param user: login
1083
1084
        :param realm: the realm sent by the server (optional)
1085
1086
        :param path: the absolute path on the server (optional)
1087
1088
        :return: The found password or the one entered by the user.
1089
        """
1090
        credentials = self.get_credentials(scheme, host, port, user, path)
1091
        if credentials is not None:
1092
            password = credentials['password']
2900.2.16 by Vincent Ladeuil
Make hhtp proxy aware of AuthenticationConfig (for password).
1093
        else:
1094
            password = None
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1095
        # Prompt user only if we could't find a password
2900.2.15 by Vincent Ladeuil
AuthenticationConfig can be queried for logins too (first step).
1096
        if password is None:
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1097
            if prompt is None:
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1098
                # Create a default prompt suitable for most of the cases
1099
                prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1100
            # Special handling for optional fields in the prompt
1101
            if port is not None:
1102
                prompt_host = '%s:%d' % (host, port)
1103
            else:
1104
                prompt_host = host
2900.2.19 by Vincent Ladeuil
Mention proxy and https in the password prompts, with tests.
1105
            password = ui.ui_factory.get_password(prompt,
1106
                                                  host=prompt_host, user=user)
2900.2.12 by Vincent Ladeuil
Since all schemes query AuthenticationConfig then prompt user, make that
1107
        return password
1108
2900.2.22 by Vincent Ladeuil
Polishing.
1109
    def decode_password(self, credentials, encoding):
1110
        return credentials
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1111
1112
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1113
class TransportConfig(object):
3242.1.5 by Aaron Bentley
Update per review comments
1114
    """A Config that reads/writes a config file on a Transport.
3242.1.4 by Aaron Bentley
Clean-up
1115
1116
    It is a low-level object that considers config data to be name/value pairs
1117
    that may be associated with a section.  Assigning meaning to the these
1118
    values is done at higher levels like TreeConfig.
1119
    """
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1120
1121
    def __init__(self, transport, filename):
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1122
        self._transport = transport
3242.1.2 by Aaron Bentley
Turn BzrDirConfig into TransportConfig, reduce code duplication
1123
        self._filename = filename
3242.1.1 by Aaron Bentley
Implement BzrDir configuration
1124
1125
    def get_option(self, name, section=None, default=None):
1126
        """Return the value associated with a named option.
1127
1128
        :param name: The name of the value
1129
        :param section: The section the option is in (if any)
1130
        :param default: The value to return if the value is not set
1131
        :return: The value or default value
1132
        """
1133
        configobj = self._get_configobj()
1134
        if section is None:
1135
            section_obj = configobj
1136
        else:
1137
            try:
1138
                section_obj = configobj[section]
1139
            except KeyError:
1140
                return default
1141
        return section_obj.get(name, default)
1142
1143
    def set_option(self, value, name, section=None):
1144
        """Set the value associated with a named option.
1145
1146
        :param value: The value to set
1147
        :param name: The name of the value to set
1148
        :param section: The section the option is in (if any)
1149
        """
1150
        configobj = self._get_configobj()
1151
        if section is None:
1152
            configobj[name] = value
1153
        else:
1154
            configobj.setdefault(section, {})[name] = value
1155
        self._set_configobj(configobj)
1156
1157
    def _get_configobj(self):
1158
        try:
1159
            return ConfigObj(self._transport.get(self._filename),
1160
                             encoding='utf-8')
1161
        except errors.NoSuchFile:
1162
            return ConfigObj(encoding='utf-8')
1163
1164
    def _set_configobj(self, configobj):
1165
        out_file = StringIO()
1166
        configobj.write(out_file)
1167
        out_file.seek(0)
1168
        self._transport.put_file(self._filename, out_file)