// Copyright 2018 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.

#ifndef CHROME_BROWSER_BADGING_BADGE_MANAGER_H_
#define CHROME_BROWSER_BADGING_BADGE_MANAGER_H_

#include <map>
#include <memory>
#include <string>

#include "base/macros.h"
#include "base/optional.h"
#include "chrome/browser/badging/badge_manager_delegate.h"
#include "components/keyed_service/core/keyed_service.h"
#include "extensions/common/extension_id.h"

class KeyedService;
class Profile;

namespace badging {

// Determines the text to put on the badge based on some badge_content.
std::string GetBadgeString(base::Optional<uint64_t> badge_content);

// Maintains a record of badge contents and dispatches badge changes to a
// delegate.
class BadgeManager : public KeyedService {
 public:
  explicit BadgeManager(Profile* profile);
  ~BadgeManager() override;

  // Records badge contents for an app and notifies the delegate if the badge
  // contents have changed.
  void UpdateBadge(const extensions::ExtensionId&, base::Optional<uint64_t>);

  // Clears badge contents for an app (if existing) and notifies the delegate.
  void ClearBadge(const extensions::ExtensionId&);

  // Sets the delegate used for setting/clearing badges.
  void SetDelegate(std::unique_ptr<BadgeManagerDelegate> delegate);

 private:
  std::unique_ptr<BadgeManagerDelegate> delegate_;

  // Maps extension id to badge contents.
  std::map<extensions::ExtensionId, base::Optional<uint64_t>> badged_apps_;

  DISALLOW_COPY_AND_ASSIGN(BadgeManager);
};

}  // namespace badging

#endif  // CHROME_BROWSER_BADGING_BADGE_MANAGER_H_