/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Andrew Bennetts
  • Date: 2008-10-30 01:05:59 UTC
  • mto: This revision was merged to the branch mainline in revision 3813.
  • Revision ID: andrew.bennetts@canonical.com-20081030010559-tumoefnsmhg4snxo
Add contrib/bzr_ssh_path_limiter.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2007, 2008 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
78
78
    errors,
79
79
    mail_client,
80
80
    osutils,
81
 
    registry,
82
81
    symbol_versioning,
83
82
    trace,
84
83
    ui,
146
145
class Config(object):
147
146
    """A configuration policy - what username, editor, gpg needs etc."""
148
147
 
149
 
    def __init__(self):
150
 
        super(Config, self).__init__()
151
 
 
152
148
    def get_editor(self):
153
149
        """Get the users pop up editor."""
154
150
        raise NotImplementedError
155
151
 
156
 
    def get_change_editor(self, old_tree, new_tree):
157
 
        from bzrlib import diff
158
 
        cmd = self._get_change_editor()
159
 
        if cmd is None:
160
 
            return None
161
 
        return diff.DiffFromTool.from_string(cmd, old_tree, new_tree,
162
 
                                             sys.stdout)
163
 
 
164
 
 
165
152
    def get_mail_client(self):
166
153
        """Get a mail client to use"""
167
154
        selected_client = self.get_user_option('mail_client')
186
173
        """Get a generic option - no special process, no default."""
187
174
        return self._get_user_option(option_name)
188
175
 
189
 
    def get_user_option_as_bool(self, option_name):
190
 
        """Get a generic option as a boolean - no special process, no default.
191
 
 
192
 
        :return None if the option doesn't exist or its value can't be
193
 
            interpreted as a boolean. Returns True or False otherwise.
194
 
        """
195
 
        s = self._get_user_option(option_name)
196
 
        if s is None:
197
 
            # The option doesn't exist
198
 
            return None
199
 
        val = ui.bool_from_string(s)
200
 
        if val is None:
201
 
            # The value can't be interpreted as a boolean
202
 
            trace.warning('Value "%s" is not a boolean for "%s"',
203
 
                          s, option_name)
204
 
        return val
205
 
 
206
 
    def get_user_option_as_list(self, option_name):
207
 
        """Get a generic option as a list - no special process, no default.
208
 
 
209
 
        :return None if the option doesn't exist. Returns the value as a list
210
 
            otherwise.
211
 
        """
212
 
        l = self._get_user_option(option_name)
213
 
        if isinstance(l, (str, unicode)):
214
 
            # A single value, most probably the user forgot the final ','
215
 
            l = [l]
216
 
        return l
217
 
 
218
176
    def gpg_signing_command(self):
219
177
        """What program should be used to sign signatures?"""
220
178
        result = self._gpg_signing_command()
237
195
        """See log_format()."""
238
196
        return None
239
197
 
 
198
    def __init__(self):
 
199
        super(Config, self).__init__()
 
200
 
240
201
    def post_commit(self):
241
202
        """An ordered list of python functions to call.
242
203
 
254
215
 
255
216
    def username(self):
256
217
        """Return email-style username.
257
 
 
 
218
    
258
219
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
259
 
 
260
 
        $BZR_EMAIL can be set to override this, then
 
220
        
 
221
        $BZR_EMAIL can be set to override this (as well as the
 
222
        deprecated $BZREMAIL), then
261
223
        the concrete policy type is checked, and finally
262
224
        $EMAIL is examined.
263
 
        If no username can be found, errors.NoWhoami exception is raised.
264
 
 
 
225
        If none is found, a reasonable default is (hopefully)
 
226
        created.
 
227
    
265
228
        TODO: Check it's reasonably well-formed.
266
229
        """
267
230
        v = os.environ.get('BZR_EMAIL')
276
239
        if v:
277
240
            return v.decode(osutils.get_user_encoding())
278
241
 
279
 
        raise errors.NoWhoami()
280
 
 
281
 
    def ensure_username(self):
282
 
        """Raise errors.NoWhoami if username is not set.
283
 
 
284
 
        This method relies on the username() function raising the error.
