Skip to content

Commit

Permalink
BAT Confirmations should not be started if Rewards is disabled
Browse files Browse the repository at this point in the history
  • Loading branch information
tmancey committed Apr 30, 2020
1 parent d0e88e8 commit bd5f7cc
Show file tree
Hide file tree
Showing 5 changed files with 68 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -950,10 +950,9 @@ void ConfirmationsImpl::UpdateAdsRewards(const bool should_refresh) {
if (!state_has_loaded_) {
BLOG(ERROR) <<
"Unable to update ads rewards as Confirmations state is not ready";
return;
}

DCHECK(wallet_info_.IsValid());

ads_rewards_->Update(wallet_info_, should_refresh);
}

Expand Down
57 changes: 53 additions & 4 deletions vendor/bat-native-ledger/src/bat/ledger/internal/ledger_impl.cc
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
#include "bat/ledger/internal/report/report.h"
#include "bat/ledger/internal/ledger_impl.h"
#include "bat/ledger/internal/media/helper.h"
#include "bat/ledger/internal/state_keys.h"
#include "bat/ledger/internal/static_values.h"
#include "net/http/http_status_code.h"

Expand Down Expand Up @@ -126,7 +127,22 @@ void LedgerImpl::Initialize(

initializing_ = true;

InitializeConfirmations(execute_create_script, callback);
MaybeInitializeConfirmations(execute_create_script, callback);
}

void LedgerImpl::MaybeInitializeConfirmations(
const bool execute_create_script,
ledger::ResultCallback callback) {
const bool is_enabled = GetBooleanState(ledger::kStateEnabled);

const bool is_enabled_migrated =
GetBooleanState(ledger::kStateEnabledMigrated);

if (is_enabled || !is_enabled_migrated) {
InitializeConfirmations(execute_create_script, callback);
} else {
InitializeDatabase(execute_create_script, callback);
}
}

void LedgerImpl::InitializeConfirmations(
Expand All @@ -142,7 +158,7 @@ void LedgerImpl::InitializeConfirmations(
this,
_1,
execute_create_script,
callback);
std::move(callback));
bat_confirmations_->Initialize(initialized_callback);
}

Expand All @@ -151,10 +167,18 @@ void LedgerImpl::OnConfirmationsInitialized(
const bool execute_create_script,
ledger::ResultCallback callback) {
if (!success) {
// If confirmations fails we should fall-through and continue initializing
// ledger
BLOG(this, ledger::LogLevel::LOG_ERROR) <<
"Failed to initialize confirmations";
}

InitializeDatabase(execute_create_script, callback);
}

void LedgerImpl::InitializeDatabase(
const bool execute_create_script,
ledger::ResultCallback callback) {
ledger::ResultCallback finish_callback =
std::bind(&LedgerImpl::OnWalletInitializedInternal,
this,
Expand All @@ -164,8 +188,9 @@ void LedgerImpl::OnConfirmationsInitialized(
auto database_callback = std::bind(&LedgerImpl::OnDatabaseInitialized,
this,
_1,
finish_callback);
bat_database_->Initialize(execute_create_script, database_callback);
std::move(finish_callback));
bat_database_->Initialize(execute_create_script,
std::move(database_callback));
}

void LedgerImpl::CreateWallet(ledger::ResultCallback callback) {
Expand Down Expand Up @@ -360,6 +385,10 @@ void LedgerImpl::OnLedgerStateLoaded(

void LedgerImpl::SetConfirmationsWalletInfo(
const ledger::WalletInfoProperties& wallet_info_properties) {
if (!bat_confirmations_) {
return;
}

if (wallet_info_properties.key_info_seed.size() != SEED_LENGTH) {
BLOG(this, ledger::LogLevel::LOG_ERROR) << "Failed to initialize "
"confirmations due to invalid wallet";
Expand Down Expand Up @@ -973,6 +1002,10 @@ void LedgerImpl::LogResponse(
}

void LedgerImpl::UpdateAdsRewards() {
if (!bat_confirmations_) {
return;
}

bat_confirmations_->UpdateAdsRewards(false);
}

Expand Down Expand Up @@ -1219,6 +1252,10 @@ void LedgerImpl::SaveNormalizedPublisherList(ledger::PublisherInfoList list) {
}

void LedgerImpl::SetCatalogIssuers(const std::string& info) {
if (!bat_confirmations_) {
return;
}

ads::IssuersInfo issuers_info_ads;
if (issuers_info_ads.FromJson(info) != ads::Result::SUCCESS)
return;
Expand All @@ -1238,6 +1275,10 @@ void LedgerImpl::SetCatalogIssuers(const std::string& info) {
void LedgerImpl::ConfirmAd(
const std::string& json,
const std::string& confirmation_type) {
if (!bat_confirmations_) {
return;
}

ads::AdInfo ad_info;
if (ad_info.FromJson(json) != ads::Result::SUCCESS) {
return;
Expand All @@ -1258,12 +1299,20 @@ void LedgerImpl::ConfirmAction(
const std::string& creative_instance_id,
const std::string& creative_set_id,
const std::string& confirmation_type) {
if (!bat_confirmations_) {
return;
}

bat_confirmations_->ConfirmAction(creative_instance_id, creative_set_id,
confirmations::ConfirmationType(confirmation_type));
}

void LedgerImpl::GetTransactionHistory(
ledger::GetTransactionHistoryCallback callback) {
if (!bat_confirmations_) {
return;
}

bat_confirmations_->GetTransactionHistory(callback);
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -733,6 +733,10 @@ class LedgerImpl : public ledger::Ledger {
ledger::ResultCallback callback);

private:
void MaybeInitializeConfirmations(
const bool execute_create_script,
ledger::ResultCallback callback);

void InitializeConfirmations(
const bool execute_create_script,
ledger::ResultCallback callback);
Expand All @@ -742,6 +746,10 @@ class LedgerImpl : public ledger::Ledger {
const bool execute_create_script,
ledger::ResultCallback callback);

void InitializeDatabase(
const bool execute_create_script,
ledger::ResultCallback callback);

void OnLoad(ledger::VisitDataPtr visit_data,
const uint64_t& current_time) override;

Expand Down
2 changes: 2 additions & 0 deletions vendor/bat-native-ledger/src/bat/ledger/internal/state_keys.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,8 @@
#include <string>

namespace ledger {
const char kStateEnabled[] = "enabled";
const char kStateEnabledMigrated[] = "enabled_migrated";
const char kStateServerPublisherListStamp[] = "server_publisher_list_stamp";
const char kStateUpholdAnonAddress[] = "uphold_anon_address";
const char kStatePromotionLastFetchStamp[] = "promotion_last_fetch_stamp";
Expand Down
4 changes: 4 additions & 0 deletions vendor/brave-ios/Ledger/BATBraveLedger.mm
Original file line number Diff line number Diff line change
Expand Up @@ -1277,6 +1277,10 @@ - (void)setBooleanState:(const std::string&)name value:(bool)value
- (bool)getBooleanState:(const std::string&)name
{
const auto key = [NSString stringWithUTF8String:name.c_str()];
if (![self.prefs objectForKey:key]) {
return NO;
}

return [self.prefs[key] boolValue];
}

Expand Down

0 comments on commit bd5f7cc

Please sign in to comment.