WSAPICall.cpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac

    This file is part of Leosac.

    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Leosac 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 Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#include "WSAPICall.hpp"
#include "core/audit/WSAPICall_odb.h"
#include "core/auth/User.hpp"
#include "tools/db/MultiplexedTransaction.hpp"
#include "tools/log.hpp"

using namespace Leosac;
using namespace Leosac::Audit;

WSAPICall::WSAPICall()
    : status_code_(APIStatusCode::UNKNOWN)
    , database_operations_(0)
{
}

WSAPICallPtr WSAPICall::create(const DBPtr &database)
{
    ASSERT_LOG(database, "Database cannot be null.");

    db::MultiplexedTransaction t(database->begin());
    WSAPICallPtr audit(new Audit::WSAPICall());
    database->persist(audit);
    t.commit();
    audit->database_ = database;
    return audit;
}

void WSAPICall::method(const std::string &str)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    api_method_ = str;
}

void WSAPICall::uuid(const std::string &str)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    uuid_ = str;
}

void WSAPICall::status_code(APIStatusCode code)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    status_code_ = code;
}

void WSAPICall::status_string(const std::string &str)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    status_string_ = str;
}

void WSAPICall::source_endpoint(const std::string &str)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    source_endpoint_ = str;
}

void WSAPICall::request_content(const std::string &str)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    request_content_ = str;
}

void WSAPICall::response_content(const std::string &str)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    response_content_ = str;
}

void WSAPICall::database_operations(uint16_t nb_operation)
{
    ASSERT_LOG(!finalized(), "Audit entry is already finalized.");
    database_operations_ = nb_operation;
}

const std::string &WSAPICall::method() const
{
    return api_method_;
}

const std::string &WSAPICall::uuid() const
{
    return uuid_;
}

APIStatusCode WSAPICall::status_code() const
{
    return status_code_;
}

const std::string &WSAPICall::status_string() const
{
    return status_string_;
}

const std::string &WSAPICall::source_endpoint() const
{
    return source_endpoint_;
}

std::string WSAPICall::generate_description() const
{
    std::stringstream ss;

    auto author = author_.load();
    if (author)
    {
        ss << "Websocket API call by user " << author->username() << " to "
           << method() << " took " << duration_ << " milliseconds.";
    }
    else
    {
        ss << "Websocket API call by anonymous to " << method() << " took "
           << duration_ << " milliseconds.";
    }
    return ss.str();
}