285
 
        """
286
 
        self.username()
 
242
        name, email = _auto_user_id()
 
243
        if name:
 
244
            return '%s <%s>' % (name, email)
 
245
        else:
 
246
            return email
287
247
 
288
248
    def signature_checking(self):
289
249
        """What is the current policy for signature checking?."""
334
294
                path = 'bzr'
335
295
            return path
336
296
 
337
 
    def suppress_warning(self, warning):
338
 
        """Should the warning be suppressed or emitted.
339
 
 
340
 
        :param warning: The name of the warning being tested.
341
 
 
342
 
        :returns: True if the warning should be suppressed, False otherwise.
343
 
        """
344
 
        warnings = self.get_user_option_as_list('suppress_warnings')
345
 
        if warnings is None or warning not in warnings:
346
 
            return False
347
 
        else:
348
 
            return True
349
 
 
350
297
 
351
298
class IniBasedConfig(Config):
352
299
    """A configuration policy that draws from ini files."""
353
300
 
354
 
    def __init__(self, get_filename):
355
 
        super(IniBasedConfig, self).__init__()
356
 
        self._get_filename = get_filename
357
 
        self._parser = None
358
 
 
359
301
    def _get_parser(self, file=None):
360
302
        if self._parser is not None:
361
303
            return self._parser
389
331
        """Return the policy for the given (section, option_name) pair."""
390
332
        return POLICY_NONE
391
333
 
392
 
    def _get_change_editor(self):
393
 
        return self.get_user_option('change_editor')
394
 
 
395
334
    def _get_signature_checking(self):
396
335
        """See Config._get_signature_checking."""
397
336
        policy = self._get_user_option('check_signatures')
441
380
        """See Config.log_format."""
442
381
        return self._get_user_option('log_format')
443
382
 
 
383
    def __init__(self, get_filename):
 
384
        super(IniBasedConfig, self).__init__()
 
385
        self._get_filename = get_filename
 
386
        self._parser = None
 
387
        
444
388
    def _post_commit(self):
445
389
        """See Config.post_commit."""
446
390
        return self._get_user_option('post_commit')
469
413
 
470
414
    def _get_alias(self, value):
471
415
        try:
472
 
            return self._get_parser().get_value("ALIASES",
 
416
            return self._get_parser().get_value("ALIASES", 
473
417
                                                value)
474
418
        except KeyError:
475
419
            pass
477
421
    def _get_nickname(self):
478
422
        return self.get_user_option('nickname')
479
423
 
480
 
    def _write_config_file(self):
481
 
        f = file(self._get_filename(), "wb")
482
 
        try:
483
 
            osutils.copy_ownership_from_path(f.name)
484
 
            self._get_parser().write(f)
485
 
        finally:
486
 
            f.close()
487
 
 
488
424
 
489
425
class GlobalConfig(IniBasedConfig):
490
426
    """The configuration that should be used for a specific location."""
526
462
        self._get_parser().setdefault(section, {})[option] = value
527
463
        self._write_config_file()
528
464
 
 
465
    def _write_config_file(self):
 
466
        f = open(self._get_filename(), 'wb')
 
467
        self._get_parser().write(f)
 
468
        f.close()
 
469
 
529
470
 
530
471
class LocationConfig(IniBasedConfig):
531
472
    """A configuration object that gives the policy for a location."""
665
606
        self._get_parser()[location][option]=value
666
607
        # the allowed values of store match the config policies
667
608
        self._set_option_policy(location, option, store)
668
 
        self._write_config_file()
 
609
        self._get_parser().write(file(self._get_filename(), 'wb'))
669
610
 
670
611
 
671
612
class BranchConfig(Config):
700
641
 
701
642
    def _get_safe_value(self, option_name):
702
643
        """This variant of get_best_value never returns untrusted values.
703
 
 
 
644
        
704
645
        It does not return values from the branch data, because the branch may
705
646
        not be controlled by the user.
706
647
 
715
656
 
716
657
    def _get_user_id(self):
717
658
        """Return the full user id for the branch.
718
 
 
 
659
    
719
660
        e.g. "John Hacker <jhacker@example.com>"
720
661
        This is looked up in the email controlfile for the branch.
721
662
        """
725
666
                    .rstrip("\r\n"))
