# Copyright (c) 2013 Rackspace, 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.

import falcon
from wsgiref import simple_server

from marconi.common import config
import marconi.openstack.common.log as logging
from marconi import transport
from marconi.transport.wsgi import claims
from marconi.transport.wsgi import messages
from marconi.transport.wsgi import queues
from marconi.transport.wsgi import stats

OPTIONS = {
    'bind': '0.0.0.0',
    'port': 8888
}

cfg = config.namespace('drivers:transport:wsgi').from_options(**OPTIONS)

LOG = logging.getLogger(__name__)


class Driver(transport.DriverBase):

    def __init__(self, storage):
        super(Driver, self).__init__(storage)

        self.app = falcon.API()

        # Queues Endpoints
        queue_controller = self.storage.queue_controller
        queue_collection = queues.CollectionResource(queue_controller)
        self.app.add_route('/v1/{project_id}/queues', queue_collection)

        queue_item = queues.ItemResource(queue_controller)
        self.app.add_route('/v1/{project_id}/queues/{queue_name}', queue_item)

        stats_endpoint = stats.Resource(queue_controller)
        self.app.add_route('/v1/{project_id}/queues/{queue_name}'
                           '/stats', stats_endpoint)

        # Messages Endpoints
        message_controller = self.storage.message_controller
        msg_collection = messages.CollectionResource(message_controller)
        self.app.add_route('/v1/{project_id}/queues/{queue_name}'
                           '/messages', msg_collection)

        msg_item = messages.ItemResource(message_controller)
        self.app.add_route('/v1/{project_id}/queues/{queue_name}'
                           '/messages/{message_id}', msg_item)

        # Claims Endpoints
        claim_controller = self.storage.claim_controller
        claim_collection = claims.CollectionResource(claim_controller)
        self.app.add_route('/v1/{project_id}/queues/{queue_name}'
                           '/claims', claim_collection)

        claim_item = claims.ItemResource(claim_controller)
        self.app.add_route('/v1/{project_id}/queues/{queue_name}'
                           '/claims/{claim_id}', claim_item)

    def listen(self):
        msg = _("Serving on host %(bind)s:%(port)s") % {"bind": cfg.bind,
                                                        "port": cfg.port}
        LOG.debug(msg)
        httpd = simple_server.make_server(cfg.bind, cfg.port, self.app)
        httpd.serve_forever()