Synnefo

# Too many lines in module pylint: disable-msg=C0302

# Too many arguments (7/5) pylint: disable-msg=R0913

"""

Fabric file for snf-deploy

"""

from __future__ import with_statement

from fabric.api import hide, env, settings, local, roles

from fabric.operations import run, put, get

import fabric

import re

import os

import shutil

import tempfile

import ast

from snfdeploy.lib import debug, Conf, Env, disable_color

from snfdeploy import massedit

def setup_env(args):

    """Setup environment"""

    print("Loading configuration for synnefo...")

    conf = Conf(args)

    env.env = Env(conf)

    env.local = args.autoconf

    env.key_inject = args.key_inject

    env.password = env.env.password

    env.user = env.env.user

    env.shell = "/bin/bash -c"

    env.key_filename = args.ssh_key

    if args.disable_colors:

        disable_color()

    if env.env.cms.hostname in \

            [env.env.accounts.hostname, env.env.cyclades.hostname,

             env.env.pithos.hostname]:

        env.cms_pass = True

    else:

        env.cms_pass = False

    if env.env.accounts.hostname in \

            [env.env.cyclades.hostname, env.env.pithos.hostname]:

        env.csrf_disable = True

    else:

        env.csrf_disable = False

    env.roledefs = {

        "nodes": env.env.ips,

        "ips": env.env.ips,

        "accounts": [env.env.accounts.ip],

        "cyclades": [env.env.cyclades.ip],

        "pithos": [env.env.pithos.ip],

        "cms": [env.env.cms.ip],

        "mq": [env.env.mq.ip],

        "db": [env.env.db.ip],

        "mq": [env.env.mq.ip],

        "db": [env.env.db.ip],

        "ns": [env.env.ns.ip],

        "client": [env.env.client.ip],

        "router": [env.env.router.ip],

        "stats": [env.env.stats.ip],

    }

    env.enable_lvm = False

    env.enable_drbd = False

    if ast.literal_eval(env.env.create_extra_disk) and env.env.extra_disk:

        env.enable_lvm = True

        env.enable_drbd = True

    env.roledefs.update({

        "ganeti": env.env.cluster_ips,

        "master": [env.env.master.ip],

    })

def install_package(package):

    debug(env.host, " * Installing package %s..." % package)

    apt_get = "export DEBIAN_FRONTEND=noninteractive ;" + \

              "apt-get install -y --force-yes "

    host_info = env.env.ips_info[env.host]

    env.env.update_packages(host_info.os)

    if ast.literal_eval(env.env.use_local_packages):

        with settings(warn_only=True):

            deb = local("ls %s/%s*%s_*.deb"

                        % (env.env.packages, package, host_info.os),

                        capture=True)

            if deb:

                debug(env.host,

                      " * Package %s found in %s..."

                      % (package, env.env.packages))

                try_put(deb, "/tmp/")

                try_run("dpkg -i /tmp/%s || "

                        % os.path.basename(deb) + apt_get + "-f")

                try_run("rm /tmp/%s" % os.path.basename(deb))

                return

    info = getattr(env.env, package)

    if info in \

            ["squeeze-backports", "squeeze", "stable",

             "testing", "unstable", "wheezy"]:

        apt_get += " -t %s %s " % (info, package)

    elif info:

        apt_get += " %s=%s " % (package, info)

    else:

        apt_get += package

    try_run(apt_get)

    return

@roles("ns")

def update_ns_for_ganeti():

    debug(env.host,

          "Updating name server entries for backend %s..."

          % env.env.cluster.fqdn)

    update_arecord(env.env.cluster)

    update_ptrrecord(env.env.cluster)

    try_run("/etc/init.d/bind9 restart")

@roles("ns")

def update_ns_for_node(node):

    info = env.env.nodes_info.get(node)

    update_arecord(info)

    update_ptrrecord(info)

    try_run("/etc/init.d/bind9 restart")

@roles("ns")

def update_arecord(host):

    filename = "/etc/bind/zones/" + env.env.domain

    cmd = """

    echo '{0}' >> {1}

    """.format(host.arecord, filename)

    try_run(cmd)

@roles("ns")

def update_cnamerecord(host):

    filename = "/etc/bind/zones/" + env.env.domain

    cmd = """

    echo '{0}' >> {1}

    """.format(host.cnamerecord, filename)

    try_run(cmd)

@roles("ns")

def update_ptrrecord(host):

    filename = "/etc/bind/rev/synnefo.in-addr.arpa.zone"

    cmd = """

    echo '{0}' >> {1}

    """.format(host.ptrrecord, filename)

    try_run(cmd)

