X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-dev.git/blobdiff_plain/c81a9160bc47b724cba617285da6fec429734a21..f67228463e6f00ade66dc069b65fe37a7bb96f09:/invirt-build-conf

diff --git a/invirt-build-conf b/invirt-build-conf
index f74b05d..d43bc7c 100755
--- a/invirt-build-conf
+++ b/invirt-build-conf
@@ -3,36 +3,61 @@
 """Re-generate the remctl configuration for build submissions.
 
 This script generates the remctl ACL and configuration for each build
-pocket defined in the configuration.
+pocket defined in the configuration. It also updates the .k5login for
+the git user that developers can push through.
 """
 
 
 import os
 import tempfile
 
-from invirt.authz import mech as authz
+from invirt import authz
+from invirt import builder
 from invirt.config import structs as config
 
 
+def userToPrinc(user):
+    """Convert an AFS principal to a Kerberos v5 principal."""
+    if '@' in user:
+        (princ, realm) = user.split('@')
+    else:
+        princ = user
+        realm = config.kerberos.realm
+
+    return princ.replace('.', '/') + '@' + realm
+
+
 def main():
+    all_devs = set()
+
     # Python could really use a file-like object that gets written to
     # a temporary path and moved to its final resting place on
     # .close(). Oh well.
-    conf = tempfile.NamedTemporaryFile(delete=False)
-    build_handler = '/usr/sbin/invirt-submit-build'
+    conf_fd, conf_name = tempfile.mkstemp()
+    conf = os.fdopen(conf_fd, 'r+')
+    build_handler = '/usr/bin/invirt-submit-build'
 
-    for pocket in config.git.pockets:
-        acl = authz.expandAdmin(getattr(config.git.pockets, pocket).acl, None)
+    for pocket in config.build.pockets:
+        acl = authz.expandAdmin(getattr(config.build.pockets, pocket).acl, None)
 
-        acl_fd = tempfile.NamedTemporaryFile(delete=False)
-        print >>acl_fd, '\n'.join(acl)
+        acl_fd, acl_name = tempfile.mkstemp()
+        acl_fd = os.fdopen(acl_fd, 'r+')
+        print >>acl_fd, '\n'.join(userToPrinc(a) for a in acl)
+
+        all_devs.update(set(userToPrinc(a) for a in acl))
 
         acl_path = os.path.join('/etc/remctl/acl/build-%s' % pocket)
 
-        os.rename(acl_fd.name, acl_path)
+        os.rename(acl_name, acl_path)
         print >>conf, 'build %s %s %s' % (pocket, build_handler, acl_path)
 
-    os.rename(conf, '/etc/remctl/conf.d/build')
+    os.rename(conf_name, '/etc/remctl/conf.d/build')
+
+    k5login_fd, k5login_name = tempfile.mkstemp()
+    k5login = os.fdopen(k5login_fd, 'r+')
+    print >>k5login, '\n'.join(all_devs)
+
+    os.rename(k5login_name, os.path.join(builder._REPO_DIR, '.k5login'))
 
 
 if __name__ == '__main__':