Synnefo

# vim: ts=4 sts=4 et ai sw=4 fileencoding=utf-8

#

# Copyright © 2010 Greek Research and Technology Network

#

from django.conf import settings

from piston.handler import BaseHandler, AnonymousBaseHandler

from synnefo.api.faults import fault, noContent, accepted, created

from synnefo.api.helpers import instance_to_server, paginator

from synnefo.util.rapi import GanetiRapiClient, GanetiApiError

from synnefo.db.models import VirtualMachine, Image, User, id_from_instance_name

from util.rapi import GanetiRapiClient

if settings.GANETI_CLUSTER_INFO:

    rapi = GanetiRapiClient(*settings.GANETI_CLUSTER_INFO)

else:

    rapi = None

ganeti_prefix_id = settings.GANETI_PREFIX_ID

VERSIONS = [

    {

        "status": "CURRENT",

        "id": "v1.0",

        "docURL" : "http://docs.rackspacecloud.com/servers/api/v1.0/cs-devguide-20090714.pdf ",

        "wadl" : "http://docs.rackspacecloud.com/servers/api/v1.0/application.wadl"

    },

]

class VersionHandler(AnonymousBaseHandler):

    allowed_methods = ('GET',)

    def read(self, request, number=None):

        if number is None:

            versions = map(lambda v: {

                        "status": v["status"],

                        "id": v["id"],

                    }, VERSIONS)

            return { "versions": versions }

        else:

            for version in VERSIONS:

                if version["id"] == number:

                    return { "version": version }

            raise fault.itemNotFound

class ServerHandler(BaseHandler):

    allowed_methods = ('GET', 'POST', 'PUT', 'DELETE')

    def read(self, request, id=None):

        from time import sleep

        sleep(1)

        #TODO: delete the sleep once the mock objects are removed

        if id is None:

            return self.read_all(request)

        elif id == "detail":

            return self.read_all(request, detail=True)

        else:

            return self.read_one(request, id)

    def read_one(self, request, id):

        if not rapi: # No ganeti backend. Return mock objects

            servers = VirtualMachine.objects.filter(owner=User.objects.all()[0])

            return { "server": servers[0] }

        try:

            instance = rapi.GetInstance(id)

            return { "server": instance_to_server(instance) }

        except GanetiApiError:

            raise fault.itemNotFound

    @paginator

    def read_all(self, request, detail=False):

        if not rapi: # No ganeti backend. Return mock objects

            if detail:

                virtual_servers = VirtualMachine.objects.filter(owner=User.objects.all()[0])

                #get the first user, since we don't have any user data yet

                virtual_servers_list = [{'status': server.state, 'flavorId': server.flavor, \

                    'name': server.name, 'id': server.id, 'imageId': server.imageid, 

                    'metadata': {'Server_Label': server.server_label, \

                    'Image_Version': server.image_version}, \

                    'hostId': '9e107d9d372bb6826bd81d3542a419d6',  \

                    'addresses': {'public': ['67.23.10.133'], 'private': ['10.176.42.17']}} \

                        for server in virtual_servers]

                #pass some fake data regarding ip, since we don't have any such data

                return { "servers":  virtual_servers_list }                

            else:

                virtual_servers = VirtualMachine.objects.filter(owner=User.objects.all()[0])

                return { "servers": [ { "id": s.id, "name": s.name } for s in virtual_servers ] }

        if not detail:

            instances = rapi.GetInstances(bulk=False)

            servers = [ { "id": id, "name": id } for id in instances ]

        else:

            instances = rapi.GetInstances(bulk=True)

            servers = []

            for instance in instances:

                servers.append(instance_to_server(instance))

        return { "servers": servers }

    def create(self, request):

        print 'create machine was called'

        rapi.CreateInstance('create', 'machine-unwebXYZ', 'plain', [{"size": 5120}], [{}], os='debootstrap+default', ip_check=False, name_check=False,pnode="store68", beparams={'auto_balance': True, 'vcpus': 2, 'memory': 1024})

        #TODO: replace with real data from request.POST

        return accepted

    def update(self, request, id):

        return noContent

    def delete(self, request, id):

        machine = 'machine-XXX' #VirtualMachine.objects.get(id=id_from_instance_name(id))

        print 'deleting machine %s' % machine

        rapi.DeleteInstance(machine.name)

        return accepted

class ServerAddressHandler(BaseHandler):

    allowed_methods = ('GET', 'PUT', 'DELETE')

    def read(self, request, id, type=None):

        """List IP addresses for a server"""

        if type is None:

            pass

        elif type == "private":

            pass

        elif type == "public":

            pass

        return {}

    def update(self, request, id, address):

        """Share an IP address to another in the group"""

        return accepted

    def delete(self, request, id, address):

        """Unshare an IP address"""

        return accepted

class ServerActionHandler(BaseHandler):

    allowed_methods = ('POST', 'DELETE', 'GET', 'PUT')

