webapi.py

# Copyright (c) 2018 Maen Artimy
# 
# 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 json
import os
import mimetypes
#import requests
#import datetime as dt
#import sys

# from ryu.base import app_manager
from ryu.app.wsgi import ControllerBase
# from ryu.app.wsgi import WSGIApplication
from ryu.app.wsgi import route
from ryu.app.wsgi import Response

# ref: https://tools.itef.org/doc/python-routes/html


class WebApi(ControllerBase):
    def __init__(self, req, link, data, **config):
        super(WebApi, self).__init__(req, link, data, **config)
        self.api = data["webctl"]
        self.lists = data["lists"]
        self.rootdir = os.path.dirname(os.path.abspath(__file__))

    def make_response(self, filename):
        filetype, encoding = mimetypes.guess_type(filename)
        if not filetype:
            filetype = 'application/octet-stream'
        res = Response(content_type=filetype)
        res.body = open(filename, 'rb').read()
        return res

    @route('monitor', '/status', methods=['GET'])
    def get_flow_stats(self, req, **_kwargs):
        """Get switch data
        """
        if req.GET['status'] and req.GET['dpid']:
            res = Response(content_type="application/json")
            res.json = self.api.get_stats(req.GET['status'], req.GET['dpid'])
            return res
        return Response(status=404) # Resource does not exist

    @route('monitor', '/data', methods=['GET'])
    def get_flow_data(self, req, **_kwargs):
        """Get switch data
        """
        # TODO: merge with get_flow_stats 
        if req.GET:
            lst = {}
            if req.GET.get("list") == "switches":
                lst = {t[0]:t[0] for t in self.api.get_switches()}
            if req.GET.get("switchdesc"):
                dpid = int(req.GET["switchdesc"])
                lst = self.api.get_switch_desc(dpid)
            if req.GET.get("portdesc"):
                dpid = int(req.GET["portdesc"])
                lst = self.api.get_port_desc(dpid)
            if req.GET.get("portstat"):
                dpid = int(req.GET["portstat"])
                lst = self.api.get_port_stat(dpid)
            if req.GET.get("flowsumm"):
                dpid = int(req.GET["flowsumm"])
                lst = self.api.get_flow_summary(dpid)
            if req.GET.get("tablestat"):
                dpid = int(req.GET["tablestat"])
                lst = self.api.get_table_stat(dpid)

            res = Response(content_type="application/json")
            res.json = lst
            return res
        return Response(status=400)  # bad request

    @route('monitor', '/topology', methods=['GET'])
    def get_topology(self, req, **_kwargs):
        """Get topology info
        """
        res = Response(content_type="application/json")
        reply = self.api.get_topology_data()
        res.json = reply
        return res       

    @route('monitor', '/meterform', methods=['GET', 'POST'])
    def get_meter_form(self, req, **_kwargs):
        """Connect with meter form
        """ 
        if req.POST:
            res = Response()
            res.body = self.api.process_meter_message(req.json)
            return res
        return Response(status=400)  # bad request

    @route('monitor', '/groupform', methods=['GET', 'POST'])
    def get_group_form(self, req, **_kwargs):
        """Connect with group form
        """ 
        if req.POST:
            res = Response()
            res.body = self.api.process_group_message(req.json)
            return res
        return Response(status=400)  # bad request

    @route('monitor', '/flowform', methods=['GET', 'POST'])
    def get_flow_form(self, req, **_kwargs):
        """Connect with flow form
        """ 
        if req.POST:
            res = Response()
            res.body = self.api.process_flow_message(req.json)
            return res
        elif req.GET and "list" in req.GET:
            lst = {}
            if req.GET["list"] == "actions":
                lst = self.lists["actions"]
            elif req.GET["list"] == "matches":
                lst = self.lists["matches"]
            elif req.GET["list"] == "switches":
                lst = {t[0]:str(t[0]) for t in self.api.get_switches()}
                #print(lst)

            res = Response(content_type="application/json")
            res.json = lst
            return res
        return Response(status=400)  # bad request

    @route('monitor', '/logs', methods=['GET'])
    def get_logs(self, req, **_kwargs):
        """Get log mesages
        """
        if req.GET:
            logs = self.api.read_logs()
            res = Response(content_type="application/json")
            res.json = logs
            return res
        return Response(status=400)  # bad request

    @route('monitor', '/home/{filename:.*}', methods=['GET'])
    def get_filename(self, req, filename, **_kwargs):
        """Get monitoring information from ofctl_rest app
        """
        if (filename == "" or filename == None):
            filename = "index.html"
        try:
            filename = os.path.join(self.rootdir, filename)
            return self.make_response(filename)
        except IOError:
            return Response(status=400)

    # def get_log_SSE(self, req, **_kwargs):
    #     # Support for SSE
    #     # https://streamdata.io/blog/server-sent-events/

    #     def eventStream():
    #         while True:
    #             msgs = self.api.get_messages()
    #             body = ['data: {}'.format(l) for l in msgs]

    #             self.api.clear_messages()
    #             yield '\n'.join(body) + '\n\n'

    #     res = Response(content_type="text/event-stream")
    #     res.body = eventStream().next()
    #     return res