[Cocoa] Add an optional width parameter to -[WKWebProcessPlugInNodeHandle renderedIma...
[WebKit-https.git] / Source / WebKit2 / WebProcess / InjectedBundle / API / Cocoa / WKWebProcessPlugInNodeHandle.mm
1 /*
2  * Copyright (C) 2013 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 "WKWebProcessPlugInNodeHandleInternal.h"
28
29 #import "WKSharedAPICast.h"
30 #import "WKWebProcessPlugInFrameInternal.h"
31 #import <WebCore/IntRect.h>
32 #import <WebKit/WebImage.h>
33
34 #if WK_API_ENABLED
35
36 using namespace WebKit;
37
38 @implementation WKWebProcessPlugInNodeHandle {
39     API::ObjectStorage<InjectedBundleNodeHandle> _nodeHandle;
40 }
41
42 - (void)dealloc
43 {
44     _nodeHandle->~InjectedBundleNodeHandle();
45     [super dealloc];
46 }
47
48 + (WKWebProcessPlugInNodeHandle *)nodeHandleWithJSValue:(JSValue *)value inContext:(JSContext *)context
49 {
50     JSContextRef contextRef = [context JSGlobalContextRef];
51     JSObjectRef objectRef = JSValueToObject(contextRef, [value JSValueRef], 0);
52     auto nodeHandle = InjectedBundleNodeHandle::getOrCreate(contextRef, objectRef);
53     if (!nodeHandle)
54         return nil;
55
56     return [wrapper(*nodeHandle.leakRef()) autorelease];
57 }
58
59 - (WKWebProcessPlugInFrame *)htmlIFrameElementContentFrame
60 {
61     auto frame = _nodeHandle->htmlIFrameElementContentFrame();
62     if (!frame)
63         return nil;
64
65     return [wrapper(*frame.leakRef()) autorelease];
66 }
67
68 #if PLATFORM(IOS)
69 - (UIImage *)renderedImageWithOptions:(WKSnapshotOptions)options
70 {
71     return [self renderedImageWithOptions:options width:nil];
72 }
73
74 - (UIImage *)renderedImageWithOptions:(WKSnapshotOptions)options width:(NSNumber *)width
75 {
76     std::optional<float> optionalWidth;
77     if (width)
78         optionalWidth = width.floatValue;
79
80     RefPtr<WebImage> image = _nodeHandle->renderedImage(toSnapshotOptions(options), options & kWKSnapshotOptionsExcludeOverflow, optionalWidth);
81     if (!image)
82         return nil;
83
84     return [[[UIImage alloc] initWithCGImage:image->bitmap().makeCGImage().get()] autorelease];
85 }
86 #endif
87
88 #if PLATFORM(MAC)
89 - (NSImage *)renderedImageWithOptions:(WKSnapshotOptions)options
90 {
91     return [self renderedImageWithOptions:options width:nil];
92 }
93
94 - (NSImage *)renderedImageWithOptions:(WKSnapshotOptions)options width:(NSNumber *)width
95 {
96     std::optional<float> optionalWidth;
97     if (width)
98         optionalWidth = width.floatValue;
99
100     RefPtr<WebImage> image = _nodeHandle->renderedImage(toSnapshotOptions(options), options & kWKSnapshotOptionsExcludeOverflow, optionalWidth);
101     if (!image)
102         return nil;
103
104     return [[[NSImage alloc] initWithCGImage:image->bitmap().makeCGImage().get() size:NSZeroSize] autorelease];
105 }
106 #endif
107
108 - (CGRect)elementBounds
109 {
110     return _nodeHandle->elementBounds();
111 }
112
113 - (BOOL)HTMLInputElementIsAutoFilled
114 {
115     return _nodeHandle->isHTMLInputElementAutoFilled();
116 }
117
118 - (void)setHTMLInputElementIsAutoFilled:(BOOL)isAutoFilled
119 {
120     _nodeHandle->setHTMLInputElementAutoFilled(isAutoFilled);
121 }
122
123 - (BOOL)HTMLInputElementIsUserEdited
124 {
125     return _nodeHandle->htmlInputElementLastChangeWasUserEdit();
126 }
127
128 - (BOOL)HTMLTextAreaElementIsUserEdited
129 {
130     return _nodeHandle->htmlTextAreaElementLastChangeWasUserEdit();
131 }
132
133 - (BOOL)isTextField
134 {
135     return _nodeHandle->isTextField();
136 }
137
138 - (WKWebProcessPlugInNodeHandle *)HTMLTableCellElementCellAbove
139 {
140     auto nodeHandle = _nodeHandle->htmlTableCellElementCellAbove();
141     if (!nodeHandle)
142         return nil;
143
144     return [wrapper(*nodeHandle.leakRef()) autorelease];
145 }
146
147 - (WKWebProcessPlugInFrame *)frame
148 {
149     return [wrapper(*_nodeHandle->document()->documentFrame().leakRef()) autorelease];
150 }
151
152 - (InjectedBundleNodeHandle&)_nodeHandle
153 {
154     return *_nodeHandle;
155 }
156
157 #pragma mark WKObject protocol implementation
158
159 - (API::Object&)_apiObject
160 {
161     return *_nodeHandle;
162 }
163
164 @end
165
166 #endif // WK_API_ENABLED