726
667
        except errors.NoSuchFile, e:
727
668
            pass
728
 
 
 
669
        
729
670
        return self._get_best_value('_get_user_id')
730
671
 
731
 
    def _get_change_editor(self):
732
 
        return self._get_best_value('_get_change_editor')
733
 
 
734
672
    def _get_signature_checking(self):
735
673
        """See Config._get_signature_checking."""
736
674
        return self._get_best_value('_get_signature_checking')
770
708
                        trace.warning('Value "%s" is masked by "%s" from'
771
709
                                      ' branch.conf', value, mask_value)
772
710
 
 
711
 
773
712
    def _gpg_signing_command(self):
774
713
        """See Config.gpg_signing_command."""
775
714
        return self._get_safe_value('_gpg_signing_command')
776
 
 
 
715
        
777
716
    def __init__(self, branch):
778
717
        super(BranchConfig, self).__init__()
779
718
        self._location_config = None
780
719
        self._branch_data_config = None
781
720
        self._global_config = None
782
721
        self.branch = branch
783
 
        self.option_sources = (self._get_location_config,
 
722
        self.option_sources = (self._get_location_config, 
784
723
                               self._get_branch_data_config,
785
724
                               self._get_global_config)
786
725
 
822
761
            os.mkdir(parent_dir)
823
762
        trace.mutter('creating config directory: %r', path)
824
763
        os.mkdir(path)
825
 
        osutils.copy_ownership_from_path(path)
826
764
 
827
765
 
828
766
def config_dir():
829
767
    """Return per-user configuration directory.
830
768
 
831
769
    By default this is ~/.bazaar/
832
 
 
 
770
    
833
771
    TODO: Global option --config-dir to override this.
834
772
    """
835
773
    base = os.environ.get('BZR_HOME', None)
843
781
                                  ' or HOME set')
844
782
        return osutils.pathjoin(base, 'bazaar', '2.0')
845
783
    else:
 
784
        # cygwin, linux, and darwin all have a $HOME directory
846
785
        if base is None:
847
786
            base = os.path.expanduser("~")
848
787
        return osutils.pathjoin(base, ".bazaar")
873
812
    return osutils.pathjoin(config_dir(), 'ignore')
874
813
 
875
814
 
876
 
def crash_dir():
877
 
    """Return the directory name to store crash files.
878
 
 
879
 
    This doesn't implicitly create it.
880
 
 
881
 
    On Windows it's in the config directory; elsewhere it's /var/crash
882
 
    which may be monitored by apport.  It can be overridden by
883
 
    $APPORT_CRASH_DIR.
 
815
def _auto_user_id():
 
816
    """Calculate automatic user identification.
 
817
 
 
818
    Returns (realname, email).
 
819
 
 
820
    Only used when none is set in the environment or the id file.
 
821
 
 
822
    This previously used the FQDN as the default domain, but that can
 
823
    be very slow on machines where DNS is broken.  So now we simply
 
824
    use the hostname.
