2007-10-01 Eric Seidel <eric@webkit.org>
[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 @implementation DOMSVGScriptElement (WebDOMSVGScriptElementOperationsPrivate)
207
208 - (NSArray *)_subresourceURLs
209 {
210     return [self _URLsFromSelectors:@selector(href), nil];
211 }
212
213 @end
214
215 @implementation DOMSVGCursorElement (WebDOMSVGCursorElementOperationsPrivate)
216
217 - (NSArray *)_subresourceURLs
218 {
219     return [self _URLsFromSelectors:@selector(href), nil];
220 }
221
222 @end
223
224 @implementation DOMSVGFEImageElement (WebDOMSVGFEImageElementOperationsPrivate)
225
226 - (NSArray *)_subresourceURLs
227 {
228     return [self _URLsFromSelectors:@selector(href), nil];
229 }
230
231 @end
232
233 #endif
234
235 @implementation DOMProcessingInstruction (WebDOMProcessingInstructionOperationsPrivate)
236
237 - (NSString *)_stylesheetURL
238 {
239     DOMStyleSheet *styleSheet = [self sheet];
240     if (styleSheet)
241         return [styleSheet href];
242     return nil;
243 }
244
245 - (NSArray *)_subresourceURLs
246 {
247     return [self _URLsFromSelectors:@selector(_stylesheetURL), nil];
248 }
249
250 @end
251
252 @implementation DOMHTMLEmbedElement (WebDOMHTMLEmbedElementOperationsPrivate)
253
254 - (NSArray *)_subresourceURLs
255 {
256     return [self _URLsFromSelectors:@selector(src), nil];
257 }
258
259 @end
260
261 @implementation DOMHTMLObjectElement (WebDOMHTMLObjectElementOperationsPrivate)
262
263 - (NSArray *)_subresourceURLs
264 {
265     SEL useMapSelector = [[self useMap] hasPrefix:@"#"] ? nil : @selector(useMap);
266     return [self _URLsFromSelectors:@selector(data), useMapSelector, nil];
267 }
268
269 @end
270
271 @implementation DOMHTMLParamElement (WebDOMHTMLParamElementOperationsPrivate)
272
273 - (NSArray *)_subresourceURLs
274 {
275     NSString *paramName = [self name];
276     if ([paramName _webkit_isCaseInsensitiveEqualToString:@"data"] ||
277         [paramName _webkit_isCaseInsensitiveEqualToString:@"movie"] ||
278         [paramName _webkit_isCaseInsensitiveEqualToString:@"src"]) {
279         return [self _URLsFromSelectors:@selector(value), nil];
280     }
281     return nil;
282 }
283
284 @end
285
286 @implementation DOMHTMLTableElement (WebDOMHTMLTableElementOperationsPrivate)
287
288 - (NSString *)_web_background
289 {
290     return [self getAttribute:@"background"];
291 }
292
293 - (NSArray *)_subresourceURLs
294 {
295     return [self _URLsFromSelectors:@selector(_web_background), nil];
296 }
297
298 @end
299
300 @implementation DOMHTMLTableCellElement (WebDOMHTMLTableCellElementOperationsPrivate)
301
302 - (NSString *)_web_background
303 {
304     return [self getAttribute:@"background"];
305 }
306
307 - (NSArray *)_subresourceURLs
308 {
309     return [self _URLsFromSelectors:@selector(_web_background), nil];
310 }
311
312 @end
313
314 @implementation DOMHTMLFrameElement (WebDOMHTMLFrameElementOperations)
315
316 - (WebFrame *)contentFrame
317 {
318     return [[self contentDocument] webFrame];
319 }
320
321 @end
322
323 @implementation DOMHTMLIFrameElement (WebDOMHTMLIFrameElementOperations)
324
325 - (WebFrame *)contentFrame
326 {
327     return [[self contentDocument] webFrame];
328 }
329
330 @end
331
332 @implementation DOMHTMLObjectElement (WebDOMHTMLObjectElementOperations)
333
334 - (WebFrame *)contentFrame
335 {
336     return [[self contentDocument] webFrame];
337 }
338
339 @end