chrome/browser/ui/views/toolbar/chrome_labs_bubble_view_model.h

// Copyright 2020 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_UI_VIEWS_TOOLBAR_CHROME_LABS_BUBBLE_VIEW_MODEL_H_
#define CHROME_BROWSER_UI_VIEWS_TOOLBAR_CHROME_LABS_BUBBLE_VIEW_MODEL_H_

#include <string>
#include <vector>
#include "components/version_info/channel.h"

struct LabInfo {
  LabInfo(
      const std::string& internal_name,
      const std::u16string& visible_name,
      const std::u16string& visible_description,
      const std::string& feedback_category_name,
      version_info::Channel allowed_channel,
      std::vector<std::u16string> translated_feature_variation_descriptions =
          std::vector<std::u16string>());
  LabInfo(const LabInfo& other);
  ~LabInfo();
  std::string internal_name;
  std::u16string visible_name;
  std::u16string visible_description;
  std::string feedback_category_name;
  // Channels that are less stable than allowed_channel will also be
  // considered allowed. ex) if BETA is specified, this feature will also be
  // shown on CANARY and DEV.
  version_info::Channel allowed_channel;
  std::vector<std::u16string> translated_feature_variation_descriptions;
};

class ChromeLabsBubbleViewModel {
 public:
  ChromeLabsBubbleViewModel();
  ~ChromeLabsBubbleViewModel();

  const std::vector<LabInfo>& GetLabInfo() const;

 private:
  const std::vector<LabInfo>& lab_info_;
};

// ScopedChromeLabsModelDataForTesting is intended to be used in test settings
// to replace the production data in ChromeLabsBubbleViewModel. Upon
// destruction, ScopedChromeLabsModelDataForTesting will remove the test data
// from ChromeLabsBubbleViewModel.

class ScopedChromeLabsModelDataForTesting {
 public:
  ScopedChromeLabsModelDataForTesting();
  ScopedChromeLabsModelDataForTesting(
      const ScopedChromeLabsModelDataForTesting&) = delete;
  ScopedChromeLabsModelDataForTesting& operator=(
      const ScopedChromeLabsModelDataForTesting&) = delete;
  ~ScopedChromeLabsModelDataForTesting();

  void SetModelDataForTesting(const std::vector<LabInfo>& test_feature_info);
};

#endif  // CHROME_BROWSER_UI_VIEWS_TOOLBAR_CHROME_LABS_BUBBLE_VIEW_MODEL_H_