components/feature_engagement/internal/availability_model_impl.cc

// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/feature_engagement/internal/availability_model_impl.h"

#include <memory>
#include <utility>

#include "base/bind.h"
#include "base/logging.h"
#include "base/memory/weak_ptr.h"
#include "components/feature_engagement/internal/persistent_availability_store.h"

namespace feature_engagement {

AvailabilityModelImpl::AvailabilityModelImpl(
    StoreLoadCallback store_load_callback)
    : ready_(false),
      store_load_callback_(std::move(store_load_callback)),
      weak_ptr_factory_(this) {}

AvailabilityModelImpl::~AvailabilityModelImpl() = default;

void AvailabilityModelImpl::Initialize(OnInitializedCallback callback,
                                       uint32_t current_day) {
  DCHECK(store_load_callback_);
  std::move(store_load_callback_)
      .Run(base::BindOnce(&AvailabilityModelImpl::OnStoreLoadComplete,
                          weak_ptr_factory_.GetWeakPtr(), std::move(callback)),
           current_day);
}

bool AvailabilityModelImpl::IsReady() const {
  return ready_;
}

base::Optional<uint32_t> AvailabilityModelImpl::GetAvailability(
    const base::Feature& feature) const {
  auto search = feature_availabilities_.find(feature.name);
  if (search == feature_availabilities_.end())
    return base::nullopt;

  return search->second;
}

void AvailabilityModelImpl::OnStoreLoadComplete(
    OnInitializedCallback on_initialized_callback,
    bool success,
    std::unique_ptr<std::map<std::string, uint32_t>> feature_availabilities) {
  if (!success) {
    std::move(on_initialized_callback).Run(false);
    return;
  }

  feature_availabilities_ = std::move(*feature_availabilities);

  ready_ = true;
  std::move(on_initialized_callback).Run(true);
}

}  // namespace feature_engagement