1
"""Serve branches at urls that mimic the file system layout."""
6
from bzrlib import branch, errors, lru_cache
8
from paste.request import path_info_pop
9
from paste import httpexceptions
11
from loggerhead.apps.branch import BranchWSGIApp
12
from loggerhead.apps import favicon_app, static_app
13
from loggerhead.controllers.directory_ui import DirectoryUI
15
sql_dir = tempfile.mkdtemp(prefix='loggerhead-cache-')
18
class BranchesFromFileSystemServer(object):
19
def __init__(self, path, root, name=None):
24
def app_for_branch(self, branch):
29
branch_app = BranchWSGIApp(
30
branch, name, {'cachepath': sql_dir}, self.root.graph_cache)
33
def app_for_non_branch(self, environ):
34
segment = path_info_pop(environ)
36
raise httpexceptions.HTTPMovedPermanently(
37
environ['SCRIPT_NAME'] + '/')
43
return DirectoryUI(environ['loggerhead.static.url'], self.path, name)
45
new_path = os.path.join(self.path, segment)
47
new_name = os.path.join(self.name, segment)
49
new_name = '/' + segment
50
return BranchesFromFileSystemServer(new_path, self.root, new_name)
52
def __call__(self, environ, start_response):
53
if not os.path.isdir(self.path):
54
raise httpexceptions.HTTPNotFound()
56
b = branch.Branch.open(self.path)
57
except errors.NotBranchError:
58
return self.app_for_non_branch(environ)(environ, start_response)
60
return self.app_for_branch(b)(environ, start_response)
63
class BranchesFromFileSystemRoot(object):
65
def __init__(self, folder):
66
self.graph_cache = lru_cache.LRUCache()
69
def __call__(self, environ, start_response):
70
environ['loggerhead.static.url'] = environ['SCRIPT_NAME']
71
if environ['PATH_INFO'].startswith('/static/'):
72
segment = path_info_pop(environ)
73
assert segment == 'static'
74
return static_app(environ, start_response)
75
elif environ['PATH_INFO'] == '/favicon.ico':
76
return favicon_app(environ, start_response)
78
return BranchesFromFileSystemServer(
79
self.folder, self)(environ, start_response)
82
class UserBranchesFromFileSystemRoot(object):
84
def __init__(self, folder, trunk_dir):
85
self.graph_cache = lru_cache.LRUCache()
87
self.trunk_dir = trunk_dir
89
def __call__(self, environ, start_response):
90
environ['loggerhead.static.url'] = environ['SCRIPT_NAME']
91
path_info= environ['PATH_INFO']
92
if path_info.startswith('/static/'):
93
segment = path_info_pop(environ)
94
assert segment == 'static'
95
return static_app(environ, start_response)
96
elif path_info == '/favicon.ico':
97
return favicon_app(environ, start_response)
99
# segments starting with ~ are user branches
100
if path_info.startswith('/~'):
101
segment = path_info_pop(environ)
102
new_path = os.path.join(self.folder, segment[1:])
103
return BranchesFromFileSystemServer(
104
new_path, self, segment)(environ, start_response)
106
new_path = os.path.join(self.folder, self.trunk_dir)
107
return BranchesFromFileSystemServer(
108
new_path, self)(environ, start_response)