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
32
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__,)
35
def command_line_parser():
36
parser = OptionParser("%prog [options] <path>")
42
parser.add_option("--user-dirs", action="store_true", dest="user_dirs",
43
help="Serve user directories as ~user.")
44
parser.add_option("--trunk-dir", metavar="DIR",
45
help="The directory that contains the trunk branches.")
46
parser.add_option("--port", dest="user_port",
47
help=("Port Loggerhead should listen on "
48
"(defaults to 8080)."))
49
parser.add_option("--host", dest="user_host",
50
help="Host Loggerhead should listen on.")
51
parser.add_option("--prefix", dest="user_prefix",
52
help="Specify host prefix.")
53
parser.add_option("--profile", action="store_true", dest="profile",
54
help="Generate callgrind profile data to "
55
"%d-stats.callgrind on each request.")
56
parser.add_option("--reload", action="store_true", dest="reload",
57
help="Restarts the application when changing python"
58
" files. Only used for development purposes.")
59
parser.add_option('--log-folder', dest="log_folder",
60
type=str, help="The directory to place log files in.")
61
parser.add_option("--version", action="store_true", dest="show_version",
62
help="Print the software version and exit")
67
parser = command_line_parser()
68
(options, args) = parser.parse_args(sys.argv[1:])
70
if options.show_version:
71
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'):
82
if not os.path.isdir(path):
83
print "%s is not a directory" % path
86
if options.trunk_dir and not options.user_dirs:
97
87
print "--trunk-dir is only valid with --user-dirs"
100
if config.get_option('reload'):
101
91
if Reloader.is_installed():
102
92
Reloader.install()
104
94
return Reloader.restart_with_reloader()
106
if config.get_option('user_dirs'):
107
if not config.get_option('trunk_dir'):
97
if not options.trunk_dir:
108
98
print "You didn't specify a directory for the trunk directories."
110
app = UserBranchesFromTransportRoot(base, config)
112
app = BranchesFromTransportRoot(base, config)
114
setup_logging(config)
116
if config.get_option('profile'):
100
app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
102
app = BranchesFromFileSystemRoot(path)
105
logging.basicConfig()
106
logging.getLogger('').setLevel(logging.DEBUG)
107
logger = getattr(app, 'log', logging.getLogger('loggerhead'))
108
if options.log_folder:
109
logfile_path = os.path.join(options.log_folder, 'serve-branches.log')
111
logfile_path = 'serve-branches.log'
112
logfile = logging.FileHandler(logfile_path, 'a')
113
formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
115
logfile.setFormatter(formatter)
116
logfile.setLevel(logging.DEBUG)
117
logger.addHandler(logfile)
118
# setup_logging() #end
119
app = ErrorHandlerApp(app)
120
app = HTTPExceptionHandler(app)
121
app = TransLogger(app, logger=logger)
117
123
from loggerhead.middleware.profile import LSProfMiddleware
118
124
app = LSProfMiddleware(app)
119
if config.get_option('memory_profile'):
120
from dozer import Dozer
123
if not config.get_option('user_prefix'):
126
if not options.user_prefix:
126
prefix = config.get_option('user_prefix')
127
if not prefix.startswith('/'):
128
prefix = '/' + prefix
129
prefix = options.user_prefix
131
132
from paste.deploy.config import PrefixMiddleware
132
133
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
136
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'):
138
if not options.user_port:
165
port = config.get_option('user_port')
141
port = options.user_port
167
if not config.get_option('user_host'):
143
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()
146
host = options.user_host
148
httpserver.serve(app, host=host, port=port)
151
if __name__ == "__main__":
added
removed
serve-branches
