24
from bzrlib.plugin import load_plugins
22
from optparse import OptionParser
26
24
from paste import httpserver
27
from paste.httpexceptions import HTTPExceptionHandler, HTTPInternalServerError
25
from paste.httpexceptions import HTTPExceptionHandler
28
26
from paste.translogger import TransLogger
30
28
from loggerhead import __version__
31
from loggerhead.apps.transport import (
32
BranchesFromTransportRoot, UserBranchesFromTransportRoot)
33
from loggerhead.config import LoggerheadConfig
29
from loggerhead.apps.filesystem import (
30
BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
34
31
from loggerhead.util import Reloader
35
from loggerhead.apps.error import ErrorHandlerApp
38
def get_config_and_path(args):
39
config = LoggerheadConfig(args)
41
if config.get_option('show_version'):
42
print "loggerhead %s" % (__version__,)
34
def command_line_parser():
35
parser = OptionParser("%prog [options] <path>")
40
parser.add_option("--user-dirs", action="store_true", dest="user_dirs",
41
help="Serve user directories as ~user.")
42
parser.add_option("--trunk-dir", metavar="DIR",
43
help="The directory that contains the trunk branches.")
44
parser.add_option("--port", dest="user_port",
45
help="Port Loggerhead should listen on (defaults to 8080).")
46
parser.add_option("--host", dest="user_host",
47
help="Host Loggerhead should listen on.")
48
parser.add_option("--prefix", dest="user_prefix",
49
help="Specify host prefix.")
50
parser.add_option("--reload", action="store_true", dest="reload",
51
help="Restarts the application when changing python"
52
" files. Only used for development purposes.")
53
parser.add_option("--version", action="store_true", dest="show_version",
54
help="Print the software version and exit")
60
logging.getLogger().setLevel(logging.DEBUG)
62
parser = command_line_parser()
63
(options, args) = parser.parse_args(sys.argv[1:])
65
if options.show_version:
66
print "loggerhead %s" % __version__
45
if config.arg_count > 1:
48
elif config.arg_count == 1:
49
base = config.get_arg(0)
53
if not config.get_option('allow_writes'):
54
base = 'readonly+' + base
59
def setup_logging(config, init_logging=True, log_file=None):
60
log_level = config.get_log_level()
63
if log_level is not None:
64
logging.getLogger('').setLevel(log_level)
65
logger = logging.getLogger('loggerhead')
66
if log_level is not None:
67
logger.setLevel(log_level)
68
if log_file is not None:
69
handler = logging.StreamHandler(log_file)
71
if config.get_option('log_folder'):
72
logfile_path = os.path.join(
73
config.get_option('log_folder'), 'serve-branches.log')
75
logfile_path = 'serve-branches.log'
76
handler = logging.FileHandler(logfile_path, 'a')
77
formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
79
handler.setFormatter(formatter)
80
# We set the handler to accept all messages, the *logger* won't emit them
81
# if it is configured to suppress it
82
handler.setLevel(logging.DEBUG)
83
logger.addHandler(handler)
84
def _restrict_logging(logger_name):
85
logger = logging.getLogger(logger_name)
86
if logger.getEffectiveLevel() < logging.INFO:
87
logger.setLevel(logging.INFO)
88
# simpleTAL is *very* verbose in DEBUG mode, which is otherwise the
89
# default. So quiet it up a bit.
90
_restrict_logging('simpleTAL')
91
_restrict_logging('simpleTALES')
95
def make_app_for_config_and_path(config, base):
96
if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
77
if not os.path.isdir(path):
78
print "%s is not a directory" % path
81
if options.trunk_dir and not options.user_dirs:
97
82
print "--trunk-dir is only valid with --user-dirs"
100
if config.get_option('reload'):
101
86
if Reloader.is_installed():
102
87
Reloader.install()
104
89
return Reloader.restart_with_reloader()
106
if config.get_option('user_dirs'):
107
if not config.get_option('trunk_dir'):
92
if not options.trunk_dir:
108
93
print "You didn't specify a directory for the trunk directories."
110
app = UserBranchesFromTransportRoot(base, config)
95
app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
112
app = BranchesFromTransportRoot(base, config)
114
setup_logging(config)
116
if config.get_option('profile'):
117
from loggerhead.middleware.profile import LSProfMiddleware
118
app = LSProfMiddleware(app)
119
if config.get_option('memory_profile'):
120
from dozer import Dozer
123
if not config.get_option('user_prefix'):
97
app = BranchesFromFileSystemRoot(path)
98
app = HTTPExceptionHandler(app)
99
app = TransLogger(app)
101
if not options.user_prefix:
126
prefix = config.get_option('user_prefix')
127
if not prefix.startswith('/'):
128
prefix = '/' + prefix
104
prefix = options.user_prefix
131
107
from paste.deploy.config import PrefixMiddleware
132
108
except ImportError:
133
cant_proxy_correctly_message = (
134
'Unsupported configuration: PasteDeploy not available, but '
135
'loggerhead appears to be behind a proxy.')
136
def check_not_proxied(app):
137
def wrapped(environ, start_response):
138
if 'HTTP_X_FORWARDED_SERVER' in environ:
139
exc = HTTPInternalServerError()
140
exc.explanation = cant_proxy_correctly_message
142
return app(environ, start_response)
144
app = check_not_proxied(app)
146
111
app = PrefixMiddleware(app, prefix=prefix)
148
app = HTTPExceptionHandler(app)
149
app = ErrorHandlerApp(app)
150
app = TransLogger(app, logger=logging.getLogger('loggerhead'))
158
config, path = get_config_and_path(args)
160
app = make_app_for_config_and_path(config, path)
162
if not config.get_option('user_port'):
113
if not options.user_port:
165
port = config.get_option('user_port')
116
port = options.user_port
167
if not config.get_option('user_host'):
118
if not options.user_host:
170
host = config.get_option('user_host')
172
if not config.get_option('protocol'):
175
protocol = config.get_option('protocol')
177
if protocol == 'http':
178
httpserver.serve(app, host=host, port=port)
180
if protocol == 'fcgi':
181
from flup.server.fcgi import WSGIServer
182
elif protocol == 'scgi':
183
from flup.server.scgi import WSGIServer
184
elif protocol == 'ajp':
185
from flup.server.ajp import WSGIServer
187
print 'Unknown protocol: %s.' % (protocol)
189
WSGIServer(app, bindAddress=(host, int(port))).run()
121
host = options.user_host
123
httpserver.serve(app, host=host, port=port)
126
if __name__ == "__main__":