~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Aaron Bentley
  • Date: 2007-08-16 00:54:00 UTC
  • mto: This revision was merged to the branch mainline in revision 2711.
  • Revision ID: aaron.bentley@utoronto.ca-20070816005400-oxxtqiy310wx10h9
Fix typo

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2007 Canonical Ltd
2
2
#   Authors: Robert Collins <robert.collins@canonical.com>
3
3
#            and others
4
4
#
14
14
#
15
15
# You should have received a copy of the GNU General Public License
16
16
# along with this program; if not, write to the Free Software
17
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
18
 
19
19
"""Configuration that affects the behaviour of Bazaar.
20
20
 
37
37
[/home/robertc/source]
38
38
recurse=False|True(default)
39
39
email= as above
40
 
check_signatures= as above
 
40
check_signatures= as above 
41
41
create_signatures= as above.
42
42
 
43
43
explanation of options
45
45
editor - this option sets the pop up editor to use during commits.
46
46
email - this option sets the user id bzr will use when committing.
47
47
check_signatures - this option controls whether bzr will require good gpg
48
 
                   signatures, ignore them, or check them if they are
 
48
                   signatures, ignore them, or check them if they are 
49
49
                   present.
50
 
create_signatures - this option controls whether bzr will always create
 
50
create_signatures - this option controls whether bzr will always create 
51
51
                    gpg signatures, never create them, or create them if the
52
52
                    branch is configured to require them.
53
53
log_format - this option sets the default log format.  Possible values are
70
70
import errno
71
71
from fnmatch import fnmatch
72
72
import re
73
 
from cStringIO import StringIO
 
73
from StringIO import StringIO
74
74
 
75
75
import bzrlib
76
76
from bzrlib import (
77
 
    atomicfile,
78
 
    debug,
79
77
    errors,
80
 
    mail_client,
81
78
    osutils,
82
 
    registry,
83
79
    symbol_versioning,
84
80
    trace,
85
 
    ui,
86
81
    urlutils,
87
82
    win32utils,
88
83
    )
89
 
from bzrlib.util.configobj import configobj
 
84
import bzrlib.util.configobj.configobj as configobj
90
85
""")
91
86
 
 
87
from bzrlib.trace import mutter, warning
 
88
 
92
89
 
93
90
CHECK_IF_POSSIBLE=0
94
91
CHECK_ALWAYS=1
123
120
STORE_BRANCH = 3
124
121
STORE_GLOBAL = 4
125
122
 
126
 
_ConfigObj = None
127
 
def ConfigObj(*args, **kwargs):
128
 
    global _ConfigObj
129
 
    if _ConfigObj is None:
130
 
        class ConfigObj(configobj.ConfigObj):
131
 
 
132
 
            def get_bool(self, section, key):
133
 
                return self[section].as_bool(key)
134
 
 
135
 
            def get_value(self, section, name):
136
 
                # Try [] for the old DEFAULT section.
137
 
                if section == "DEFAULT":
138
 
                    try:
139
 
                        return self[name]
140
 
                    except KeyError:
141
 
                        pass
142
 
                return self[section][name]
143
 
        _ConfigObj = ConfigObj
144
 
    return _ConfigObj(*args, **kwargs)
 
123
 
 
124
class ConfigObj(configobj.ConfigObj):
 
125
 
 
126
    def get_bool(self, section, key):
 
127
        return self[section].as_bool(key)
 
128
 
 
129
    def get_value(self, section, name):
 
130
        # Try [] for the old DEFAULT section.
 
131
        if section == "DEFAULT":
 
132
            try:
 
133
                return self[name]
 
134
            except KeyError:
 
135
                pass
 
136
        return self[section][name]
145
137
 
146
138
 
147
139
class Config(object):
148
140
    """A configuration policy - what username, editor, gpg needs etc."""
149
141
 
150
 
    def __init__(self):
151
 
        super(Config, self).__init__()
152
 
 
153
142
    def get_editor(self):
154
143
        """Get the users pop up editor."""
155
144
        raise NotImplementedError
156
145
 
157
 
    def get_change_editor(self, old_tree, new_tree):
158
 
        from bzrlib import diff
159
 
        cmd = self._get_change_editor()
160
 
        if cmd is None:
161
 
            return None
162
 
        return diff.DiffFromTool.from_string(cmd, old_tree, new_tree,
163
 
                                             sys.stdout)
164
 
 
165
 
 
166
 
    def get_mail_client(self):
167
 
        """Get a mail client to use"""
168
 
        selected_client = self.get_user_option('mail_client')
169
 
        _registry = mail_client.mail_client_registry
170
 
        try:
171
 
            mail_client_class = _registry.get(selected_client)
172
 
        except KeyError:
173
 
            raise errors.UnknownMailClient(selected_client)
174
 
        return mail_client_class(self)
175
 
 
176
146
    def _get_signature_checking(self):
177
147
        """Template method to override signature checking policy."""
178
148
 
187
157
        """Get a generic option - no special process, no default."""
188
158
        return self._get_user_option(option_name)
189
159
 
190
 
    def get_user_option_as_bool(self, option_name):
191
 
        """Get a generic option as a boolean - no special process, no default.
192
 
 
193
 
        :return None if the option doesn't exist or its value can't be
194
 
            interpreted as a boolean. Returns True or False otherwise.
195
 
        """
196
 
        s = self._get_user_option(option_name)
197
 
        if s is None:
198
 
            # The option doesn't exist
199
 
            return None
200
 
        val = ui.bool_from_string(s)
201
 
        if val is None:
202
 
            # The value can't be interpreted as a boolean
203
 
            trace.warning('Value "%s" is not a boolean for "%s"',
204
 
                          s, option_name)
205
 
        return val
206
 
 
207
 
    def get_user_option_as_list(self, option_name):
208
 
        """Get a generic option as a list - no special process, no default.
209
 
 
210
 
        :return None if the option doesn't exist. Returns the value as a list
211
 
            otherwise.
212
 
        """
