AccountService.cpp

Go to the documentation of this file.

/*
 * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
 *
 * 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; either version 2 of the License, or (at your
 * option) any later version.
 *
 * 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 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 "AccountService.h"
#include "BattlenetRpcErrorCodes.h"
#include "Session.h"
 
namespace Battlenet::Services
{
namespace V1
{
Account::Account(Session* session) : AccountService(session)
{
}
 
uint32 Account::HandleGetAccountState(account::v1::GetAccountStateRequest const* request, account::v1::GetAccountStateResponse* response, std::function<void(ServiceBase*, uint32, ::google::protobuf::Message const*)>& /*continuation*/)
{
    if (!_session->IsAuthed())
        return ERROR_DENIED;
 
    if (request->options().field_privacy_info())
    {
        response->mutable_state()->mutable_privacy_info()->set_is_using_rid(false);
        response->mutable_state()->mutable_privacy_info()->set_is_visible_for_view_friends(false);
        response->mutable_state()->mutable_privacy_info()->set_is_hidden_from_friend_finder(true);
 
        response->mutable_tags()->set_privacy_info_tag(0xD7CA834D);
    }
 
    return ERROR_OK;
}
 
uint32 Account::HandleGetGameAccountState(account::v1::GetGameAccountStateRequest const* request, account::v1::GetGameAccountStateResponse* response, std::function<void(ServiceBase*, uint32, ::google::protobuf::Message const*)>& /*continuation*/)
{
    if (!_session->IsAuthed())
        return ERROR_DENIED;
 
    if (request->options().field_game_level_info())
    {
        if (GameAccountInfo const* gameAccountInfo = _session->GetGameAccountInfo(request->game_account_id().low()))
        {
            response->mutable_state()->mutable_game_level_info()->set_name(gameAccountInfo->DisplayName);
            response->mutable_state()->mutable_game_level_info()->set_program(ClientBuild::Program::WoW);
        }
 
        response->mutable_tags()->set_game_level_info_tag(0x5C46D483);
    }
 
    if (request->options().field_game_status())
    {
        if (GameAccountInfo const* gameAccountInfo = _session->GetGameAccountInfo(request->game_account_id().low()))
        {
            response->mutable_state()->mutable_game_status()->set_is_suspended(gameAccountInfo->IsBanned);
            response->mutable_state()->mutable_game_status()->set_is_banned(gameAccountInfo->IsPermanenetlyBanned);
            response->mutable_state()->mutable_game_status()->set_suspension_expires(uint64(gameAccountInfo->UnbanDate) * 1000000);
        }
 
        response->mutable_state()->mutable_game_status()->set_program(ClientBuild::Program::WoW);
        response->mutable_tags()->set_game_status_tag(0x98B75F99);
    }
 
    return ERROR_OK;
}
}
 
namespace V2
{
Account::Account(Session* session) : AccountService(session)
{
}
 
uint32 Account::HandleGetAccountInfo(account::v2::client::GetAccountInfoRequest const* /*request*/,
    account::v2::client::GetAccountInfoResponse* response,
    std::function<void(ServiceBase*, uint32, google::protobuf::Message const*)>& /*continuation*/)
{
    account::v2::AccountInfo* info = response->mutable_info();
 
    info->set_account_id(_session->GetAccountId());
    info->add_flags(account::v2::AccountInfo::FLAG_IS_HIDDEN_FROM_FRIEND_FINDER);
 
    return ERROR_OK;
}
 
uint32 Account::HandleGetRestriction(account::v2::client::GetRestrictionRequest const* /*request*/,
    account::v2::client::GetRestrictionResponse* /*response*/,
    std::function<void(ServiceBase*, uint32, google::protobuf::Message const*)>& /*continuation*/)
{
    return ERROR_OK;
}
 
uint32 Account::HandleGetGameAccountInfo(account::v2::client::GetGameAccountInfoRequest const* request,
    account::v2::client::GetGameAccountInfoResponse* response,
    std::function<void(ServiceBase*, uint32, google::protobuf::Message const*)>& /*continuation*/)
{
    if (request->has_game_account())
    {
        if (GameAccountInfo const* gameAccountInfo = _session->GetGameAccountInfo(request->game_account().id()))
        {
            account::v2::GameAccountInfo* info = response->mutable_info();
            info->set_account_id(request->game_account().id());
            info->set_name(gameAccountInfo->DisplayName);
        }
    }
 
    return ERROR_OK;
}
 
uint32 Account::HandleGetGameAccountRestriction(account::v2::client::GetGameAccountRestrictionRequest const* request,
    account::v2::client::GetGameAccountRestrictionResponse* response,
    std::function<void(ServiceBase*, uint32, google::protobuf::Message const*)>& /*continuation*/)
{
    if (request->has_game_account())
    {
        if (GameAccountInfo const* gameAccountInfo = _session->GetGameAccountInfo(request->game_account().id()))
        {
            if (gameAccountInfo->IsPermanenetlyBanned)
            {
                account::v2::Restriction* restriction = response->add_restrictions();
                restriction->set_title_id(ClientBuild::Program::WoW);
                restriction->set_type(account::v2::RESTRICTION_TYPE_LOGIN_BANNED);
                restriction->set_created_time_ms(gameAccountInfo->BanDate * IN_MILLISECONDS);
            }
 
            if (gameAccountInfo->IsBanned)
            {
                account::v2::Restriction* restriction = response->add_restrictions();
                restriction->set_title_id(ClientBuild::Program::WoW);
                restriction->set_type(account::v2::RESTRICTION_TYPE_LOGIN_SUSPENDED);
                restriction->set_created_time_ms(gameAccountInfo->BanDate * IN_MILLISECONDS);
                restriction->set_expire_time_ms(gameAccountInfo->UnbanDate * IN_MILLISECONDS);
            }
        }
    }
 
    return ERROR_OK;
}
}
}