~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/apps/config.py

  • Committer: Jelmer Vernooij
  • Date: 2011-09-08 10:30:14 UTC
  • mfrom: (456.1.1 trunk)
  • Revision ID: jelmer@samba.org-20110908103014-homoebi42li9eq62
Merge 1.18.1 release.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
"""A server that uses a loggerhead.conf file.
2
 
 
3
 
We recreate the branch discovery and url scheme of the old branchview
4
 
code.  It's all a bit horrible really.
5
 
"""
6
 
 
7
 
import logging
8
 
import os
9
 
import posixpath
10
 
 
11
 
import bzrlib.lru_cache
12
 
 
13
 
from bzrlib.util.configobj.configobj import ConfigObj
14
 
 
15
 
from paste.request import path_info_pop
16
 
from paste import httpexceptions
17
 
from paste.wsgiwrappers import WSGIResponse
18
 
 
19
 
from loggerhead.apps.branch import BranchWSGIApp
20
 
from loggerhead.apps import favicon_app, static_app, robots_app
21
 
from loggerhead.templatefunctions import templatefunctions
22
 
from loggerhead.zptsupport import load_template
23
 
from loggerhead import util
24
 
 
25
 
log = logging.getLogger("loggerhead.controllers")
26
 
 
27
 
from loggerhead.history import is_branch
28
 
 
29
 
 
30
 
class Project(object):
31
 
    """A project contains the branches.
32
 
 
33
 
    There is some complication because we don't want to hold on to the
34
 
    branches when they are not being browsed."""
35
 
 
36
 
    def __init__(self, name, config, root_config, graph_cache):
37
 
        self.name = name
38
 
        self.friendly_name = config.get('name', name)
39
 
        self.description = config.get('description', '')
40
 
        self.long_description = config.get('long_description', '')
41
 
        self._config = config
42
 
        self._root_config = root_config
43
 
        self.graph_cache = graph_cache
44
 
 
45
 
        self.view_names = []
46
 
        self.view_data_by_name = {}
47
 
        for view_name in config.sections:
48
 
            log.debug('Configuring (project %s) branch %s...', name, view_name)
49
 
            self._add_view(
50
 
                view_name, config[view_name], config[view_name].get('folder'))
51
 
 
52
 
        self._auto_folder = config.get('auto_publish_folder', None)
53
 
        self._auto_list = []
54
 
        if self._auto_folder is not None:
55
 
            self._recheck_auto_folders()
56
 
 
57
 
    def _recheck_auto_folders(self):
58
 
        if self._auto_folder is None:
59
 
            return
60
 
        auto_list = []
61
 
        # scan a folder for bazaar branches, and add them automatically
62
 
        for path, folders, filenames in os.walk(self._auto_folder):
63
 
            for folder in folders:
64
 
                folder = os.path.join(path, folder)
65
 
                if is_branch(folder):
66
 
                    auto_list.append(folder)
67
 
        auto_list.sort()
68
 
        if auto_list == self._auto_list:
69
 
            # nothing has changed; do nothing.
70
 
            return
71
 
 
72
 
        # rebuild views:
73
 
        self.view_names = []
74
 
        log.debug('Rescanning auto-folder for project %s ...', self.name)
75
 
        for folder in auto_list:
76
 
            view_name = os.path.basename(folder)
77
 
            log.debug('Auto-configuring (project %s) branch %s...',
78
 
                      self.name,
79
 
                      view_name)
80
 
            self._add_view(view_name, ConfigObj(), folder)
81
 
        self._auto_list = auto_list
82
 
 
83
 
    def _get_branch_url(self, view, view_config, folder):
84
 
        url = view_config.get('url', None)
85
 
        if url is not None:
86
 
            return url
87
 
        url = self._config.get('url_prefix', None)
88
 
        if url is not None:
89
 
            return posixpath.join(url, folder) + '/'
90
 
        return None
91
 
 
92
 
    def _get_description(self, view, view_config, history):
93
 
        description = view_config.get('description', None)
94
 
        if description is not None:
95
 
            return description
96
 
        description = history._branch.get_config().get_user_option(
97
 
                          'description')
98
 
        return description
99
 
 
100
 
    def _add_view(self, view_name, view_config, folder):
101
 
        b = bzrlib.branch.Branch.open(folder)
102
 
        view = BranchWSGIApp(b, view_name, view_config, self.graph_cache)
103
 
        b.lock_read()
104
 
        try:
105
 
            history = view.get_history()
106
 
            friendly_name = view_config.get('branch_name', None)
107
 
            if friendly_name is None:
108
 
                friendly_name = history.get_config().get_nickname()
109
 
                if friendly_name is None:
