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
|
// 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/lens/lens_overlay_request_id_generator.h"
#include "base/check.h"
#include "base/containers/span.h"
#include "base/rand_util.h"
#include "components/base32/base32.h"
#include "components/lens/lens_features.h"
#include "lens_overlay_request_id_generator.h"
#include "third_party/lens_server_proto/lens_overlay_request_id.pb.h"
namespace lens {
// The number of bytes to use in an analytics id.
constexpr size_t kAnalyticsIdBytesSize = 16;
LensOverlayRequestIdGenerator::LensOverlayRequestIdGenerator() {
LensOverlayRequestIdGenerator::ResetRequestId();
}
LensOverlayRequestIdGenerator::~LensOverlayRequestIdGenerator() = default;
void LensOverlayRequestIdGenerator::ResetRequestId() {
uuid_ = base::RandUint64();
sequence_id_ = 0;
image_sequence_id_ = 0;
long_context_id_ = 0;
analytics_id_ = base::RandBytesAsString(kAnalyticsIdBytesSize);
routing_info_.reset();
}
std::unique_ptr<lens::LensOverlayRequestId>
LensOverlayRequestIdGenerator::GetNextRequestId(
RequestIdUpdateMode update_mode) {
// Verify that the initial request id is only generated once.
CHECK(update_mode != RequestIdUpdateMode::kInitialRequest ||
sequence_id_ == 0);
bool increment_image_sequence =
update_mode == RequestIdUpdateMode::kFullImageRequest ||
update_mode == RequestIdUpdateMode::kInitialRequest;
bool increment_sequence = update_mode != RequestIdUpdateMode::kOpenInNewTab;
bool increment_long_context =
lens::features::PageContentUploadRequestIdFixEnabled() &&
(update_mode == RequestIdUpdateMode::kPageContentRequest ||
update_mode == RequestIdUpdateMode::kInitialRequest);
bool create_analytics_id =
update_mode != RequestIdUpdateMode::kSearchUrl &&
update_mode != RequestIdUpdateMode::kPartialPageContentRequest;
bool store_analytics_id = update_mode != RequestIdUpdateMode::kOpenInNewTab;
// The server currently expects the image sequence id to be incremented for
// every page content request. This is a temporary fix until the server
// changes to index by sequence id instead of image sequence id.
if (!lens::features::PageContentUploadRequestIdFixEnabled() &&
update_mode == RequestIdUpdateMode::kPageContentRequest) {
increment_image_sequence = true;
}
if (increment_image_sequence) {
image_sequence_id_++;
}
if (increment_sequence) {
sequence_id_++;
}
if (increment_long_context) {
long_context_id_++;
}
std::string analytics_id_to_set = analytics_id_;
if (create_analytics_id) {
analytics_id_to_set = base::RandBytesAsString(kAnalyticsIdBytesSize);
if (store_analytics_id) {
analytics_id_ = analytics_id_to_set;
}
}
std::unique_ptr<lens::LensOverlayRequestId> request_id =
GetCurrentRequestId();
request_id->set_analytics_id(analytics_id_to_set);
return request_id;
}
std::string LensOverlayRequestIdGenerator::GetBase32EncodedAnalyticsId() {
return base32::Base32Encode(base::as_byte_span(analytics_id_),
base32::Base32EncodePolicy::OMIT_PADDING);
}
std::unique_ptr<lens::LensOverlayRequestId>
LensOverlayRequestIdGenerator::SetRoutingInfo(
lens::LensOverlayRoutingInfo routing_info) {
routing_info_ = routing_info;
return GetCurrentRequestId();
}
std::unique_ptr<lens::LensOverlayRequestId>
LensOverlayRequestIdGenerator::GetCurrentRequestId() {
auto request_id = std::make_unique<lens::LensOverlayRequestId>();
request_id->set_uuid(uuid_);
request_id->set_sequence_id(sequence_id_);
request_id->set_analytics_id(analytics_id_);
request_id->set_long_context_id(long_context_id_);
request_id->set_image_sequence_id(image_sequence_id_);
if (routing_info_.has_value()) {
request_id->mutable_routing_info()->CopyFrom(routing_info_.value());
}
return request_id;
}
} // namespace lens
|
