28
28
from paste.translogger import TransLogger
30
30
from loggerhead import __version__
31
from loggerhead.apps.transport import (
32
BranchesFromTransportRoot, UserBranchesFromTransportRoot)
33
from loggerhead.config import LoggerheadConfig
31
from loggerhead.apps.filesystem import (
32
BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
34
33
from loggerhead.util import Reloader
35
34
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__,)
37
def command_line_parser():
38
parser = OptionParser("%prog [options] <path>")
44
parser.add_option("--user-dirs", action="store_true", dest="user_dirs",
45
help="Serve user directories as ~user.")
46
parser.add_option("--trunk-dir", metavar="DIR",
47
help="The directory that contains the trunk branches.")
48
parser.add_option("--port", dest="user_port",
49
help=("Port Loggerhead should listen on "
50
"(defaults to 8080)."))
51
parser.add_option("--host", dest="user_host",
52
help="Host Loggerhead should listen on.")
53
parser.add_option("--prefix", dest="user_prefix",
54
help="Specify host prefix.")
55
parser.add_option("--profile", action="store_true", dest="profile",
56
help="Generate callgrind profile data to "
57
"%d-stats.callgrind on each request.")
58
parser.add_option("--reload", action="store_true", dest="reload",
59
help="Restarts the application when changing python"
60
" files. Only used for development purposes.")
61
parser.add_option('--log-folder', dest="log_folder",
62
type=str, help="The directory to place log files in.")
63
parser.add_option("--version", action="store_true", dest="show_version",
64
help="Print the software version and exit")
69
parser = command_line_parser()
70
(options, args) = parser.parse_args(sys.argv[1:])
72
if options.show_version:
73
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):
84
if not os.path.isdir(path):
85
print "%s is not a directory" % path
88
if options.trunk_dir and not options.user_dirs:
89
print "--trunk-dir is only valid with --user-dirs"
93
if Reloader.is_installed():
96
return Reloader.restart_with_reloader()
99
if not options.trunk_dir:
100
print "You didn't specify a directory for the trunk directories."
102
app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
104
app = BranchesFromFileSystemRoot(path)
60
107
logging.basicConfig()
61
108
logging.getLogger('').setLevel(logging.DEBUG)
62
logger = logging.getLogger('loggerhead')
63
if config.get_option('log_folder'):
64
logfile_path = os.path.join(
65
config.get_option('log_folder'), 'serve-branches.log')
109
logger = getattr(app, 'log', logging.getLogger('loggerhead'))
110
if options.log_folder:
111
logfile_path = os.path.join(options.log_folder, 'serve-branches.log')
67
113
logfile_path = 'serve-branches.log'
68
114
logfile = logging.FileHandler(logfile_path, 'a')
71
117
logfile.setFormatter(formatter)
72
118
logfile.setLevel(logging.DEBUG)
73
119
logger.addHandler(logfile)
77
def make_app_for_config_and_path(config, base):
78
if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
79
print "--trunk-dir is only valid with --user-dirs"
82
if config.get_option('reload'):
83
if Reloader.is_installed():
86
return Reloader.restart_with_reloader()
88
if config.get_option('user_dirs'):
89
if not config.get_option('trunk_dir'):
90
print "You didn't specify a directory for the trunk directories."
92
app = UserBranchesFromTransportRoot(base, config)
94
app = BranchesFromTransportRoot(base, config)
98
if config.get_option('profile'):
120
# setup_logging() #end
121
app = TransLogger(app, logger=logger)
99
123
from loggerhead.middleware.profile import LSProfMiddleware
100
124
app = LSProfMiddleware(app)
101
if config.get_option('memory_profile'):
102
from dozer import Dozer
105
if not config.get_option('user_prefix'):
126
if not options.user_prefix:
108
prefix = config.get_option('user_prefix')
129
prefix = options.user_prefix
109
130
if not prefix.startswith('/'):
110
131
prefix = '/' + prefix
130
151
app = HTTPExceptionHandler(app)
131
152
app = ErrorHandlerApp(app)
132
app = TransLogger(app, logger=logging.getLogger('loggerhead'))
140
config, path = get_config_and_path(args)
142
app = make_app_for_config_and_path(config, path)
144
if not config.get_option('user_port'):
154
if not options.user_port:
147
port = config.get_option('user_port')
157
port = options.user_port
149
if not config.get_option('user_host'):
159
if not options.user_host:
152
host = config.get_option('user_host')
154
if not config.get_option('protocol'):
157
protocol = config.get_option('protocol')
159
if protocol == 'http':
160
httpserver.serve(app, host=host, port=port)
162
if protocol == 'fcgi':
163
from flup.server.fcgi import WSGIServer
164
elif protocol == 'scgi':
165
from flup.server.scgi import WSGIServer
166
elif protocol == 'ajp':
167
from flup.server.ajp import WSGIServer
169
print 'Unknown protocol: %s.' % (protocol)
171
WSGIServer(app, bindAddress=(host, int(port))).run()
162
host = options.user_host
166
httpserver.serve(app, host=host, port=port)
169
if __name__ == "__main__":