884
825
    """
 
826
    import socket
 
827
 
885
828
    if sys.platform == 'win32':
886
 
        return osutils.pathjoin(config_dir(), 'Crash')
887
 
    else:
888
 
        # XXX: hardcoded in apport_python_hook.py; therefore here too -- mbp
889
 
        # 2010-01-31
890
 
        return os.environ.get('APPORT_CRASH_DIR', '/var/crash')
891
 
 
892
 
 
893
 
def xdg_cache_dir():
894
 
    # See http://standards.freedesktop.org/basedir-spec/latest/ar01s03.html
895
 
    # Possibly this should be different on Windows?
896
 
    e = os.environ.get('XDG_CACHE_DIR', None)
897
 
    if e:
898
 
        return e
899
 
    else:
900
 
        return os.path.expanduser('~/.cache')
 
829
        name = win32utils.get_user_name_unicode()
 
830
        if name is None:
 
831
            raise errors.BzrError("Cannot autodetect user name.\n"
 
832
                                  "Please, set your name with command like:\n"
 
833
                                  'bzr whoami "Your Name <name@domain.com>"')
 
834
        host = win32utils.get_host_name_unicode()
 
835
        if host is None:
 
836
            host = socket.gethostname()
 
837
        return name, (name + '@' + host)
 
838
 
 
839
    try:
 
840
        import pwd
 
841
        uid = os.getuid()
 
842
        w = pwd.getpwuid(uid)
 
843
 
 
844
        # we try utf-8 first, because on many variants (like Linux),
 
845
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
846
        # false positives.  (many users will have their user encoding set to
 
847
        # latin-1, which cannot raise UnicodeError.)
 
848
        try:
 
849
            gecos = w.pw_gecos.decode('utf-8')
 
850
            encoding = 'utf-8'
 
851
        except UnicodeError:
 
852
            try:
 
853
                encoding = osutils.get_user_encoding()
 
854
                gecos = w.pw_gecos.decode(encoding)
 
855
            except UnicodeError:
 
856
                raise errors.BzrCommandError('Unable to determine your name.  '
 
857
                   'Use "bzr whoami" to set it.')
 
858
        try:
 
859
            username = w.pw_name.decode(encoding)
 
860
        except UnicodeError:
 
861
            raise errors.BzrCommandError('Unable to determine your name.  '
 
862
                'Use "bzr whoami" to set it.')
 
863
 
 
864
        comma = gecos.find(',')
 
865
        if comma == -1:
 
866
            realname = gecos
 
867
        else:
 
868
            realname = gecos[:comma]
 
869
        if not realname:
 
870
            realname = username
 
871
 
 
872
    except ImportError:
 
873
        import getpass
 
874
        try:
 
875
            user_encoding = osutils.get_user_encoding()
 
876
            realname = username = getpass.getuser().decode(user_encoding)
 
877
        except UnicodeDecodeError:
 
878
            raise errors.BzrError("Can't decode username as %s." % \
 
879
                    user_encoding)
 
880
 
 
881
    return realname, (username + '@' + socket.gethostname())
901
882
 
902
883
 
903
884
def parse_username(username):
912
893
def extract_email_address(e):
913
894
    """Return just the address part of an email string.
914
895
 
915
 
    That is just the user@domain part, nothing else.
 
896
    That is just the user@domain part, nothing else. 
916
897
    This part is required to contain only ascii characters.
917
898
    If it can't be extracted, raises an error.
918
899
 
931
912
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
932
913
 
933
914
    def __init__(self, branch):
934
 
        self._config = branch._get_config()
 
915
        # XXX: Really this should be asking the branch for its configuration
 
916
        # data, rather than relying on a Transport, so that it can work 
 
917
        # more cleanly with a RemoteBranch that has no transport.
 
918
        self._config = TransportConfig(branch._transport, 'branch.conf')
935
919
        self.branch = branch
936
920
 
937
921
    def _get_parser(self, file=None):
945
929
            return self._config.get_option(name, section, default)
946
930
        finally:
947
931
            self.branch.unlock()
 
932
        return result
948
933
 
949
934
    def set_option(self, value, name, section=None):
950
935
        """Set a per-branch configuration option"""
991
976
        """Save the config file, only tests should use it for now."""
992
977
        conf_dir = os.path.dirname(self._filename)
993
978
        ensure_config_dir_exists(conf_dir)
994
 
        f = file(self._filename, 'wb')
995
 
        try:
996
 
            self._get_config().write(f)
997
 
        finally:
998
 
            f.close()
 
979
        self._get_config().write(file(self._filename, 'wb'))
999
980
 
1000
981
    def _set_option(self, section_name, option_name, value):
1001
982
        """Set an authentication configuration option"""
1007
988
        section[option_name] = value
1008
989
        self._save()
1009
990
 
1010
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
1011
 
                        realm=None):
 
991
    def get_credentials(self, scheme, host, port=None, user=None, path=None):
