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