213
 
        l = self._get_user_option(option_name)
214
 
        if isinstance(l, (str, unicode)):
215
 
            # A single value, most probably the user forgot the final ','
216
 
            l = [l]
217
 
        return l
218
 
 
219
160
    def gpg_signing_command(self):
220
161
        """What program should be used to sign signatures?"""
221
162
        result = self._gpg_signing_command()
238
179
        """See log_format()."""
239
180
        return None
240
181
 
 
182
    def __init__(self):
 
183
        super(Config, self).__init__()
 
184
 
241
185
    def post_commit(self):
242
186
        """An ordered list of python functions to call.
243
187
 
255
199
 
256
200
    def username(self):
257
201
        """Return email-style username.
258
 
 
 
202
    
259
203
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
260
 
 
261
 
        $BZR_EMAIL can be set to override this, then
 
204
        
 
205
        $BZR_EMAIL can be set to override this (as well as the
 
206
        deprecated $BZREMAIL), then
262
207
        the concrete policy type is checked, and finally
263
208
        $EMAIL is examined.
264
 
        If no username can be found, errors.NoWhoami exception is raised.
265
 
 
 
209
        If none is found, a reasonable default is (hopefully)
 
210
        created.
 
211
    
266
212
        TODO: Check it's reasonably well-formed.
267
213
        """
268
214
        v = os.environ.get('BZR_EMAIL')
269
215
        if v:
270
 
            return v.decode(osutils.get_user_encoding())
271
 
 
 
216
            return v.decode(bzrlib.user_encoding)
 
217
        v = os.environ.get('BZREMAIL')
 
218
        if v:
 
219
            warning('BZREMAIL is deprecated in favor of BZR_EMAIL. Please update your configuration.')
 
220
            return v.decode(bzrlib.user_encoding)
 
221
    
272
222
        v = self._get_user_id()
273
223
        if v:
274
224
            return v
275
 
 
 
225
        
276
226
        v = os.environ.get('EMAIL')
277
227
        if v:
278
 
            return v.decode(osutils.get_user_encoding())
279
 
 
280
 
        raise errors.NoWhoami()
281
 
 
282
 
    def ensure_username(self):
283
 
        """Raise errors.NoWhoami if username is not set.
284
 
 
285
 
        This method relies on the username() function raising the error.
286
 
        """
287
 
        self.username()
 
228
            return v.decode(bzrlib.user_encoding)
 
229
 
 
230
        name, email = _auto_user_id()
 
231
        if name:
 
232
            return '%s <%s>' % (name, email)
 
233
        else:
 
234
            return email
288
235
 
289
236
    def signature_checking(self):
290
237
        """What is the current policy for signature checking?."""
306
253
        if policy is None:
307
254
            policy = self._get_signature_checking()
308
255
            if policy is not None:
309
 
                trace.warning("Please use create_signatures,"
310
 
                              " not check_signatures to set signing policy.")
 
256
                warning("Please use create_signatures, not check_signatures "
 
257
                        "to set signing policy.")
311
258
            if policy == CHECK_ALWAYS:
312
259
                return True
313
260
        elif policy == SIGN_ALWAYS:
326
273
    def _get_nickname(self):
327
274
        return None
328
275
 
329
 
    def get_bzr_remote_path(self):
330
 
        try:
331
 
            return os.environ['BZR_REMOTE_PATH']
332
 
        except KeyError:
333
 
            path = self.get_user_option("bzr_remote_path")
334
 
            if path is None:
335
 
                path = 'bzr'
336
 
            return path
337
 
 
338
 
    def suppress_warning(self, warning):
339
 
        """Should the warning be suppressed or emitted.
340
 
 
341
 
        :param warning: The name of the warning being tested.
342
 
 
343
 
        :returns: True if the warning should be suppressed, False otherwise.
