1
# Copyright (C) 2008-2011 Canonical Ltd.
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
1
"""The WSGI application for serving a Bazaar branch."""
23
6
import bzrlib.branch
25
from bzrlib.hooks import Hooks
26
7
import bzrlib.lru_cache
28
9
from paste import request
29
10
from paste import httpexceptions
31
12
from loggerhead.apps import static_app
13
from loggerhead.controllers.changelog_ui import ChangeLogUI
14
from loggerhead.controllers.inventory_ui import InventoryUI
32
15
from loggerhead.controllers.annotate_ui import AnnotateUI
33
from loggerhead.controllers.view_ui import ViewUI
16
from loggerhead.controllers.revision_ui import RevisionUI
34
17
from loggerhead.controllers.atom_ui import AtomUI
35
from loggerhead.controllers.changelog_ui import ChangeLogUI
36
from loggerhead.controllers.diff_ui import DiffUI
37
18
from loggerhead.controllers.download_ui import DownloadUI
38
from loggerhead.controllers.filediff_ui import FileDiffUI
39
from loggerhead.controllers.inventory_ui import InventoryUI
40
from loggerhead.controllers.revision_ui import RevisionUI
41
from loggerhead.controllers.revlog_ui import RevLogUI
42
19
from loggerhead.controllers.search_ui import SearchUI
43
20
from loggerhead.history import History
44
21
from loggerhead import util
49
24
class BranchWSGIApp(object):
51
26
def __init__(self, branch, friendly_name=None, config={},
52
graph_cache=None, branch_link=None, is_root=False,
53
served_url=_DEFAULT, use_cdn=False, private=False):
27
graph_cache=None, branch_link=None):
54
28
self.branch = branch
55
29
self._config = config
56
30
self.friendly_name = friendly_name
57
31
self.branch_link = branch_link # Currently only used in Launchpad
58
32
self.log = logging.getLogger('loggerhead.%s' % (friendly_name,))
59
33
if graph_cache is None:
60
graph_cache = bzrlib.lru_cache.LRUCache(10)
34
graph_cache = bzrlib.lru_cache.LRUCache()
61
35
self.graph_cache = graph_cache
62
self.is_root = is_root
63
self.served_url = served_url
64
self.use_cdn = use_cdn
65
self.private = private
67
def public_private_css(self):
73
37
def get_history(self):
74
revinfo_disk_cache = None
38
_history = History(self.branch, self.graph_cache)
75
39
cache_path = self._config.get('cachepath', None)
76
40
if cache_path is not None:
77
41
# Only import the cache if we're going to use it.
78
42
# This makes sqlite optional
80
from loggerhead.changecache import RevInfoDiskCache
44
from loggerhead.changecache import FileChangeCache
81
45
except ImportError:
82
46
self.log.debug("Couldn't load python-sqlite,"
83
47
" continuing without using a cache")
85
revinfo_disk_cache = RevInfoDiskCache(cache_path)
87
self.branch, self.graph_cache,
88
revinfo_disk_cache=revinfo_disk_cache, cache_key=self.friendly_name)
49
_history.use_file_cache(
50
FileChangeCache(_history, cache_path))
90
53
def url(self, *args, **kw):
91
54
if isinstance(args[0], list):
94
57
for k, v in kw.iteritems():
96
qs.append('%s=%s' % (k, urllib.quote(v)))
59
qs.append('%s=%s'%(k, urllib.quote(v)))
98
path_info = urllib.quote(
99
unicode('/'.join(args)).encode('utf-8'), safe='/~:')
101
path_info += '?' + qs
102
return self._url_base + path_info
104
def absolute_url(self, *args, **kw):
105
rel_url = self.url(*args, **kw)
106
return request.resolve_relative_url(rel_url, self._environ)
61
return request.construct_url(
62
self._environ, script_name=self._url_base,
63
path_info='/'.join(args),
108
66
def context_url(self, *args, **kw):
109
67
kw = util.get_context(**kw)
112
70
def static_url(self, path):
113
71
return self._static_url_base + path
115
def yui_url(self, path):
117
base = 'http://yui.yahooapis.com/3.0.0pr2/build/'
119
base = self.static_url('/static/javascript/yui/build/')
122
73
controllers_dict = {
123
'+filediff': FileDiffUI,
125
74
'annotate': AnnotateUI,
127
75
'changes': ChangeLogUI,
129
'download': DownloadUI,
130
76
'files': InventoryUI,
131
77
'revision': RevisionUI,
78
'download': DownloadUI,
132
80
'search': SearchUI,
136
83
def last_updated(self):
137
84
h = self.get_history()
138
change = h.get_changes([h.last_revid])[0]
85
change = h.get_changes([ h.last_revid ])[0]
139
86
return change.date
141
def public_branch_url(self):
142
89
return self.branch.get_config().get_user_option('public_branch')
144
def lookup_app(self, environ):
145
# Check again if the branch is blocked from being served, this is
146
# mostly for tests. It's already checked in apps/transport.py
147
if self.branch.get_config().get_user_option('http_serve') == 'False':
148
raise httpexceptions.HTTPNotFound()
91
def app(self, environ, start_response):
149
92
self._url_base = environ['SCRIPT_NAME']
150
93
self._static_url_base = environ.get('loggerhead.static.url')
151
94
if self._static_url_base is None:
152
95
self._static_url_base = self._url_base
153
96
self._environ = environ
154
if self.served_url is _DEFAULT:
155
public_branch = self.public_branch_url()
156
if public_branch is not None:
157
self.served_url = public_branch
159
# Loggerhead only supports serving .bzr/ on local branches, so
160
# we shouldn't suggest something that won't work.
162
util.local_path_from_url(self.branch.base)
163
self.served_url = self.url([])
164
except bzrlib.errors.InvalidURL:
165
self.served_url = None
166
for hook in self.hooks['controller']:
167
controller = hook(self, environ)
168
if controller is not None:
170
97
path = request.path_info_pop(environ)
172
99
raise httpexceptions.HTTPMovedPermanently(
173
self.absolute_url('/changes'))
100
self._url_base + '/changes')
174
101
if path == 'static':
176
elif path == '+json':
177
environ['loggerhead.as_json'] = True
178
path = request.path_info_pop(environ)
102
return static_app(environ, start_response)
179
103
cls = self.controllers_dict.get(path)
181
return cls(self, self.get_history)
182
raise httpexceptions.HTTPNotFound()
184
def app(self, environ, start_response):
105
raise httpexceptions.HTTPNotFound()
185
106
self.branch.lock_read()
188
c = self.lookup_app(environ)
189
return c(environ, start_response)
191
environ['exc_info'] = sys.exc_info()
192
environ['branch'] = self
108
c = cls(self, self.get_history())
109
return c(environ, start_response)
195
111
self.branch.unlock()
198
class BranchWSGIAppHooks(Hooks):
199
"""A dictionary mapping hook name to a list of callables for WSGI app branch hooks.
203
"""Create the default hooks.
205
Hooks.__init__(self, "bzrlib.plugins.loggerhead.apps.branch",
206
"BranchWSGIApp.hooks")
207
self.add_hook('controller',
208
"Invoked when looking for the controller to use for a "
209
"branch subpage. The api signature is (branch_app, environ)."
210
"If a hook can provide a controller, it should return one, "
211
"as a standard WSGI app. If it can't provide a controller, "
212
"it should return None", (1, 19))
215
BranchWSGIApp.hooks = BranchWSGIAppHooks()
added
removed
loggerhead/apps/branch.py
