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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// Copyright 2022 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/feed/web_feed_tab_helper.h"
#include <utility>
#include "base/functional/bind.h"
#include "chrome/browser/feed/web_feed_util.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/web_contents.h"
namespace feed {
namespace {
class WebFeedInfoFinderImpl : public WebFeedTabHelper::WebFeedInfoFinder {
public:
WebFeedInfoFinderImpl() = default;
~WebFeedInfoFinderImpl() override = default;
void FindForPage(
content::WebContents* web_contents,
base::OnceCallback<void(WebFeedMetadata)> callback) override {
FindWebFeedInfoForPage(
web_contents,
WebFeedPageInformationRequestReason::kMenuItemPresentation,
std::move(callback));
}
};
} // namespace
WebFeedTabHelper::WebFeedTabHelper(content::WebContents* web_contents)
: content::WebContentsObserver(web_contents),
content::WebContentsUserData<WebFeedTabHelper>(*web_contents) {
DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
web_feed_info_finder_ = std::make_unique<WebFeedInfoFinderImpl>();
}
WebFeedTabHelper::~WebFeedTabHelper() = default;
void WebFeedTabHelper::PrimaryPageChanged(content::Page& page) {
// This is a new navigation so we can invalidate any previously scheduled
// operations.
weak_ptr_factory_.InvalidateWeakPtrs();
web_feed_info_finder_->FindForPage(
web_contents(), base::BindOnce(&WebFeedTabHelper::OnWebFeedInfoRetrieved,
weak_ptr_factory_.GetWeakPtr(),
web_contents()->GetLastCommittedURL()));
}
void WebFeedTabHelper::OnWebFeedInfoRetrieved(const GURL& url,
WebFeedMetadata metadata) {
url_ = url;
if (metadata.subscription_status == WebFeedSubscriptionStatus::kSubscribed) {
follow_state_ = TabWebFeedFollowState::kFollowed;
web_feed_id_ = metadata.web_feed_id;
} else if (metadata.subscription_status ==
WebFeedSubscriptionStatus::kNotSubscribed) {
follow_state_ = TabWebFeedFollowState::kNotFollowed;
} else {
follow_state_ = TabWebFeedFollowState::kUnknown;
}
}
void WebFeedTabHelper::UpdateWebFeedInfo(const GURL& url,
TabWebFeedFollowState follow_state,
const std::string& web_feed_id) {
if (url != url_)
return;
follow_state_ = follow_state;
web_feed_id_ = web_feed_id;
}
void WebFeedTabHelper::SetWebFeedInfoForTesting(
const GURL& url,
TabWebFeedFollowState follow_state,
const std::string& web_feed_id) {
url_ = url;
follow_state_ = follow_state;
web_feed_id_ = web_feed_id;
}
void WebFeedTabHelper::SetWebFeedInfoFinderForTesting(
std::unique_ptr<WebFeedInfoFinder> finder) {
web_feed_info_finder_ = std::move(finder);
}
WEB_CONTENTS_USER_DATA_KEY_IMPL(WebFeedTabHelper);
} // namespace feed
|
