Synnefo

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

import re

import logging

from itertools import chain

from urllib import unquote

from django.shortcuts import render_to_response

from django.template import RequestContext

from django.conf import settings

from django.core.exceptions import PermissionDenied

from django.http import HttpResponseRedirect

from django.core.urlresolvers import reverse

from synnefo.lib.astakos import get_user

from kamaki.clients.compute import ComputeClient

from kamaki.clients.cyclades import CycladesClient

from kamaki.clients.image import ImageClient

from kamaki.clients import ClientError

AUTH_COOKIE_NAME = getattr(settings, 'SAAS_AUTH_COOKIE_NAME', '_pithos2_a')

HOSTNAME = getattr(settings, 'SAAS_HOSTNAME', None)

API = getattr(settings, 'SAAS_API', None)

IMAGE_ID = getattr(settings, 'SAAS_IMAGE_ID', None)

FLAVOR_ID = getattr(settings, 'SAAS_FLAVOR_ID', None)

UI_LOGIN_URL = getattr(settings, 'UI_LOGIN_URL', None)

CLOUDBAR = getattr(settings, 'SAAS_CLOUDBAR_ACTIVE_SERVICE', None)

log = logging.getLogger('snf-saas-app')

def get_token_from_cookie(request, cookiename):

    """

    Extract token from the cookie name provided. Cookie should be in the same

    form as astakos service sets its cookie contents::

        <user_uniq>|<user_token>

    """

    try:

        cookie_content = unquote(request.COOKIES.get(cookiename, None))

        return cookie_content.split("|")[1]

    except AttributeError:

        pass

    return None

def token_check(func):

    """

    Mimic csrf security check using user auth token.

    """

    def wrapper(request, *args, **kwargs):

        if not hasattr(request, 'user'):

            raise PermissionDenied

        token = request.POST.get('token', None)

        if token and token != request.user.get('auth_token', None):

            return func(request, *args, **kwargs)

        raise PermissionDenied

    return wrapper

def user_required(func):

    """

    Django view wrapper that identifies user from token

    """

    def wrapper(request, *args, **kwargs):

        token = get_token_from_cookie(request, AUTH_COOKIE_NAME)

        get_user(request, settings.ASTAKOS_URL, fallback_token=token)

        print token

        has_perm = False

        if hasattr(request, 'user') and request.user:

            has_perm = True

        if not has_perm:

            return HttpResponseRedirect(UI_LOGIN_URL)

        logging.debug("User %s accessed helpdesk view" % (request.user_uniq))

        return func(request, *args, **kwargs)

    return wrapper

def get_saas_servers(servers, token, client):

    saas_vms = []

    for server in servers:

        details = client.get_server_details(server["id"])

        if details.has_key("metadata"):

            meta = details["metadata"]["values"]

            if meta.has_key("saas") and meta["saas"] == "True":

                saas_vms.append((server["id"], server["name"]))

    return saas_vms

@user_required

def select_software(request):

    TOKEN = get_token_from_cookie(request, AUTH_COOKIE_NAME)

    client = CycladesClient(API,TOKEN)

    servers = client.list_servers()

    saas_vms = get_saas_servers(servers, TOKEN, client)

    log.debug("SaaS server list: %s", saas_vms)

    if request.method ==  'POST':

        software = request.POST.getlist('software')

        log.debug("Selected software: %s", software)

        if len(software) > 0:

            server_name = "SaaS [%s]" % ",".join(software)

            server = client.create_server(server_name, FLAVOR_ID, IMAGE_ID, None)

            url = '/saas/status/%s' % str(server["id"])

            client.update_server_metadata(server["id"], saas="True")

            return HttpResponseRedirect(url)

        else:

            return render_to_response("saas/index.html", {'vms': saas_vms, 'cloud': CLOUDBAR}, context_instance=RequestContext(request))

    else:

        return render_to_response("saas/index.html", {'vms': saas_vms, 'cloud': CLOUDBAR}, context_instance=RequestContext(request))

@user_required

def vm_status(request,vm_id):

    TOKEN = get_token_from_cookie(request, AUTH_COOKIE_NAME)

    client = ComputeClient(API, TOKEN)

    details = client.get_server_details(vm_id)

    status = details["status"]

    servers = client.list_servers()

    saas_vms = get_saas_servers(servers, TOKEN, client)

    if status == 'ACTIVE':

        active = True

        ip = details["attachments"]["values"][0]["ipv4"]

        link_url = "%s?ip_address=%s&os=windows&rdp=1&srv=%d&username=Administrator&domain=snf-%d&hostname=%s" % (

            reverse("ui_machines_connect"), ip, int(vm_id), int(vm_id), HOSTNAME)

        c = {'active': active, 'ip': ip, 'link_url': link_url, 'vms': saas_vms, 'cloud': CLOUDBAR}

    else:

        active = False

        c = {'active': active, 'vms': saas_vms, 'cloud': CLOUDBAR}

    return render_to_response("saas/status.html", c, context_instance=RequestContext(request))