1
1
#!/usr/bin/env python
3
# Copyright (C) 2008, 2009 Canonical Ltd
5
2
# This program is free software; you can redistribute it and/or modify
6
3
# it under the terms of the GNU General Public License as published by
7
4
# the Free Software Foundation; either version 2 of the License, or
25
22
from bzrlib.plugin import load_plugins
26
from bzrlib.transport import get_transport
28
24
from paste import httpserver
29
25
from paste.httpexceptions import HTTPExceptionHandler, HTTPInternalServerError
30
26
from paste.translogger import TransLogger
32
28
from loggerhead import __version__
33
from loggerhead.apps.transport import (
34
BranchesFromTransportRoot, UserBranchesFromTransportRoot)
29
from loggerhead.apps.filesystem import (
30
BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
35
31
from loggerhead.config import LoggerheadConfig
36
32
from loggerhead.util import Reloader
37
33
from loggerhead.apps.error import ErrorHandlerApp
40
def get_config_and_path(args):
41
37
config = LoggerheadConfig()
43
39
if config.get_option('show_version'):
48
44
config.print_help()
50
46
elif config.arg_count == 1:
51
base = config.get_arg(0)
55
if not config.get_option('allow_writes'):
56
base = 'readonly+' + base
61
def setup_logging(config):
47
path = config.get_arg(0)
51
if not os.path.isdir(path):
52
print "%s is not a directory" % path
55
if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
56
print "--trunk-dir is only valid with --user-dirs"
59
if config.get_option('reload'):
60
if Reloader.is_installed():
63
return Reloader.restart_with_reloader()
65
if config.get_option('user_dirs'):
66
if not config.get_option('trunk_dir'):
67
print "You didn't specify a directory for the trunk directories."
69
app = UserBranchesFromFileSystemRoot(path, config)
71
app = BranchesFromFileSystemRoot(path, config)
62
74
logging.basicConfig()
63
75
logging.getLogger('').setLevel(logging.DEBUG)
64
logger = logging.getLogger('loggerhead')
76
logger = getattr(app, 'log', logging.getLogger('loggerhead'))
65
77
if config.get_option('log_folder'):
66
78
logfile_path = os.path.join(
67
79
config.get_option('log_folder'), 'serve-branches.log')
73
85
logfile.setFormatter(formatter)
74
86
logfile.setLevel(logging.DEBUG)
75
87
logger.addHandler(logfile)
79
def make_app_for_config_and_path(config, base):
80
if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
81
print "--trunk-dir is only valid with --user-dirs"
84
if config.get_option('reload'):
85
if Reloader.is_installed():
88
return Reloader.restart_with_reloader()
90
if config.get_option('user_dirs'):
91
if not config.get_option('trunk_dir'):
92
print "You didn't specify a directory for the trunk directories."
94
app = UserBranchesFromTransportRoot(base, config)
96
app = BranchesFromTransportRoot(base, config)
89
if config.get_option('memory_profile'):
90
memprofile = logging.getLogger('loggerhead-memprofile')
91
memprofile.setLevel(logging.DEBUG)
92
memprofile.addHandler(logging.FileHandler('loggerhead-memprofile'))
94
# setup_logging() #end
96
app = TransLogger(app, logger=logger)
100
97
if config.get_option('profile'):
101
98
from loggerhead.middleware.profile import LSProfMiddleware
102
99
app = LSProfMiddleware(app)
103
100
if config.get_option('memory_profile'):
104
from dozer import Dozer
101
from loggerhead.middleware.profile import MemoryProfileMiddleware
102
app = MemoryProfileMiddleware(app)
107
104
if not config.get_option('user_prefix'):
132
129
app = HTTPExceptionHandler(app)
133
130
app = ErrorHandlerApp(app)
134
app = TransLogger(app, logger=logging.getLogger('loggerhead'))
142
config, path = get_config_and_path(args)
144
app = make_app_for_config_and_path(config, path)
146
132
if not config.get_option('user_port'):