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();
}