← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/apps/filesystem.py

  • Committer: Michael Hudson
  • Date: 2008-09-30 03:05:31 UTC
  • mfrom: (226.1.3 fix-breadcrumbs)
  • Revision ID: michael.hudson@canonical.com-20080930030531-at8220tnkx9pvqcw
small fixes to the breadcrumbs in the --prefix case.
(thanks Matt Nordhoff for the prods)

Show diffs side-by-side

added added

removed removed

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