344
 
        """
345
 
        warnings = self.get_user_option_as_list('suppress_warnings')
346
 
        if warnings is None or warning not in warnings:
347
 
            return False
348
 
        else:
349
 
            return True
350
 
 
351
276
 
352
277
class IniBasedConfig(Config):
353
278
    """A configuration policy that draws from ini files."""
354
279
 
355
 
    def __init__(self, get_filename):
356
 
        super(IniBasedConfig, self).__init__()
357
 
        self._get_filename = get_filename
358
 
        self._parser = None
359
 
 
360
280
    def _get_parser(self, file=None):
361
281
        if self._parser is not None:
362
282
            return self._parser
390
310
        """Return the policy for the given (section, option_name) pair."""
391
311
        return POLICY_NONE
392
312
 
393
 
    def _get_change_editor(self):
394
 
        return self.get_user_option('change_editor')
395
 
 
396
313
    def _get_signature_checking(self):
397
314
        """See Config._get_signature_checking."""
398
315
        policy = self._get_user_option('check_signatures')
442
359
        """See Config.log_format."""
443
360
        return self._get_user_option('log_format')
444
361
 
 
362
    def __init__(self, get_filename):
 
363
        super(IniBasedConfig, self).__init__()
 
364
        self._get_filename = get_filename
 
365
        self._parser = None
 
366
        
445
367
    def _post_commit(self):
446
368
        """See Config.post_commit."""
447
369
        return self._get_user_option('post_commit')
470
392
 
471
393
    def _get_alias(self, value):
472
394
        try:
473
 
            return self._get_parser().get_value("ALIASES",
 
395
            return self._get_parser().get_value("ALIASES", 
474
396
                                                value)
475
397
        except KeyError:
476
398
            pass
478
400
    def _get_nickname(self):
479
401
        return self.get_user_option('nickname')
480
402
 
481
 
    def _write_config_file(self):
482
 
        filename = self._get_filename()
483
 
        atomic_file = atomicfile.AtomicFile(filename)
484
 
        self._get_parser().write(atomic_file)
485
 
        atomic_file.commit()
486
 
        atomic_file.close()
487
 
        osutils.copy_ownership_from_path(filename)
488
 
 
489
403
 
490
404
class GlobalConfig(IniBasedConfig):
491
405
    """The configuration that should be used for a specific location."""
498
412
 
499
413
    def set_user_option(self, option, value):
500
414
        """Save option and its value in the configuration."""
501
 
        self._set_option(option, value, 'DEFAULT')
502
 
 
503
 
    def get_aliases(self):
504
 
        """Return the aliases section."""
505
 
        if 'ALIASES' in self._get_parser():
506
 
            return self._get_parser()['ALIASES']
507
 
        else:
508
 
            return {}
509
 
 
510
 
    def set_alias(self, alias_name, alias_command):
511
 
        """Save the alias in the configuration."""
512
 
        self._set_option(alias_name, alias_command, 'ALIASES')
513
 
 
514
 
    def unset_alias(self, alias_name):
515
 
        """Unset an existing alias."""
516
 
        aliases = self._get_parser().get('ALIASES')
517
 
        if not aliases or alias_name not in aliases:
518
 
            raise errors.NoSuchAlias(alias_name)
519
 
        del aliases[alias_name]
520
 
        self._write_config_file()
521
 
 
522
 
    def _set_option(self, option, value, section):
523
415
        # FIXME: RBC 20051029 This should refresh the parser and also take a
524
416
        # file lock on bazaar.conf.
525
417
        conf_dir = os.path.dirname(self._get_filename())
526
418
        ensure_config_dir_exists(conf_dir)
527
 
        self._get_parser().setdefault(section, {})[option] = value
528
 
        self._write_config_file()
 
419
        if 'DEFAULT' not in self._get_parser():
 
420
            self._get_parser()['DEFAULT'] = {}
 
421
        self._get_parser()['DEFAULT'][option] = value
 
422
        f = open(self._get_filename(), 'wb')
 
423
        self._get_parser().write(f)
 
424
        f.close()
529
425
 
530
426
 
531
427
class LocationConfig(IniBasedConfig):
533
429
 
534
430
    def __init__(self, location):
535
431
        name_generator = locations_config_filename
536
 
        if (not os.path.exists(name_generator()) and
 
432
        if (not os.path.exists(name_generator()) and 
537
433
                os.path.exists(branches_config_filename())):
538
434
            if sys.platform == 'win32':
539
 
                trace.warning('Please rename %s to %s'
540
 
                              % (branches_config_filename(),
541
 
                                 locations_config_filename()))
 
435
                warning('Please rename %s to %s' 
 
436
                         % (branches_config_filename(),
 
437
                            locations_config_filename()))
542
438
            else:
543
 
                trace.warning('Please rename ~/.bazaar/branches.conf'
544
 
                              ' to ~/.bazaar/locations.conf')
 
439
                warning('Please rename ~/.bazaar/branches.conf'
 
440
                        ' to ~/.bazaar/locations.conf')
545
441
            name_generator = branches_config_filename
546
442
        super(LocationConfig, self).__init__(name_generator)
547
443
        # local file locations are looked up by local path, rather than
646
542
 
647
543
    def set_user_option(self, option, value, store=STORE_LOCATION):
648
544
        """Save option and its value in the configuration."""
649
 
        if store not in [STORE_LOCATION,
 
545
        assert store in [STORE_LOCATION,
650
546
                         STORE_LOCATION_NORECURSE,
651
 
                         STORE_LOCATION_APPENDPATH]:
652
 
            raise ValueError('bad storage policy %r for %r' %
653
 
                (store, option))
 
547
                         STORE_LOCATION_APPENDPATH], 'bad storage policy'
654
548
        # FIXME: RBC 20051029 This should refresh the parser and also take a
655
549
        # file lock on locations.conf.
656
550
        conf_dir = os.path.dirname(self._get_filename())
666
560
        self._get_parser()[location][option]=value
667
561
        # the allowed values of store match the config policies
668
562
        self._set_option_policy(location, option, store)
669
 
        self._write_config_file()
 
563
        self._get_parser().write(file(self._get_filename(), 'wb'))
670
564
 
671
565
 
672
566
class BranchConfig(Config):
701
595
 
702
596
    def _get_safe_value(self, option_name):
703
597
        """This variant of get_best_value never returns untrusted values.
704
 
 
 
598
        
705
599
        It does not return values from the branch data, because the branch may
706
600
        not be controlled by the user.
707
601
 
716
610
 
717
611
    def _get_user_id(self):
718
612
        """Return the full user id for the branch.
719
 
 
720
 
        e.g. "John Hacker <jhacker@example.com>"
 
613
    
 
614
        e.g. "John Hacker <jhacker@foo.org>"
721
615
        This is looked up in the email controlfile for the branch.
722
616
        """
723
617
        try:
