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
|
// 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.
#ifndef CHROME_BROWSER_CHROMEOS_PRINTING_PRINT_PREVIEW_PRINT_PREVIEW_WEBCONTENTS_MANAGER_H_
#define CHROME_BROWSER_CHROMEOS_PRINTING_PRINT_PREVIEW_PRINT_PREVIEW_WEBCONTENTS_MANAGER_H_
#include <map>
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/unguessable_token.h"
#include "chromeos/crosapi/mojom/print_preview_cros.mojom.h"
#include "components/printing/common/print.mojom-forward.h"
#include "content/public/browser/web_contents.h"
namespace chromeos {
// Manages a 1:1 relationship between a printing source's webcontents and a
// base::UnguessableToken. Each token represent a webcontent and is used as a
// proxy for determining which print preview is relevant.
// Communicates to ash via crosapi.
class PrintPreviewWebcontentsManager
: public crosapi::mojom::PrintPreviewCrosClient {
public:
PrintPreviewWebcontentsManager();
PrintPreviewWebcontentsManager(const PrintPreviewWebcontentsManager&) =
delete;
PrintPreviewWebcontentsManager& operator=(
const PrintPreviewWebcontentsManager&) = delete;
~PrintPreviewWebcontentsManager() override;
static PrintPreviewWebcontentsManager* Get();
static void SetInstanceForTesting(PrintPreviewWebcontentsManager* manager);
static void ResetInstanceForTesting();
void Initialize();
// Establishes new mappings of webcontents and token, then requests a new
// print preview dialog to appear.
void RequestPrintPreview(
const base::UnguessableToken& token,
content::WebContents* webcontents,
::printing::mojom::RequestPrintPreviewParamsPtr params);
// Handles removing the webcontents mapping and informing the ash client
// of the removed webcontent. This can happen if the initiating source
// (e.g. tab) closes/crashes.
void PrintPreviewDone(const base::UnguessableToken& token);
// crosapi::mojom::PrintPreviewCrosClient:
void GeneratePrintPreview(const base::UnguessableToken& token,
crosapi::mojom::PrintSettingsPtr settings,
GeneratePrintPreviewCallback callback) override;
// Handles ash -> chrome requests when the print dialog is closed.
void HandleDialogClosed(const base::UnguessableToken& token,
HandleDialogClosedCallback callback) override;
static void SetPrintPreviewCrosDelegateForTesting(
crosapi::mojom::PrintPreviewCrosDelegate* delegate);
private:
friend class MockPrintPreviewWebcontentsManager;
friend class PrintPreviewWebContentsManagerBrowserTest;
void OnRequestPrintPreviewCallback(bool success);
void OnPrintPreviewDoneCallback(bool success);
// Remove the entry from lookup maps keyed by `token`.
// Returns the webcontents tied to `token` if removal was successful.
content::WebContents* RemoveTokenMapping(const base::UnguessableToken& token);
// Mapping a unique ID to its webcontents.
std::map<base::UnguessableToken,
raw_ptr<content::WebContents, CtnExperimental>>
token_to_webcontents_;
base::WeakPtrFactory<PrintPreviewWebcontentsManager> weak_ptr_factory_{this};
};
} // namespace chromeos
#endif // CHROME_BROWSER_CHROMEOS_PRINTING_PRINT_PREVIEW_PRINT_PREVIEW_WEBCONTENTS_MANAGER_H_
|
