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 #endif
207
208 @implementation DOMHTMLEmbedElement (WebDOMHTMLEmbedElementOperationsPrivate)
209
210 - (NSArray *)_subresourceURLs
211 {
212     return [self _URLsFromSelectors:@selector(src), nil];
213 }
214
215 @end
216
217 @implementation DOMHTMLObjectElement (WebDOMHTMLObjectElementOperationsPrivate)
218
219 - (NSArray *)_subresourceURLs
220 {
221     SEL useMapSelector = [[self useMap] hasPrefix:@"#"] ? nil : @selector(useMap);
222     return [self _URLsFromSelectors:@selector(data), useMapSelector, nil];
223 }
224
225 @end
226
227 @implementation DOMHTMLParamElement (WebDOMHTMLParamElementOperationsPrivate)
228
229 - (NSArray *)_subresourceURLs
230 {
231     NSString *paramName = [self name];
232     if ([paramName _webkit_isCaseInsensitiveEqualToString:@"data"] ||
233         [paramName _webkit_isCaseInsensitiveEqualToString:@"movie"] ||
234         [paramName _webkit_isCaseInsensitiveEqualToString:@"src"]) {
235         return [self _URLsFromSelectors:@selector(value), nil];
236     }
237     return nil;
238 }
239
240 @end
241
242 @implementation DOMHTMLTableElement (WebDOMHTMLTableElementOperationsPrivate)
243
244 - (NSString *)_web_background
245 {
246     return [self getAttribute:@"background"];
247 }
248
249 - (NSArray *)_subresourceURLs
250 {
251     return [self _URLsFromSelectors:@selector(_web_background), nil];
252 }
253
254 @end
255
256 @implementation DOMHTMLTableCellElement (WebDOMHTMLTableCellElementOperationsPrivate)
257
258 - (NSString *)_web_background
259 {
260     return [self getAttribute:@"background"];
261 }
262
263 - (NSArray *)_subresourceURLs
264 {
265     return [self _URLsFromSelectors:@selector(_web_background), nil];
266 }
267
268 @end
269
270 @implementation DOMHTMLFrameElement (WebDOMHTMLFrameElementOperations)
271
272 - (WebFrame *)contentFrame
273 {
274     return [[self contentDocument] webFrame];
275 }
276
277 @end
278
279 @implementation DOMHTMLIFrameElement (WebDOMHTMLIFrameElementOperations)
280
281 - (WebFrame *)contentFrame
282 {
283     return [[self contentDocument] webFrame];
284 }
285
286 @end
287
288 @implementation DOMHTMLObjectElement (WebDOMHTMLObjectElementOperations)
289
290 - (WebFrame *)contentFrame
291 {
292     return [[self contentDocument] webFrame];
293 }
294
295 @end