110
 
                    friendly_name = view_name
111
 
            branch_url = self._get_branch_url(view, view_config, view_name)
112
 
            description = self._get_description(view, view_config, history)
113
 
            self.view_data_by_name[view_name] = {
114
 
                'branch_path': folder,
115
 
                'config': view_config,
116
 
                'description': description,
117
 
                'friendly_name': friendly_name,
118
 
                'graph_cache': self.graph_cache,
119
 
                'name': view_name,
120
 
                'served_url': branch_url,
121
 
                }
122
 
            self.view_names.append(view_name)
123
 
        finally:
124
 
            b.unlock()
125
 
 
126
 
    def view_named(self, name):
127
 
        view_data = self.view_data_by_name.get(name)
128
 
        if view_data is None:
129
 
            return None
130
 
        view_data = view_data.copy()
131
 
        branch_path = view_data.pop('branch_path')
132
 
        description = view_data.pop('description')
133
 
        name = view_data.pop('name')
134
 
        b = bzrlib.branch.Branch.open(branch_path)
135
 
        b.lock_read()
136
 
        view = BranchWSGIApp(b, **view_data)
137
 
        view.description = description
138
 
        view.name = name
139
 
        return view
140
 
 
141
 
    def call(self, environ, start_response):
142
 
        segment = path_info_pop(environ)
143
 
        if not segment:
144
 
            raise httpexceptions.HTTPNotFound()
145
 
        else:
146
 
            view = self.view_named(segment)
147
 
            if view is None:
148
 
                raise httpexceptions.HTTPNotFound()
149
 
            try:
150
 
                return view.app(environ, start_response)
151
 
            finally:
152
 
                view.branch.unlock()
153
 
 
154
 
 
155
 
class Root(object):
156
 
    """The root of the server -- renders as the browse view,
157
 
    dispatches to Project above for each 'project'."""
158
 
 
159
 
    def __init__(self, config):
160
 
        self.projects = []
161
 
        self.config = config
162
 
        self.projects_by_name = {}
163
 
        graph_cache = bzrlib.lru_cache.LRUCache()
164
 
        for project_name in self.config.sections:
165
 
            project = Project(project_name, self.config[project_name],
166
 
                              self.config, graph_cache)
167
 
            self.projects.append(project)
168
 
            self.projects_by_name[project_name] = project
169
 
 
170
 
    def browse(self, response):
171
 
        # This is insanely complicated because we want to open and
172
 
        # lock all the branches, render the view and then unlock the
173
 
        # branches again.
174
 
        for p in self.projects:
175
 
            p._recheck_auto_folders()
176
 
 
177
 
        class branch(object):
178
 
 
179
 
            @staticmethod
180
 
            def static_url(path):
181
 
                return self._static_url_base + path
182
 
        views_by_project = {}
183
 
        all_views = []
184
 
        try:
185
 
            for p in self.projects:
186
 
                views_by_project[p] = []
187
 
                for vn in p.view_names:
188
 
                    v = p.view_named(vn)
189
 
                    all_views.append(v)
190
 
                    views_by_project[p].append(v)
191
 
            vals = {
192
 
                'projects': self.projects,
193
 
                'util': util,
194
 
                'title': self.config.get('title', None),
195
 
                'branch': branch,
196
 
                'views_by_project': views_by_project,
197
 
            }
198
 
            vals.update(templatefunctions)
199
 
            response.headers['Content-Type'] = 'text/html'
200
 
            template = load_template('loggerhead.templates.browse')
201
 
            template.expand_into(response, **vals)
202
 
        finally:
203
 
            for v in all_views:
204
 
                v.branch.unlock()
205
 
 
206
 
    def __call__(self, environ, start_response):
207
 
        self._static_url_base = environ['loggerhead.static.url'] = \
208
 
                                environ['SCRIPT_NAME']
209
 
        segment = path_info_pop(environ)
210
 
        if segment is None:
211
 
            raise httpexceptions.HTTPMovedPermanently.relative_redirect(
212
 
                environ['SCRIPT_NAME'] + '/', environ)
213
 
        elif segment == '':
214
 
            response = WSGIResponse()
215
 
            self.browse(response)
216
 
            return response(environ, start_response)
217
 
        elif segment == 'robots.txt':
218
 
            return robots_app(environ, start_response)
219
 
        elif segment == 'static':
220
 
            return static_app(environ, start_response)
221
 
        elif segment == 'favicon.ico':
222
 
            return favicon_app(environ, start_response)
223
 
        else:
224
 
            project = self.projects_by_name.get(segment)
225
 
            if project is None:
226
 
                raise httpexceptions.HTTPNotFound()
227
 
            return project.call(environ, start_response)