mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2024-11-08 16:57:16 +01:00
123 lines
3.8 KiB
C++
123 lines
3.8 KiB
C++
// ------------------------------------------------------------------------------------------------
|
|
#include "Account.hpp"
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
namespace SqMod {
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
Account::Account(CSStr host, CSStr user, CSStr pass, CSStr schema, Uint16 port, CSStr socket)
|
|
: m_Host(host ? host : _SC(""))
|
|
, m_User(user ? user : _SC(""))
|
|
, m_Pass(pass ? pass : _SC(""))
|
|
, m_Port(port)
|
|
, m_Schema(schema ? schema : _SC(""))
|
|
, m_Socket(socket ? socket : _SC(""))
|
|
, m_SSL_Key()
|
|
, m_SSL_Cert()
|
|
, m_SSL_CA()
|
|
, m_SSL_CA_Path()
|
|
, m_SSL_Cipher()
|
|
, m_Autocommit()
|
|
, m_Options()
|
|
{
|
|
// Attempt to extract the port number from the host
|
|
String::size_type pos = m_Host.find(':');
|
|
// Was it specified in the host
|
|
if (pos != String::npos)
|
|
{
|
|
try
|
|
{
|
|
// Attempt to extract the numeric value
|
|
ULong num = std::stol(m_Host.substr(pos + 1));
|
|
// Is the port number withing range?
|
|
if (num > 0xFFFF)
|
|
{
|
|
SHROWF("Port number out of range: %lu");
|
|
}
|
|
// Assign the extracted port number
|
|
m_Port = static_cast< Uint16 >(num);
|
|
// Remove everything after the port number
|
|
m_Host = m_Host.substr(0, pos);
|
|
}
|
|
catch (const std::exception & e)
|
|
{
|
|
SHROWF("Cannot extract the port number [%s]", e.what());
|
|
}
|
|
}
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
void Account::SetSSL(CSStr key, CSStr cert, CSStr ca, CSStr ca_path, CSStr cipher)
|
|
{
|
|
if (!key || *key == '\0')
|
|
{
|
|
m_SSL_Key.clear();
|
|
m_SSL_Cert.clear();
|
|
m_SSL_CA.clear();
|
|
m_SSL_CA_Path.clear();
|
|
m_SSL_Cipher.clear();
|
|
}
|
|
else
|
|
{
|
|
m_SSL_Key.assign(key ? key : _SC(""));
|
|
m_SSL_Cert.assign(cert ? cert : _SC(""));
|
|
m_SSL_CA.assign(ca ? ca : _SC(""));
|
|
m_SSL_CA_Path.assign(ca_path ? ca_path : _SC(""));
|
|
m_SSL_Cipher.assign(cipher ? cipher : _SC(""));
|
|
}
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
Table Account::GetOptions() const
|
|
{
|
|
// Allocate an empty table
|
|
Table tbl(DefaultVM::Get());
|
|
// Insert every option into the table
|
|
for (const auto & opt : m_Options)
|
|
{
|
|
tbl.SetValue(opt.first.c_str(), opt.second);
|
|
}
|
|
// Return the resulted table
|
|
return tbl;
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
const String & Account::GetOption(CSStr name) const
|
|
{
|
|
// Make sure the specified name is valid
|
|
if (!name || *name == '\0')
|
|
{
|
|
SHROWF("Invalid or empty option name");
|
|
}
|
|
// Attempt to find the requested option
|
|
const Options::const_iterator itr = m_Options.find(name);
|
|
// Return the result
|
|
return (itr == m_Options.cend()) ? s_String : itr->second;
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
void Account::SetOption(CSStr name, CSStr value)
|
|
{
|
|
// Make sure the specified name is valid
|
|
if (!name || *name == '\0')
|
|
{
|
|
SHROWF("Invalid or empty option name");
|
|
}
|
|
// Assign the specified value
|
|
m_Options[name] = value ? value : _SC("");
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
void Account::RemoveOption(CSStr name) const
|
|
{
|
|
// Make sure the specified name is valid
|
|
if (!name || *name == '\0')
|
|
{
|
|
SHROWF("Invalid or empty option name");
|
|
}
|
|
// Erase the specified value
|
|
m_Options.erase(name);
|
|
}
|
|
|
|
} // Namespace:: SqMod
|