1012
992
        """Returns the matching credentials from authentication.conf file.
1013
993
 
1014
994
        :param scheme: protocol
1020
1000
        :param user: login (optional)
1021
1001
 
1022
1002
        :param path: the absolute path on the server (optional)
1023
 
        
1024
 
        :param realm: the http authentication realm (optional)
1025
1003
 
1026
1004
        :return: A dict containing the matching credentials or None.
1027
1005
           This includes:
1028
1006
           - name: the section name of the credentials in the
1029
1007
             authentication.conf file,
1030
 
           - user: can't be different from the provided user if any,
1031
 
           - scheme: the server protocol,
1032
 
           - host: the server address,
1033
 
           - port: the server port (can be None),
1034
 
           - path: the absolute server path (can be None),
1035
 
           - realm: the http specific authentication realm (can be None),
 
1008
           - user: can't de different from the provided user if any,
1036
1009
           - password: the decoded password, could be None if the credential
1037
1010
             defines only the user
1038
1011
           - verify_certificates: https specific, True if the server
1079
1052
            if a_user is None:
1080
1053
                # Can't find a user
1081
1054
                continue
1082
 
            # Prepare a credentials dictionary with additional keys
1083
 
            # for the credential providers
1084
1055
            credentials = dict(name=auth_def_name,
1085
1056
                               user=a_user,
1086
 
                               scheme=a_scheme,
1087
 
                               host=host,
1088
 
                               port=port,
1089
 
                               path=path,
1090
 
                               realm=realm,
1091
1057
                               password=auth_def.get('password', None),
1092
1058
                               verify_certificates=a_verify_certificates)
1093
 
            # Decode the password in the credentials (or get one)
1094
1059
            self.decode_password(credentials,
1095
1060
                                 auth_def.get('password_encoding', None))
1096
1061
            if 'auth' in debug.debug_flags:
1097
1062
                trace.mutter("Using authentication section: %r", auth_def_name)
1098
1063
            break
1099
1064
 
1100
 
        if credentials is None:
1101
 
            # No credentials were found in authentication.conf, try the fallback
1102
 
            # credentials stores.
1103
 
            credentials = credential_store_registry.get_fallback_credentials(
1104
 
                scheme, host, port, user, path, realm)
1105
 
 
1106
1065
        return credentials
1107
1066
 
1108
1067
    def set_credentials(self, name, host, user, scheme=None, password=None,
1109
 
                        port=None, path=None, verify_certificates=None,
1110
 
                        realm=None):
 
1068
                        port=None, path=None, verify_certificates=None):
1111
1069
        """Set authentication credentials for a host.
1112
1070
 
1113
1071
        Any existing credentials with matching scheme, host, port and path
1124
1082
            apply to.
1125
1083
        :param verify_certificates: On https, verify server certificates if
1126
1084
            True.
1127
 
        :param realm: The http authentication realm (optional).
1128
1085
        """
1129
1086
        values = {'host': host, 'user': user}
1130
1087
        if password is not None:
1137
1094
            values['path'] = path
1138
1095
        if verify_certificates is not None:
1139
1096
            values['verify_certificates'] = str(verify_certificates)
1140
 
        if realm is not None:
1141
 
            values['realm'] = realm
1142
1097
        config = self._get_config()
1143
1098
        for_deletion = []
1144
1099
        for section, existing_values in config.items():
1145
 
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
 
1100
            for key in ('scheme', 'host', 'port', 'path'):
1146
1101
                if existing_values.get(key) != values.get(key):
1147
1102
                    break
1148
1103
            else:
1150
1105
        config.update({name: values})
1151
1106
        self._save()
1152
1107
 
1153
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1154
 
                 prompt=None, ask=False, default=None):
 
1108
    def get_user(self, scheme, host, port=None,
 
1109
                 realm=None, path=None, prompt=None):
1155
1110
        """Get a user from authentication file.
1156
1111
 
1157
1112
        :param scheme: protocol
1164
1119
 
1165
1120
        :param path: the absolute path on the server (optional)
1166
1121
 
1167
 
        :param ask: Ask the user if there is no explicitly configured username 
1168
 
                    (optional)
1169
 
 
1170
 
        :param default: The username returned if none is defined (optional).
1171
 
 
1172
1122
        :return: The found user.
