1 // Copyright 2020 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "content/browser/eye_dropper_chooser_impl.h"
6
7 #include "base/callback.h"
8 #include "content/browser/renderer_host/frame_tree_node.h"
9 #include "content/browser/renderer_host/render_frame_host_impl.h"
10 #include "content/public/browser/eye_dropper.h"
11 #include "content/public/browser/eye_dropper_listener.h"
12 #include "content/public/browser/web_contents.h"
13 #include "content/public/browser/web_contents_delegate.h"
14 #include "third_party/blink/public/mojom/choosers/color_chooser.mojom.h"
15
16 namespace content {
17
18 // static
Create(RenderFrameHost * render_frame_host,mojo::PendingReceiver<blink::mojom::EyeDropperChooser> receiver)19 void EyeDropperChooserImpl::Create(
20 RenderFrameHost* render_frame_host,
21 mojo::PendingReceiver<blink::mojom::EyeDropperChooser> receiver) {
22 DCHECK(render_frame_host);
23
24 // Renderer process should already check for user activation before sending
25 // this request. Double check in case of compromised renderer and consume
26 // the activation.
27 if (!static_cast<RenderFrameHostImpl*>(render_frame_host)
28 ->frame_tree_node()
29 ->UpdateUserActivationState(
30 blink::mojom::UserActivationUpdateType::
31 kConsumeTransientActivation,
32 blink::mojom::UserActivationNotificationType::kNone)) {
33 return;
34 }
35
36 new EyeDropperChooserImpl(render_frame_host, std::move(receiver));
37 }
38
EyeDropperChooserImpl(RenderFrameHost * render_frame_host,mojo::PendingReceiver<blink::mojom::EyeDropperChooser> receiver)39 EyeDropperChooserImpl::EyeDropperChooserImpl(
40 RenderFrameHost* render_frame_host,
41 mojo::PendingReceiver<blink::mojom::EyeDropperChooser> receiver)
42 : FrameServiceBase(render_frame_host, std::move(receiver)) {}
43
~EyeDropperChooserImpl()44 EyeDropperChooserImpl::~EyeDropperChooserImpl() {
45 if (callback_)
46 std::move(callback_).Run(/*success=*/false, /*color=*/0);
47 }
48
Choose(ChooseCallback callback)49 void EyeDropperChooserImpl::Choose(ChooseCallback callback) {
50 if (!render_frame_host() || callback_ || eye_dropper_) {
51 std::move(callback).Run(/*success=*/false, /*color=*/0);
52 return;
53 }
54
55 callback_ = std::move(callback);
56 WebContents* web_contents =
57 WebContents::FromRenderFrameHost(render_frame_host());
58 if (WebContentsDelegate* delegate = web_contents->GetDelegate())
59 eye_dropper_ = delegate->OpenEyeDropper(render_frame_host(), this);
60
61 if (!eye_dropper_) {
62 // Color selection wasn't successful since the eye dropper can't be opened.
63 ColorSelectionCanceled();
64 }
65 }
66
ColorSelected(SkColor color)67 void EyeDropperChooserImpl::ColorSelected(SkColor color) {
68 eye_dropper_.reset();
69 std::move(callback_).Run(/*success=*/true, color);
70 }
71
ColorSelectionCanceled()72 void EyeDropperChooserImpl::ColorSelectionCanceled() {
73 eye_dropper_.reset();
74 std::move(callback_).Run(/*success=*/false, /*color=*/0);
75 }
76
77 } // namespace content
78