← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to serve-branches

  • Committer: Matt Nordhoff
  • Date: 2009-06-17 23:10:07 UTC
  • mto: This revision was merged to the branch mainline in revision 367.
  • Revision ID: mnordhoff@mattnordhoff.com-20090617231007-09y132sof4ix98po
Obey the http_serve setting for serving over hpss

Show diffs side-by-side

added added

removed removed

Lines of Context:
serve-branches
1
1
#!/usr/bin/env python
 
2
#
 
3
# Copyright (C) 2008, 2009 Canonical Ltd
 
4
#
2
5
# This program is free software; you can redistribute it and/or modify
3
6
# it under the terms of the GNU General Public License as published by
4
7
# the Free Software Foundation; either version 2 of the License, or
19
22
import os
20
23
import sys
21
24
 
22
 
from optparse import OptionParser
 
25
from bzrlib.plugin import load_plugins
 
26
from bzrlib.transport import get_transport
23
27
 
24
28
from paste import httpserver
25
 
from paste.httpexceptions import HTTPExceptionHandler
 
29
from paste.httpexceptions import HTTPExceptionHandler, HTTPInternalServerError
26
30
from paste.translogger import TransLogger
27
31
 
28
32
from loggerhead import __version__
29
 
from loggerhead.apps.filesystem import (
30
 
    BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
31
 
 
32
 
 
33
 
def command_line_parser():
34
 
    parser = OptionParser("%prog [options] <path>")
35
 
    parser.set_defaults(
36
 
        user_dirs=False,
37
 
        show_version=False,
38
 
        )
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")
45
 
    return parser
 
33
from loggerhead.apps.transport import (
 
34
    BranchesFromTransportRoot, UserBranchesFromTransportRoot)
 
35
from loggerhead.config import LoggerheadConfig
 
36
from loggerhead.util import Reloader
 
37
from loggerhead.apps.error import ErrorHandlerApp
46
38
 
47
39
 
48
40
def main(args):
49
 
    logging.basicConfig()
50
 
    logging.getLogger().setLevel(logging.DEBUG)
51
 
 
52
 
    parser = command_line_parser()
53
 
    (options, args) = parser.parse_args(sys.argv[1:])
54
 
 
55
 
    if options.show_version:
 
41
    config = LoggerheadConfig()
 
42
 
 
43
    if config.get_option('show_version'):
56
44
        print "loggerhead %s" % __version__
57
45
        sys.exit(0)
58
46
 
59
 
    if len(args) > 1:
60
 
        parser.print_help()
 
47
    if config.arg_count > 1:
 
48
        config.print_help()
61
49
        sys.exit(1)
62
 
    elif len(args) == 1:
63
 
        [path] = args
 
50
    elif config.arg_count == 1:
 
51
        path = config.get_arg(0)
64
52
    else:
65
53
        path = '.'
66
54
 
67
 
    if not os.path.isdir(path):
68
 
        print "%s is not a directory" % path
69
 
        sys.exit(1)
70
 
 
71
 
    if options.trunk_dir and not options.user_dirs:
 
55
    load_plugins()
 
56
 
 
57
    if config.get_option('allow_writes'):
 
58
        transport = get_transport(path)
 
59
    else:
 
60
        transport = get_transport('readonly+' + path)
 
61
 
 
62
    if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
72
63
        print "--trunk-dir is only valid with --user-dirs"
73
64
        sys.exit(1)
74
65
 
75
 
    if options.user_dirs:
76
 
        if not options.trunk_dir:
 
66
    if config.get_option('reload'):
 
67
        if Reloader.is_installed():
 
68
            Reloader.install()
 
69
        else:
 
70
            return Reloader.restart_with_reloader()
 
71
 
 
72
    if config.get_option('user_dirs'):
 
73
        if not config.get_option('trunk_dir'):
77
74
            print "You didn't specify a directory for the trunk directories."
78
75
            sys.exit(1)
79
 
        app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
80
 
    else:
81
 
        app = BranchesFromFileSystemRoot(path)
82
 
    app = HTTPExceptionHandler(app)
83
 
    app = TransLogger(app)
 
76
        app = UserBranchesFromTransportRoot(transport, config)
 
77
    else:
 
78
        app = BranchesFromTransportRoot(transport, config)
 
79
 
 
80
    # setup_logging()
 
81
    logging.basicConfig()
 
82
    logging.getLogger('').setLevel(logging.DEBUG)
 
83
    logger = getattr(app, 'log', logging.getLogger('loggerhead'))
 
84
    if config.get_option('log_folder'):
 
85
        logfile_path = os.path.join(
 
86
            config.get_option('log_folder'), 'serve-branches.log')
 
87
    else:
 
88
        logfile_path = 'serve-branches.log'
 
89
    logfile = logging.FileHandler(logfile_path, 'a')
 
90
    formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
 
91
                                  ' %(message)s')
 
92
    logfile.setFormatter(formatter)
 
93
    logfile.setLevel(logging.DEBUG)
 
94
    logger.addHandler(logfile)
 
95
 
 
96
    # setup_logging() #end
 
97
 
 
98
    if config.get_option('profile'):
 
99
        from loggerhead.middleware.profile import LSProfMiddleware
 
100
        app = LSProfMiddleware(app)
 
101
    if config.get_option('memory_profile'):
 
102
        from dozer import Dozer
 
103
        app = Dozer(app)
 
104
 
 
105
    if not config.get_option('user_prefix'):
 
106
        prefix = '/'
 
107
    else:
 
108
        prefix = config.get_option('user_prefix')
 
109
        if not prefix.startswith('/'):
 
110
            prefix = '/' + prefix
84
111
 
85
112
    try:
86
113
        from paste.deploy.config import PrefixMiddleware
87
114
    except ImportError:
88
 
        pass
89
 
    else:
90
 
        app = PrefixMiddleware(app)
91
 
 
92
 
    httpserver.serve(app, host='0.0.0.0', port='8080')
 
115
        cant_proxy_correctly_message = (
 
116
            'Unsupported configuration: PasteDeploy not available, but '
 
117
            'loggerhead appears to be behind a proxy.')
 
118
        def check_not_proxied(app):
 
119
            def wrapped(environ, start_response):
 
120
                if 'HTTP_X_FORWARDED_SERVER' in environ:
 
121
                    exc = HTTPInternalServerError()
 
122
                    exc.explanation = cant_proxy_correctly_message
 
123
                    raise exc
 
124
                return app(environ, start_response)
 
125
            return wrapped
 
126
        app = check_not_proxied(app)
 
127
    else:
 
128
        app = PrefixMiddleware(app, prefix=prefix)
 
129
 
 
130
    app = HTTPExceptionHandler(app)
 
131
    app = ErrorHandlerApp(app)
 
132
    app = TransLogger(app, logger=logger)
 
133
 
 
134
    if not config.get_option('user_port'):
 
135
        port = '8080'
 
136
    else:
 
137
        port = config.get_option('user_port')
 
138
 
 
139
    if not config.get_option('user_host'):
 
140
        host = '0.0.0.0'
 
141
    else:
 
142
        host = config.get_option('user_host')
 
143
 
 
144
    httpserver.serve(app, host=host, port=port)
93
145
 
94
146
 
95
147
if __name__ == "__main__":