724
 
            return (self.branch._transport.get_bytes("email")
725
 
                    .decode(osutils.get_user_encoding())
 
618
            return (self.branch.control_files.get_utf8("email") 
 
619
                    .read()
 
620
                    .decode(bzrlib.user_encoding)
726
621
                    .rstrip("\r\n"))
727
622
        except errors.NoSuchFile, e:
728
623
            pass
729
 
 
 
624
        
730
625
        return self._get_best_value('_get_user_id')
731
626
 
732
 
    def _get_change_editor(self):
733
 
        return self._get_best_value('_get_change_editor')
734
 
 
735
627
    def _get_signature_checking(self):
736
628
        """See Config._get_signature_checking."""
737
629
        return self._get_best_value('_get_signature_checking')
771
663
                        trace.warning('Value "%s" is masked by "%s" from'
772
664
                                      ' branch.conf', value, mask_value)
773
665
 
 
666
 
774
667
    def _gpg_signing_command(self):
775
668
        """See Config.gpg_signing_command."""
776
669
        return self._get_safe_value('_gpg_signing_command')
777
 
 
 
670
        
778
671
    def __init__(self, branch):
779
672
        super(BranchConfig, self).__init__()
780
673
        self._location_config = None
781
674
        self._branch_data_config = None
782
675
        self._global_config = None
783
676
        self.branch = branch
784
 
        self.option_sources = (self._get_location_config,
 
677
        self.option_sources = (self._get_location_config, 
785
678
                               self._get_branch_data_config,
786
679
                               self._get_global_config)
787
680
 
819
712
        if sys.platform == 'win32':
820
713
            parent_dir = os.path.dirname(path)
821
714
            if not os.path.isdir(parent_dir):
822
 
                trace.mutter('creating config parent directory: %r', parent_dir)
 
715
                mutter('creating config parent directory: %r', parent_dir)
823
716
            os.mkdir(parent_dir)
824
 
        trace.mutter('creating config directory: %r', path)
 
717
        mutter('creating config directory: %r', path)
825
718
        os.mkdir(path)
826
 
        osutils.copy_ownership_from_path(path)
827
719
 
828
720
 
829
721
def config_dir():
830
722
    """Return per-user configuration directory.
831
723
 
832
724
    By default this is ~/.bazaar/
833
 
 
 
725
    
834
726
    TODO: Global option --config-dir to override this.
835
727
    """
836
728
    base = os.environ.get('BZR_HOME', None)
840
732
        if base is None:
841
733
            base = os.environ.get('HOME', None)
842
734
        if base is None:
843
 
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
844
 
                                  ' or HOME set')
 
735
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA, or HOME set')
845
736
        return osutils.pathjoin(base, 'bazaar', '2.0')
846
737
    else:
 
738
        # cygwin, linux, and darwin all have a $HOME directory
847
739
        if base is None:
848
740
            base = os.path.expanduser("~")
849
741
        return osutils.pathjoin(base, ".bazaar")
864
756
    return osutils.pathjoin(config_dir(), 'locations.conf')
865
757
 
866
758
 
867
 
def authentication_config_filename():
868
 
    """Return per-user authentication ini file filename."""
869
 
    return osutils.pathjoin(config_dir(), 'authentication.conf')
870
 
 
871
 
 
872
759
def user_ignore_config_filename():
873
760
    """Return the user default ignore filename"""
874
761
    return osutils.pathjoin(config_dir(), 'ignore')
875
762
 
876
763
 
877
 
def crash_dir():
878
 
    """Return the directory name to store crash files.
879
 
 
880
 
    This doesn't implicitly create it.
881
 
 
882
 
    On Windows it's in the config directory; elsewhere it's /var/crash
883
 
    which may be monitored by apport.  It can be overridden by
884
 
    $APPORT_CRASH_DIR.
 
764
def _auto_user_id():
 
765
    """Calculate automatic user identification.
 
766
 
 
767
    Returns (realname, email).
 
768
 
 
769
    Only used when none is set in the environment or the id file.
 
770
 
 
771
    This previously used the FQDN as the default domain, but that can
 
772
    be very slow on machines where DNS is broken.  So now we simply
 
773
    use the hostname.
885
774
    """
 
775
    import socket
 
776
 
886
777
    if sys.platform == 'win32':
887
 
        return osutils.pathjoin(config_dir(), 'Crash')
888
 
    else:
889
 
        # XXX: hardcoded in apport_python_hook.py; therefore here too -- mbp
890
 
        # 2010-01-31
891
 
        return os.environ.get('APPORT_CRASH_DIR', '/var/crash')
892
 
 
893
 
 
894
 
def xdg_cache_dir():
895
 
    # See http://standards.freedesktop.org/basedir-spec/latest/ar01s03.html
896
 
    # Possibly this should be different on Windows?
897
 
    e = os.environ.get('XDG_CACHE_DIR', None)
898
 
    if e:
899
 
        return e
900
 
    else:
901
 
        return os.path.expanduser('~/.cache')
902
 
 
903
 
 
904
 
def parse_username(username):
905
 
    """Parse e-mail username and return a (name, address) tuple."""
906
 
    match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
907
 
    if match is None:
908
 
        return (username, '')
909
 
    else:
910
 
        return (match.group(1), match.group(2))
 
778
        name = win32utils.get_user_name_unicode()
 
779
        if name is None:
 
780
            raise errors.BzrError("Cannot autodetect user name.\n"
 
781
                                  "Please, set your name with command like:\n"
 
782
                                  'bzr whoami "Your Name <name@domain.com>"')
 
783
        host = win32utils.get_host_name_unicode()
 
784
        if host is None:
 
785
            host = socket.gethostname()
 
786
        return name, (name + '@' + host)
 
787
 
 
788
    try:
 
789
        import pwd
 
790
        uid = os.getuid()
 
791
        w = pwd.getpwuid(uid)
 
792
 
 
793
        # we try utf-8 first, because on many variants (like Linux),
 
794
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
795
        # false positives.  (many users will have their user encoding set to
 
796
        # latin-1, which cannot raise UnicodeError.)
 
797
        try:
 
798
            gecos = w.pw_gecos.decode('utf-8')
 
799
            encoding = 'utf-8'
 
800
        except UnicodeError:
 
801
            try:
 
802
                gecos = w.pw_gecos.decode(bzrlib.user_encoding)
 
803
                encoding = bzrlib.user_encoding
 
804
            except UnicodeError:
 
805
                raise errors.BzrCommandError('Unable to determine your name.  '
 
806
                   'Use "bzr whoami" to set it.')
 
807
        try:
 
808
            username = w.pw_name.decode(encoding)
 
809
        except UnicodeError:
 
810
            raise errors.BzrCommandError('Unable to determine your name.  '
 
811
                'Use "bzr whoami" to set it.')
 
812
 
 
813
        comma = gecos.find(',')
 
814
        if comma == -1:
 
815
            realname = gecos
 
816
        else:
 
817
            realname = gecos[:comma]
 
818
        if not realname:
 
819
            realname = username
 
820
 
 
821
    except ImportError:
 
822
        import getpass
 
823
        try:
 
824
            realname = username = getpass.getuser().decode(bzrlib.user_encoding)
 
825
        except UnicodeDecodeError:
 
826
            raise errors.BzrError("Can't decode username as %s." % \
 
827
                    bzrlib.user_encoding)
 
828
 
 
829
    return realname, (username + '@' + socket.gethostname())
911
830
 
912
831
 
913
832
def extract_email_address(e):
914
833
    """Return just the address part of an email string.
915
 
 
916
 
    That is just the user@domain part, nothing else.
 
834
    
 
835
    That is just the user@domain part, nothing else. 
917
836
    This part is required to contain only ascii characters.
918
837
    If it can't be extracted, raises an error.
919
 
 
 
838
    
920
839
    >>> extract_email_address('Jane Tester <jane@test.com>')
921
840
    "jane@test.com"
922
841
    """
923
 
    name, email = parse_username(e)
924
 
    if not email:
 
842
    m = re.search(r'[\w+.-]+@[\w+.-]+', e)
 
843
    if not m:
925
844
        raise errors.NoEmailInUsername(e)
926
 
    return email
 
845
    return m.group(0)
927
846
 
928
847
 
929
848
class TreeConfig(IniBasedConfig):
930
849
    """Branch configuration data associated with its contents, not location"""
931
 
 
932
 
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
933
 
 
934
850
    def __init__(self, branch):
935
 
        self._config = branch._get_config()
936
851
        self.branch = branch
937
852
 
938
853
    def _get_parser(self, file=None):
939
854
        if file is not None:
940
855
            return IniBasedConfig._get_parser(file)
941
 
        return self._config._get_configobj()
 
856
        return self._get_config()
 
857
 
 
858
    def _get_config(self):
 
859
        try:
 
860
            obj = ConfigObj(self.branch.control_files.get('branch.conf'),
 
861
                            encoding='utf-8')
 
862
        except errors.NoSuchFile:
 
863
            obj = ConfigObj(encoding='utf=8')
 
864
        return obj
942
865
 
943
866
    def get_option(self, name, section=None, default=None):
944
867
        self.branch.lock_read()
945
868
        try:
946
 
            return self._config.get_option(name, section, default)
 
869
            obj = self._get_config()
 
870
            try:
 
871
                if section is not None:
 
872
                    obj = obj[section]
 
873
                result = obj[name]
 
874
            except KeyError:
 
875
                result = default
947
876
        finally:
948
877
            self.branch.unlock()
 
878
        return result
949
879
 
950
880
    def set_option(self, value, name, section=None):
951
881
        """Set a per-branch configuration option"""
952
882
        self.branch.lock_write()
953
883
        try:
954
 
            self._config.set_option(value, name, section)
 
884
            cfg_obj = self._get_config()
 
885
            if section is None:
 
886
                obj = cfg_obj
 
887
            else:
 
888
                try:
 
889
                    obj = cfg_obj[section]
 
890
                except KeyError:
 
891
                    cfg_obj[section] = {}
 
892
                    obj = cfg_obj[section]
 
893
            obj[name] = value
 
894
            out_file = StringIO()
 
895
            cfg_obj.write(out_file)
 
896
            out_file.seek(0)
 
897
            self.branch.control_files.put('branch.conf', out_file)
955
898
        finally:
956
899
            self.branch.unlock()
957
 
 
958
 
 
959
 
class AuthenticationConfig(object):
960
 
    """The authentication configuration file based on a ini file.
961
 
 
962
 
    Implements the authentication.conf file described in
963
 
    doc/developers/authentication-ring.txt.
964
 
    """
965
 
 
966
 
    def __init__(self, _file=None):
967
 
        self._config = None # The ConfigObj
968
 
        if _file is None:
969
 
            self._filename = authentication_config_filename()
970
 
            self._input = self._filename = authentication_config_filename()
971
 
        else:
972
 
            # Tests can provide a string as _file
973
 
            self._filename = None
974
 
            self._input = _file
975
 
 
976
 
    def _get_config(self):
977
 
        if self._config is not None:
978
 
            return self._config
979
 
        try:
980
 
            # FIXME: Should we validate something here ? Includes: empty
981
 
            # sections are useless, at least one of
982
 
            # user/password/password_encoding should be defined, etc.
983
 
 
984
 
            # Note: the encoding below declares that the file itself is utf-8
985
 
            # encoded, but the values in the ConfigObj are always Unicode.
986
 
            self._config = ConfigObj(self._input, encoding='utf-8')
987
 
        except configobj.ConfigObjError, e:
988
 
            raise errors.ParseConfigError(e.errors, e.config.filename)
989
 
        return self._config
990
 
 
991
 
    def _save(self):
992
 
        """Save the config file, only tests should use it for now."""
993
 
        conf_dir = os.path.dirname(self._filename)
994
 
        ensure_config_dir_exists(conf_dir)
995
 
        f = file(self._filename, 'wb')
996
 
        try:
997
 
            self._get_config().write(f)
998
 
        finally:
999
 
            f.close()
1000
 
 
1001
 
    def _set_option(self, section_name, option_name, value):
1002
 
        """Set an authentication configuration option"""
1003
 
        conf = self._get_config()
1004
 
        section = conf.get(section_name)
1005
 
        if section is None:
1006
 
            conf[section] = {}
1007
 
            section = conf[section]
1008
 
        section[option_name] = value
1009
 
        self._save()
1010
 
 
1011
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
1012
 
                        realm=None):
1013
 
        """Returns the matching credentials from authentication.conf file.
1014
 
 
1015
 
        :param scheme: protocol
1016
 
 
1017
 
        :param host: the server address
1018
 
 
1019
 
        :param port: the associated port (optional)
1020
 
 
1021
 
        :param user: login (optional)
1022
 
 
1023
 
        :param path: the absolute path on the server (optional)
1024
 
        
1025
 
        :param realm: the http authentication realm (optional)
1026
 
 
1027
 
        :return: A dict containing the matching credentials or None.
1028
 
           This includes:
1029
 
           - name: the section name of the credentials in the
1030
 
             authentication.conf file,
1031
 
           - user: can't be different from the provided user if any,
1032
 
           - scheme: the server protocol,
1033
 
           - host: the server address,
1034
 
           - port: the server port (can be None),
1035
 
           - path: the absolute server path (can be None),
1036
 
           - realm: the http specific authentication realm (can be None),
1037
 
           - password: the decoded password, could be None if the credential
1038
 
             defines only the user
1039
 
           - verify_certificates: https specific, True if the server
1040
 
             certificate should be verified, False otherwise.
1041
 
        """
1042
 
        credentials = None
1043
 
        for auth_def_name, auth_def in self._get_config().items():
1044
 
            if type(auth_def) is not configobj.Section:
1045
 
                raise ValueError("%s defined outside a section" % auth_def_name)
1046
 
 
1047
 
            a_scheme, a_host, a_user, a_path = map(
1048
 
                auth_def.get, ['scheme', 'host', 'user', 'path'])
1049
 
 
1050
 
            try:
1051
 
                a_port = auth_def.as_int('port')
1052
 
            except KeyError:
1053
 
                a_port = None
1054
 
            except ValueError:
1055
 
                raise ValueError("'port' not numeric in %s" % auth_def_name)
1056
 
            try:
1057
 
                a_verify_certificates = auth_def.as_bool('verify_certificates')
1058
 
            except KeyError:
1059
 
                a_verify_certificates = True
1060
 
            except ValueError:
1061
 
                raise ValueError(
1062
 
                    "'verify_certificates' not boolean in %s" % auth_def_name)
1063
 
 
1064
 
            # Attempt matching
1065
 
            if a_scheme is not None and scheme != a_scheme:
1066
 
                continue
1067
 
            if a_host is not None:
1068
 
                if not (host == a_host
1069
 
                        or (a_host.startswith('.') and host.endswith(a_host))):
1070
 
                    continue
1071
 
            if a_port is not None and port != a_port:
1072
 
                continue
1073
 
            if (a_path is not None and path is not None
1074
 
                and not path.startswith(a_path)):
1075
 
                continue
1076
 
            if (a_user is not None and user is not None
1077
 
                and a_user != user):
1078
 
                # Never contradict the caller about the user to be used
1079
 
                continue
1080
 
            if a_user is None:
1081
 
                # Can't find a user
1082
 
                continue
1083
 
            # Prepare a credentials dictionary with additional keys
1084
 
            # for the credential providers
1085
 
            credentials = dict(name=auth_def_name,
1086
 
                               user=a_user,
1087
 
                               scheme=a_scheme,
1088
 
                               host=host,
1089
 
                               port=port,
1090
 
                               path=path,
1091
 
                               realm=realm,
1092
 
                               password=auth_def.get('password', None),
1093
 
                               verify_certificates=a_verify_certificates)
1094
 
            # Decode the password in the credentials (or get one)
1095
 
            self.decode_password(credentials,
1096
 
                                 auth_def.get('password_encoding', None))
1097
 
            if 'auth' in debug.debug_flags:
1098
 
                trace.mutter("Using authentication section: %r", auth_def_name)
1099
 
            break
1100
 
 
1101
 
        if credentials is None:
1102
 
            # No credentials were found in authentication.conf, try the fallback
1103
 
            # credentials stores.
1104
 
            credentials = credential_store_registry.get_fallback_credentials(
1105
 
                scheme, host, port, user, path, realm)
1106
 
 
1107
 
        return credentials
1108
 
 
1109
 
    def set_credentials(self, name, host, user, scheme=None, password=None,
1110
 
                        port=None, path=None, verify_certificates=None,
1111
 
                        realm=None):
1112
 
        """Set authentication credentials for a host.
1113
 
 
1114
 
        Any existing credentials with matching scheme, host, port and path
1115
 
        will be deleted, regardless of name.
1116
 
 
1117
 
        :param name: An arbitrary name to describe this set of credentials.
1118
 
        :param host: Name of the host that accepts these credentials.
1119
 
        :param user: The username portion of these credentials.
1120
 
        :param scheme: The URL scheme (e.g. ssh, http) the credentials apply
1121
 
            to.
1122
 
        :param password: Password portion of these credentials.
1123
 
        :param port: The IP port on the host that these credentials apply to.
1124
 
        :param path: A filesystem path on the host that these credentials
1125
 
            apply to.
1126
 
        :param verify_certificates: On https, verify server certificates if
1127
 
            True.
1128
 
        :param realm: The http authentication realm (optional).
1129
 
        """
1130
 
        values = {'host': host, 'user': user}
1131
 
        if password is not None:
1132
 
            values['password'] = password
1133
 
        if scheme is not None:
1134
 
            values['scheme'] = scheme
1135
 
        if port is not None:
1136
 
            values['port'] = '%d' % port
1137
 
        if path is not None:
1138
 
            values['path'] = path
1139
 
        if verify_certificates is not None:
1140
 
            values['verify_certificates'] = str(verify_certificates)
1141
 
        if realm is not None:
1142
 
            values['realm'] = realm
1143
 
        config = self._get_config()
1144
 
        for_deletion = []
1145
 
        for section, existing_values in config.items():
1146
 
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
1147
 
                if existing_values.get(key) != values.get(key):
1148
 
                    break
1149
 
            else:
1150
 
                del config[section]
1151
 
        config.update({name: values})
1152
 
        self._save()
1153
 
 
1154
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1155
 
                 prompt=None, ask=False, default=None):
1156
 
        """Get a user from authentication file.
1157
 
 
1158
 
        :param scheme: protocol
1159
 
 
1160
 
        :param host: the server address
1161
 
 
1162
 
        :param port: the associated port (optional)
1163
 
 
1164
 
        :param realm: the realm sent by the server (optional)
1165
 
 
1166
 
        :param path: the absolute path on the server (optional)
1167
 
 
1168
 
        :param ask: Ask the user if there is no explicitly configured username 
1169
 
                    (optional)
1170
 
 
1171
 
        :param default: The username returned if none is defined (optional).
1172
 
 
1173
 
        :return: The found user.
1174
 
        """
1175
 
        credentials = self.get_credentials(scheme, host, port, user=None,
1176
 
                                           path=path, realm=realm)
1177
 
        if credentials is not None:
1178
 
            user = credentials['user']
1179
 
        else:
1180
 
            user = None
1181
 
        if user is None:
1182
 
            if ask:
1183
 
                if prompt is None:
1184
 
                    # Create a default prompt suitable for most cases
1185
 
                    prompt = scheme.upper() + ' %(host)s username'
1186
 
                # Special handling for optional fields in the prompt
1187
 
                if port is not None:
1188
 
                    prompt_host = '%s:%d' % (host, port)
1189
 
                else:
1190
 
                    prompt_host = host
1191
 
                user = ui.ui_factory.get_username(prompt, host=prompt_host)
1192
 
            else:
1193
 
                user = default
1194
 
        return user
1195
 
 
1196
 
    def get_password(self, scheme, host, user, port=None,
1197
 
                     realm=None, path=None, prompt=None):
1198
 
        """Get a password from authentication file or prompt the user for one.
1199
 
 
1200
 
        :param scheme: protocol
1201
 
 
1202
 
        :param host: the server address
1203
 
 
1204
 
        :param port: the associated port (optional)
1205
 
 
1206
 
        :param user: login
1207
 
 
1208
 
        :param realm: the realm sent by the server (optional)
1209
 
 
1210
 
        :param path: the absolute path on the server (optional)
1211
 
 
1212
 
        :return: The found password or the one entered by the user.
1213
 
        """
1214
 
        credentials = self.get_credentials(scheme, host, port, user, path,
1215
 
                                           realm)
1216
 
        if credentials is not None:
1217
 
            password = credentials['password']
1218
 
            if password is not None and scheme is 'ssh':
1219
 
                trace.warning('password ignored in section [%s],'
1220
 
                              ' use an ssh agent instead'
1221
 
                              % credentials['name'])
1222
 
                password = None
1223
 
        else:
1224
 
            password = None
1225
 
        # Prompt user only if we could't find a password
1226
 
        if password is None:
1227
 
            if prompt is None:
1228
 
                # Create a default prompt suitable for most cases
1229
 
                prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
1230
 
            # Special handling for optional fields in the prompt
1231
 
            if port is not None:
1232
 
                prompt_host = '%s:%d' % (host, port)
1233
 
            else:
1234
 
                prompt_host = host
1235
 
            password = ui.ui_factory.get_password(prompt,
1236
 
                                                  host=prompt_host, user=user)
1237
 
        return password
1238
 
 
1239
 
    def decode_password(self, credentials, encoding):
1240
 
        try:
1241
 
            cs = credential_store_registry.get_credential_store(encoding)
1242
 
        except KeyError:
1243
 
            raise ValueError('%r is not a known password_encoding' % encoding)
1244
 
        credentials['password'] = cs.decode_password(credentials)
1245
 
        return credentials
1246
 
 
1247
 
 
1248
 
class CredentialStoreRegistry(registry.Registry):
1249
 
    """A class that registers credential stores.
1250
 
 
1251
 
    A credential store provides access to credentials via the password_encoding
1252
 
    field in authentication.conf sections.
1253
 
 
1254
 
    Except for stores provided by bzr itself, most stores are expected to be
1255
 
    provided by plugins that will therefore use
1256
 
    register_lazy(password_encoding, module_name, member_name, help=help,
1257
 
    fallback=fallback) to install themselves.
1258
 
 
1259
 
    A fallback credential store is one that is queried if no credentials can be
1260
 
    found via authentication.conf.
1261
 
    """
1262
 
 
1263
 
    def get_credential_store(self, encoding=None):
1264
 
        cs = self.get(encoding)
1265
 
        if callable(cs):
1266
 
            cs = cs()
1267
 
        return cs
1268
 
 
1269
 
    def is_fallback(self, name):
1270
 
        """Check if the named credentials store should be used as fallback."""
1271
 
        return self.get_info(name)
1272
 
 
1273
 
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
1274
 
                                 path=None, realm=None):
1275
 
        """Request credentials from all fallback credentials stores.
1276
 
 
1277
 
        The first credentials store that can provide credentials wins.
1278
 
        """
1279
 
        credentials = None
1280
 
        for name in self.keys():
1281
 
            if not self.is_fallback(name):
1282
 
                continue
1283
 
            cs = self.get_credential_store(name)
1284
 
            credentials = cs.get_credentials(scheme, host, port, user,
1285
 
                                             path, realm)
1286
 
            if credentials is not None:
1287
 
                # We found some credentials
1288
 
                break
1289
 
        return credentials
1290
 
 
1291
 
    def register(self, key, obj, help=None, override_existing=False,
1292
 
                 fallback=False):
1293
 
        """Register a new object to a name.
1294
 
 
1295
 
        :param key: This is the key to use to request the object later.
1296
 
        :param obj: The object to register.
1297
 
        :param help: Help text for this entry. This may be a string or
1298
 
                a callable. If it is a callable, it should take two
1299
 
                parameters (registry, key): this registry and the key that
1300
 
                the help was registered under.
1301
 
        :param override_existing: Raise KeyErorr if False and something has
1302
 
                already been registered for that key. If True, ignore if there
1303
 
                is an existing key (always register the new value).
1304
 
        :param fallback: Whether this credential store should be 
1305
 
                used as fallback.
1306
 
        """
1307
 
        return super(CredentialStoreRegistry,
1308
 
                     self).register(key, obj, help, info=fallback,
1309
 
                                    override_existing=override_existing)
1310
 
 
1311
 
    def register_lazy(self, key, module_name, member_name,
1312
 
                      help=None, override_existing=False,
1313
 
                      fallback=False):
1314
 
        """Register a new credential store to be loaded on request.
1315
 
 
1316
 
        :param module_name: The python path to the module. Such as 'os.path'.
1317
 
        :param member_name: The member of the module to return.  If empty or
1318
 
                None, get() will return the module itself.
1319
 
        :param help: Help text for this entry. This may be a string or
1320
 
                a callable.
1321
 
        :param override_existing: If True, replace the existing object
1322
 
                with the new one. If False, if there is already something
1323
 
                registered with the same key, raise a KeyError
1324
 
        :param fallback: Whether this credential store should be 
1325
 
                used as fallback.
1326
 
        """
1327
 
        return super(CredentialStoreRegistry, self).register_lazy(
1328
 
            key, module_name, member_name, help,
1329
 
            info=fallback, override_existing=override_existing)
1330
 
 
1331
 
 
1332
 
credential_store_registry = CredentialStoreRegistry()
1333
 
 
1334
 
 
1335
 
class CredentialStore(object):
1336
 
    """An abstract class to implement storage for credentials"""
1337
 
 
1338
 
    def decode_password(self, credentials):
1339
 
        """Returns a clear text password for the provided credentials."""
1340
 
        raise NotImplementedError(self.decode_password)
1341
 
 
1342
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
1343
 
                        realm=None):
1344
 
        """Return the matching credentials from this credential store.
1345
 
 
1346
 
        This method is only called on fallback credential stores.
1347
 
        """
1348
 
        raise NotImplementedError(self.get_credentials)
1349
 
 
1350
 
 
1351
 
 
1352
 
class PlainTextCredentialStore(CredentialStore):
1353
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
1354
 
 
1355
 
    def decode_password(self, credentials):
1356
 
        """See CredentialStore.decode_password."""
1357
 
        return credentials['password']
1358
 
 
1359
 
 
1360
 
credential_store_registry.register('plain', PlainTextCredentialStore,
1361
 
                                   help=PlainTextCredentialStore.__doc__)
1362
 
credential_store_registry.default_key = 'plain'
1363
 
 
1364
 
 
1365
 
class BzrDirConfig(object):
1366
 
 
1367
 
    def __init__(self, bzrdir):
1368
 
        self._bzrdir = bzrdir
1369
 
        self._config = bzrdir._get_config()
1370
 
 
1371
 
    def set_default_stack_on(self, value):
1372
 
        """Set the default stacking location.
1373
 
 
1374
 
        It may be set to a location, or None.
1375
 
 
1376
 
        This policy affects all branches contained by this bzrdir, except for
1377
 
        those under repositories.
1378
 
        """
1379
 
        if self._config is None:
1380
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1381
 
        if value is None:
1382
 
            self._config.set_option('', 'default_stack_on')
1383
 
        else:
1384
 
            self._config.set_option(value, 'default_stack_on')
1385
 
 
1386
 
    def get_default_stack_on(self):
1387
 
        """Return the default stacking location.
1388
 
 
1389
 
        This will either be a location, or None.
1390
 
 
1391
 
        This policy affects all branches contained by this bzrdir, except for
1392
 
        those under repositories.
1393
 
        """
1394
 
        if self._config is None:
1395
 
            return None
1396
 
        value = self._config.get_option('default_stack_on')
1397
 
        if value == '':
1398
 
            value = None
1399
 
        return value
1400
 
 
1401
 
 
1402
 
class TransportConfig(object):
1403
 
    """A Config that reads/writes a config file on a Transport.
1404
 
 
1405
 
    It is a low-level object that considers config data to be name/value pairs
1406
 
    that may be associated with a section.  Assigning meaning to the these
1407
 
    values is done at higher levels like TreeConfig.
1408
 
    """
1409
 
 
1410
 
    def __init__(self, transport, filename):
1411
 
        self._transport = transport
1412
 
        self._filename = filename
1413
 
 
1414
 
    def get_option(self, name, section=None, default=None):
1415
 
        """Return the value associated with a named option.
1416
 
 
1417
 
        :param name: The name of the value
1418
 
        :param section: The section the option is in (if any)
1419
 
        :param default: The value to return if the value is not set
1420
 
        :return: The value or default value
1421
 
        """
1422
 
        configobj = self._get_configobj()
1423
 
        if section is None:
1424
 
            section_obj = configobj
1425
 
        else:
1426
 
            try:
1427
 
                section_obj = configobj[section]
1428
 
            except KeyError:
1429
 
                return default
1430
 
        return section_obj.get(name, default)
1431
 
 
1432
 
    def set_option(self, value, name, section=None):
1433
 
        """Set the value associated with a named option.
1434
 
 
1435
 
        :param value: The value to set
1436
 
        :param name: The name of the value to set
1437
 
        :param section: The section the option is in (if any)
1438
 
        """
1439
 
        configobj = self._get_configobj()
1440
 
        if section is None:
1441
 
            configobj[name] = value
1442
 
        else:
1443
 
            configobj.setdefault(section, {})[name] = value
1444
 
        self._set_configobj(configobj)
1445
 
 
1446
 
    def _get_config_file(self):
1447
 
        try:
1448
 
            return StringIO(self._transport.get_bytes(self._filename))
1449
 
        except errors.NoSuchFile:
1450
 
            return StringIO()
1451
 
 
1452
 
    def _get_configobj(self):
1453
 
        f = self._get_config_file()
1454
 
        try:
1455
 
            return ConfigObj(f, encoding='utf-8')
1456
 
        finally:
1457
 
            f.close()
1458
 
 
1459
 
    def _set_configobj(self, configobj):
1460
 
        out_file = StringIO()
1461
 
        configobj.write(out_file)
1462
 
        out_file.seek(0)
1463
 
        self._transport.put_file(self._filename, out_file)