# Copyright 2014
# The Cloudscaling Group, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from gceapi.api import common as gce_common
from gceapi.api import machine_type_api
from gceapi.api import wsgi as gce_wsgi


class Controller(gce_common.Controller):
    """GCE Machine types controller"""

    def __init__(self, *args, **kwargs):
        super(Controller, self).__init__(machine_type_api.API(),
                                         *args, **kwargs)

    def format_item(self, request, flavor, scope):
        result_dict = {
            "name": flavor["name"],
            "description": "",
            "guestCpus": flavor["vcpus"],
            "memoryMb": flavor["ram"],
            "imageSpaceGb": flavor["disk"],
            # NOTE(Alex): Is not supported by Openstack
            "maximumPersistentDisks": 0,
            # NOTE(Alex): Is not supported by Openstack
            "maximumPersistentDisksSizeGb": '0',
            }

        if "OS-FLV-EXT-DATA:ephemeral" in flavor:
            size = flavor["OS-FLV-EXT-DATA:ephemeral"]
            if size > 0:
                result_dict["scratchDisks"] = [{"diskGb": size}]

        return self._format_item(request, result_dict, scope)


def create_resource():
    return gce_wsgi.GCEResource(Controller())