@roles("nodes")

def apt_get_update():

    debug(env.host, "apt-get update....")

    try_run("apt-get update")

@roles("ns")

def setup_ns():

    debug(env.host, "Setting up name server..")

    #WARNING: this should be remove after we are done

    # because gevent does pick randomly nameservers and google does

    # not know our setup!!!!!

    apt_get_update()

    install_package("bind9")

    tmpl = "/etc/bind/named.conf.local"

    replace = {

        "domain": env.env.domain,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl)

    try_run("mkdir -p /etc/bind/zones")

    tmpl = "/etc/bind/zones/example.com"

    replace = {

        "domain": env.env.domain,

        "ns_node_ip": env.env.ns.ip,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    remote = "/etc/bind/zones/" + env.env.domain

    try_put(custom, remote)

    try_run("mkdir -p /etc/bind/rev")

    tmpl = "/etc/bind/rev/synnefo.in-addr.arpa.zone"

    replace = {

        "domain": env.env.domain,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl)

    tmpl = "/etc/bind/named.conf.options"

    replace = {

        "NODE_IPS": ";".join(env.env.ips),

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl, mode=0644)

    for role, info in env.env.roles.iteritems():

        if role == "ns":

            continue

        update_cnamerecord(info)

    for node, info in env.env.nodes_info.iteritems():

        update_arecord(info)

        update_ptrrecord(info)

    try_run("/etc/init.d/bind9 restart")

@roles("nodes")

def check_dhcp():

    debug(env.host, "Checking IPs for synnefo..")

    for n, info in env.env.nodes_info.iteritems():

        try_run("ping -c 1 " + info.ip)

@roles("nodes")

def check_dns():

    debug(env.host, "Checking fqdns for synnefo..")

    for n, info in env.env.nodes_info.iteritems():

        try_run("ping -c 1 " + info.fqdn)

    for n, info in env.env.roles.iteritems():

        try_run("ping -c 1 " + info.fqdn)

@roles("nodes")

def check_connectivity():

    debug(env.host, "Checking internet connectivity..")

    try_run("ping -c 1 www.google.com")

@roles("nodes")

def check_ssh():

    debug(env.host, "Checking password-less ssh..")

    for n, info in env.env.nodes_info.iteritems():

        try_run("ssh " + info.fqdn + "  date")

@roles("ips")

def add_keys():

    if not env.key_inject:

        debug(env.host, "Skipping ssh keys injection..")

        return

    else:

        debug(env.host, "Adding rsa/dsa keys..")

    try_run("mkdir -p /root/.ssh")

    cmd = """

for f in $(ls /root/.ssh/*); do

  cp $f $f.bak

done

    """

    try_run(cmd)

    files = ["authorized_keys", "id_dsa", "id_dsa.pub",

             "id_rsa", "id_rsa.pub"]

    for f in files:

        tmpl = "/root/.ssh/" + f

        replace = {}

        custom = customize_settings_from_tmpl(tmpl, replace)

        try_put(custom, tmpl, mode=0600)

    cmd = """

if [ -e /root/.ssh/authorized_keys.bak ]; then

  cat /root/.ssh/authorized_keys.bak >> /root/.ssh/authorized_keys

fi

    """

    debug(env.host, "Updating exising authorized keys..")

    try_run(cmd)

@roles("ips")

def setup_resolv_conf():

    debug(env.host, "Tweak /etc/resolv.conf...")

    try_run("/etc/init.d/network-manager stop", abort=False)

    tmpl = "/etc/dhcp/dhclient-enter-hooks.d/nodnsupdate"

    replace = {}

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl, mode=0644)

    try_run("cp /etc/resolv.conf /etc/resolv.conf.bak")

    tmpl = "/etc/resolv.conf"

    replace = {

        "domain": env.env.domain,

        "ns_node_ip": env.env.ns.ip,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try:

        try_put(custom, tmpl)

        cmd = """

        echo "\

# This has been generated automatically by snf-deploy, at

# $(date).

# The immutable bit (+i attribute) has been used to avoid it being

# overwritten by software such as NetworkManager or resolvconf.

# Use lsattr/chattr to view or modify its file attributes.

$(cat {0})" > {0}

""".format(tmpl)

        try_run(cmd)

    except:

        pass

    try_run("chattr +i /etc/resolv.conf")

@roles("ips")

def setup_hosts():

    debug(env.host, "Tweaking /etc/hosts and ssh_config files...")

    try_run("echo StrictHostKeyChecking no >> /etc/ssh/ssh_config")

    cmd = "sed -i 's/^127.*$/127.0.0.1 localhost/g' /etc/hosts "

    try_run(cmd)

    host_info = env.env.ips_info[env.host]

    cmd = "hostname %s" % host_info.hostname

    try_run(cmd)

    cmd = "echo %s > /etc/hostname" % host_info.hostname

    try_run(cmd)

def try_run(cmd, abort=True):

    try:

        if env.local:

            return local(cmd, capture=True)

        else:

            return run(cmd)

    except BaseException as e:

        if abort:

            fabric.utils.abort(e)

        else:

            debug(env.host, "WARNING: command failed. Continuing anyway...")

def try_put(local_path=None, remote_path=None, abort=True, **kwargs):

    try:

        put(local_path=local_path, remote_path=remote_path, **kwargs)

    except BaseException as e:

        if abort:

            fabric.utils.abort(e)

        else:

            debug(env.host, "WARNING: command failed. Continuing anyway...")

def try_get(remote_path, local_path=None, abort=True, **kwargs):

    try:

        get(remote_path, local_path=local_path, **kwargs)

    except BaseException as e:

        if abort:

            fabric.utils.abort(e)

        else:

            debug(env.host, "WARNING: command failed. Continuing anyway...")

def create_bridges():

    debug(env.host, " * Creating bridges...")

    install_package("bridge-utils")

    cmd = """

    brctl addbr {0} ; ip link set {0} up

    """.format(env.env.common_bridge)

    try_run(cmd)

def connect_bridges():

    debug(env.host, " * Connecting bridges...")

    #cmd = """

    #brctl addif {0} {1}

    #""".format(env.env.common_bridge, env.env.public_iface)

    #try_run(cmd)

@roles("ganeti")

def setup_net_infra():

    debug(env.host, "Setup networking infrastracture..")

    create_bridges()

    connect_bridges()

@roles("ganeti")

def setup_lvm():

    debug(env.host, "create volume group %s for ganeti.." % env.env.vg)

    if env.enable_lvm:

        install_package("lvm2")

        cmd = """

        pvcreate {0}

        vgcreate {1} {0}

        """.format(env.env.extra_disk, env.env.vg)

        try_run(cmd)

def customize_settings_from_tmpl(tmpl, replace):

    debug(env.host, " * Customizing template %s..." % tmpl)

    local = env.env.templates + tmpl

    _, custom = tempfile.mkstemp()

    shutil.copyfile(local, custom)

    for k, v in replace.iteritems():

        regex = "re.sub('%{0}%', '{1}', line)".format(k.upper(), v)

        massedit.edit_files([custom], [regex], dry_run=False)

    return custom

@roles("nodes")

def setup_apt():

    debug(env.host, "Setting up apt sources...")

    install_package("curl")

    cmd = """

    echo 'APT::Install-Suggests "false";' >> /etc/apt/apt.conf

    curl -k https://dev.grnet.gr/files/apt-grnetdev.pub | apt-key add -

    """

    try_run(cmd)

    host_info = env.env.ips_info[env.host]

    if host_info.os == "squeeze":

        tmpl = "/etc/apt/sources.list.d/synnefo.squeeze.list"

    else:

        tmpl = "/etc/apt/sources.list.d/synnefo.wheezy.list"

    replace = {}

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl)

    apt_get_update()

