assistant_suggestions_model.cc 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. // Copyright 2018 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "ash/assistant/model/assistant_suggestions_model.h"
  5. #include <algorithm>
  6. #include "ash/assistant/model/assistant_suggestions_model_observer.h"
  7. #include "base/unguessable_token.h"
  8. namespace ash {
  9. AssistantSuggestionsModel::AssistantSuggestionsModel() = default;
  10. AssistantSuggestionsModel::~AssistantSuggestionsModel() = default;
  11. void AssistantSuggestionsModel::AddObserver(
  12. AssistantSuggestionsModelObserver* observer) const {
  13. observers_.AddObserver(observer);
  14. }
  15. void AssistantSuggestionsModel::RemoveObserver(
  16. AssistantSuggestionsModelObserver* observer) const {
  17. observers_.RemoveObserver(observer);
  18. }
  19. const chromeos::assistant::AssistantSuggestion*
  20. AssistantSuggestionsModel::GetSuggestionById(
  21. const base::UnguessableToken& id) const {
  22. for (auto& conversation_starter : conversation_starters_) {
  23. if (conversation_starter.id == id)
  24. return &conversation_starter;
  25. }
  26. for (auto& onboarding_suggestion : onboarding_suggestions_) {
  27. if (onboarding_suggestion.id == id)
  28. return &onboarding_suggestion;
  29. }
  30. return nullptr;
  31. }
  32. void AssistantSuggestionsModel::SetConversationStarters(
  33. std::vector<AssistantSuggestion>&& conversation_starters) {
  34. conversation_starters_ = std::move(conversation_starters);
  35. NotifyConversationStartersChanged();
  36. }
  37. const std::vector<chromeos::assistant::AssistantSuggestion>&
  38. AssistantSuggestionsModel::GetConversationStarters() const {
  39. return conversation_starters_;
  40. }
  41. void AssistantSuggestionsModel::SetOnboardingSuggestions(
  42. std::vector<AssistantSuggestion>&& onboarding_suggestions) {
  43. onboarding_suggestions_ = std::move(onboarding_suggestions);
  44. NotifyOnboardingSuggestionsChanged();
  45. }
  46. const std::vector<chromeos::assistant::AssistantSuggestion>&
  47. AssistantSuggestionsModel::GetOnboardingSuggestions() const {
  48. return onboarding_suggestions_;
  49. }
  50. void AssistantSuggestionsModel::NotifyConversationStartersChanged() {
  51. for (AssistantSuggestionsModelObserver& observer : observers_)
  52. observer.OnConversationStartersChanged(conversation_starters_);
  53. }
  54. void AssistantSuggestionsModel::NotifyOnboardingSuggestionsChanged() {
  55. for (AssistantSuggestionsModelObserver& observer : observers_)
  56. observer.OnOnboardingSuggestionsChanged(onboarding_suggestions_);
  57. }
  58. } // namespace ash