|
1072
by matt.giuca
Renamed scripts to services. |
1 |
#!/usr/bin/python
|
2 |
||
3 |
import os |
|
4 |
import sys |
|
5 |
import logging |
|
6 |
||
7 |
import conf |
|
8 |
import common.db |
|
9 |
import common.chat |
|
10 |
import common.makeuser |
|
11 |
import common.studpath |
|
12 |
||
13 |
# usage:
|
|
14 |
# usrmgt-server <port> <magic>
|
|
15 |
||
16 |
# User management operations:
|
|
17 |
# - Create local user
|
|
18 |
# - [Re]Create jail for a user
|
|
19 |
# - Create a svn repository for a user
|
|
20 |
# - create repository
|
|
21 |
# - svn config
|
|
22 |
# - svn auth
|
|
23 |
# - /etc/passwd entry
|
|
24 |
# - Disable a user's account
|
|
25 |
# - Enable a user's account
|
|
26 |
# - Remove a user
|
|
27 |
# - Rebuild svn config
|
|
28 |
# - Rebuild svn auth file
|
|
29 |
# - Rebuild passwd + push to nodes.
|
|
30 |
||
31 |
def activate_user(props): |
|
32 |
"""Create the on-disk stuff for the given user.
|
|
33 |
Sets the state of the user in the db from pending to enabled.
|
|
34 |
Expected properties:
|
|
35 |
login - the user name for the jail
|
|
36 |
STRING REQUIRED
|
|
37 |
Return Value: None
|
|
38 |
"""
|
|
39 |
||
40 |
os.umask(0022) # Bad, but start_server sets it worse. |
|
41 |
||
42 |
login = props['login'] |
|
43 |
||
44 |
db = common.db.DB() |
|
45 |
||
46 |
try: |
|
47 |
||
48 |
# FIXME: check we're pending
|
|
49 |
||
50 |
details = db.get_user(login) |
|
51 |
||
52 |
# make svn config/auth
|
|
53 |
||
54 |
repopath = os.path.join(conf.svn_repo_path, 'users', login) |
|
55 |
logging.debug("Creating user's Subversion repository") |
|
|
1073
by matt.giuca
usrmgt-server: When creating svn repo, now explicitly asks for the exception |
56 |
common.makeuser.make_svn_repo(repopath, throw_on_error=True) |
|
1072
by matt.giuca
Renamed scripts to services. |
57 |
|
58 |
rebuild_svn_config(props) |
|
59 |
||
60 |
logging.debug("Adding Subversion authentication") |
|
|
1073
by matt.giuca
usrmgt-server: When creating svn repo, now explicitly asks for the exception |
61 |
passwd = common.makeuser.make_svn_auth(login, throw_on_error=True) |
|
1072
by matt.giuca
Renamed scripts to services. |
62 |
logging.debug("passwd: %s" % passwd) |
63 |
||
64 |
logging.debug("Creating jail") |
|
65 |
common.makeuser.make_jail(login, details.unixid, svn_pass=passwd) |
|
66 |
||
67 |
logging.info("Enabling user") |
|
68 |
db.update_user(login, state='enabled') |
|
69 |
||
70 |
return {"response": "okay"} |
|
71 |
||
72 |
finally: |
|
73 |
db.close() |
|
74 |
||
75 |
def rebuild_svn_config(props): |
|
76 |
"""Rebuilds the svn config file
|
|
77 |
Return value:
|
|
78 |
response (okay, failure)
|
|
79 |
"""
|
|
80 |
try: |
|
81 |
common.makeuser.rebuild_svn_config() |
|
82 |
except Exception, e: |
|
83 |
logging.warning('Rebuild of Subversion authorization config failed!') |
|
84 |
return{'response': 'failure', 'msg': repr(e)} |
|
85 |
||
86 |
return {'response': 'okay'} |
|
87 |
||
88 |
def rebuild_svn_group_config(props): |
|
89 |
"""Rebuilds the svn group config file
|
|
90 |
Return value:
|
|
91 |
response (okay, failure)
|
|
92 |
"""
|
|
93 |
try: |
|
94 |
common.makeuser.rebuild_svn_group_config() |
|
95 |
except Exception, e: |
|
96 |
logging.warning( |
|
97 |
'Rebuild of Subversion group authorization config failed!') |
|
98 |
return{'response': 'failure', 'msg': repr(e)} |
|
99 |
||
100 |
return {'response': 'okay'} |
|
101 |
||
102 |
def create_group_repository(props): |
|
103 |
"""Creates on disk repository for the given group
|
|
104 |
Expected properties:
|
|
105 |
subj_short_name, year, semester, groupnm
|
|
106 |
Return value:
|
|
107 |
response (okay, failure)
|
|
108 |
"""
|
|
109 |
||
110 |
subj_short_name = props['subj_short_name'] |
|
111 |
year = props['year'] |
|
112 |
semester = props['semester'] |
|
113 |
groupnm = props['groupnm'] |
|
114 |
||
115 |
namespace = "_".join([subj_short_name, year, semester, groupnm]) |
|
116 |
repopath = os.path.join(conf.svn_repo_path, 'groups', namespace) |
|
117 |
logging.debug("Creating Subversion repository %s"%repopath) |
|
118 |
try: |
|
119 |
common.makeuser.make_svn_repo(repopath) |
|
120 |
except Exception, e: |
|
121 |
logging.error("Failed to create Subversion repository %s: %s"% |
|
122 |
(repopath,repr(e))) |
|
123 |
return {'response': 'failure', 'msg': repr(e)} |
|
124 |
||
125 |
return {'response': 'okay'} |
|
126 |
||
127 |
actions = { |
|
128 |
'activate_user':activate_user, |
|
129 |
'create_group_repository':create_group_repository, |
|
130 |
'rebuild_svn_config':rebuild_svn_config, |
|
131 |
'rebuild_svn_group_config':rebuild_svn_group_config, |
|
132 |
}
|
|
133 |
||
134 |
def initializer(): |
|
135 |
try: |
|
136 |
pidfile = open('/var/run/usrmgt-server.pid', 'w') |
|
137 |
pidfile.write('%d\n' % os.getpid()) |
|
138 |
pidfile.close() |
|
139 |
except IOError, (errno, strerror): |
|
140 |
print "Couldn't write PID file. IO error(%s): %s" % (errno, strerror) |
|
141 |
sys.exit(1) |
|
142 |
||
143 |
def dispatch(props): |
|
144 |
logging.debug(repr(props)) |
|
145 |
action = props.keys()[0] |
|
146 |
return actions[action](props[action]) |
|
147 |
||
148 |
if __name__ == "__main__": |
|
149 |
pid = os.getpid() |
|
150 |
||
151 |
logging.basicConfig(filename="/var/log/usrmgt.log", level=logging.INFO) |
|
152 |
logging.info("Starting usrmgt server on port %d (pid = %d)" % |
|
153 |
(conf.usrmgt_port, pid)) |
|
154 |
||
155 |
common.chat.start_server(conf.usrmgt_port, conf.usrmgt_magic, True, dispatch, initializer) |