@roles("cyclades", "cms", "pithos", "accounts")

def restart_services():

    debug(env.host, " * Restarting apache2 and gunicorn...")

    try_run("/etc/init.d/gunicorn restart")

    try_run("/etc/init.d/apache2 restart")

def setup_gunicorn():

    debug(env.host, " * Setting up gunicorn...")

    install_package("gunicorn")

    try_run("chown root.www-data /var/log/gunicorn")

    tmpl = "/etc/gunicorn.d/synnefo"

    replace = {}

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl, mode=0644)

    try_run("/etc/init.d/gunicorn restart")

def setup_apache():

    debug(env.host, " * Setting up apache2...")

    host_info = env.env.ips_info[env.host]

    install_package("apache2")

    tmpl = "/etc/apache2/sites-available/synnefo"

    replace = {

        "HOST": host_info.fqdn,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl)

    tmpl = "/etc/apache2/sites-available/synnefo-ssl"

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl)

    cmd = """

    a2enmod ssl

    a2enmod rewrite

    a2dissite default

    a2ensite synnefo

    a2ensite synnefo-ssl

    a2enmod headers

    a2enmod proxy_http

    a2dismod autoindex

    """

    try_run(cmd)

    try_run("/etc/init.d/apache2 restart")

@roles("mq")

