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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMEOS_COMPONENTS_QUICK_ANSWERS_RESULT_LOADER_H_
#define CHROMEOS_COMPONENTS_QUICK_ANSWERS_RESULT_LOADER_H_
#include <memory>
#include <string>
#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "chromeos/components/quick_answers/quick_answers_model.h"
#include "chromeos/components/quick_answers/search_result_parsers/search_response_parser.h"
namespace network {
class SharedURLLoaderFactory;
class SimpleURLLoader;
struct ResourceRequest;
} // namespace network
namespace quick_answers {
enum class IntentType;
struct PreprocessedOutput;
class ResultLoader {
public:
// A delegate interface for the ResultLoader.
class ResultLoaderDelegate {
public:
ResultLoaderDelegate(const ResultLoaderDelegate&) = delete;
ResultLoaderDelegate& operator=(const ResultLoaderDelegate&) = delete;
// Invoked when there is a network error.
virtual void OnNetworkError() {}
// Invoked when the `quick_answers_session` is received. Note that
// `quick_answers_session` may be `nullptr` if no answer found for the
// selected content.
virtual void OnQuickAnswerReceived(
std::unique_ptr<QuickAnswersSession> quick_answers_session) {}
protected:
ResultLoaderDelegate() = default;
virtual ~ResultLoaderDelegate() = default;
};
// Callback used when parsing for `quick_answers_session` is complete. Note
// that `quick_answers_session` may be `nullptr`.
using ResponseParserCallback = base::OnceCallback<void(
std::unique_ptr<QuickAnswersSession> quick_answers_session)>;
using BuildRequestCallback = base::OnceCallback<void(
std::unique_ptr<network::ResourceRequest> resource_request,
const std::string& request_body)>;
ResultLoader(
scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory,
ResultLoaderDelegate* delegate);
ResultLoader(const ResultLoader&) = delete;
ResultLoader& operator=(const ResultLoader&) = delete;
// Virtual for testing.
virtual ~ResultLoader();
// Creates ResultLoader based on the |intent_type|.
static std::unique_ptr<ResultLoader> Create(
IntentType intent_type,
scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory,
ResultLoader::ResultLoaderDelegate* delegate);
// Starts downloading of |quick_answers| associated with |selected_text|,
// calling |ResultLoaderDelegate| methods when finished.
// Note that delegate methods should be called only once per
// ResultLoader instance. Virtual for testing.
virtual void Fetch(const PreprocessedOutput& preprocessed_output);
protected:
// Builds the request URL from |selected_text|.
virtual void BuildRequest(const PreprocessedOutput& preprocessed_output,
BuildRequestCallback callback) const = 0;
// Process the |response_body| and invoked the callback with |QuickAnswer|.
virtual void ProcessResponse(const PreprocessedOutput& preprocessed_output,
std::unique_ptr<std::string> response_body,
ResponseParserCallback complete_callback) = 0;
ResultLoaderDelegate* delegate() const { return delegate_; }
private:
scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory_;
std::unique_ptr<network::SimpleURLLoader> loader_;
const raw_ptr<ResultLoaderDelegate> delegate_;
void OnBuildRequestComplete(
const PreprocessedOutput& preprocessed_output,
std::unique_ptr<network::ResourceRequest> resource_request,
const std::string& request_body);
void OnSimpleURLLoaderComplete(const PreprocessedOutput& preprocessed_output,
std::unique_ptr<std::string> response_body);
void OnResultParserComplete(
std::unique_ptr<QuickAnswersSession> quick_answers_session);
// Time when the query is issued.
base::TimeTicks fetch_start_time_;
base::WeakPtrFactory<ResultLoader> weak_factory_{this};
};
} // namespace quick_answers
#endif // CHROMEOS_COMPONENTS_QUICK_ANSWERS_RESULT_LOADER_H_
|
