← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/apps/transport.py

this shouldn't have been commented out

Show diffs side-by-side

added added

removed removed

Lines of Context:
loggerhead/apps/transport.py
1
 
"""Serve branches at urls that mimic a transport's file system layout."""
2
 
 
3
 
from bzrlib import branch, errors, lru_cache, urlutils
4
 
 
5
 
from paste.request import path_info_pop
6
 
from paste import httpexceptions
7
 
from paste import urlparser
8
 
 
9
 
from loggerhead.apps.branch import BranchWSGIApp
10
 
from loggerhead.apps import favicon_app, static_app
11
 
from loggerhead.config import LoggerheadConfig
12
 
from loggerhead.controllers.directory_ui import DirectoryUI
13
 
 
14
 
 
15
 
class BranchesFromTransportServer(object):
16
 
 
17
 
    def __init__(self, transport, root, name=None):
18
 
        self.transport = transport
19
 
        self.root = root
20
 
        self.name = name
21
 
        self._config = root._config
22
 
 
23
 
    def app_for_branch(self, branch):
24
 
        if not self.name:
25
 
            name = branch._get_nick(local=True)
26
 
            is_root = True
27
 
        else:
28
 
            name = self.name
29
 
            is_root = False
30
 
        branch_app = BranchWSGIApp(
31
 
            branch, name,
32
 
            {'cachepath': self._config.SQL_DIR},
33
 
            self.root.graph_cache, is_root=is_root,
34
 
            use_cdn=self._config.get_option('use_cdn'))
35
 
        return branch_app.app
36
 
 
37
 
    def app_for_non_branch(self, environ):
38
 
        segment = path_info_pop(environ)
39
 
        if segment is None:
40
 
            raise httpexceptions.HTTPMovedPermanently(
41
 
                environ['SCRIPT_NAME'] + '/')
42
 
        elif segment == '':
43
 
            if self.name:
44
 
                name = self.name
45
 
            else:
46
 
                name = '/'
47
 
            return DirectoryUI(environ['loggerhead.static.url'],
48
 
                               self.transport,
49
 
                               name)
50
 
        else:
51
 
            new_transport = self.transport.clone(segment)
52
 
            if self.name:
53
 
                new_name = urlutils.join(self.name, segment)
54
 
            else:
55
 
                new_name = '/' + segment
56
 
            return BranchesFromTransportServer(new_transport, self.root, new_name)
57
 
 
58
 
    def __call__(self, environ, start_response):
59
 
        try:
60
 
            b = branch.Branch.open_from_transport(self.transport)
61
 
        except errors.NotBranchError:
62
 
            if not self.transport.listable() or not self.transport.has('.'):
63
 
                raise httpexceptions.HTTPNotFound()
64
 
            return self.app_for_non_branch(environ)(environ, start_response)
65
 
        else:
66
 
            return self.app_for_branch(b)(environ, start_response)
67
 
 
68
 
 
69
 
class BranchesFromTransportRoot(object):
70
 
 
71
 
    def __init__(self, transport, config):
72
 
        self.graph_cache = lru_cache.LRUCache(10)
73
 
        self.transport = transport
74
 
        self._config = config
75
 
 
76
 
    def __call__(self, environ, start_response):
77
 
        environ['loggerhead.static.url'] = environ['SCRIPT_NAME']
78
 
        if environ['PATH_INFO'].startswith('/static/'):
79
 
            segment = path_info_pop(environ)
80
 
            assert segment == 'static'
81
 
            return static_app(environ, start_response)
82
 
        elif environ['PATH_INFO'] == '/favicon.ico':
83
 
            return favicon_app(environ, start_response)
84
 
        elif '/.bzr/' in environ['PATH_INFO']:
85
 
            app = urlparser.make_static(None, self.transport)
86
 
            return app(environ, start_response)
87
 
        else:
88
 
            return BranchesFromTransportServer(
89
 
                self.transport, self)(environ, start_response)
90
 
 
91
 
 
92
 
class UserBranchesFromTransportRoot(object):
93
 
 
94
 
    def __init__(self, transport, config):
95
 
        self.graph_cache = lru_cache.LRUCache(10)
96
 
        self.transport = transport
97
 
        self._config = config
98
 
        self.trunk_dir = config.get_option('trunk_dir')
99
 
 
100
 
    def __call__(self, environ, start_response):
101
 
        environ['loggerhead.static.url'] = environ['SCRIPT_NAME']
102
 
        path_info = environ['PATH_INFO']
103
 
        if path_info.startswith('/static/'):
104
 
            segment = path_info_pop(environ)
105
 
            assert segment == 'static'
106
 
            return static_app(environ, start_response)
107
 
        elif path_info == '/favicon.ico':
108
 
            return favicon_app(environ, start_response)
109
 
        else:
110
 
            # segments starting with ~ are user branches
111
 
            if path_info.startswith('/~'):
112
 
                segment = path_info_pop(environ)
113
 
                new_transport = self.transport.clone(segment[1:])
114
 
                return BranchesFromTransportServer(
115
 
                    new_transport, self, segment)(environ, start_response)
116
 
            else:
117
 
                new_transport = self.transport.clone(self.trunk_dir)
118
 
                return BranchesFromTransportServer(
119
 
                    new_transport, self)(environ, start_response)