def setup_mq():

    debug(env.host, "Setting up RabbitMQ...")

    install_package("rabbitmq-server")

    cmd = """

    rabbitmqctl add_user {0} {1}

    rabbitmqctl set_permissions {0} ".*" ".*" ".*"

    rabbitmqctl delete_user guest

    rabbitmqctl set_user_tags {0} administrator

    """.format(env.env.synnefo_user, env.env.synnefo_rabbitmq_passwd)

    try_run(cmd)

    try_run("/etc/init.d/rabbitmq-server restart")

@roles("db")

def allow_access_in_db(ip, user="all", method="md5"):

    cmd = """

    pg_hba=$(ls /etc/postgresql/*/main/pg_hba.conf)

    echo host all {0} {1}/32 {2} >> $pg_hba

    """.format(user, ip, method)

    try_run(cmd)

    cmd = """

    pg_hba=$(ls /etc/postgresql/*/main/pg_hba.conf)

    sed -i 's/\(host.*127.0.0.1.*\)md5/\\1trust/' $pg_hba

    """

    try_run(cmd)

    try_run("/etc/init.d/postgresql restart")

@roles("db")

def setup_db():

    debug(env.host, "Setting up DataBase server...")

    install_package("postgresql")

    tmpl = "/tmp/db-init.psql"

    replace = {

        "synnefo_user": env.env.synnefo_user,

        "synnefo_db_passwd": env.env.synnefo_db_passwd,

        }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl)

    cmd = 'su - postgres -c "psql -w -f %s" ' % tmpl

    try_run(cmd)

    cmd = """

    conf=$(ls /etc/postgresql/*/main/postgresql.conf)

    echo "listen_addresses = '*'" >> $conf

    """

    try_run(cmd)

    if env.env.testing_vm:

        cmd = """

        conf=$(ls /etc/postgresql/*/main/postgresql.conf)

        echo "fsync=off\nsynchronous_commit=off\nfull_page_writes=off" >> $conf

        """

        try_run(cmd)

    allow_access_in_db(env.host, "all", "trust")

    try_run("/etc/init.d/postgresql restart")

@roles("db")

def destroy_db():

    try_run("""su - postgres -c ' psql -w -c "drop database snf_apps" '""")

    try_run("""su - postgres -c ' psql -w -c "drop database snf_pithos" '""")

def setup_webproject():

    debug(env.host, " * Setting up snf-webproject...")

    with settings(hide("everything")):

        try_run("ping -c1 " + env.env.db.ip)

    setup_common()

    install_package("snf-webproject")

    install_package("python-psycopg2")

    install_package("python-gevent")

    install_package("python-django")

    tmpl = "/etc/synnefo/webproject.conf"

    replace = {

        "synnefo_user": env.env.synnefo_user,

        "synnefo_db_passwd": env.env.synnefo_db_passwd,

        "db_node": env.env.db.ip,

        "domain": env.env.domain,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl, mode=0644)

    with settings(host_string=env.env.db.ip):

        host_info = env.env.ips_info[env.host]

        allow_access_in_db(host_info.ip, "all", "trust")

    try_run("/etc/init.d/gunicorn restart")

def setup_common():

    debug(env.host, " * Setting up snf-common...")

    host_info = env.env.ips_info[env.host]

    install_package("python-objpool")

    install_package("snf-common")

    install_package("python-astakosclient")

    install_package("snf-django-lib")

    install_package("snf-branding")

    tmpl = "/etc/synnefo/common.conf"

    replace = {

        #FIXME:

        "EMAIL_SUBJECT_PREFIX": env.host,

        "domain": env.env.domain,

        "HOST": host_info.fqdn,

        "MAIL_DIR": env.env.mail_dir,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl, mode=0644)

    try_run("mkdir -p {0}; chmod 777 {0}".format(env.env.mail_dir))

    try_run("/etc/init.d/gunicorn restart")

@roles("accounts")

def astakos_loaddata():

    debug(env.host, " * Loading initial data to astakos...")

    cmd = """

    snf-manage loaddata groups

    """

    try_run(cmd)

@roles("accounts")

def astakos_register_components():

    debug(env.host, " * Register services in astakos...")

    cyclades_base_url = "https://%s/cyclades" % env.env.cyclades.fqdn

    pithos_base_url = "https://%s/pithos" % env.env.pithos.fqdn

    astakos_base_url = "https://%s/astakos" % env.env.accounts.fqdn

    cmd = """

    snf-manage component-add "home" --ui-url https://{0}

    snf-manage component-add "cyclades" --base-url {1} --ui-url {1}/ui

    snf-manage component-add "pithos" --base-url {2} --ui-url {2}/ui

    snf-manage component-add "astakos" --base-url {3} --ui-url {3}/ui

    """.format(env.env.cms.fqdn, cyclades_base_url,

               pithos_base_url, astakos_base_url)

    try_run(cmd)

