00f1a6e6d20e79a8b2213797d2db764742acae7c
[matthijs/servers/drsnuggles.git] / etc / init.d / fastcgi
1 #!/usr/bin/python
2 import os
3 import re
4 import pwd
5 import grp
6 import shutil
7 import stat
8
9 ROOT_DIR="/data/www"
10
11 # SITES = [(sitename, application_list)]
12 # application_list = [application_name, (application_name, command, ...)]
13 # Here, sitename is the name of the site. This folder name should exist below ROOT_DIR and
14 # is also used below SOCKET_DIR. The site name is also translated to a user and
15 # group name by replacing dots by dashes and prepending USER_PREFIX and
16 # GROUP_PREFIX.
17 #
18 # application_list specifies the applications to start for this site. These can
19 # be generic (when only application_name is given), in which case the command
20 # is looked up in APPLICATIONS using the application_name. For a site-specific
21 # application, command is the command that should be run. It will be prefixed
22 # with the site's root dir, if is not an absolute path.
23
24  
25 SITES=[
26         ('stderr.nl',             ['php', ('trac', 'applications/trac/trac.fcgi')]),
27         ('stdin.nl',              ['php']),
28         ('stdout.nl',             ['php']),
29         ('ninniach.nl',           ['php']),
30         ('blues-brothers.eu',     ['php']),
31         ('evolution-events.nl',   ['php',
32             ('wipi', 'applications/wipi/wipi.fcgi'),
33             ('hunternet', 'applications/lexnet/manage.py runfcgi'),
34             ('xerxes', 'applications/xerxes/manage.py runfcgi'),
35             ('dorestad-bookings', 'applications/dorestad-bookings/manage.py.local runfcgi debug=true'),
36         ]),
37 #       ('stdio.flexvps.nl',      ['php']),
38 #       ('foresightsecurity.nl',  ['php']),
39 ]
40
41 # Generic applications that can be run for any site
42 # Maps application_name to application_command. application_command will be
43 # prefixed with the site's root dir, if it is not an absolute path.
44 APPLICATIONS={"php": "/usr/bin/php-cgi"}
45
46 # Kill these procs before starting new ones. Only processes of these names that
47 # are run by the sites in SITES are killed. This is a bit hackish, we should
48 # really be using pidfiles...
49 KILL_PROCS=['php-cgi', 'manage.py', 'manage.py.local', 'trac.fcgi', 'wipi.fcgi']
50
51 ## ABSOLUTE path to the spawn-fcgi binary
52 SPAWNFCGI="/usr/bin/spawn-fcgi"
53
54 ## Dir in which to create the UNIX sockets to listen on
55 SOCKET_DIR="%s/var/fcgi" % (ROOT_DIR)
56
57 ## number of PHP children to spawn
58 PHP_FCGI_CHILDREN=2
59
60 ## maximum number of requests a single PHP process can serve before it is restarted
61 PHP_FCGI_MAX_REQUESTS=1000
62
63 # The user to run as, will be prefixed to the sitename
64 USER_PREFIX="httpd-"
65 # The group to run as.
66 SCRIPT_GROUP="httpd-users"
67 # The group that should be able to use the sockets created
68 HTTPD_GROUP="www-data"
69
70 # Will be postfixed to the site's root and exported in the PHPRC variable.
71 PHPRC_DIR="conf"
72
73 #### END OF CONFIG ####
74
75 for (site, apps) in SITES:
76         site_name = re.sub('\.', '-', site)
77
78         ## switch to the following user / group
79         user_id    = "%s%s" % (USER_PREFIX, site_name)
80         
81         # Find the site dir
82         site_dir   = os.path.join(ROOT_DIR, site)
83         socket_dir = os.path.join(SOCKET_DIR, site_name)
84
85         # Pass the site dir to all fastcgi processes
86         os.environ['SITE_DIR'] = site_dir
87
88         if not site_dir:
89                 raise Exception("Site dir does not exist: %s" % (site_dir))
90         
91         
92         # Kill existing processes first
93         for procname in KILL_PROCS:
94                 os.system('killall --user %s %s' % (user_id, procname))
95
96         # Remove old sockets
97         if os.path.exists(socket_dir): 
98                 shutil.rmtree(socket_dir)
99
100         # Create dir for sockets. Make owning group root and set group write
101         # permissions, so the mask field in the acl will not block out anything.
102         os.makedirs(socket_dir)
103         os.chown(socket_dir, pwd.getpwnam(user_id)[2], grp.getgrnam(HTTPD_GROUP)[2])
104         #os.chmod(socket_dir, stat.S_IRWXU)
105         
106         for app in apps:
107                 # Unpack app tuple or lookup app command in APPLICATIONS
108                 if isinstance(app, tuple):
109                         if len(app) == 2:
110                                 (app_name, app_command) = app
111                         else:
112                                 raise Exception("Wrong number of elements in site tuple: %s", app)
113                 else:
114                         app_name = app
115                         app_command = APPLICATIONS[app_name]
116         
117                 # Prefix with site dir if not an absolute path
118                 if not os.path.isabs(app_command):
119                         app_command = os.path.join(site_dir, app_command)
120
121                 # Create socket filename
122                 socket = os.path.join(socket_dir, app_name)
123
124                 # Build the command
125                 # TODO: Wrap this in env to clear up the environment
126                 spawnfcgi = '%s -s "%s" -u "%s" -g "%s"' % (SPAWNFCGI, socket, user_id, SCRIPT_GROUP)
127                 fcgiapp = ' -- %s' % (app_command)
128
129                 if app_name == 'php':
130                         os.environ['PHP_FCGI_MAX_REQUESTS'] = str(PHP_FCGI_MAX_REQUESTS)
131                         phprc = os.path.join(site_dir, PHPRC_DIR, 'php.ini')
132                         if os.path.exists(phprc):
133                                 #os.environ['PHPRC'] = phprc
134                                 fcgiapp   += ' -c %s' % (phprc)
135                         spawnfcgi += ' -C %s' % (PHP_FCGI_CHILDREN)
136
137
138                 print spawnfcgi + fcgiapp
139                 os.system(spawnfcgi + fcgiapp)
140
141                 # Ensure www-data can write to the socket :-S
142                 # Spawn-fcgi explicitely chmods the socket after creation, very
143                 # annoying
144                 os.chmod(socket, stat.S_IRWXU | stat.S_IRWXG)