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