165 lines
4.3 KiB
C++
165 lines
4.3 KiB
C++
/*
|
|
* Copyright (c) 2018, evilny0
|
|
*
|
|
* This file is part of cpfm.
|
|
*
|
|
* cpfm 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* cpm 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 cpfm. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include "pf.h"
|
|
#include "wallet.h"
|
|
#include "exchange.h"
|
|
#include "exchanges/kraken.h"
|
|
#include "pricesource.h"
|
|
|
|
|
|
PortfolioManager::PortfolioManager()
|
|
{
|
|
|
|
}
|
|
|
|
PortfolioManager::~PortfolioManager()
|
|
{
|
|
|
|
}
|
|
|
|
void PortfolioManager::doTestStuff()
|
|
{
|
|
// Test user 1
|
|
emptyUserBalances(1);
|
|
|
|
ExchangesManager exchangesManager(1);
|
|
exchangesManager.analyzeUserAccounts();
|
|
|
|
WalletsManager walletsManager(1);
|
|
walletsManager.analyzeUserWallets();
|
|
|
|
displayUserBalances(1);
|
|
|
|
// Test user 2
|
|
emptyUserBalances(2);
|
|
|
|
ExchangesManager exchangesManagerTwo(2);
|
|
exchangesManagerTwo.analyzeUserAccounts();
|
|
|
|
WalletsManager walletsManagerTwo(2);
|
|
walletsManagerTwo.analyzeUserWallets();
|
|
|
|
displayUserBalances(2);
|
|
}
|
|
|
|
void PortfolioManager::emptyUserBalances(int userId)
|
|
{
|
|
mysql::connection db(getMysqlConfig());
|
|
const auto wallets_balances = TableWalletsBalances{};
|
|
const auto exchanges_balances = TableExchangesBalances{};
|
|
|
|
db.run(remove_from(wallets_balances).unconditionally());
|
|
linfo << "Emptied wallets balances DB table for user " << userId;
|
|
|
|
db.run(remove_from(exchanges_balances).unconditionally());
|
|
linfo << "Emptied exchanges balances DB table for user " << userId;
|
|
}
|
|
|
|
void PortfolioManager::displayUserBalances(int userId)
|
|
{
|
|
mysql::connection db(getMysqlConfig());
|
|
const auto wallets = TableWallets{};
|
|
const auto wallets_balances = TableWalletsBalances{};
|
|
const auto exchanges_accounts = TableExchangesAccounts{};
|
|
const auto exchanges_balances = TableExchangesBalances{};
|
|
|
|
map<int,Money> totalBalances;
|
|
for (const auto& row: db.run(select(wallets_balances.balance, wallets_balances.coin_id).from(wallets_balances.cross_join(wallets)).where(wallets_balances.wallet_id == wallets.wallet_id and wallets.user_id == userId)))
|
|
{
|
|
Money m(row.balance);
|
|
totalBalances[row.coin_id] += m;
|
|
}
|
|
|
|
for (const auto& row: db.run(select(exchanges_balances.balance, exchanges_balances.coin_id).from(exchanges_balances.cross_join(exchanges_accounts)).where(exchanges_balances.account_id == exchanges_accounts.account_id and exchanges_accounts.user_id == userId)))
|
|
{
|
|
Money m(row.balance);
|
|
totalBalances[row.coin_id] += m;
|
|
}
|
|
|
|
Money total;
|
|
PriceSourceCryptoWatch ps;
|
|
for (const auto& balance: totalBalances)
|
|
{
|
|
if (balance.second != 0)
|
|
{
|
|
cout << getCoinName(balance.first) << " : " << balance.second << " | " << balance.second * ps.getCoinPrice(balance.first) << endl;
|
|
total += balance.second * ps.getCoinPrice(balance.first);
|
|
}
|
|
}
|
|
cout << "---------------" << endl;
|
|
cout << "Total : " << total << " EUR." << endl;
|
|
}
|
|
|
|
void PortfolioManager::run()
|
|
{
|
|
while (!m_bStopThread)
|
|
{
|
|
if (m_bStopThread)
|
|
break;
|
|
|
|
#ifdef _WIN32
|
|
SwitchToThread();
|
|
Sleep (1000);
|
|
#else
|
|
pthread_yield();
|
|
sleep (1);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
string getCoinName(int coinId)
|
|
{
|
|
string s;
|
|
|
|
switch(coinId)
|
|
{
|
|
case CPFM_COIN_ID_BTC:
|
|
s = "BTC";
|
|
break;
|
|
case CPFM_COIN_ID_LTC:
|
|
s = "LTC";
|
|
break;
|
|
case CPFM_COIN_ID_EUR:
|
|
s = "EUR";
|
|
break;
|
|
case CPFM_COIN_ID_ICN:
|
|
s = "ICN";
|
|
break;
|
|
case CPFM_COIN_ID_ETC:
|
|
s = "ETC";
|
|
break;
|
|
case CPFM_COIN_ID_ETH:
|
|
s = "ETH";
|
|
break;
|
|
case CPFM_COIN_ID_BCH:
|
|
s = "BCH";
|
|
break;
|
|
case CPFM_COIN_ID_BTG:
|
|
s = "BTG";
|
|
break;
|
|
default:
|
|
s = "Error!";
|
|
break;
|
|
}
|
|
|
|
return s;
|
|
}
|