@roles("accounts")

def astakos_register_pithos_view():

    debug(env.host, " * Register pithos view as oauth2 client...")

    pithos_base_url = "https://%s/pithos" % env.env.pithos.fqdn

    cmd = """

    snf-manage oauth2-client-add pithos-view --secret=12345 --is-trusted \

    --url {0}

    """.format('%s/ui/view' % pithos_base_url)

    try_run(cmd)

@roles("accounts")

def add_user():

    debug(env.host, " * adding user %s to astakos..." % env.env.user_email)

    email = env.env.user_email

    name = env.env.user_name

    lastname = env.env.user_lastname

    passwd = env.env.user_passwd

    cmd = """

    snf-manage user-add {0} {1} {2}

    """.format(email, name, lastname)

    try_run(cmd)

    with settings(host_string=env.env.db.ip):

        uid, user_auth_token, user_uuid = get_auth_token_from_db(email)

    cmd = """

    snf-manage user-modify --password {0} {1}

    """.format(passwd, uid)

    try_run(cmd)

@roles("accounts")

def activate_user(user_email=None):

    if not user_email:

        user_email = env.env.user_email

    debug(env.host, " * Activate user %s..." % user_email)

    with settings(host_string=env.env.db.ip):

        uid, user_auth_token, user_uuid = get_auth_token_from_db(user_email)

    cmd = """

    snf-manage user-modify --verify {0}

    snf-manage user-modify --accept {0}

    """.format(uid)

    try_run(cmd)

@roles("accounts")

def setup_astakos():

    debug(env.host, "Setting up snf-astakos-app...")

    setup_gunicorn()

    setup_apache()

    setup_webproject()

    install_package("python-django-south")

    install_package("snf-astakos-app")

    install_package("kamaki")

    tmpl = "/etc/synnefo/astakos.conf"

    replace = {

        "ACCOUNTS": env.env.accounts.fqdn,

        "domain": env.env.domain,

        "CYCLADES": env.env.cyclades.fqdn,

        "PITHOS": env.env.pithos.fqdn,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl, mode=0644)

    if env.csrf_disable:

        cmd = """

cat <<EOF >> /etc/synnefo/astakos.conf

try:

  MIDDLEWARE_CLASSES.remove('django.middleware.csrf.CsrfViewMiddleware')

except:

  pass

EOF

"""

        try_run(cmd)

    try_run("/etc/init.d/gunicorn restart")

    cmd = """

    snf-manage syncdb --noinput

    snf-manage migrate im --delete-ghost-migrations

    snf-manage migrate quotaholder_app

    snf-manage migrate oa2

    """

    try_run(cmd)

@roles("accounts")

def get_service_details(service="pithos"):

    debug(env.host,

          " * Getting registered details for %s service..." % service)

    result = try_run("snf-manage component-list -o id,name,token")

    r = re.compile(r".*%s.*" % service, re.M)

    service_id, _, service_token = r.search(result).group().split()

    # print("%s: %s %s" % (service, service_id, service_token))

    return (service_id, service_token)

@roles("db")

def get_auth_token_from_db(user_email=None):

    if not user_email:

        user_email = env.env.user_email

    debug(env.host,

          " * Getting authentication token and uuid for user %s..."

          % user_email)

    cmd = """

echo "select id, auth_token, uuid, email from auth_user, im_astakosuser \

where auth_user.id = im_astakosuser.user_ptr_id and auth_user.email = '{0}';" \

> /tmp/psqlcmd

su - postgres -c  "psql -w -d snf_apps -f /tmp/psqlcmd"

""".format(user_email)

    result = try_run(cmd)

    r = re.compile(r"(\d+)[ |]*(\S+)[ |]*(\S+)[ |]*" + user_email, re.M)

    match = r.search(result)

    uid, user_auth_token, user_uuid = match.groups()

    # print("%s: %s %s %s" % ( user_email, uid, user_auth_token, user_uuid))

    return (uid, user_auth_token, user_uuid)

@roles("cms")

def cms_loaddata():

    debug(env.host, " * Loading cms initial data...")

    if env.cms_pass:

        debug(env.host, "Aborting. Prerequisites not met.")

        return

    tmpl = "/tmp/sites.json"

    replace = {}

    custom = customize_settings_from_tmpl(tmpl, replace)

    try_put(custom, tmpl)

    tmpl = "/tmp/page.json"

    replace = {}