← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/smart/bzrdir.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2009-04-30 15:00:23 UTC
  • mfrom: (4273.1.21 branch-subtree-locations)
  • Revision ID: pqm@pqm.ubuntu.com-20090430150023-1cw4lwqf312vpuu8
(abentley) Implement references command.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/smart/bzrdir.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
17
17
"""Server-side bzrdir related request implmentations."""
18
18
 
19
19
 
20
 
from bzrlib import branch, errors, repository, urlutils
 
20
from bzrlib import branch, errors, repository
21
21
from bzrlib.bzrdir import (
22
22
    BzrDir,
23
23
    BzrDirFormat,
24
24
    BzrDirMetaFormat1,
25
 
    BzrProber,
26
 
    )
27
 
from bzrlib.controldir import (
28
25
    network_format_registry,
29
26
    )
30
27
from bzrlib.smart.request import (
37
34
class SmartServerRequestOpenBzrDir(SmartServerRequest):
38
35
 
39
36
    def do(self, path):
 
37
        from bzrlib.bzrdir import BzrDirFormat
40
38
        try:
41
39
            t = self.transport_from_client_path(path)
42
40
        except errors.PathNotChild:
47
45
            # clients that don't anticipate errors from this method.
48
46
            answer = 'no'
49
47
        else:
50
 
            bzr_prober = BzrProber()
 
48
            default_format = BzrDirFormat.get_default_format()
 
49
            real_bzrdir = default_format.open(t, _found=True)
51
50
            try:
52
 
                bzr_prober.probe_transport(t)
 
51
                real_bzrdir._format.probe_transport(t)
53
52
            except (errors.NotBranchError, errors.UnknownFormatError):
54
53
                answer = 'no'
55
54
            else:
57
56
        return SuccessfulSmartServerResponse((answer,))
58
57
 
59
58
 
60
 
class SmartServerRequestOpenBzrDir_2_1(SmartServerRequest):
61
 
 
62
 
    def do(self, path):
63
 
        """Is there a BzrDir present, and if so does it have a working tree?
64
 
 
65
 
        New in 2.1.
66
 
        """
67
 
        try:
68
 
            t = self.transport_from_client_path(path)
69
 
        except errors.PathNotChild:
70
 
            # The client is trying to ask about a path that they have no access
71
 
            # to.
72
 
            return SuccessfulSmartServerResponse(('no',))
73
 
        try:
74
 
            bd = BzrDir.open_from_transport(t)
75
 
        except errors.NotBranchError:
76
 
            answer = ('no',)
77
 
        else:
78
 
            answer = ('yes',)
79
 
            if bd.has_workingtree():
80
 
                answer += ('yes',)
81
 
            else:
82
 
                answer += ('no',)
83
 
        return SuccessfulSmartServerResponse(answer)
84
 
 
85
 
 
86
59
class SmartServerRequestBzrDir(SmartServerRequest):
87
60
 
88
61
    def do(self, path, *args):
90
63
        try:
91
64
            self._bzrdir = BzrDir.open_from_transport(
92
65
                self.transport_from_client_path(path))
93
 
        except errors.NotBranchError, e:
94
 
            return FailedSmartServerResponse(('nobranch',))
 
66
        except errors.NotBranchError:
 
67
            return FailedSmartServerResponse(('nobranch', ))
95
68
        return self.do_bzrdir_request(*args)
96
69
 
97
70
    def _boolean_to_yes_no(self, a_boolean):
112
85
        """Get the relative path for repository from current_transport."""
113
86
        # the relpath of the bzrdir in the found repository gives us the
114
87
        # path segments to pop-out.
115
 
        relpath = repository.user_transport.relpath(
 
88
        relpath = repository.bzrdir.root_transport.relpath(
116
89
            current_transport.base)
117
90
        if len(relpath):
118
91
            segments = ['..'] * len(relpath.split('/'))
381
354
    def do(self, bzrdir_network_name, path, use_existing_dir, create_prefix,
382
355
        force_new_repo, stacked_on, stack_on_pwd, repo_format_name,
383
356
        make_working_trees, shared_repo):
384
 
        """Initialize a bzrdir at path as per
385
 
        BzrDirFormat.initialize_on_transport_ex.
386
 
 
387
 
        New in 1.16.  (Replaces BzrDirFormat.initialize_ex verb from 1.15).
388
 
 
389
 
        :return: return SuccessfulSmartServerResponse((repo_path, rich_root,
390
 
            tree_ref, external_lookup, repo_network_name,
391
 
            repo_bzrdir_network_name, bzrdir_format_network_name,
392
 
            NoneTrueFalse(stacking), final_stack, final_stack_pwd,
393
 
            repo_lock_token))
 
357
        """Initialize a bzrdir at path as per BzrDirFormat.initialize_ex
 
358
 
 
359
        :return: SmartServerResponse()
394
360
        """
395
361
        target_transport = self.transport_from_client_path(path)
396
362
        format = network_format_registry.get(bzrdir_network_name)
417
383
            repo_bzrdir_name = ''
418
384
            final_stack = None
419
385
            final_stack_pwd = None
420
 
            repo_lock_token = ''
421
386
        else:
422
387
            repo_path = self._repo_relpath(bzrdir.root_transport, repo)
423
388
            if repo_path == '':
428
393
            repo_bzrdir_name = repo.bzrdir._format.network_name()
429
394
            final_stack = repository_policy._stack_on
430
395
            final_stack_pwd = repository_policy._stack_on_pwd
431
 
            # It is returned locked, but we need to do the lock to get the lock
432
 
            # token.
433
 
            repo.unlock()
434
 
            repo_lock_token = repo.lock_write().repository_token or ''
435
 
            if repo_lock_token:
436
 
                repo.leave_lock_in_place()
437
 
            repo.unlock()
438
396
        final_stack = final_stack or ''
439
397
        final_stack_pwd = final_stack_pwd or ''
440
 
 
441
 
        # We want this to be relative to the bzrdir.
442
 
        if final_stack_pwd:
443
 
            final_stack_pwd = urlutils.relative_url(
444
 
                target_transport.base, final_stack_pwd)
445
 
 
446
 
        # Can't meaningfully return a root path.
447
 
        if final_stack.startswith('/'):
448
 
            client_path = self._root_client_path + final_stack[1:]
449
 
            final_stack = urlutils.relative_url(
450
 
                self._root_client_path, client_path)
451
 
            final_stack_pwd = '.'
452
 
 
453
398
        return SuccessfulSmartServerResponse((repo_path, rich_root, tree_ref,
454
399
            external_lookup, repo_name, repo_bzrdir_name,
455
400
            bzrdir._format.network_name(),
456
401
            self._serialize_NoneTrueFalse(stacking), final_stack,
457
 
            final_stack_pwd, repo_lock_token))
 
402
            final_stack_pwd))
458
403
 
459
404
 
460
405
class SmartServerRequestOpenBranch(SmartServerRequestBzrDir):
467
412
                return SuccessfulSmartServerResponse(('ok', ''))
468
413
            else:
469
414
                return SuccessfulSmartServerResponse(('ok', reference_url))
470
 
        except errors.NotBranchError, e:
471
 
            return FailedSmartServerResponse(('nobranch',))
 
415
        except errors.NotBranchError:
 
416
            return FailedSmartServerResponse(('nobranch', ))
472
417
 
473
418
 
474
419
class SmartServerRequestOpenBranchV2(SmartServerRequestBzrDir):
483
428
                return SuccessfulSmartServerResponse(('branch', format))
484
429
            else:
485
430
                return SuccessfulSmartServerResponse(('ref', reference_url))
486
 
        except errors.NotBranchError, e:
487
 
            return FailedSmartServerResponse(('nobranch',))
488
 
 
489
 
 
490
 
class SmartServerRequestOpenBranchV3(SmartServerRequestBzrDir):
491
 
 
492
 
    def do_bzrdir_request(self):
493
 
        """Open a branch at path and return the reference or format.
494
 
        
495
 
        This version introduced in 2.1.
496
 
 
497
 
        Differences to SmartServerRequestOpenBranchV2:
498
 
          * can return 2-element ('nobranch', extra), where 'extra' is a string
499
 
            with an explanation like 'location is a repository'.  Previously
500
 
            a 'nobranch' response would never have more than one element.
501
 
        """
502
 
        try:
503
 
            reference_url = self._bzrdir.get_branch_reference()
504
 
            if reference_url is None:
505
 
                br = self._bzrdir.open_branch(ignore_fallbacks=True)
506
 
                format = br._format.network_name()
507
 
                return SuccessfulSmartServerResponse(('branch', format))
508
 
            else:
509
 
                return SuccessfulSmartServerResponse(('ref', reference_url))
510
 
        except errors.NotBranchError, e:
511
 
            # Stringify the exception so that its .detail attribute will be
512
 
            # filled out.
513
 
            str(e)
514
 
            resp = ('nobranch',)
515
 
            detail = e.detail
516
 
            if detail:
517
 
                if detail.startswith(': '):
518
 
                    detail = detail[2:]
519
 
                resp += (detail,)
520
 
            return FailedSmartServerResponse(resp)
521
 
 
 
431
        except errors.NotBranchError:
 
432
            return FailedSmartServerResponse(('nobranch', ))