22
from optparse import OptionParser
22
from bzrlib.plugin import load_plugins
24
24
from paste import httpserver
25
from paste.httpexceptions import HTTPExceptionHandler
25
from paste.httpexceptions import HTTPExceptionHandler, HTTPInternalServerError
26
26
from paste.translogger import TransLogger
28
28
from loggerhead import __version__
29
29
from loggerhead.apps.filesystem import (
30
30
BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
33
def command_line_parser():
34
parser = OptionParser("%prog [options] <path>")
39
parser.add_option("--user-dirs", action="store_true", dest="user_dirs",
40
help="Serve user directories as ~user.")
41
parser.add_option("--trunk-dir", metavar="DIR",
42
help="The directory that contains the trunk branches.")
43
parser.add_option("--version", action="store_true", dest="show_version",
44
help="Print the software version and exit")
31
from loggerhead.config import LoggerheadConfig
32
from loggerhead.util import Reloader
33
from loggerhead.apps.error import ErrorHandlerApp
50
logging.getLogger().setLevel(logging.DEBUG)
52
parser = command_line_parser()
53
(options, args) = parser.parse_args(sys.argv[1:])
55
if options.show_version:
37
config = LoggerheadConfig()
39
if config.get_option('show_version'):
56
40
print "loggerhead %s" % __version__
43
if config.arg_count > 1:
46
elif config.arg_count == 1:
47
path = config.get_arg(0)
68
52
print "%s is not a directory" % path
71
if options.trunk_dir and not options.user_dirs:
55
if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
72
56
print "--trunk-dir is only valid with --user-dirs"
76
if not options.trunk_dir:
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'):
77
67
print "You didn't specify a directory for the trunk directories."
79
app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
81
app = BranchesFromFileSystemRoot(path)
82
app = HTTPExceptionHandler(app)
83
app = TransLogger(app)
69
app = UserBranchesFromFileSystemRoot(path, config)
71
app = BranchesFromFileSystemRoot(path, config)
75
logging.getLogger('').setLevel(logging.DEBUG)
76
logger = getattr(app, 'log', logging.getLogger('loggerhead'))
77
if config.get_option('log_folder'):
78
logfile_path = os.path.join(
79
config.get_option('log_folder'), 'serve-branches.log')
81
logfile_path = 'serve-branches.log'
82
logfile = logging.FileHandler(logfile_path, 'a')
83
formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
85
logfile.setFormatter(formatter)
86
logfile.setLevel(logging.DEBUG)
87
logger.addHandler(logfile)
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)
97
if config.get_option('profile'):
98
from loggerhead.middleware.profile import LSProfMiddleware
99
app = LSProfMiddleware(app)
100
if config.get_option('memory_profile'):
101
from loggerhead.middleware.profile import MemoryProfileMiddleware
102
app = MemoryProfileMiddleware(app)
104
if not config.get_option('user_prefix'):
107
prefix = config.get_option('user_prefix')
108
if not prefix.startswith('/'):
109
prefix = '/' + prefix
86
112
from paste.deploy.config import PrefixMiddleware
87
113
except ImportError:
90
app = PrefixMiddleware(app)
92
httpserver.serve(app, host='0.0.0.0', port='8080')
114
cant_proxy_correctly_message = (
115
'Unsupported configuration: PasteDeploy not available, but '
116
'loggerhead appears to be behind a proxy.')
117
def check_not_proxied(app):
118
def wrapped(environ, start_response):
119
if 'HTTP_X_FORWARDED_SERVER' in environ:
120
exc = HTTPInternalServerError()
121
exc.explanation = cant_proxy_correctly_message
123
return app(environ, start_response)
125
app = check_not_proxied(app)
127
app = PrefixMiddleware(app, prefix=prefix)
129
app = HTTPExceptionHandler(app)
130
app = ErrorHandlerApp(app)
132
if not config.get_option('user_port'):
135
port = config.get_option('user_port')
137
if not config.get_option('user_host'):
140
host = config.get_option('user_host')
144
httpserver.serve(app, host=host, port=port)
95
147
if __name__ == "__main__":