6ed465090e17e27ccdbf9f1a87608ef89fbe0f75
[WebKit-https.git] / WebKit / DOM / WebDOMOperations.mm
1 /*
2  * Copyright (C) 2005 Apple Computer, 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  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer. 
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution. 
13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission. 
16  *
17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #import <WebKit/WebDOMOperationsPrivate.h>
30
31 #import <WebKit/DOMExtensions.h>
32 #import <WebKit/DOMHTML.h>
33 #import <JavaScriptCore/Assertions.h>
34 #import <WebKit/WebFrameBridge.h>
35 #import <WebKit/WebDataSourcePrivate.h>
36 #import <WebKit/WebFramePrivate.h>
37 #import <WebKit/WebKitNSStringExtras.h>
38 #import <WebKit/WebArchiver.h>
39
40 #if ENABLE(SVG)
41 #import <WebKit/DOMSVG.h>
42 #endif
43
44 @implementation DOMNode (WebDOMNodeOperations)
45
46 - (WebFrameBridge *)_bridge
47 {
48     return (WebFrameBridge *)[WebFrameBridge bridgeForDOMDocument:[self ownerDocument]];
49 }
50
51 - (WebArchive *)webArchive
52 {
53     return [WebArchiver archiveNode:self];
54 }
55
56 - (NSString *)markupString
57 {
58     return [[self _bridge] markupStringFromNode:self nodes:nil];
59 }
60
61 - (NSArray *)_URLsFromSelectors:(SEL)firstSel, ...
62 {
63     NSMutableArray *URLs = [NSMutableArray array];
64     
65     va_list args;
66     va_start(args, firstSel);
67     
68     SEL selector = firstSel;
69     do {
70 #if ENABLE(SVG)
71         NSString *string;
72         id attributeValue = [self performSelector:selector];
73         if ([attributeValue isKindOfClass:[DOMSVGAnimatedString class]])
74             string = [(DOMSVGAnimatedString*)attributeValue animVal];
75         else
76             string = attributeValue;
77 #else
78         NSString *string = [self performSelector:selector];
79 #endif
80         if ([string length] > 0)
81             [URLs addObject:[[self ownerDocument] URLWithAttributeString:string]];
82     } while ((selector = va_arg(args, SEL)) != nil);
83     
84     va_end(args);
85     
86     return URLs;
87 }
88
89 - (NSArray *)_subresourceURLs
90 {
91     return nil;
92 }
93
94 @end
95
96 @implementation DOMDocument (WebDOMDocumentOperations)
97
98 - (WebFrame *)webFrame
99 {
100     return [[self _bridge] webFrame];
101 }
102
103 - (NSURL *)URLWithAttributeString:(NSString *)string
104 {
105     return [[self _bridge] URLWithAttributeString:string];
106 }
107
108 @end
109
110 @implementation DOMDocument (WebDOMDocumentOperationsPrivate)
111
112 - (DOMRange *)_createRangeWithNode:(DOMNode *)node
113 {
114     DOMRange *range = [self createRange];
115     [range selectNode:node];
116     return range;
117 }
118
119 - (DOMRange *)_documentRange
120 {
121     return [self _createRangeWithNode:[self documentElement]];
122 }
123
124 @end
125
126 @implementation DOMRange (WebDOMRangeOperations)
127
128 - (WebFrameBridge *)_bridge
129 {
130     return [[self startContainer] _bridge];
131 }
132
133 - (WebArchive *)webArchive
134 {
135     return [WebArchiver archiveRange:self];
136 }
137
138 - (NSString *)markupString
139 {
140     return [[self _bridge] markupStringFromRange:self nodes:nil];
141 }
142
143 @end
144
145 @implementation DOMHTMLBodyElement (WebDOMHTMLBodyElementOperationsPrivate)
146
147 - (NSArray *)_subresourceURLs
148 {
149     return [self _URLsFromSelectors:@selector(background), nil];
150 }
151
152 @end
153
154 @implementation DOMHTMLInputElement (WebDOMHTMLInputElementOperationsPrivate)
155
156 - (NSArray *)_subresourceURLs
157 {
158     return [self _URLsFromSelectors:@selector(src), nil];
159 }
160
161 @end
162
163 @implementation DOMHTMLLinkElement (WebDOMHTMLLinkElementOperationsPrivate)
164
165 - (NSArray *)_subresourceURLs
166 {
167     NSString *relName = [self rel];
168     if ([relName _webkit_isCaseInsensitiveEqualToString:@"stylesheet"] || [relName _webkit_isCaseInsensitiveEqualToString:@"icon"]) {
169         return [self _URLsFromSelectors:@selector(href), nil];
170     }
171     return nil;
172 }
173
174 @end
175
176 @implementation DOMHTMLScriptElement (WebDOMHTMLScriptElementOperationsPrivate)
177
178 - (NSArray *)_subresourceURLs
179 {
180     return [self _URLsFromSelectors:@selector(src), nil];
181 }
182
183 @end
184
185 @implementation DOMHTMLImageElement (WebDOMHTMLImageElementOperationsPrivate)
186
187 - (NSArray *)_subresourceURLs
188 {
189     SEL useMapSelector = [[self useMap] hasPrefix:@"#"] ? nil : @selector(useMap);
190     return [self _URLsFromSelectors:@selector(src), useMapSelector, nil];
191 }
192
193 @end
194
195 #if ENABLE(SVG)
196
197 @implementation DOMSVGImageElement (WebDOMSVGImageElementOperationsPrivate)
198
199 - (NSArray *)_subresourceURLs
200 {
201     return [self _URLsFromSelectors:@selector(href), nil];
202 }
203
204 @end
205
206 #endif
207
208 @implementation DOMProcessingInstruction (WebDOMProcessingInstructionOperationsPrivate)
209
210 - (NSString *)_stylesheetURL
211 {
212     DOMStyleSheet *styleSheet = [self sheet];
213     if (styleSheet)
214         return [styleSheet href];
215     return nil;
216 }
217
218 - (NSArray *)_subresourceURLs
219 {
220     return [self _URLsFromSelectors:@selector(_stylesheetURL), nil];
221 }
222
223 @end
224
225 @implementation DOMHTMLEmbedElement (WebDOMHTMLEmbedElementOperationsPrivate)
226
227 - (NSArray *)_subresourceURLs
228 {
229     return [self _URLsFromSelectors:@selector(src), nil];
230 }
231
232 @end
233
234 @implementation DOMHTMLObjectElement (WebDOMHTMLObjectElementOperationsPrivate)
235
236 - (NSArray *)_subresourceURLs
237 {
238     SEL useMapSelector = [[self useMap] hasPrefix:@"#"] ? nil : @selector(useMap);
239     return [self _URLsFromSelectors:@selector(data), useMapSelector, nil];
240 }
241
242 @end
243
244 @implementation DOMHTMLParamElement (WebDOMHTMLParamElementOperationsPrivate)
245
246 - (NSArray *)_subresourceURLs
247 {
248     NSString *paramName = [self name];
249     if ([paramName _webkit_isCaseInsensitiveEqualToString:@"data"] ||
250         [paramName _webkit_isCaseInsensitiveEqualToString:@"movie"] ||
251         [paramName _webkit_isCaseInsensitiveEqualToString:@"src"]) {
252         return [self _URLsFromSelectors:@selector(value), nil];
253     }
254     return nil;
255 }
256
257 @end
258
259 @implementation DOMHTMLTableElement (WebDOMHTMLTableElementOperationsPrivate)
260
261 - (NSString *)_web_background
262 {
263     return [self getAttribute:@"background"];
264 }
265
266 - (NSArray *)_subresourceURLs
267 {
268     return [self _URLsFromSelectors:@selector(_web_background), nil];
269 }
270
271 @end
272
273 @implementation DOMHTMLTableCellElement (WebDOMHTMLTableCellElementOperationsPrivate)
274
275 - (NSString *)_web_background
276 {
277     return [self getAttribute:@"background"];
278 }
279
280 - (NSArray *)_subresourceURLs
281 {
282     return [self _URLsFromSelectors:@selector(_web_background), nil];
283 }
284
285 @end
286
287 @implementation DOMHTMLFrameElement (WebDOMHTMLFrameElementOperations)
288
289 - (WebFrame *)contentFrame
290 {
291     return [[self contentDocument] webFrame];
292 }
293
294 @end
295
296 @implementation DOMHTMLIFrameElement (WebDOMHTMLIFrameElementOperations)
297
298 - (WebFrame *)contentFrame
299 {
300     return [[self contentDocument] webFrame];
301 }
302
303 @end
304
305 @implementation DOMHTMLObjectElement (WebDOMHTMLObjectElementOperations)
306
307 - (WebFrame *)contentFrame
308 {
309     return [[self contentDocument] webFrame];
310 }
311
312 @end