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 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, HttpResponse

from django.core.urlresolvers import reverse

from synnefo.lib.astakos import get_user, get_token_from_cookie

from kamaki.clients import ClientError

import saas.ui.functions as functions

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

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

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

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

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

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

# --------------------------------------------------------------------

# Authendicate user wrapper

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)

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

            pass

        else:

            uri = request.build_absolute_uri()

            if '?' in uri:

                uri = uri[:uri.find('?')]

            uri = UI_LOGIN_URL + '?next=' + uri

            return HttpResponseRedirect(uri)

        log.debug("User %s accessed SaaS" % (request.user_uniq))

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

    return wrapper

# --------------------------------------------------------------------

# SaaS's views

# ----------------------------

@user_required

def select_software(request):

    token = get_token_from_cookie(request, AUTH_COOKIE_NAME)

    try:

        saas_vm = functions.get_saas_server(token)

        # There exists a saas server (redirect to status)

        log.debug("Server %s found in `select_software', redirecting.."

                  % saas_vm)

        url = reverse("saas:vm-status")

        return HttpResponseRedirect(url)

    except LookupError:

        pass

    log.debug("No SaaS Server found, let's create one..")

    if request.method == 'POST':

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

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

        if len(software) > 0:

            server = functions.create_saas_server(token,software)

            log.debug("Server %s created, redirecting.." % server)

            url = reverse("saas:vm-status")

            return HttpResponseRedirect(url)

        else:

            log.debug("No software selected, nothing to do")

            return render_to_response("saas/index.html",

                    {'cloud': CLOUDBAR},

                    context_instance=RequestContext(request))

    else:

        return render_to_response("saas/index.html",

                {'cloud': CLOUDBAR},

                context_instance=RequestContext(request))

# ----------------------------

@user_required

def vm_status(request):

    token = get_token_from_cookie(request, AUTH_COOKIE_NAME)

    try:

        saas_server = functions.get_saas_server(token)

    except LookupError:

        log.debug("SaaS Server does not exist")

        url = reverse("saas:saas-software")

        return HttpResponseRedirect(url)

    server_id = saas_server['id']

    status = saas_server['status']

    if status == 'ACTIVE':

        ip = saas_server['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(server_id),

                int(server_id), HOSTNAME))

        enc_password = saas_server['metadata']['values']['saasPass']

        c = {'status': status, 'ip': ip, 'link_url': link_url,

             'cloud': CLOUDBAR, 'vm_id': server_id,

             'vm_username': "Administrator",

             'vm_password': functions.decrypt(enc_password)}

    elif status == 'BUILD':

        c = {'status': status, 'cloud': CLOUDBAR}

    else:

        raise PemissionDenied

    return render_to_response("saas/status.html", c,

            context_instance=RequestContext(request))

# ----------------------------

@user_required

def index(request):

    token = get_token_from_cookie(request, AUTH_COOKIE_NAME)

    try:

        saas_vm = functions.get_saas_server(token)

        url = reverse("saas:vm-status")

        return HttpResponseRedirect(url)

    except LookupError:

        url = reverse("saas:saas-software")

        return HttpResponseRedirect(url)

# ----------------------------

@user_required

def vm_destroy(request,vm_id):

    log.debug("Destroying server %s.." % vm_id)

    token = get_token_from_cookie(request, AUTH_COOKIE_NAME)

    functions.delete_saas_server(token, vm_id)

    url = reverse("saas:saas-index")

    return HttpResponseRedirect(url)