blob: c96be0f3710b0eff1998d1f8502b3f930eff53a6 [file] [log] [blame]
#!/usr/bin/env python
#
# Copyright 2016 the original author or authors.
#
# 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 os
from klein import Klein
from simplejson import dumps, load
from structlog import get_logger
from twisted.internet import reactor, endpoints
from twisted.internet.defer import inlineCallbacks, returnValue
from twisted.internet.tcp import Port
from twisted.web.server import Site
from twisted.web.static import File
from werkzeug.exceptions import BadRequest
log = get_logger()
class WebServer(object):
app = Klein()
def __init__(self, port, work_dir, grpc_client):
self.port = port
self.site = None
self.work_dir = work_dir
self.grpc_client = grpc_client
self.swagger_ui_root_dir = os.path.abspath(
os.path.join(os.path.dirname(__file__), '../swagger_ui'))
self.tcp_port = None
self.shutting_down = False
@inlineCallbacks
def start(self):
log.debug('starting')
yield self._open_endpoint()
log.info('started')
returnValue(self)
@inlineCallbacks
def stop(self):
log.debug('stopping')
self.shutting_down = True
if self.tcp_port is not None:
assert isinstance(self.tcp_port, Port)
yield self.tcp_port.socket.close()
log.info('stopped')
@inlineCallbacks
def _open_endpoint(self):
endpoint = endpoints.TCP4ServerEndpoint(reactor, self.port)
self.site = Site(self.app.resource())
self.tcp_port = yield endpoint.listen(self.site)
log.info('web-server-started', port=self.port)
self.endpoint = endpoint
def reload_generated_routes(self):
for fname in os.listdir(self.work_dir):
if fname.endswith('_gw.py'):
module_name = fname.replace('.py', '')
m = __import__(module_name)
assert hasattr(m, 'add_routes')
m.add_routes(self.app, self.grpc_client)
log.info('routes-loaded', module=module_name)
# static swagger_ui website as landing page (for now)
@app.route('/', branch=True)
def static(self, request):
try:
log.debug(request=request)
return File(self.swagger_ui_root_dir)
except Exception, e:
log.exception('file-not-found', request=request)
# static swagger.json file to serve the schema
@app.route('/v1/swagger.json')
def swagger_json(self, request):
try:
return File(os.path.join(self.work_dir, 'swagger.json'))
except Exception, e:
log.exception('file-not-found', request=request)