Synnefo

# Copyright 2013 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.

from itertools import ifilter

from logging import getLogger

from django.http import HttpResponse

from django.utils import simplejson as json

import datetime

from dateutil.parser import parse as date_parse

from snf_django.lib import api

from snf_django.lib.api import faults, utils

from synnefo.volume import volumes, snapshots, util

from synnefo.db.models import Volume

from synnefo.plankton.utils import image_backend

log = getLogger('synnefo.volume')

def display_null_field(field):

    if field is None:

        return None

    else:

        str(field)

def volume_to_dict(volume, detail=True):

    data = {

        "id": str(volume.id),

        "display_name": display_null_field(volume.name),

        "links": util.volume_to_dict(volume.id),

    }

    if detail:

        details = {

            "status": volume.status.lower(),

            "size": volume.size,

            "display_description": volume.description,

            "created_at": utils.isoformat(volume.created),

            "metadata": dict((m.key, m.value) for m in volume.metadata.all()),

            "snapshot_id": display_null_field(volume.source_snapshot_id),

            "source_volid": display_null_field(volume.source_volume_id),

            "image_id": display_null_field(volume.source_image_id),

            "attachments": get_volume_attachments(volume),

            # TODO:

            "volume_type": None,

            "delete_on_termination": volume.delete_on_termination,

            #"availabilit_zone": None,

            #"bootable": None,

            #"os-vol-tenant-attr:tenant_id": None,

            #"os-vol-host-attr:host": None,

            #"os-vol-mig-status-attr:name_id": None,

            #"os-vol-mig-status-attr:migstat": None,

        }

        data.update(details)

    return data

def get_volume_attachments(volume):

    if volume.machine_id is None:

        return []

    else:

        return [{"server_id": volume.machine_id,

                 "volume_id": volume.id,

                 "device_index": volume.index}]

@api.api_method(http_method="POST", user_required=True, logger=log)

def create_volume(request):

    """Create a new Volume."""

    req = utils.get_request_dict(request)

    log.debug("create_volume %s", req)

    user_id = request.user_uniq

    new_volume = req.get("volume")

    if new_volume is None:

        raise faults.BadRequest("Missing 'volume' attribute.")

    # Get and validate 'name' parameter

    # TODO: auto generate name

    name = new_volume.get("display_name", None)

    if name is None:

        raise faults.BadRequest("Volume 'name' is needed.")

    # Get and validate 'size' parameter

    size = new_volume.get("size")

    if size is None:

        raise faults.BadRequest("Volume 'size' is needed.")

    try:

        size = int(size)

        if size <= 0:

            raise ValueError

    except ValueError:

        raise faults.BadRequest("Volume 'size' needs to be a positive integer"

                                " value. '%s' cannot be accepted." % size)

    # TODO: Fix volume type, validate, etc..

    volume_type = new_volume.get("volume_type", None)

    # Optional parameters

    description = new_volume.get("display_description", "")

    metadata = new_volume.get("metadata", {})

    if not isinstance(metadata, dict):

        msg = "Volume 'metadata' needs to be a dictionary of key-value pairs."\

              " '%s' can not be accepted." % metadata

        raise faults.BadRequest(msg)

    # Id of the volume to clone from

    source_volume_id = new_volume.get("source_volid")

    # Id of the snapshot to create the volume from

    source_snapshot_id = new_volume.get("snapshot_id")

    # Reference to an Image stored in Glance

    source_image_id = new_volume.get("imageRef")

    # TODO: Check that not all of them are used

    server_id = new_volume.get("server_id")

    if server_id is None:

        raise faults.BadRequest("Attribute 'server_id' is mandatory")

    # Create the volume

    volume = volumes.create(user_id=user_id, size=size, name=name,

                            source_volume_id=source_volume_id,

                            source_snapshot_id=source_snapshot_id,

                            source_image_id=source_image_id,

                            volume_type=volume_type, description=description,

                            metadata=metadata, server_id=server_id)

    # Render response

    data = json.dumps(dict(volume=volume_to_dict(volume, detail=False)))

    return HttpResponse(data, status=200)

@api.api_method(http_method="GET", user_required=True, logger=log)

def list_volumes(request, detail=False):

    log.debug('list_volumes detail=%s', detail)

    volumes = Volume.objects.filter(userid=request.user_uniq).order_by("id")

    volumes = utils.filter_modified_since(request, objects=volumes)

    volumes = [volume_to_dict(v, detail) for v in volumes]

    data = json.dumps({'volumes': volumes})

    return HttpResponse(data, content_type="application/json", status=200)

@api.api_method(http_method="DELETE", user_required=True, logger=log)

def delete_volume(request, volume_id):

    log.debug("delete_volume volume_id: %s", volume_id)

    volume = util.get.volume(request.user_uniq, volume_id, for_update=True)

    volumes.delete(volume)

    return HttpResponse(status=202)

@api.api_method(http_method="GET", user_required=True, logger=log)

def get_volume(request, volume_id):

    log.debug('get_volume volume_id: %s', volume_id)

    volume = util.get.volume(request.user_uniq, volume_id)

    data = json.dumps({'volume': volume_to_dict(volume, detail=True)})

    return HttpResponse(data, content_type="application/json", status=200)

@api.api_method(http_method="PUT", user_required=True, logger=log)

