1 files changed, 90 insertions, 0 deletions
diff --git a/bin/rainbow-sugarize b/bin/rainbow-sugarize
new file mode 100755
index 0000000..ea0e312
--- /dev/null
+++ b/bin/rainbow-sugarize
@@ -0,0 +1,90 @@
+#!/usr/bin/env python
+
+import sys
+import pwd
+
+from os import getuid, getgid, environ, chmod, chown, symlink
+from os.path import join, isdir, dirname
+from optparse import OptionParser
+from shutil import copyfile
+
+from rainbow.util import make_reporter, trace, make_dirs
+
+sys.excepthook = trace
+
+def main():
+    parser = OptionParser(version='0.1')
+    parser.add_option('-v', '--verbose', default=0, action='count',
+                      help='Verbosity. Repeat for more verbose output.')
+    parser.add_option('-q', '--quiet', default=False, action='store_true',
+                      help='Quiet. Disable all output.')
+    parser.add_option('-u', '--user', default=None,
+                      help="Isolated user.")
+    parser.add_option('-e', '--envdir', default=None,
+                      help="Envdir to be used for launching.")
+    opts, args = parser.parse_args()
+    if not opts.user or not opts.envdir:
+        parser.print_help()
+        exit(1)
+
+    report = make_reporter(opts.verbose, opts.quiet, sys.stdout)
+
+    def check_user(report, opts):
+        report(1, 'Sugarizing isolated uid %s.', opts.user)
+        return opts.user
+
+    def check_envdir(report, envdir):
+        report(1, 'Sugarizing envdir %s.', envdir)
+        assert isdir(envdir)
+        def write_envvar(k, v):
+            report(1, '-E %s=%s', k, v)
+            open(join(envdir, k), 'w').write(v)
+        return write_envvar
+
+    user = check_user(report, opts)
+    write_envvar = check_envdir(report, opts.envdir)
+
+    o = pwd.getpwuid(getuid())
+    i = pwd.getpwnam(user)
+    h_o = o.pw_dir
+    h_i = i.pw_dir
+
+    # We want to use xauth generate $DISPLAY . untrusted, but we don't have
+    # XSECURITY enabled. <MS>
+    #environ.setdefault('XAUTHORITY', join(h_o, '.Xauthority'))
+    environ.setdefault('ICEAUTHORITY', join(h_o, '.ICEauthority'))
+
+    x_cookie_path = join(h_i, '.Xauthority')
+    make_dirs(dirname(x_cookie_path), getuid(), getgid(), 0777)
+    #copyfile(environ['XAUTHORITY'], x_cookie_path)
+    #chmod(x_cookie_path, 0666)
+    #chown(x_cookie_path, o.pw_uid, i.pw_gid)
+
+    ice_cookie_path = join(h_i, '.ICEauthority')
+    make_dirs(dirname(ice_cookie_path), getuid(), getgid(), 0777)
+    copyfile(environ['ICEAUTHORITY'], ice_cookie_path)
+    chmod(ice_cookie_path, 0666)
+    chown(ice_cookie_path, o.pw_uid, i.pw_gid)
+
+    for frag in ['owner.key.pub']:
+        path = join('.sugar/default/', frag)
+        make_dirs(dirname(join(h_i, path)), getuid(), getgid(), 0777)
+        copyfile(join(h_o, path), join(h_i, path))
+        chmod(join(h_i, path), 0666)
+
+    write_envvar('USER', i.pw_name)
+    write_envvar('HOME', h_i)
+    #write_envvar('XAUTHORITY', x_cookie_path)
+    write_envvar('ICEAUTHORITY', ice_cookie_path)
+    write_envvar('SUGAR_ACTIVITY_ROOT', h_i)
+    write_envvar('TMPDIR', join(h_i, 'tmp'))
+    write_envvar('DISPLAY', environ['DISPLAY'])
+
+    symlink(".", join(h_i, "instance"))
+    symlink(environ["SUGAR_BUNDLE_ID"], join(h_i, "data"))
+    symlink(environ.get("TMPDIR", "/tmp"), join(h_i, "tmp"))
+
+if __name__ == '__main__':
+    main()
+
+# vim : et sw=4 ts=4 sts=4 :