/*
 * Copyright (c) 2016 MariaDB Corporation Ab
 *
 * Use of this software is governed by the Business Source License included
 * in the LICENSE.TXT file and at www.mariadb.com/bsl11.
 *
 * Change Date: 2019-07-01
 *
 * On the date above, in accordance with the Business Source License, use
 * of this software will be governed by version 2 or later of the General
 * Public License.
 */

#include "maxscale/adminclient.hh"

#include <string>
#include <sstream>

#include <maxscale/atomic.h>
#include <maxscale/hk_heartbeat.h>
#include <maxscale/log_manager.h>
#include <maxscale/thread.h>
#include <maxscale/spinlock.hh>

#include "maxscale/httprequest.hh"
#include "maxscale/httpresponse.hh"
#include "maxscale/resource.hh"

using std::string;
using std::stringstream;
using mxs::SpinLockGuard;

AdminClient::AdminClient(int fd, const struct sockaddr_storage& addr, int timeout):
    m_fd(fd),
    m_last_activity(atomic_read_int64(&hkheartbeat)),
    m_addr(addr)
{
}

void AdminClient::close_connection()
{
    SpinLockGuard guard(m_lock);

    if (m_fd != -1)
    {
        close(m_fd);
        m_fd = -1;
    }
}

AdminClient::~AdminClient()
{
    close_connection();
}

static bool read_request(int fd, string& output)
{
    while (true)
    {
        char buf[1024];
        int bufsize = sizeof(buf) - 1;

        int rc = read(fd, buf, bufsize);

        if (rc == -1)
        {
            return false;
        }

        buf[rc] = '\0';
        output += buf;

        if (rc < bufsize)
        {
            /** Complete request read */
            break;
        }
    }

    return true;
}

static bool write_response(int fd, const string& body)
{
    return write(fd, body.c_str(), body.length()) != -1;
}

void AdminClient::process()
{
    string data;
    atomic_write_int64(&m_last_activity, hkheartbeat);

    if (read_request(m_fd, data))
    {
        HttpResponse response(HTTP_400_BAD_REQUEST);
        SHttpRequest request(HttpRequest::parse(data));
        atomic_write_int64(&m_last_activity, hkheartbeat);

        if (request.get())
        {
            response = resource_handle_request(*request);
        }

        if (!write_response(m_fd, response.get_response()))
        {
            MXS_ERROR("Failed to write response to client: %d, %s", errno, mxs_strerror(errno));
        }
    }
    else
    {
        MXS_ERROR("Failed to read client request: %d, %s", errno, mxs_strerror(errno));
    }
}