sdrangel/sdrbase/webapi/webapirequestmapper.cpp

///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2017 Edouard Griffiths, F4EXB.                                  //
//                                                                               //
// Swagger server adapter interface                                              //
//                                                                               //
// This program is free software; you can redistribute it and/or modify          //
// it under the terms of the GNU General Public License as published by          //
// the Free Software Foundation as version 3 of the License, or                  //
//                                                                               //
// This program is distributed in the hope that it will be useful,               //
// but WITHOUT ANY WARRANTY; without even the implied warranty of                //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the                  //
// GNU General Public License V3 for more details.                               //
//                                                                               //
// You should have received a copy of the GNU General Public License             //
// along with this program. If not, see <http://www.gnu.org/licenses/>.          //
///////////////////////////////////////////////////////////////////////////////////

//#include <QDirIterator>

#include "httpdocrootsettings.h"
#include "webapirequestmapper.h"
#include "SWGInstanceSummaryResponse.h"
#include "SWGInstanceDevicesResponse.h"
#include "SWGInstanceChannelsResponse.h"
#include "SWGErrorResponse.h"

WebAPIRequestMapper::WebAPIRequestMapper(QObject* parent) :
    HttpRequestHandler(parent),
    m_adapter(0)
{
    qtwebapp::HttpDocrootSettings docrootSettings;
    docrootSettings.path = ":/";
    m_staticFileController = new qtwebapp::StaticFileController(docrootSettings, parent);
}

WebAPIRequestMapper::~WebAPIRequestMapper()
{
    delete m_staticFileController;
}

void WebAPIRequestMapper::service(qtwebapp::HttpRequest& request, qtwebapp::HttpResponse& response)
{
    if (m_adapter == 0) // format service unavailable if adapter is null
    {
        response.write("Service not available");
        response.setStatus(500,"Service not available");
    }
    else // normal processing
    {
        QByteArray path=request.getPath();

        if (path == WebAPIAdapterInterface::instanceSummaryURL)
        {
            if (request.getMethod() == "GET")
            {
                Swagger::SWGInstanceSummaryResponse normalResponse;
                Swagger::SWGErrorResponse errorResponse;

                int status = m_adapter->instanceSummary(normalResponse, errorResponse);

                if (status == 200) {
                    response.write(normalResponse.asJson().toUtf8());
                } else {
                    response.write(errorResponse.asJson().toUtf8());
                }

                response.setStatus(status);
            }
            else
            {
                response.write("Invalid HTTP method");
                response.setStatus(405,"Invalid HTTP method");
            }
        }
        else if (path == WebAPIAdapterInterface::instanceDevicesURL)
        {
            Swagger::SWGInstanceDevicesResponse normalResponse;
            Swagger::SWGErrorResponse errorResponse;

            if (request.getMethod() == "GET")
            {
                QByteArray txStr = request.getParameter("tx");
                bool tx = (txStr == "true");

                int status = m_adapter->instanceDevices(tx, normalResponse, errorResponse);

                if (status == 200) {
                    response.write(normalResponse.asJson().toUtf8());
                } else {
                    response.write(errorResponse.asJson().toUtf8());
                }

                response.setStatus(status);
            }
            else
            {
                response.write("Invalid HTTP method");
                response.setStatus(405,"Invalid HTTP method");
            }
        }
        else if (path == WebAPIAdapterInterface::instanceChannelsURL)
        {
            Swagger::SWGInstanceChannelsResponse normalResponse;
            Swagger::SWGErrorResponse errorResponse;

            if (request.getMethod() == "GET")
            {
                QByteArray txStr = request.getParameter("tx");
                bool tx = (txStr == "true");

                int status = m_adapter->instanceChannels(tx, normalResponse, errorResponse);

                if (status == 200) {
                    response.write(normalResponse.asJson().toUtf8());
                } else {
                    response.write(errorResponse.asJson().toUtf8());
                }

                response.setStatus(status);
            }
            else
            {
                response.write("Invalid HTTP method");
                response.setStatus(405,"Invalid HTTP method");
            }
        }
        else if (path == WebAPIAdapterInterface::instanceLoggingURL)
        {
            Swagger::SWGLoggingInfo normalResponse;
            Swagger::SWGErrorResponse errorResponse;

            if (request.getMethod() == "GET")
            {
                int status = m_adapter->instanceLoggingGet(normalResponse, errorResponse);

                if (status == 200) {
                    response.write(normalResponse.asJson().toUtf8());
                } else {
                    response.write(errorResponse.asJson().toUtf8());
                }

                response.setStatus(status);
            }
            else if (request.getMethod() == "PUT")
            {
                try
                {
                    QString jsonStr = request.getBody();
                    qDebug("WebAPIRequestMapper::service: /sdrangel/logging (PUT): %s", qPrintable(jsonStr));
                    normalResponse.fromJson(jsonStr);
                    int status = m_adapter->instanceLoggingPut(normalResponse, errorResponse);

                    if (status == 200) {
                        response.write(normalResponse.asJson().toUtf8());
                    } else {
                        response.write(errorResponse.asJson().toUtf8());
                    }

                    response.setStatus(status);
                }
                catch (const std::exception& ex)
                {
                    response.write("Invalid input format");
                    response.setStatus(400,"Invalid input format");
                }
            }
            else
            {
                response.write("Invalid HTTP method");
                response.setStatus(405,"Invalid HTTP method");
            }
        }
        else
        {
//            QDirIterator it(":", QDirIterator::Subdirectories);
//            while (it.hasNext()) {
//                qDebug() << "WebAPIRequestMapper::service: " << it.next();
//            }

            QByteArray path = "/index.html";
            m_staticFileController->service(path, response);
            //response.setStatus(404,"Not found");
        }
    }
}