If there are no services available, do not show the service controls UI
[WebKit-https.git] / Source / WebKit2 / UIProcess / mac / ServicesController.mm
1 /*
2  * Copyright (C) 2014 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "config.h"
27 #import "ServicesController.h"
28
29 #if ENABLE(SERVICE_CONTROLS)
30
31 #import "WebContext.h"
32 #import "WebProcessMessages.h"
33 #import <wtf/NeverDestroyed.h>
34
35 #if __has_include(<AppKit/NSSharingService_Private.h>)
36 #import <AppKit/NSSharingService_Private.h>
37 #else
38 typedef enum {
39     NSSharingServicePickerStyleMenu = 0,
40     NSSharingServicePickerStyleRollover = 1,
41     NSSharingServicePickerStyleTextSelection = 2,
42     NSSharingServicePickerStyleDataDetector = 3
43 } NSSharingServicePickerStyle;
44
45 @interface NSSharingServicePicker (Details)
46 @property NSSharingServicePickerStyle style;
47 - (NSMenu *)menu;
48 @end
49 #endif
50
51 namespace WebKit {
52
53 ServicesController& ServicesController::shared()
54 {
55     static NeverDestroyed<ServicesController> sharedController;
56     return sharedController;
57 }
58
59 ServicesController::ServicesController()
60     : m_refreshExistingServicesTimer(RunLoop::main(), this, &ServicesController::refreshExistingServicesTimerFired)
61     , m_hasImageServices(false)
62     , m_hasSelectionServices(false)
63 {
64     m_refreshExistingServicesTimer.startOneShot(0);
65 }
66
67 void ServicesController::refreshExistingServices(WebContext* context)
68 {
69     ASSERT(context);
70
71     m_contextsToNotify.add(context);
72     m_refreshExistingServicesTimer.startOneShot(0);
73 }
74
75 void ServicesController::refreshExistingServicesTimerFired()
76 {
77     static NeverDestroyed<NSImage *> image([[NSImage alloc] init]);
78     RetainPtr<NSSharingServicePicker>  picker = adoptNS([[NSSharingServicePicker alloc] initWithItems:@[ image ]]);
79     [picker setStyle:NSSharingServicePickerStyleRollover];
80
81     bool hasImageServices = picker.get().menu;
82
83     static NeverDestroyed<NSAttributedString *> attributedString([[NSAttributedString alloc] initWithString:@"a"]);
84     picker = adoptNS([[NSSharingServicePicker alloc] initWithItems:@[ attributedString ]]);
85     [picker setStyle:NSSharingServicePickerStyleTextSelection];
86
87     bool hasSelectionServices = picker.get().menu;
88
89     bool notifyContexts = (hasImageServices != m_hasImageServices) || (hasSelectionServices != m_hasSelectionServices);
90     m_hasSelectionServices = hasSelectionServices;
91     m_hasImageServices = hasImageServices;
92
93     if (notifyContexts) {
94         for (const RefPtr<WebContext>& context : m_contextsToNotify)
95             context->sendToAllProcesses(Messages::WebProcess::SetEnabledServices(m_hasImageServices, m_hasSelectionServices));
96     }
97
98     m_contextsToNotify.clear();
99 }
100
101 } // namespace WebKit
102
103 #endif // ENABLE(SERVICE_CONTROLS)