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 redirect, get_object_or_404, render_to_response

from django.template import Context, loader

from django.conf import settings

from django.views.generic.simple import direct_to_template

from django.core.exceptions import PermissionDenied

from django.http import Http404, HttpResponse, HttpResponseRedirect

from django.core.urlresolvers import reverse

from synnefo.lib.astakos import get_user

from synnefo.saas.forms import VMSettings

from itertools import chain

from urllib import unquote

from kamaki.clients.compute import ComputeClient

from kamaki.clients.cyclades import CycladesClient

from kamaki.clients.image import ImageClient

from kamaki.clients import ClientError

# Demo settings

API = "https://cyclades.okeanos.io/api/v1.1"

TOKEN = "OwmGvNejXe3KQSh4/F2X+g=="

IMAGE_ID = "cba883ae-c261-4df6-9f14-37b39cbf56da"

FLAVOR_ID = 81

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, ASTAKOS_URL, fallback_token=token)

        print token

        has_perm = False

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

            has_perm = True

        if not has_perm:

            raise PermissionDenied

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

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

    return wrapper

def index(request):

    form = VMSettings()

    return render_to_response("saas/index.html", {'form': form})

def select_software(request):

    if request.method ==  'POST':

        form = VMSettings(request.POST)

        if form.is_valid():

            client = ComputeClient(API, TOKEN)

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

            server_name = form.cleaned_data['name']

            #Logging

            log.info("Software selected")

            log.info(software)

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

            log.info(server)

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

            return HttpResponseRedirect(url)

        else:

            return render_to_response("saas/index.html",{'form': form})

    else:

        return HttpResponseRedirect("/saas/software/")

def vm_status(request,vm_id):

    log.info(vm_id)

    client = ComputeClient(API, TOKEN)

    details = client.get_server_details(vm_id)

    status = details["status"]

    #Logging

    log.info(status)

    if status == 'ACTIVE':

        active = True

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

        rdp_tuple = (ip,int(vm_id),int(vm_id))

        rdp = "https://cyclades.okeanos.io/machines/connect?ip_address=%s&os=windows&rdp=1&srv=%d&username=Administrator&domain=snf-%d&hostname=gate.okeanos.io"%rdp_tuple

        template = loader.get_template('saas/status.html')

        context = Context({'active': active, 'ip': ip, 'rdp': rdp})

    else:

        active = False

        template = loader.get_template('saas/status.html')

        context = Context({'active': active,})

    return HttpResponse(template.render(context))