blob: 3b81edb22199abd015b491cef4d2ad46c753690d (
plain)
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
|
// Copyright 2017 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.
#include "components/offline_pages/core/renovations/page_renovator.h"
#include <utility>
#include "base/bind.h"
#include "base/callback.h"
#include "base/logging.h"
namespace offline_pages {
PageRenovator::PageRenovator(PageRenovationLoader* renovation_loader,
std::unique_ptr<ScriptInjector> script_injector,
const GURL& request_url)
: renovation_loader_(renovation_loader),
script_injector_(std::move(script_injector)) {
DCHECK(renovation_loader);
PrepareScript(request_url);
}
PageRenovator::~PageRenovator() {}
void PageRenovator::RunRenovations(CompletionCallback callback) {
// Prepare callback and inject combined script.
base::OnceCallback<void(const base::Value&)> cb = base::BindOnce(
[](base::OnceClosure callback, const base::Value&) {
if (callback)
std::move(callback).Run();
},
std::move(callback));
script_injector_->Inject(script_, std::move(cb));
}
void PageRenovator::PrepareScript(const GURL& url) {
std::vector<std::string> renovation_keys;
// Pick which renovations to run.
for (const std::unique_ptr<PageRenovation>& renovation :
renovation_loader_->renovations()) {
if (renovation->ShouldRun(url)) {
renovation_keys.push_back(renovation->GetID());
}
}
// Store combined renovation script. TODO(crbug.com/736933): handle
// failed GetRenovationScript call.
renovation_loader_->GetRenovationScript(renovation_keys, &script_);
}
} // namespace offline_pages
|
