# vim: tabstop=4 shiftwidth=4 softtabstop=4

# Copyright 2011 OpenStack LLC.
# All Rights Reserved.
#
#    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.

import webob.exc

from nova.api.openstack.v2 import extensions


class FoxInSocksController(object):

    def index(self, req):
        return "Try to say this Mr. Knox, sir..."


class Foxinsocks(extensions.ExtensionDescriptor):
    """The Fox In Socks Extension"""

    name = "Fox In Socks"
    alias = "FOXNSOX"
    namespace = "http://www.fox.in.socks/api/ext/pie/v1.0"
    updated = "2011-01-22T13:25:27-06:00"

    def __init__(self, ext_mgr):
        ext_mgr.register(self)

    def get_resources(self):
        resources = []
        resource = extensions.ResourceExtension('foxnsocks',
                                               FoxInSocksController())
        resources.append(resource)
        return resources

    def get_actions(self):
        actions = []
        actions.append(extensions.ActionExtension('servers', 'add_tweedle',
                                                    self._add_tweedle))
        actions.append(extensions.ActionExtension('servers', 'delete_tweedle',
                                                    self._delete_tweedle))
        actions.append(extensions.ActionExtension('servers', 'fail',
                                                    self._fail))
        return actions

    def get_request_extensions(self):
        request_exts = []

        def _goose_handler(req, res, body):
            #NOTE: This only handles JSON responses.
            # You can use content type header to test for XML.
            body['flavor']['googoose'] = req.GET.get('chewing')
            return res

        req_ext1 = extensions.RequestExtension('GET',
                                     '/v2/:(project_id)/flavors/:(id)',
                                     _goose_handler)
        request_exts.append(req_ext1)

        def _bands_handler(req, res, body):
            #NOTE: This only handles JSON responses.
            # You can use content type header to test for XML.
            body['big_bands'] = 'Pig Bands!'
            return res

        req_ext2 = extensions.RequestExtension('GET',
                                     '/v2/:(project_id)/flavors/:(id)',
                                     _bands_handler)
        request_exts.append(req_ext2)
        return request_exts

    def _add_tweedle(self, input_dict, req, id):

        return "Tweedle Beetle Added."

    def _delete_tweedle(self, input_dict, req, id):

        return "Tweedle Beetle Deleted."

    def _fail(self, input_dict, req, id):

        raise webob.exc.HTTPBadRequest(explanation='Tweedle fail')
