add rgw module

Signed-off-by: Travis Rhoden <trhoden@redhat.com>

diff --git a/ceph_deploy/rgw.py b/ceph_deploy/rgw.py
new file mode 100644 (file)
index 0000000..cfc8fbf
--- /dev/null
+++ b/ceph_deploy/rgw.py
@@ -0,0 +1,208 @@
+from cStringIO import StringIO
+import errno
+import logging
+import os
+
+from ceph_deploy import conf
+from ceph_deploy import exc
+from ceph_deploy import hosts
+from ceph_deploy.util import system
+from ceph_deploy.lib import remoto
+from ceph_deploy.cliutil import priority
+
+
+LOG = logging.getLogger(__name__)
+
+
+def get_bootstrap_rgw_key(cluster):
+    """
+    Read the bootstrap-rgw key for `cluster`.
+    """
+    path = '{cluster}.bootstrap-rgw.keyring'.format(cluster=cluster)
+    try:
+        with file(path, 'rb') as f:
+            return f.read()
+    except IOError:
+        raise RuntimeError('bootstrap-rgw keyring not found; run \'gatherkeys\'')
+
+
+def create_rgw(distro, name, cluster, init):
+    conn = distro.conn
+
+    path = '/var/lib/ceph/rgw/{cluster}-{name}'.format(
+        cluster=cluster,
+        name=name
+        )
+
+    conn.remote_module.safe_mkdir(path)
+
+    bootstrap_keyring = '/var/lib/ceph/bootstrap-rgw/{cluster}.keyring'.format(
+        cluster=cluster
+        )
+
+    keypath = os.path.join(path, 'keyring')
+
+    stdout, stderr, returncode = remoto.process.check(
+        conn,
+        [
+            'ceph',
+            '--cluster', cluster,
+            '--name', 'client.bootstrap-rgw',
+            '--keyring', bootstrap_keyring,
+            'auth', 'get-or-create', 'client.rgw.{name}'.format(name=name),
+            'osd', 'allow rwx',
+            'mon', 'allow rwx',
+            '-o',
+            os.path.join(keypath),
+        ]
+    )
+    if returncode > 0 and returncode != errno.EACCES:
+        for line in stderr:
+            conn.logger.error(line)
+        for line in stdout:
+            # yes stdout as err because this is an error
+            conn.logger.error(line)
+        conn.logger.error('exit code from command was: %s' % returncode)
+        raise RuntimeError('could not create rgw')
+
+        remoto.process.check(
+            conn,
+            [
+                'ceph',
+                '--cluster', cluster,
+                '--name', 'client.bootstrap-rgw',
+                '--keyring', bootstrap_keyring,
+                'auth', 'get-or-create', 'client.rgw.{name}'.format(name=name),
+                'osd', 'allow *',
+                'mon', 'allow *',
+                '-o',
+                os.path.join(keypath),
+            ]
+        )
+
+    conn.remote_module.touch_file(os.path.join(path, 'done'))
+    conn.remote_module.touch_file(os.path.join(path, init))
+
+    if init == 'upstart':
+        remoto.process.run(
+            conn,
+            [
+                'initctl',
+                'emit',
+                'ceph-rgw',
+                'cluster={cluster}'.format(cluster=cluster),
+                'id={name}'.format(name=name),
+            ],
+            timeout=7
+        )
+    elif init == 'sysvinit':
+        remoto.process.run(
+            conn,
+            [
+                'service',
+                'ceph',
+                'start',
+                'rgw.{name}'.format(name=name),
+            ],
+            timeout=7
+        )
+
+    if distro.is_el:
+        system.enable_service(distro.conn)
+
+
+def rgw_create(args):
+    cfg = conf.ceph.load(args)
+    LOG.debug(
+        'Deploying rgw, cluster %s hosts %s',
+        args.cluster,
+        ' '.join(':'.join(x or '' for x in t) for t in args.rgw),
+        )
+
+    if not args.rgw:
+        raise exc.NeedHostError()
+
+    key = get_bootstrap_rgw_key(cluster=args.cluster)
+
+    bootstrapped = set()
+    errors = 0
+    for hostname, name in args.rgw:
+        try:
+            distro = hosts.get(hostname, username=args.username)
+            rlogger = distro.conn.logger
+            LOG.info(
+                'Distro info: %s %s %s',
+                distro.name,
+                distro.release,
+                distro.codename
+            )
+            LOG.debug('remote host will use %s', distro.init)
+
+            if hostname not in bootstrapped:
+                bootstrapped.add(hostname)
+                LOG.debug('deploying rgw bootstrap to %s', hostname)
+                conf_data = StringIO()
+                cfg.write(conf_data)
+                distro.conn.remote_module.write_conf(
+                    args.cluster,
+                    conf_data.getvalue(),
+                    args.overwrite_conf,
+                )
+
+                path = '/var/lib/ceph/bootstrap-rgw/{cluster}.keyring'.format(
+                    cluster=args.cluster,
+                )
+
+                if not distro.conn.remote_module.path_exists(path):
+                    rlogger.warning('rgw keyring does not exist yet, creating one')
+                    distro.conn.remote_module.write_keyring(path, key)
+
+            create_rgw(distro, name, args.cluster, distro.init)
+            distro.conn.exit()
+        except RuntimeError as e:
+            LOG.error(e)
+            errors += 1
+
+    if errors:
+        raise exc.GenericError('Failed to create %d RGWs' % errors)
+
+
+def rgw(args):
+    if args.subcommand == 'create':
+        rgw_create(args)
+    else:
+        LOG.error('subcommand %s not implemented', args.subcommand)
+
+
+def colon_separated(s):
+    host = s
+    name = s
+    if s.count(':') == 1:
+        (host, name) = s.split(':')
+    return (host, name)
+
+
+@priority(30)
+def make(parser):
+    """
+    Deploy ceph RGW on remote hosts.
+    """
+    parser.add_argument(
+        'subcommand',
+        metavar='SUBCOMMAND',
+        choices=[
+            'create',
+            'destroy',
+            ],
+        help='create or destroy',
+        )
+    parser.add_argument(
+        'rgw',
+        metavar='HOST[:NAME]',
+        nargs='*',
+        type=colon_separated,
+        help='host (and optionally the daemon name) to deploy on',
+        )
+    parser.set_defaults(
+        func=rgw,
+        )