def update_volume(request, volume_id):

    req = utils.get_request_dict(request)

    log.debug('update_volume volume_id: %s, request: %s', volume_id, req)

    volume = util.get.volume(request.user_uniq, volume_id, for_update=True)

    new_name = req.get("display_name")

    description = req.get("display_description")

    if new_name is None and description is None:

        raise faults.BadRequest("Nothing to update.")

    if new_name is not None:

        volume = volumes.rename(volume, new_name)

    if description is not None:

        volume = volumes.update_description(volume, description)

    data = json.dumps({'volume': volume_to_dict(volume, detail=True)})

    return HttpResponse(data, content_type="application/json", status=200)

def snapshot_to_dict(snapshot, detail=True):

    owner = snapshot['owner']

    status = snapshot['status']

    progress = "%s%%" % 100 if status == "ACTIVE" else 0

    data = {

        "id": snapshot["uuid"],

        "size": int(snapshot["size"]) >> 30,  # gigabytes

        "display_name": snapshot["name"],

        "display_description": snapshot["description"],

        "status": status,

        "user_id": owner,

        "tenant_id": owner,

        "os-extended-snapshot-attribute:progress": progress,

        #"os-extended-snapshot-attribute:project_id": project,

        "created_at": utils.isoformat(date_parse(snapshot["created_at"])),

        "metadata": snapshot.get("metadata", {}),

        "volume_id": snapshot.get("volume_id"),

        "links": util.snapshot_to_links(snapshot["uuid"])

    }

    return data

@api.api_method(http_method="POST", user_required=True, logger=log)

def create_snapshot(request):

    """Create a new Snapshot."""

    req = utils.get_request_dict(request)

    log.debug("create_snapshot %s", req)

    user_id = request.user_uniq

    new_snapshot = req.get("snapshot")

    if new_snapshot is None:

        raise faults.BadRequest("Missing 'snapshot' attribute.")

    # Get and validate 'name' parameter

    # TODO: auto generate name

    metadata = new_snapshot.get("metadata", {})

    if not isinstance(metadata, dict):

        msg = "Snapshot 'metadata' needs to be a dictionary of key-value"\

              " pairs. '%s' can not be accepted." % metadata

        raise faults.BadRequest(msg)

    volume_id = new_snapshot.get("volume_id", None)

    if volume_id is None:

        raise faults.BadRequest("'volume_id' attribute is missing.")

    volume = util.get_volume(user_id, volume_id, for_update=True,

                             exception=faults.BadRequest)

    name = new_snapshot.get("display_name", None)

    if name is None:

        name = "snapshot_volume_%s_%s" %\

            (volume.id, str(datetime.datetime.now()))

    description = new_snapshot.get("display_description", "")

    # TODO: What to do with force ?

    force = new_snapshot.get("force", False)

    if not isinstance(force, bool):

        raise faults.BadRequest("Invalid value for 'force' attribute.")

    snapshot = snapshots.create(user_id=user_id, volume=volume, name=name,

                                description=description, metadata=metadata,

                                force=force)

    # Render response

    data = json.dumps(dict(snapshot=snapshot_to_dict(snapshot, detail=False)))

    return HttpResponse(data, status=202)

@api.api_method(http_method="GET", user_required=True, logger=log)

def list_snapshots(request, detail=False):

    log.debug('list_snapshots detail=%s', detail)

    since = utils.isoparse(request.GET.get('changes-since'))

    with image_backend(request.user_uniq) as backend:

        snapshots = backend.list_snapshots()

        if since:

            updated_since = lambda snap:\

                date_parse(snap["updated_at"]) >= since

            snapshots = ifilter(updated_since, snapshots)

            if not snapshots:

                return HttpResponse(status=304)

    snapshots = sorted(snapshots, key=lambda x: x['id'])

    snapshots_dict = [snapshot_to_dict(snapshot, detail)

                      for snapshot in snapshots]

    data = json.dumps(dict(snapshots=snapshots_dict))

    return HttpResponse(data, status=200)

@api.api_method(http_method="DELETE", user_required=True, logger=log)

def delete_snapshot(request, snapshot_id):

    log.debug("delete_snapshot snapshot_id: %s", snapshot_id)

    snapshot = util.get_snapshot(request.user_uniq, snapshot_id)

    snapshots.delete(snapshot)

    return HttpResponse(status=202)

@api.api_method(http_method="GET", user_required=True, logger=log)

def get_snapshot(request, snapshot_id):

    log.debug('get_snapshot snapshot_id: %s', snapshot_id)

    snapshot = util.get_snapshot(request.user_uniq, snapshot_id)

    data = json.dumps({'snapshot': snapshot_to_dict(snapshot, detail=True)})

    return HttpResponse(data, content_type="application/json", status=200)

@api.api_method(http_method="PUT", user_required=True, logger=log)

def update_snapshot(request, snapshot_id):

    req = utils.get_request_dict(request)

    log.debug('update_snapshot snapshot_id: %s, request: %s', snapshot_id, req)

    snapshot = util.get_snapshot(request.user_uniq, snapshot_id)

    new_name = req.get("display_name")

    new_description = req.get("display_description")

    if new_name is None and new_description is None:

        raise faults.BadRequest("Nothing to update.")

    if new_name is not None:

        snapshot = snapshots.rename(snapshot, new_name)

    if new_description is not None:

        snapshot = snapshots.update_description(snapshot, new_description)

    data = json.dumps({'snapshot': snapshot_to_dict(snapshot, detail=True)})

    return HttpResponse(data, content_type="application/json", status=200)