#TODO: remove GET/PUT

    def read(self, request, id):

        return accepted

    def create(self, request, id):

        """Reboot, rebuild, resize, confirm resized, revert resized"""

        machine = 'machine-XXX' #VirtualMachine.objects.get(id=id_from_instance_name(id))

        reboot_request = request.POST.get('reboot', None)

        shutdown_request = request.POST.get('shutdown', None)

        if reboot_request:

            print 'reboot was asked, with options: %s' % reboot_request   

            rapi.RebootInstance(machine)

        elif shutdown_request:

            print 'shutdown was asked, with options: %s' % shutdown_request               

            rapi.ShutdownInstance(machine)

        return accepted

    def delete(self, request, id):

        """Delete an Instance"""

        return accepted

    def update(self, request, id):

        return noContent

#read is called on GET requests

#create is called on POST, and creates new objects, and should return them (or rc.CREATED.)

#update is called on PUT, and should update an existing product and return them (or rc.ALL_OK.)

#delete is called on DELETE, and should delete an existing object. Should not return anything, just rc.DELETED.'''

class ServerBackupHandler(BaseHandler):

    """ Backup Schedules are not implemented yet, return notImplemented """

    allowed_methods = ('GET', 'POST', 'DELETE')

    def read(self, request, id):

        raise fault.notImplemented

    def create(self, request, id):

        raise fault.notImplemented

    def delete(self, request, id):

        raise fault.notImplemented

class FlavorHandler(BaseHandler):

    allowed_methods = ('GET',)

    flavors = [

          {

            "id" : 1,

            "name" : "256 MB Server",

            "ram" : 256,

            "disk" : 10

          },

          {

            "id" : 2,

            "name" : "512 MB Server",

            "ram" : 512,

            "disk" : 20

          }

        ]

    def read(self, request, id=None):

        """

        List flavors or retrieve one

        Returns: OK

        Faults: cloudServersFault, serviceUnavailable, unauthorized,

                badRequest, itemNotFound

        """

        if id is None:

            simple = map(lambda v: {

                        "id": v["id"],

                        "name": v["name"],

                    }, self.flavors)

            return { "flavors": simple }

        elif id == "detail":

            return { "flavors": self.flavors }

        else:

            for flavor in self.flavors:

                if str(flavor["id"]) == id:

                    return { "flavor": flavor }

            raise fault.itemNotFound

class ImageHandler(BaseHandler):

    allowed_methods = ('GET', 'POST')

    def read(self, request, id=None):

        """

        List images or retrieve one

        Returns: OK

        Faults: cloudServersFault, serviceUnavailable, unauthorized,

                badRequest, itemNotFound

        """

        images = Image.objects.all()

        if not rapi: # No ganeti backend. Return mock objects

            if id == "detail":

                return { "images": images }

            elif id is None:

                return { "images": [ { "id": s.id, "name": s.name } for s in images ] }

            else:

                return { "image": images[0] }

        if id is None:

            return {}

        elif id == "detail":

            return {}

        else:

            raise fault.itemNotFound

    def create(self, request):

        """Create a new image"""

        return accepted

class SharedIPGroupHandler(BaseHandler):

    allowed_methods = ('GET', 'POST', 'DELETE')

    def read(self, request, id=None):

        """List Shared IP Groups"""

        if id is None:

            return {}

        elif id == "detail":

            return {}

        else:

            raise fault.itemNotFound

    def create(self, request, id):

        """Creates a new Shared IP Group"""

        return created

    def delete(self, request, id):

        """Deletes a Shared IP Group"""

        return noContent

class LimitHandler(BaseHandler):

    allowed_methods = ('GET',)

    # XXX: hookup with @throttle

    rate = [

        {

           "verb" : "POST",

           "URI" : "*",

           "regex" : ".*",

           "value" : 10,

           "remaining" : 2,

           "unit" : "MINUTE",

           "resetTime" : 1244425439

        },

        {

           "verb" : "POST",

           "URI" : "*/servers",

           "regex" : "^/servers",

           "value" : 25,

           "remaining" : 24,

           "unit" : "DAY",

           "resetTime" : 1244511839

        },

        {

           "verb" : "PUT",

           "URI" : "*",

           "regex" : ".*",

           "value" : 10,

           "remaining" : 2,

           "unit" : "MINUTE",

           "resetTime" : 1244425439

        },

        {

           "verb" : "GET",

           "URI" : "*",

           "regex" : ".*",

           "value" : 3,

           "remaining" : 3,

           "unit" : "MINUTE",

           "resetTime" : 1244425439

        },

        {

           "verb" : "DELETE",

           "URI" : "*",

           "regex" : ".*",

           "value" : 100,

           "remaining" : 100,

           "unit" : "MINUTE",

           "resetTime" : 1244425439

        }

    ]

    absolute = {

        "maxTotalRAMSize" : 51200,

        "maxIPGroups" : 50,

        "maxIPGroupMembers" : 25

    }

    def read(self, request):

        return { "limits": {

                "rate": self.rate,

                "absolute": self.absolute,

               }

            }