1173
1123
        """
1174
1124
        credentials = self.get_credentials(scheme, host, port, user=None,
1175
 
                                           path=path, realm=realm)
 
1125
                                           path=path)
1176
1126
        if credentials is not None:
1177
1127
            user = credentials['user']
1178
1128
        else:
1179
1129
            user = None
1180
 
        if user is None:
1181
 
            if ask:
1182
 
                if prompt is None:
1183
 
                    # Create a default prompt suitable for most cases
1184
 
                    prompt = scheme.upper() + ' %(host)s username'
1185
 
                # Special handling for optional fields in the prompt
1186
 
                if port is not None:
1187
 
                    prompt_host = '%s:%d' % (host, port)
1188
 
                else:
1189
 
                    prompt_host = host
1190
 
                user = ui.ui_factory.get_username(prompt, host=prompt_host)
1191
 
            else:
1192
 
                user = default
1193
1130
        return user
1194
1131
 
1195
1132
    def get_password(self, scheme, host, user, port=None,
1210
1147
 
1211
1148
        :return: The found password or the one entered by the user.
1212
1149
        """
1213
 
        credentials = self.get_credentials(scheme, host, port, user, path,
1214
 
                                           realm)
 
1150
        credentials = self.get_credentials(scheme, host, port, user, path)
1215
1151
        if credentials is not None:
1216
1152
            password = credentials['password']
1217
1153
            if password is not None and scheme is 'ssh':
1236
1172
        return password
1237
1173
 
1238
1174
    def decode_password(self, credentials, encoding):
1239
 
        try:
1240
 
            cs = credential_store_registry.get_credential_store(encoding)
1241
 
        except KeyError:
1242
 
            raise ValueError('%r is not a known password_encoding' % encoding)
1243
 
        credentials['password'] = cs.decode_password(credentials)
1244
 
        return credentials
1245
 
 
1246
 
 
1247
 
class CredentialStoreRegistry(registry.Registry):
1248
 
    """A class that registers credential stores.
1249
 
 
1250
 
    A credential store provides access to credentials via the password_encoding
1251
 
    field in authentication.conf sections.
1252
 
 
1253
 
    Except for stores provided by bzr itself, most stores are expected to be
1254
 
    provided by plugins that will therefore use
1255
 
    register_lazy(password_encoding, module_name, member_name, help=help,
1256
 
    fallback=fallback) to install themselves.
1257
 
 
1258
 
    A fallback credential store is one that is queried if no credentials can be
1259
 
    found via authentication.conf.
1260
 
    """
1261
 
 
1262
 
    def get_credential_store(self, encoding=None):
1263
 
        cs = self.get(encoding)
1264
 
        if callable(cs):
1265
 
            cs = cs()
1266
 
        return cs
1267
 
 
1268
 
    def is_fallback(self, name):
1269
 
        """Check if the named credentials store should be used as fallback."""
1270
 
        return self.get_info(name)
1271
 
 
1272
 
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
1273
 
                                 path=None, realm=None):
1274
 
        """Request credentials from all fallback credentials stores.
1275
 
 
1276
 
        The first credentials store that can provide credentials wins.
1277
 
        """
1278
 
        credentials = None
1279
 
        for name in self.keys():
1280
 
            if not self.is_fallback(name):
1281
 
                continue
1282
 
            cs = self.get_credential_store(name)
1283
 
            credentials = cs.get_credentials(scheme, host, port, user,
1284
 
                                             path, realm)
1285
 
            if credentials is not None:
1286
 
                # We found some credentials
1287
 
                break
1288
 
        return credentials
1289
 
 
1290
 
    def register(self, key, obj, help=None, override_existing=False,
1291
 
                 fallback=False):
1292
 
        """Register a new object to a name.
1293
 
 
1294
 
        :param key: This is the key to use to request the object later.
1295
 
        :param obj: The object to register.
1296
 
        :param help: Help text for this entry. This may be a string or
1297
 
                a callable. If it is a callable, it should take two
1298
 
                parameters (registry, key): this registry and the key that
1299
 
                the help was registered under.
1300
 
        :param override_existing: Raise KeyErorr if False and something has
1301
 
                already been registered for that key. If True, ignore if there
1302
 
                is an existing key (always register the new value).
1303
 
        :param fallback: Whether this credential store should be 
1304
 
                used as fallback.
1305
 
        """
1306
 
        return super(CredentialStoreRegistry,
1307
 
                     self).register(key, obj, help, info=fallback,
1308
 
                                    override_existing=override_existing)
