Synnefo

from __future__ import with_statement

from fabric.api import *

from fabric.contrib.console import confirm

from random import choice

from fabric.operations import run, put

import re

import shutil, os

from functools import wraps

import imp

import ConfigParser

import sys

import tempfile

import ast

from snfdeploy.lib import *

from snfdeploy import massedit

def setup_env(confdir="conf", packages="packages",

              templates="files", cluster_name="ganeti1", autoconf=False, disable_colors=False, key_inject=False):

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

    print(" * Using config files under %s..." % confdir)

    print(" * Using %s and %s for packages and templates accordingly..." % (packages, templates))

    autoconf = ast.literal_eval(autoconf)

    disable_colors = ast.literal_eval(disable_colors)

    env.key_inject = ast.literal_eval(key_inject)

    conf = Conf.configure(confdir=confdir, cluster_name=cluster_name, autoconf=autoconf)

    env.env = Env(conf)

    env.local = autoconf

    env.password = env.env.password

    env.user = env.env.user

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

    if 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],

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

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

        "router": [env.env.router.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_all.deb" % (env.env.packages, package, host_info.os))

            if deb:

                debug(env.host, " * Package %s found in %s..." % (package, env.env.packages))

                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", "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)

    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

    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)

    put(custom, tmpl)

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

    replace = {

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

      }

    custom = customize_settings_from_tmpl(tmpl, replace)

    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, True)

@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, True)

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

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

@roles("nodes")

def check_connectivity():

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

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

@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", True)

@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)

      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")

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

    replace = {}

    custom = customize_settings_from_tmpl(tmpl, replace)

    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:

      put(custom, tmpl)

    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/' /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=False):

    try:

      if env.local:

        return local(cmd, capture=True)

      else:

        return run(cmd)

    except:

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

      if abort:

        raise

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)

    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")

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

    replace = {}

    custom = customize_settings_from_tmpl(tmpl, replace)

    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)

    put(custom, tmpl)

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

    custom = customize_settings_from_tmpl(tmpl, replace)

    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 = """

    echo host all {0} {1}/32 {2} >> /etc/postgresql/8.4/main/pg_hba.conf

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

    try_run(cmd)

    cmd = """

    sed -i 's/\(host.*127.0.0.1.*\)md5/\\1trust/' /etc/postgresql/8.4/main/pg_hba.conf

    """

    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)

    put(custom, tmpl)

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

    try_run(cmd)

    cmd = """

    echo "listen_addresses = '*'" >> /etc/postgresql/8.4/main/postgresql.conf

    """

    try_run(cmd)

    if env.env.testing_vm:

        cmd = """

        echo "fsync=off\nsynchronous_commit=off\nfull_page_writes=off" >> /etc/postgresql/8.4/main/postgresql.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")

    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)

    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,

    }

    custom = customize_settings_from_tmpl(tmpl, replace)

    put(custom, tmpl, mode=0644)

    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_services():

    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" https://{0} home-icon.png

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

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

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

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

               pithos_base_url, astakos_base_url)

    try_run(cmd)

    import_service("astakos", astakos_base_url)

    import_service("pithos", pithos_base_url)

    import_service("cyclades", cyclades_base_url)

    cmd = """

    snf-manage resource-modify --limit 40G pithos.diskspace

    snf-manage resource-modify --limit 2 astakos.pending_app

    snf-manage resource-modify --limit 4 cyclades.vm

    snf-manage resource-modify --limit 40G cyclades.disk

    snf-manage resource-modify --limit 8G cyclades.ram

    snf-manage resource-modify --limit 16 cyclades.cpu

    snf-manage resource-modify --limit 4 cyclades.network.private

    """

    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)

    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

    """

    try_run(cmd)

def import_service(service, base_url):

    try_run("snf-service-export %s %s | snf-manage service-import --json -" %

            (service, base_url))

@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")

    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)

    put(custom, tmpl)

    tmpl = "/tmp/page.json"

    replace = {}

    custom = customize_settings_from_tmpl(tmpl, replace)

    put(custom, tmpl)

    cmd = """

    snf-manage loaddata /tmp/sites.json

    snf-manage loaddata /tmp/page.json

    snf-manage createsuperuser --username=admin --email=admin@{0} --noinput

    """.format(env.env.domain)

    try_run(cmd)

@roles("cms")

def setup_cms():

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

    if env.cms_pass:

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

      return

    with settings(hide("everything")):

        try_run("ping -c1 accounts." + env.env.domain)

    setup_gunicorn()

    setup_apache()

    setup_webproject()

    install_package("snf-cloudcms")

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

    replace = {

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

        }

    custom = customize_settings_from_tmpl(tmpl, replace)

    put(custom, tmpl, mode=0644)

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

    cmd = """

    snf-manage syncdb

    snf-manage migrate --delete-ghost-migrations

    """.format(env.env.domain)

    try_run(cmd)

def setup_nfs_dirs():

    debug(env.host, " * Creating NFS mount point for pithos and ganeti...")

    cmd = """

    mkdir -p {0}

    cd {0}

    mkdir -p data

    chown www-data:www-data data

    chmod g+ws data

    mkdir -p /srv/okeanos

    """.format(env.env.pithos_dir)

    try_run(cmd)

@roles("nodes")

def setup_nfs_clients():

    if env.host == env.env.pithos.ip:

      return

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

    debug(env.host, " * Mounting pithos NFS mount point...")

    with settings(hide("everything")):