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
|
// 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.
#include "components/paint_preview/common/recording_map.h"
#include <string_view>
#include "base/files/file_path.h"
#include "base/strings/utf_string_conversions.h"
#include "build/build_config.h"
namespace paint_preview {
namespace {
base::FilePath ToFilePath(std::string_view path_str) {
#if BUILDFLAG(IS_WIN)
return base::FilePath(base::UTF8ToWide(path_str));
#else
return base::FilePath(path_str);
#endif
}
RecordingMap RecordingMapFromBufferMap(
base::flat_map<base::UnguessableToken, mojo_base::BigBuffer>&& buffer_map) {
std::vector<std::pair<base::UnguessableToken, SerializedRecording>> entries;
for (auto& entry : buffer_map) {
entries.emplace_back(entry.first,
SerializedRecording(std::move(entry.second)));
}
return RecordingMap(std::move(entries));
}
} // namespace
std::pair<RecordingMap, PaintPreviewProto> RecordingMapFromCaptureResult(
CaptureResult&& capture_result) {
switch (capture_result.persistence) {
case RecordingPersistence::kFileSystem:
return {RecordingMapFromPaintPreviewProto(capture_result.proto),
std::move(capture_result.proto)};
case RecordingPersistence::kMemoryBuffer:
return {
RecordingMapFromBufferMap(std::move(capture_result.serialized_skps)),
capture_result.proto};
}
NOTREACHED();
}
RecordingMap RecordingMapFromPaintPreviewProto(const PaintPreviewProto& proto) {
std::vector<std::pair<base::UnguessableToken, SerializedRecording>> entries;
entries.reserve(1 + proto.subframes_size());
SerializedRecording root_frame_recording(
ToFilePath(proto.root_frame().file_path()));
// We can't composite anything with an invalid SKP file path for the root
// frame.
if (!root_frame_recording.IsValid())
return {};
std::optional<base::UnguessableToken> root_frame_embedding_token =
base::UnguessableToken::Deserialize(
proto.root_frame().embedding_token_high(),
proto.root_frame().embedding_token_low());
// TODO(crbug.com/40252979): Investigate whether a deserialization
// failure can actually occur here and if it can, add a comment discussing
// how this can happen.
if (!root_frame_embedding_token.has_value()) {
return {};
}
entries.emplace_back(root_frame_embedding_token.value(),
std::move(root_frame_recording));
for (const auto& subframe : proto.subframes()) {
SerializedRecording frame_recording(ToFilePath(subframe.file_path()));
// Skip this frame if it doesn't have a valid SKP file path.
if (!frame_recording.IsValid())
continue;
std::optional<base::UnguessableToken> subframe_embedding_token =
base::UnguessableToken::Deserialize(subframe.embedding_token_high(),
subframe.embedding_token_low());
// TODO(crbug.com/40252979): Investigate whether a deserialization
// failure can actually occur here and if it can, add a comment discussing
// how this can happen.
if (!subframe_embedding_token.has_value()) {
continue;
}
entries.emplace_back(subframe_embedding_token.value(),
std::move(frame_recording));
}
return RecordingMap(std::move(entries));
}
} // namespace paint_preview
|