1309
 
 
1310
 
    def register_lazy(self, key, module_name, member_name,
1311
 
                      help=None, override_existing=False,
1312
 
                      fallback=False):
1313
 
        """Register a new credential store to be loaded on request.
1314
 
 
1315
 
        :param module_name: The python path to the module. Such as 'os.path'.
1316
 
        :param member_name: The member of the module to return.  If empty or
1317
 
                None, get() will return the module itself.
1318
 
        :param help: Help text for this entry. This may be a string or
1319
 
                a callable.
1320
 
        :param override_existing: If True, replace the existing object
1321
 
                with the new one. If False, if there is already something
1322
 
                registered with the same key, raise a KeyError
1323
 
        :param fallback: Whether this credential store should be 
1324
 
                used as fallback.
1325
 
        """
1326
 
        return super(CredentialStoreRegistry, self).register_lazy(
1327
 
            key, module_name, member_name, help,
1328
 
            info=fallback, override_existing=override_existing)
1329
 
 
1330
 
 
1331
 
credential_store_registry = CredentialStoreRegistry()
1332
 
 
1333
 
 
1334
 
class CredentialStore(object):
1335
 
    """An abstract class to implement storage for credentials"""
1336
 
 
1337
 
    def decode_password(self, credentials):
1338
 
        """Returns a clear text password for the provided credentials."""
1339
 
        raise NotImplementedError(self.decode_password)
1340
 
 
1341
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
1342
 
                        realm=None):
1343
 
        """Return the matching credentials from this credential store.
1344
 
 
1345
 
        This method is only called on fallback credential stores.
1346
 
        """
1347
 
        raise NotImplementedError(self.get_credentials)
1348
 
 
1349
 
 
1350
 
 
1351
 
class PlainTextCredentialStore(CredentialStore):
1352
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
1353
 
 
1354
 
    def decode_password(self, credentials):
1355
 
        """See CredentialStore.decode_password."""
1356
 
        return credentials['password']
1357
 
 
1358
 
 
1359
 
credential_store_registry.register('plain', PlainTextCredentialStore,
1360
 
                                   help=PlainTextCredentialStore.__doc__)
1361
 
credential_store_registry.default_key = 'plain'
 
1175
        return credentials
1362
1176
 
1363
1177
 
1364
1178
class BzrDirConfig(object):
1365
1179
 
1366
 
    def __init__(self, bzrdir):
1367
 
        self._bzrdir = bzrdir
1368
 
        self._config = bzrdir._get_config()
 
1180
    def __init__(self, transport):
 
1181
        self._config = TransportConfig(transport, 'control.conf')
1369
1182
 
1370
1183
    def set_default_stack_on(self, value):
1371
1184
        """Set the default stacking location.
1375
1188
        This policy affects all branches contained by this bzrdir, except for
1376
1189
        those under repositories.
1377
1190
        """
1378
 
        if self._config is None:
1379
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1380
1191
        if value is None:
1381
1192
            self._config.set_option('', 'default_stack_on')
1382
1193
        else:
1390
1201
        This policy affects all branches contained by this bzrdir, except for
1391
1202
        those under repositories.
1392
1203
        """
1393
 
        if self._config is None:
1394
 
            return None
1395
1204
        value = self._config.get_option('default_stack_on')
1396
1205
        if value == '':
1397
1206
            value = None
1442
1251
            configobj.setdefault(section, {})[name] = value
1443
1252
        self._set_configobj(configobj)
1444
1253
 
1445
 
    def _get_config_file(self):
 
1254
    def _get_configobj(self):
1446
1255
        try:
1447
 
            return StringIO(self._transport.get_bytes(self._filename))
 
1256
            return ConfigObj(self._transport.get(self._filename),
 
1257
                             encoding='utf-8')
1448
1258
        except errors.NoSuchFile:
1449
 
            return StringIO()
1450
 
 
1451
 
    def _get_configobj(self):
1452
 
        f = self._get_config_file()
1453
 
        try:
1454
 
            return ConfigObj(f, encoding='utf-8')
1455
 
        finally:
1456
 
            f.close()
 
1259
            return ConfigObj(encoding='utf-8')
1457
1260
 
1458
1261
    def _set_configobj(self, configobj):
1459
1262
        out_file = StringIO()