22
from bzrlib.plugin import load_plugins
22
from optparse import OptionParser
24
24
from paste import httpserver
25
from paste.httpexceptions import HTTPExceptionHandler, HTTPInternalServerError
25
from paste.httpexceptions import HTTPExceptionHandler
26
26
from paste.translogger import TransLogger
28
28
from loggerhead import __version__
29
29
from loggerhead.apps.filesystem import (
30
30
BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
31
from loggerhead.config import LoggerheadConfig
32
31
from loggerhead.util import Reloader
33
32
from loggerhead.apps.error import ErrorHandlerApp
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("--reload", action="store_true", dest="reload",
54
help="Restarts the application when changing python"
55
" files. Only used for development purposes.")
56
parser.add_option('--log-folder', dest="log_folder",
57
type=str, help="The directory to place log files in.")
58
parser.add_option("--version", action="store_true", dest="show_version",
59
help="Print the software version and exit")
37
config = LoggerheadConfig()
64
parser = command_line_parser()
65
(options, args) = parser.parse_args(sys.argv[1:])
39
if config.get_option('show_version'):
67
if options.show_version:
40
68
print "loggerhead %s" % __version__
43
if config.arg_count > 1:
46
elif config.arg_count == 1:
47
path = config.get_arg(0)
52
80
print "%s is not a directory" % path
55
if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
83
if options.trunk_dir and not options.user_dirs:
56
84
print "--trunk-dir is only valid with --user-dirs"
59
if config.get_option('reload'):
60
88
if Reloader.is_installed():
63
91
return Reloader.restart_with_reloader()
65
if config.get_option('user_dirs'):
66
if not config.get_option('trunk_dir'):
94
if not options.trunk_dir:
67
95
print "You didn't specify a directory for the trunk directories."
69
app = UserBranchesFromFileSystemRoot(path, config)
97
app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
71
app = BranchesFromFileSystemRoot(path, config)
99
app = BranchesFromFileSystemRoot(path)
74
102
logging.basicConfig()
75
103
logging.getLogger('').setLevel(logging.DEBUG)
76
104
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')
105
if options.log_folder:
106
logfile_path = os.path.join(options.log_folder, 'serve-branches.log')
81
108
logfile_path = 'serve-branches.log'
82
109
logfile = logging.FileHandler(logfile_path, 'a')
85
112
logfile.setFormatter(formatter)
86
113
logfile.setLevel(logging.DEBUG)
87
114
logger.addHandler(logfile)
89
115
# setup_logging() #end
116
app = ErrorHandlerApp(app)
117
app = HTTPExceptionHandler(app)
91
118
app = TransLogger(app, logger=logger)
92
if config.get_option('profile'):
93
from loggerhead.middleware.profile import LSProfMiddleware
94
app = LSProfMiddleware(app)
95
if config.get_option('memory_profile'):
96
from dozer import Dozer
99
if not config.get_option('user_prefix'):
120
if not options.user_prefix:
102
prefix = config.get_option('user_prefix')
103
if not prefix.startswith('/'):
104
prefix = '/' + prefix
123
prefix = options.user_prefix
107
126
from paste.deploy.config import PrefixMiddleware
108
127
except ImportError:
109
cant_proxy_correctly_message = (
110
'Unsupported configuration: PasteDeploy not available, but '
111
'loggerhead appears to be behind a proxy.')
112
def check_not_proxied(app):
113
def wrapped(environ, start_response):
114
if 'HTTP_X_FORWARDED_SERVER' in environ:
115
exc = HTTPInternalServerError()
116
exc.explanation = cant_proxy_correctly_message
118
return app(environ, start_response)
120
app = check_not_proxied(app)
122
130
app = PrefixMiddleware(app, prefix=prefix)
124
app = HTTPExceptionHandler(app)
125
app = ErrorHandlerApp(app)
127
if not config.get_option('user_port'):
132
if not options.user_port:
130
port = config.get_option('user_port')
135
port = options.user_port
132
if not config.get_option('user_host'):
137
if not options.user_host:
135
host = config.get_option('user_host')
140
host = options.user_host
139
142
httpserver.serve(app, host=host, port=port)