1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/ash/birch/birch_file_suggest_provider.h"
#include <optional>
#include <vector>
#include "ash/birch/birch_item.h"
#include "ash/birch/birch_model.h"
#include "ash/shell.h"
#include "base/functional/bind.h"
#include "base/time/time.h"
#include "chrome/browser/ash/file_suggest/file_suggest_keyed_service_factory.h"
#include "chrome/browser/ash/file_suggest/file_suggest_util.h"
#include "chrome/browser/profiles/profile.h"
namespace ash {
BirchFileSuggestProvider::BirchFileSuggestProvider(Profile* profile)
: file_suggest_service_(
FileSuggestKeyedServiceFactory::GetInstance()->GetService(profile)) {
file_suggest_service_observation_.Observe(file_suggest_service_);
}
BirchFileSuggestProvider::~BirchFileSuggestProvider() = default;
void BirchFileSuggestProvider::RequestBirchDataFetch() {
file_suggest_service_->GetSuggestFileData(
FileSuggestionType::kDriveFile,
base::BindOnce(&BirchFileSuggestProvider::OnSuggestedFileDataUpdated,
weak_factory_.GetWeakPtr()));
}
void BirchFileSuggestProvider::OnFileSuggestionUpdated(
FileSuggestionType type) {
weak_factory_.InvalidateWeakPtrs();
if (type == FileSuggestionType::kDriveFile) {
RequestBirchDataFetch();
}
}
void BirchFileSuggestProvider::OnSuggestedFileDataUpdated(
const std::optional<std::vector<FileSuggestData>>& suggest_results) {
if (!Shell::HasInstance()) {
return;
}
if (!suggest_results.has_value()) {
// No suggestion results exist yet, so avoid setting file suggest items and
// return early.
return;
}
if (!suggest_results) {
Shell::Get()->birch_model()->SetFileSuggestItems({});
return;
}
std::vector<BirchFileItem> file_items;
for (const auto& suggestion : *suggest_results) {
base::Time timestamp = suggestion.modified_time.value_or(base::Time());
timestamp =
std::max(timestamp, suggestion.viewed_time.value_or(base::Time()));
timestamp =
std::max(timestamp, suggestion.shared_time.value_or(base::Time()));
file_items.emplace_back(suggestion.file_path, suggestion.title,
suggestion.prediction_reason.value_or(u""),
timestamp, suggestion.drive_file_id.value_or(""),
suggestion.icon_url.value_or(""));
}
Shell::Get()->birch_model()->SetFileSuggestItems(std::move(file_items));
}
} // namespace ash
|
