9a987beab160be3f07361aa367704301330f28fe
[WebKit-https.git] / WebKit / mac / WebView / WebFrame.mm
1 /*
2  * Copyright (C) 2005, 2006, 2007, 2008 Apple 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 "WebFrameInternal.h"
30
31 #import "DOMCSSStyleDeclarationInternal.h"
32 #import "DOMDocumentFragmentInternal.h"
33 #import "DOMDocumentInternal.h"
34 #import "DOMElementInternal.h"
35 #import "DOMHTMLElementInternal.h"
36 #import "DOMNodeInternal.h"
37 #import "DOMRangeInternal.h"
38 #import "WebArchiveInternal.h"
39 #import "WebChromeClient.h"
40 #import "WebDataSourceInternal.h"
41 #import "WebDocumentLoaderMac.h"
42 #import "WebDynamicScrollBarsView.h"
43 #import "WebFrameLoaderClient.h"
44 #import "WebFrameViewInternal.h"
45 #import "WebHTMLView.h"
46 #import "WebHTMLViewInternal.h"
47 #import "WebIconFetcherInternal.h"
48 #import "WebKitStatisticsPrivate.h"
49 #import "WebKitVersionChecks.h"
50 #import "WebNSObjectExtras.h"
51 #import "WebNSURLExtras.h"
52 #import "WebScriptDebugger.h"
53 #import "WebScriptWorldInternal.h"
54 #import "WebViewInternal.h"
55 #import <JavaScriptCore/APICast.h>
56 #import <WebCore/AXObjectCache.h>
57 #import <WebCore/AccessibilityObject.h>
58 #import <WebCore/AnimationController.h>
59 #import <WebCore/CSSMutableStyleDeclaration.h>
60 #import <WebCore/Chrome.h>
61 #import <WebCore/ColorMac.h>
62 #import <WebCore/DOMImplementation.h>
63 #import <WebCore/CachedResourceLoader.h>
64 #import <WebCore/DocumentFragment.h>
65 #import <WebCore/EventHandler.h>
66 #import <WebCore/EventNames.h>
67 #import <WebCore/Frame.h>
68 #import <WebCore/FrameLoader.h>
69 #import <WebCore/FrameLoaderStateMachine.h>
70 #import <WebCore/FrameTree.h>
71 #import <WebCore/GraphicsContext.h>
72 #import <WebCore/HTMLFrameOwnerElement.h>
73 #import <WebCore/HistoryItem.h>
74 #import <WebCore/HitTestResult.h>
75 #import <WebCore/LegacyWebArchive.h>
76 #import <WebCore/Page.h>
77 #import <WebCore/PluginData.h>
78 #import <WebCore/PrintContext.h>
79 #import <WebCore/RenderLayer.h>
80 #import <WebCore/RenderPart.h>
81 #import <WebCore/RenderView.h>
82 #import <WebCore/ReplaceSelectionCommand.h>
83 #import <WebCore/RuntimeApplicationChecks.h>
84 #import <WebCore/ScriptValue.h>
85 #import <WebCore/SmartReplace.h>
86 #import <WebCore/SVGSMILElement.h>
87 #import <WebCore/TextIterator.h>
88 #import <WebCore/ThreadCheck.h>
89 #import <WebCore/TypingCommand.h>
90 #import <WebCore/htmlediting.h>
91 #import <WebCore/markup.h>
92 #import <WebCore/visible_units.h>
93 #import <WebKitSystemInterface.h>
94 #import <runtime/JSLock.h>
95 #import <runtime/JSObject.h>
96 #import <runtime/JSValue.h>
97 #import <wtf/CurrentTime.h>
98
99 using namespace std;
100 using namespace WebCore;
101 using namespace HTMLNames;
102
103 using JSC::JSGlobalObject;
104 using JSC::JSLock;
105 using JSC::JSValue;
106 using JSC::SilenceAssertionsOnly;
107
108 /*
109 Here is the current behavior matrix for four types of navigations:
110
111 Standard Nav:
112
113  Restore form state:   YES
114  Restore scroll and focus state:  YES
115  Cache policy: NSURLRequestUseProtocolCachePolicy
116  Add to back/forward list: YES
117  
118 Back/Forward:
119
120  Restore form state:   YES
121  Restore scroll and focus state:  YES
122  Cache policy: NSURLRequestReturnCacheDataElseLoad
123  Add to back/forward list: NO
124
125 Reload (meaning only the reload button):
126
127  Restore form state:   NO
128  Restore scroll and focus state:  YES
129  Cache policy: NSURLRequestReloadIgnoringCacheData
130  Add to back/forward list: NO
131
132 Repeat load of the same URL (by any other means of navigation other than the reload button, including hitting return in the location field):
133
134  Restore form state:   NO
135  Restore scroll and focus state:  NO, reset to initial conditions
136  Cache policy: NSURLRequestReloadIgnoringCacheData
137  Add to back/forward list: NO
138 */
139
140 NSString *WebPageCacheEntryDateKey = @"WebPageCacheEntryDateKey";
141 NSString *WebPageCacheDataSourceKey = @"WebPageCacheDataSourceKey";
142 NSString *WebPageCacheDocumentViewKey = @"WebPageCacheDocumentViewKey";
143
144 NSString *WebFrameMainDocumentError = @"WebFrameMainDocumentErrorKey";
145 NSString *WebFrameHasPlugins = @"WebFrameHasPluginsKey";
146 NSString *WebFrameHasUnloadListener = @"WebFrameHasUnloadListenerKey";
147 NSString *WebFrameUsesDatabases = @"WebFrameUsesDatabasesKey";
148 NSString *WebFrameUsesGeolocation = @"WebFrameUsesGeolocationKey";
149 NSString *WebFrameUsesApplicationCache = @"WebFrameUsesApplicationCacheKey";
150 NSString *WebFrameCanSuspendActiveDOMObjects = @"WebFrameCanSuspendActiveDOMObjectsKey";
151
152 // FIXME: Remove when this key becomes publicly defined
153 NSString *NSAccessibilityEnhancedUserInterfaceAttribute = @"AXEnhancedUserInterface";
154
155 @implementation WebFramePrivate
156
157 - (void)dealloc
158 {
159     [webFrameView release];
160
161     delete scriptDebugger;
162
163     [super dealloc];
164 }
165
166 - (void)finalize
167 {
168     delete scriptDebugger;
169
170     [super finalize];
171 }
172
173 - (void)setWebFrameView:(WebFrameView *)v 
174
175     [v retain];
176     [webFrameView release];
177     webFrameView = v;
178 }
179
180 @end
181
182 EditableLinkBehavior core(WebKitEditableLinkBehavior editableLinkBehavior)
183 {
184     switch (editableLinkBehavior) {
185         case WebKitEditableLinkDefaultBehavior:
186             return EditableLinkDefaultBehavior;
187         case WebKitEditableLinkAlwaysLive:
188             return EditableLinkAlwaysLive;
189         case WebKitEditableLinkOnlyLiveWithShiftKey:
190             return EditableLinkOnlyLiveWithShiftKey;
191         case WebKitEditableLinkLiveWhenNotFocused:
192             return EditableLinkLiveWhenNotFocused;
193         case WebKitEditableLinkNeverLive:
194             return EditableLinkNeverLive;
195     }
196     ASSERT_NOT_REACHED();
197     return EditableLinkDefaultBehavior;
198 }
199
200 WebCore::EditingBehaviorType core(WebKitEditingBehavior behavior)
201 {
202     switch (behavior) {
203         case WebKitEditingMacBehavior:
204             return WebCore::EditingMacBehavior;
205         case WebKitEditingWinBehavior:
206             return WebCore::EditingWindowsBehavior;
207     }
208     ASSERT_NOT_REACHED();
209     return WebCore::EditingMacBehavior;
210 }
211
212 TextDirectionSubmenuInclusionBehavior core(WebTextDirectionSubmenuInclusionBehavior behavior)
213 {
214     switch (behavior) {
215         case WebTextDirectionSubmenuNeverIncluded:
216             return TextDirectionSubmenuNeverIncluded;
217         case WebTextDirectionSubmenuAutomaticallyIncluded:
218             return TextDirectionSubmenuAutomaticallyIncluded;
219         case WebTextDirectionSubmenuAlwaysIncluded:
220             return TextDirectionSubmenuAlwaysIncluded;
221     }
222     ASSERT_NOT_REACHED();
223     return TextDirectionSubmenuNeverIncluded;
224 }
225
226 @implementation WebFrame (WebInternal)
227
228 Frame* core(WebFrame *frame)
229 {
230     return frame ? frame->_private->coreFrame : 0;
231 }
232
233 WebFrame *kit(Frame* frame)
234 {
235     return frame ? static_cast<WebFrameLoaderClient*>(frame->loader()->client())->webFrame() : nil;
236 }
237
238 Page* core(WebView *webView)
239 {
240     return [webView page];
241 }
242
243 WebView *kit(Page* page)
244 {
245     return page ? static_cast<WebChromeClient*>(page->chrome()->client())->webView() : nil;
246 }
247
248 WebView *getWebView(WebFrame *webFrame)
249 {
250     Frame* coreFrame = core(webFrame);
251     if (!coreFrame)
252         return nil;
253     return kit(coreFrame->page());
254 }
255
256 + (PassRefPtr<Frame>)_createFrameWithPage:(Page*)page frameName:(const String&)name frameView:(WebFrameView *)frameView ownerElement:(HTMLFrameOwnerElement*)ownerElement
257 {
258     WebView *webView = kit(page);
259
260     WebFrame *frame = [[self alloc] _initWithWebFrameView:frameView webView:webView];
261     RefPtr<Frame> coreFrame = Frame::create(page, ownerElement, new WebFrameLoaderClient(frame));
262     [frame release];
263     frame->_private->coreFrame = coreFrame.get();
264
265     coreFrame->tree()->setName(name);
266     if (ownerElement) {
267         ASSERT(ownerElement->document()->frame());
268         ownerElement->document()->frame()->tree()->appendChild(coreFrame.get());
269     }
270
271     coreFrame->init();
272
273     [webView _setZoomMultiplier:[webView _realZoomMultiplier] isTextOnly:[webView _realZoomMultiplierIsTextOnly]];
274
275     return coreFrame.release();
276 }
277
278 + (void)_createMainFrameWithPage:(Page*)page frameName:(const String&)name frameView:(WebFrameView *)frameView
279 {
280     [self _createFrameWithPage:page frameName:name frameView:frameView ownerElement:0];
281 }
282
283 + (PassRefPtr<WebCore::Frame>)_createSubframeWithOwnerElement:(HTMLFrameOwnerElement*)ownerElement frameName:(const String&)name frameView:(WebFrameView *)frameView
284 {
285     return [self _createFrameWithPage:ownerElement->document()->frame()->page() frameName:name frameView:frameView ownerElement:ownerElement];
286 }
287
288 - (BOOL)_isIncludedInWebKitStatistics
289 {
290     return _private && _private->includedInWebKitStatistics;
291 }
292
293 - (void)_attachScriptDebugger
294 {
295     ScriptController* scriptController = _private->coreFrame->script();
296
297     // Calling ScriptController::globalObject() would create a window shell, and dispatch corresponding callbacks, which may be premature
298     // if the script debugger is attached before a document is created.  These calls use the debuggerWorld(), we will need to pass a world
299     // to be able to debug isolated worlds.
300     if (!scriptController->existingWindowShell(debuggerWorld()))
301         return;
302
303     JSGlobalObject* globalObject = scriptController->globalObject(debuggerWorld());
304     if (!globalObject)
305         return;
306
307     if (_private->scriptDebugger) {
308         ASSERT(_private->scriptDebugger == globalObject->debugger());
309         return;
310     }
311
312     _private->scriptDebugger = new WebScriptDebugger(globalObject);
313 }
314
315 - (void)_detachScriptDebugger
316 {
317     if (!_private->scriptDebugger)
318         return;
319
320     delete _private->scriptDebugger;
321     _private->scriptDebugger = 0;
322 }
323
324 - (id)_initWithWebFrameView:(WebFrameView *)fv webView:(WebView *)v
325 {
326     self = [super init];
327     if (!self)
328         return nil;
329
330     _private = [[WebFramePrivate alloc] init];
331
332     // Set includedInWebKitStatistics before calling WebFrameView _setWebFrame, since
333     // it calls WebFrame _isIncludedInWebKitStatistics.
334     if ((_private->includedInWebKitStatistics = [[v class] shouldIncludeInWebKitStatistics]))
335         ++WebFrameCount;
336
337     if (fv) {
338         [_private setWebFrameView:fv];
339         [fv _setWebFrame:self];
340     }
341
342     _private->shouldCreateRenderers = YES;
343
344     return self;
345 }
346
347 - (void)_clearCoreFrame
348 {
349     _private->coreFrame = 0;
350 }
351
352 - (void)_updateBackgroundAndUpdatesWhileOffscreen
353 {
354     WebView *webView = getWebView(self);
355     BOOL drawsBackground = [webView drawsBackground];
356     NSColor *backgroundColor = [webView backgroundColor];
357
358     Frame* coreFrame = _private->coreFrame;
359     for (Frame* frame = coreFrame; frame; frame = frame->tree()->traverseNext(coreFrame)) {
360         if ([webView _usesDocumentViews]) {
361             // Don't call setDrawsBackground:YES here because it may be NO because of a load
362             // in progress; WebFrameLoaderClient keeps it set to NO during the load process.
363             WebFrame *webFrame = kit(frame);
364             if (!drawsBackground)
365                 [[[webFrame frameView] _scrollView] setDrawsBackground:NO];
366             [[[webFrame frameView] _scrollView] setBackgroundColor:backgroundColor];
367             id documentView = [[webFrame frameView] documentView];
368             if ([documentView respondsToSelector:@selector(setDrawsBackground:)])
369                 [documentView setDrawsBackground:drawsBackground];
370             if ([documentView respondsToSelector:@selector(setBackgroundColor:)])
371                 [documentView setBackgroundColor:backgroundColor];
372         }
373
374         if (FrameView* view = frame->view()) {
375             view->setTransparent(!drawsBackground);
376             view->setBaseBackgroundColor(colorFromNSColor([backgroundColor colorUsingColorSpaceName:NSDeviceRGBColorSpace]));
377             view->setShouldUpdateWhileOffscreen([webView shouldUpdateWhileOffscreen]);
378         }
379     }
380 }
381
382 - (void)_setInternalLoadDelegate:(id)internalLoadDelegate
383 {
384     _private->internalLoadDelegate = internalLoadDelegate;
385 }
386
387 - (id)_internalLoadDelegate
388 {
389     return _private->internalLoadDelegate;
390 }
391
392 #ifndef BUILDING_ON_TIGER
393 - (void)_unmarkAllBadGrammar
394 {
395     Frame* coreFrame = _private->coreFrame;
396     for (Frame* frame = coreFrame; frame; frame = frame->tree()->traverseNext(coreFrame)) {
397         if (Document* document = frame->document())
398             document->markers()->removeMarkers(DocumentMarker::Grammar);
399     }
400 }
401 #endif
402
403 - (void)_unmarkAllMisspellings
404 {
405     Frame* coreFrame = _private->coreFrame;
406     for (Frame* frame = coreFrame; frame; frame = frame->tree()->traverseNext(coreFrame)) {
407         if (Document* document = frame->document())
408             document->markers()->removeMarkers(DocumentMarker::Spelling);
409     }
410 }
411
412 - (BOOL)_hasSelection
413 {
414     if ([getWebView(self) _usesDocumentViews]) {
415         id documentView = [_private->webFrameView documentView];    
416
417         // optimization for common case to avoid creating potentially large selection string
418         if ([documentView isKindOfClass:[WebHTMLView class]])
419             if (Frame* coreFrame = _private->coreFrame)
420                 return coreFrame->selection()->isRange();
421
422         if ([documentView conformsToProtocol:@protocol(WebDocumentText)])
423             return [[documentView selectedString] length] > 0;
424         
425         return NO;
426     }
427
428     Frame* coreFrame = _private->coreFrame;
429     return coreFrame && coreFrame->selection()->isRange();
430 }
431
432 - (void)_clearSelection
433 {
434     ASSERT([getWebView(self) _usesDocumentViews]);
435     id documentView = [_private->webFrameView documentView];    
436     if ([documentView conformsToProtocol:@protocol(WebDocumentText)])
437         [documentView deselectAll];
438 }
439
440 #if !ASSERT_DISABLED
441 - (BOOL)_atMostOneFrameHasSelection
442 {
443     // FIXME: 4186050 is one known case that makes this debug check fail.
444     BOOL found = NO;
445     Frame* coreFrame = _private->coreFrame;
446     for (Frame* frame = coreFrame; frame; frame = frame->tree()->traverseNext(coreFrame))
447         if ([kit(frame) _hasSelection]) {
448             if (found)
449                 return NO;
450             found = YES;
451         }
452     return YES;
453 }
454 #endif
455
456 - (WebFrame *)_findFrameWithSelection
457 {
458     Frame* coreFrame = _private->coreFrame;
459     for (Frame* frame = coreFrame; frame; frame = frame->tree()->traverseNext(coreFrame)) {
460         WebFrame *webFrame = kit(frame);
461         if ([webFrame _hasSelection])
462             return webFrame;
463     }
464     return nil;
465 }
466
467 - (void)_clearSelectionInOtherFrames
468 {
469     // We rely on WebDocumentSelection protocol implementors to call this method when they become first 
470     // responder. It would be nicer to just notice first responder changes here instead, but there's no 
471     // notification sent when the first responder changes in general (Radar 2573089).
472     WebFrame *frameWithSelection = [[getWebView(self) mainFrame] _findFrameWithSelection];
473     if (frameWithSelection != self)
474         [frameWithSelection _clearSelection];
475
476     // While we're in the general area of selection and frames, check that there is only one now.
477     ASSERT([[getWebView(self) mainFrame] _atMostOneFrameHasSelection]);
478 }
479
480 static inline WebDataSource *dataSource(DocumentLoader* loader)
481 {
482     return loader ? static_cast<WebDocumentLoaderMac*>(loader)->dataSource() : nil;
483 }
484
485 - (WebDataSource *)_dataSource
486 {
487     return dataSource(_private->coreFrame->loader()->documentLoader());
488 }
489
490 - (void)_addData:(NSData *)data
491 {
492     Document* document = _private->coreFrame->document();
493
494     document->setShouldCreateRenderers(_private->shouldCreateRenderers);
495     _private->coreFrame->loader()->documentLoader()->addData((const char *)[data bytes], [data length]);
496 }
497
498 - (NSString *)_stringWithDocumentTypeStringAndMarkupString:(NSString *)markupString
499 {
500     return _private->coreFrame->documentTypeString() + markupString;
501 }
502
503 - (NSArray *)_nodesFromList:(Vector<Node*> *)nodesVector
504 {
505     size_t size = nodesVector->size();
506     NSMutableArray *nodes = [NSMutableArray arrayWithCapacity:size];
507     for (size_t i = 0; i < size; ++i)
508         [nodes addObject:kit((*nodesVector)[i])];
509     return nodes;
510 }
511
512 - (NSString *)_markupStringFromRange:(DOMRange *)range nodes:(NSArray **)nodes
513 {
514     // FIXME: This is always "for interchange". Is that right? See the previous method.
515     Vector<Node*> nodeList;
516     NSString *markupString = createMarkup(core(range), nodes ? &nodeList : 0, AnnotateForInterchange);
517     if (nodes)
518         *nodes = [self _nodesFromList:&nodeList];
519
520     return [self _stringWithDocumentTypeStringAndMarkupString:markupString];
521 }
522
523 - (NSString *)_selectedString
524 {
525     return _private->coreFrame->displayStringModifiedByEncoding(_private->coreFrame->editor()->selectedText());
526 }
527
528 - (NSString *)_stringForRange:(DOMRange *)range
529 {
530     // This will give a system malloc'd buffer that can be turned directly into an NSString
531     unsigned length;
532     UChar* buf = plainTextToMallocAllocatedBuffer(core(range), length, true);
533     
534     if (!buf)
535         return [NSString string];
536
537     // Transfer buffer ownership to NSString
538     return [[[NSString alloc] initWithCharactersNoCopy:buf length:length freeWhenDone:YES] autorelease];
539 }
540
541 - (void)_drawRect:(NSRect)rect contentsOnly:(BOOL)contentsOnly
542 {
543     ASSERT([[NSGraphicsContext currentContext] isFlipped]);
544
545     CGContextRef ctx = static_cast<CGContextRef>([[NSGraphicsContext currentContext] graphicsPort]);
546     GraphicsContext context(ctx);
547
548     FrameView* view = _private->coreFrame->view();
549     
550     bool shouldFlatten = false;
551     if (Frame* parentFrame = _private->coreFrame->tree()->parent()) {
552         // For subframes, we need to inherit the paint behavior from our parent
553         FrameView* parentView = parentFrame ? parentFrame->view() : 0;
554         if (parentView)
555             shouldFlatten = parentView->paintBehavior() & PaintBehaviorFlattenCompositingLayers;
556     } else
557         shouldFlatten = WKCGContextIsBitmapContext(ctx) && [getWebView(self) _includesFlattenedCompositingLayersWhenDrawingToBitmap];
558
559     PaintBehavior oldBehavior = PaintBehaviorNormal;
560     if (shouldFlatten) {
561         oldBehavior = view->paintBehavior();
562         view->setPaintBehavior(oldBehavior | PaintBehaviorFlattenCompositingLayers);
563     }
564     
565     if (contentsOnly)
566         _private->coreFrame->view()->paintContents(&context, enclosingIntRect(rect));
567     else
568         _private->coreFrame->view()->paint(&context, enclosingIntRect(rect));
569
570     if (shouldFlatten)
571         view->setPaintBehavior(oldBehavior);
572 }
573
574 // Used by pagination code called from AppKit when a standalone web page is printed.
575 - (NSArray*)_computePageRectsWithPrintWidthScaleFactor:(float)printWidthScaleFactor printHeight:(float)printHeight
576 {
577     NSMutableArray* pages = [NSMutableArray arrayWithCapacity:5];
578     if (printWidthScaleFactor <= 0) {
579         LOG_ERROR("printWidthScaleFactor has bad value %.2f", printWidthScaleFactor);
580         return pages;
581     }
582     
583     if (printHeight <= 0) {
584         LOG_ERROR("printHeight has bad value %.2f", printHeight);
585         return pages;
586     }
587
588     if (!_private->coreFrame || !_private->coreFrame->document() || !_private->coreFrame->view()) return pages;
589     RenderView* root = toRenderView(_private->coreFrame->document()->renderer());
590     if (!root) return pages;
591     
592     FrameView* view = _private->coreFrame->view();
593     if (!view)
594         return pages;
595
596     NSView* documentView = view->documentView();
597     if (!documentView)
598         return pages;
599
600     float docWidth = root->layer()->width();
601     float printWidth = docWidth / printWidthScaleFactor;
602
603     PrintContext printContext(_private->coreFrame);
604     printContext.computePageRectsWithPageSize(FloatSize(printWidth, printHeight), true);
605
606     const Vector<IntRect>& pageRects = printContext.pageRects();
607     const size_t pageCount = pageRects.size();
608     for (size_t pageNumber = 0; pageNumber < pageCount; ++pageNumber)
609         [pages addObject: [NSValue valueWithRect: NSRect(pageRects[pageNumber])]];
610     return pages;
611 }
612
613 - (BOOL)_getVisibleRect:(NSRect*)rect
614 {
615     ASSERT_ARG(rect, rect);
616     if (RenderPart* ownerRenderer = _private->coreFrame->ownerRenderer()) {
617         if (ownerRenderer->needsLayout())
618             return NO;
619         *rect = ownerRenderer->absoluteClippedOverflowRect();
620         return YES;
621     }
622
623     return NO;
624 }
625
626 - (NSString *)_stringByEvaluatingJavaScriptFromString:(NSString *)string
627 {
628     return [self _stringByEvaluatingJavaScriptFromString:string forceUserGesture:true];
629 }
630
631 - (NSString *)_stringByEvaluatingJavaScriptFromString:(NSString *)string forceUserGesture:(BOOL)forceUserGesture
632 {
633     ASSERT(_private->coreFrame->document());
634     
635     JSValue result = _private->coreFrame->script()->executeScript(string, forceUserGesture).jsValue();
636
637     if (!_private->coreFrame) // In case the script removed our frame from the page.
638         return @"";
639
640     // This bizarre set of rules matches behavior from WebKit for Safari 2.0.
641     // If you don't like it, use -[WebScriptObject evaluateWebScript:] or 
642     // JSEvaluateScript instead, since they have less surprising semantics.
643     if (!result || !result.isBoolean() && !result.isString() && !result.isNumber())
644         return @"";
645
646     JSLock lock(SilenceAssertionsOnly);
647     return ustringToString(result.toString(_private->coreFrame->script()->globalObject(mainThreadNormalWorld())->globalExec()));
648 }
649
650 - (NSRect)_caretRectAtNode:(DOMNode *)node offset:(int)offset affinity:(NSSelectionAffinity)affinity
651 {
652     VisiblePosition visiblePosition(core(node), offset, static_cast<EAffinity>(affinity));
653     return visiblePosition.absoluteCaretBounds();
654 }
655
656 - (NSRect)_firstRectForDOMRange:(DOMRange *)range
657 {
658    return _private->coreFrame->editor()->firstRectForRange(core(range));
659 }
660
661 - (void)_scrollDOMRangeToVisible:(DOMRange *)range
662 {
663     NSRect rangeRect = [self _firstRectForDOMRange:range];    
664     Node *startNode = core([range startContainer]);
665         
666     if (startNode && startNode->renderer()) {
667         RenderLayer *layer = startNode->renderer()->enclosingLayer();
668         if (layer)
669             layer->scrollRectToVisible(enclosingIntRect(rangeRect), false, ScrollAlignment::alignToEdgeIfNeeded, ScrollAlignment::alignToEdgeIfNeeded);
670     }
671 }
672
673 - (BOOL)_needsLayout
674 {
675     return _private->coreFrame->view() ? _private->coreFrame->view()->needsLayout() : false;
676 }
677
678 - (id)_accessibilityTree
679 {
680 #if HAVE(ACCESSIBILITY)
681     if (!AXObjectCache::accessibilityEnabled()) {
682         AXObjectCache::enableAccessibility();
683         if ([[NSApp accessibilityAttributeValue:NSAccessibilityEnhancedUserInterfaceAttribute] boolValue])
684             AXObjectCache::enableEnhancedUserInterfaceAccessibility();
685     }
686
687     if (!_private->coreFrame || !_private->coreFrame->document())
688         return nil;
689     RenderView* root = toRenderView(_private->coreFrame->document()->renderer());
690     if (!root)
691         return nil;
692     return _private->coreFrame->document()->axObjectCache()->getOrCreate(root)->wrapper();
693 #else
694     return nil;
695 #endif
696 }
697
698 - (DOMRange *)_rangeByAlteringCurrentSelection:(SelectionController::EAlteration)alteration direction:(SelectionController::EDirection)direction granularity:(TextGranularity)granularity
699 {
700     if (_private->coreFrame->selection()->isNone())
701         return nil;
702
703     SelectionController selection;
704     selection.setSelection(_private->coreFrame->selection()->selection());
705     selection.modify(alteration, direction, granularity);
706     return kit(selection.toNormalizedRange().get());
707 }
708
709 - (TextGranularity)_selectionGranularity
710 {
711     return _private->coreFrame->selectionGranularity();
712 }
713
714 - (NSRange)_convertToNSRange:(Range *)range
715 {
716     if (!range || !range->startContainer())
717         return NSMakeRange(NSNotFound, 0);
718
719     Element* selectionRoot = _private->coreFrame->selection()->rootEditableElement();
720     Element* scope = selectionRoot ? selectionRoot : _private->coreFrame->document()->documentElement();
721     
722     // Mouse events may cause TSM to attempt to create an NSRange for a portion of the view
723     // that is not inside the current editable region.  These checks ensure we don't produce
724     // potentially invalid data when responding to such requests.
725     if (range->startContainer() != scope && !range->startContainer()->isDescendantOf(scope))
726         return NSMakeRange(NSNotFound, 0);
727     if (range->endContainer() != scope && !range->endContainer()->isDescendantOf(scope))
728         return NSMakeRange(NSNotFound, 0);
729     
730     RefPtr<Range> testRange = Range::create(scope->document(), scope, 0, range->startContainer(), range->startOffset());
731     ASSERT(testRange->startContainer() == scope);
732     int startPosition = TextIterator::rangeLength(testRange.get());
733
734     ExceptionCode ec;
735     testRange->setEnd(range->endContainer(), range->endOffset(), ec);
736     ASSERT(testRange->startContainer() == scope);
737     int endPosition = TextIterator::rangeLength(testRange.get());
738
739     return NSMakeRange(startPosition, endPosition - startPosition);
740 }
741
742 - (PassRefPtr<Range>)_convertToDOMRange:(NSRange)nsrange
743 {
744     if (nsrange.location > INT_MAX)
745         return 0;
746     if (nsrange.length > INT_MAX || nsrange.location + nsrange.length > INT_MAX)
747         nsrange.length = INT_MAX - nsrange.location;
748
749     // our critical assumption is that we are only called by input methods that
750     // concentrate on a given area containing the selection
751     // We have to do this because of text fields and textareas. The DOM for those is not
752     // directly in the document DOM, so serialization is problematic. Our solution is
753     // to use the root editable element of the selection start as the positional base.
754     // That fits with AppKit's idea of an input context.
755     Element* selectionRoot = _private->coreFrame->selection()->rootEditableElement();
756     Element* scope = selectionRoot ? selectionRoot : _private->coreFrame->document()->documentElement();
757     return TextIterator::rangeFromLocationAndLength(scope, nsrange.location, nsrange.length);
758 }
759
760 - (DOMRange *)convertNSRangeToDOMRange:(NSRange)nsrange
761 {
762     // This method exists to maintain compatibility with Leopard's Dictionary.app. <rdar://problem/6002160>
763     return [self _convertNSRangeToDOMRange:nsrange];
764 }
765
766 - (DOMRange *)_convertNSRangeToDOMRange:(NSRange)nsrange
767 {
768     return kit([self _convertToDOMRange:nsrange].get());
769 }
770
771 - (NSRange)convertDOMRangeToNSRange:(DOMRange *)range
772 {
773     // This method exists to maintain compatibility with Leopard's Dictionary.app. <rdar://problem/6002160>
774     return [self _convertDOMRangeToNSRange:range];
775 }
776
777 - (NSRange)_convertDOMRangeToNSRange:(DOMRange *)range
778 {
779     return [self _convertToNSRange:core(range)];
780 }
781
782 - (DOMRange *)_markDOMRange
783 {
784     return kit(_private->coreFrame->editor()->mark().toNormalizedRange().get());
785 }
786
787 // Given proposedRange, returns an extended range that includes adjacent whitespace that should
788 // be deleted along with the proposed range in order to preserve proper spacing and punctuation of
789 // the text surrounding the deletion.
790 - (DOMRange *)_smartDeleteRangeForProposedRange:(DOMRange *)proposedRange
791 {
792     Node* startContainer = core([proposedRange startContainer]);
793     Node* endContainer = core([proposedRange endContainer]);
794     if (startContainer == nil || endContainer == nil)
795         return nil;
796
797     ASSERT(startContainer->document() == endContainer->document());
798     
799     _private->coreFrame->document()->updateLayoutIgnorePendingStylesheets();
800
801     Position start(startContainer, [proposedRange startOffset]);
802     Position end(endContainer, [proposedRange endOffset]);
803     Position newStart = start.upstream().leadingWhitespacePosition(DOWNSTREAM, true);
804     if (newStart.isNull())
805         newStart = start;
806     Position newEnd = end.downstream().trailingWhitespacePosition(DOWNSTREAM, true);
807     if (newEnd.isNull())
808         newEnd = end;
809
810     newStart = rangeCompliantEquivalent(newStart);
811     newEnd = rangeCompliantEquivalent(newEnd);
812
813     RefPtr<Range> range = _private->coreFrame->document()->createRange();
814     int exception = 0;
815     range->setStart(newStart.node(), newStart.deprecatedEditingOffset(), exception);
816     range->setEnd(newStart.node(), newStart.deprecatedEditingOffset(), exception);
817     return kit(range.get());
818 }
819
820 - (DOMDocumentFragment *)_documentFragmentWithMarkupString:(NSString *)markupString baseURLString:(NSString *)baseURLString 
821 {
822     if (!_private->coreFrame || !_private->coreFrame->document())
823         return nil;
824
825     return kit(createFragmentFromMarkup(_private->coreFrame->document(), markupString, baseURLString, FragmentScriptingNotAllowed).get());
826 }
827
828 - (DOMDocumentFragment *)_documentFragmentWithNodesAsParagraphs:(NSArray *)nodes
829 {
830     if (!_private->coreFrame || !_private->coreFrame->document())
831         return nil;
832     
833     NSEnumerator *nodeEnum = [nodes objectEnumerator];
834     Vector<Node*> nodesVector;
835     DOMNode *node;
836     while ((node = [nodeEnum nextObject]))
837         nodesVector.append(core(node));
838     
839     return kit(createFragmentFromNodes(_private->coreFrame->document(), nodesVector).get());
840 }
841
842 - (void)_replaceSelectionWithNode:(DOMNode *)node selectReplacement:(BOOL)selectReplacement smartReplace:(BOOL)smartReplace matchStyle:(BOOL)matchStyle
843 {
844     DOMDocumentFragment *fragment = kit(_private->coreFrame->document()->createDocumentFragment().get());
845     [fragment appendChild:node];
846     [self _replaceSelectionWithFragment:fragment selectReplacement:selectReplacement smartReplace:smartReplace matchStyle:matchStyle];
847 }
848
849 - (void)_insertParagraphSeparatorInQuotedContent
850 {
851     if (_private->coreFrame->selection()->isNone())
852         return;
853     
854     TypingCommand::insertParagraphSeparatorInQuotedContent(_private->coreFrame->document());
855     _private->coreFrame->revealSelection(ScrollAlignment::alignToEdgeIfNeeded);
856 }
857
858 - (VisiblePosition)_visiblePositionForPoint:(NSPoint)point
859 {
860     // FIXME: Someone with access to Apple's sources could remove this needless wrapper call.
861     return _private->coreFrame->visiblePositionForPoint(IntPoint(point));
862 }
863
864 - (DOMRange *)_characterRangeAtPoint:(NSPoint)point
865 {
866     VisiblePosition position = [self _visiblePositionForPoint:point];
867     if (position.isNull())
868         return nil;
869     
870     VisiblePosition previous = position.previous();
871     if (previous.isNotNull()) {
872         DOMRange *previousCharacterRange = kit(makeRange(previous, position).get());
873         NSRect rect = [self _firstRectForDOMRange:previousCharacterRange];
874         if (NSPointInRect(point, rect))
875             return previousCharacterRange;
876     }
877
878     VisiblePosition next = position.next();
879     if (next.isNotNull()) {
880         DOMRange *nextCharacterRange = kit(makeRange(position, next).get());
881         NSRect rect = [self _firstRectForDOMRange:nextCharacterRange];
882         if (NSPointInRect(point, rect))
883             return nextCharacterRange;
884     }
885     
886     return nil;
887 }
888
889 - (DOMCSSStyleDeclaration *)_typingStyle
890 {
891     if (!_private->coreFrame || !_private->coreFrame->typingStyle())
892         return nil;
893     return kit(_private->coreFrame->typingStyle()->copy().get());
894 }
895
896 - (void)_setTypingStyle:(DOMCSSStyleDeclaration *)style withUndoAction:(EditAction)undoAction
897 {
898     if (!_private->coreFrame)
899         return;
900     _private->coreFrame->editor()->computeAndSetTypingStyle(core(style), undoAction);
901 }
902
903 - (void)_dragSourceEndedAt:(NSPoint)windowLoc operation:(NSDragOperation)operation
904 {
905     if (!_private->coreFrame)
906         return;
907     FrameView* view = _private->coreFrame->view();
908     if (!view)
909         return;
910     ASSERT([getWebView(self) _usesDocumentViews]);
911     // FIXME: These are fake modifier keys here, but they should be real ones instead.
912     PlatformMouseEvent event(IntPoint(windowLoc), globalPoint(windowLoc, [view->platformWidget() window]),
913         LeftButton, MouseEventMoved, 0, false, false, false, false, currentTime());
914     _private->coreFrame->eventHandler()->dragSourceEndedAt(event, (DragOperation)operation);
915 }
916
917 - (BOOL)_canProvideDocumentSource
918 {
919     Frame* frame = _private->coreFrame;
920     String mimeType = frame->loader()->writer()->mimeType();
921     PluginData* pluginData = frame->page() ? frame->page()->pluginData() : 0;
922
923     if (WebCore::DOMImplementation::isTextMIMEType(mimeType) ||
924         Image::supportsType(mimeType) ||
925         (pluginData && pluginData->supportsMimeType(mimeType)))
926         return NO;
927
928     return YES;
929 }
930
931 - (BOOL)_canSaveAsWebArchive
932 {
933     // Currently, all documents that we can view source for
934     // (HTML and XML documents) can also be saved as web archives
935     return [self _canProvideDocumentSource];
936 }
937
938 - (void)_receivedData:(NSData *)data textEncodingName:(NSString *)textEncodingName
939 {
940     // Set the encoding. This only needs to be done once, but it's harmless to do it again later.
941     String encoding = _private->coreFrame->loader()->documentLoader()->overrideEncoding();
942     bool userChosen = !encoding.isNull();
943     if (encoding.isNull())
944         encoding = textEncodingName;
945     _private->coreFrame->loader()->writer()->setEncoding(encoding, userChosen);
946     [self _addData:data];
947 }
948
949 @end
950
951 @implementation WebFrame (WebPrivate)
952
953 // FIXME: This exists only as a convenience for Safari, consider moving there.
954 - (BOOL)_isDescendantOfFrame:(WebFrame *)ancestor
955 {
956     Frame* coreFrame = _private->coreFrame;
957     return coreFrame && coreFrame->tree()->isDescendantOf(core(ancestor));
958 }
959
960 - (void)_setShouldCreateRenderers:(BOOL)shouldCreateRenderers
961 {
962     _private->shouldCreateRenderers = shouldCreateRenderers;
963 }
964
965 - (NSColor *)_bodyBackgroundColor
966 {
967     Document* document = _private->coreFrame->document();
968     if (!document)
969         return nil;
970     HTMLElement* body = document->body();
971     if (!body)
972         return nil;
973     RenderObject* bodyRenderer = body->renderer();
974     if (!bodyRenderer)
975         return nil;
976     Color color = bodyRenderer->style()->visitedDependentColor(CSSPropertyBackgroundColor);
977     if (!color.isValid())
978         return nil;
979     return nsColor(color);
980 }
981
982 - (BOOL)_isFrameSet
983 {
984     Document* document = _private->coreFrame->document();
985     return document && document->isFrameSet();
986 }
987
988 - (BOOL)_firstLayoutDone
989 {
990     return _private->coreFrame->loader()->stateMachine()->firstLayoutDone();
991 }
992
993 - (WebFrameLoadType)_loadType
994 {
995     return (WebFrameLoadType)_private->coreFrame->loader()->loadType();
996 }
997
998 - (NSRange)_selectedNSRange
999 {
1000     return [self _convertToNSRange:_private->coreFrame->selection()->toNormalizedRange().get()];
1001 }
1002
1003 - (void)_selectNSRange:(NSRange)range
1004 {
1005     RefPtr<Range> domRange = [self _convertToDOMRange:range];
1006     if (domRange)
1007         _private->coreFrame->selection()->setSelection(VisibleSelection(domRange.get(), SEL_DEFAULT_AFFINITY));
1008 }
1009
1010 - (BOOL)_isDisplayingStandaloneImage
1011 {
1012     Document* document = _private->coreFrame->document();
1013     return document && document->isImageDocument();
1014 }
1015
1016 - (unsigned)_pendingFrameUnloadEventCount
1017 {
1018     return _private->coreFrame->domWindow()->pendingUnloadEventListeners();
1019 }
1020
1021 - (WebIconFetcher *)fetchApplicationIcon:(id)target
1022                                 selector:(SEL)selector
1023 {
1024     return [WebIconFetcher _fetchApplicationIconForFrame:self
1025                                                   target:target
1026                                                 selector:selector];
1027 }
1028
1029 - (void)_setIsDisconnected:(bool)isDisconnected
1030 {
1031     _private->coreFrame->setIsDisconnected(isDisconnected);
1032 }
1033
1034 - (void)_setExcludeFromTextSearch:(bool)exclude
1035 {
1036     _private->coreFrame->setExcludeFromTextSearch(exclude);
1037 }
1038
1039 #if ENABLE(NETSCAPE_PLUGIN_API)
1040 - (void)_recursive_resumeNullEventsForAllNetscapePlugins
1041 {
1042     Frame* coreFrame = core(self);
1043     for (Frame* frame = coreFrame; frame; frame = frame->tree()->traverseNext(coreFrame)) {
1044         NSView <WebDocumentView> *documentView = [[kit(frame) frameView] documentView];
1045         if ([documentView isKindOfClass:[WebHTMLView class]])
1046             [(WebHTMLView *)documentView _resumeNullEventsForAllNetscapePlugins];
1047     }
1048 }
1049
1050 - (void)_recursive_pauseNullEventsForAllNetscapePlugins
1051 {
1052     Frame* coreFrame = core(self);
1053     for (Frame* frame = coreFrame; frame; frame = frame->tree()->traverseNext(coreFrame)) {
1054         NSView <WebDocumentView> *documentView = [[kit(frame) frameView] documentView];
1055         if ([documentView isKindOfClass:[WebHTMLView class]])
1056             [(WebHTMLView *)documentView _pauseNullEventsForAllNetscapePlugins];
1057     }
1058 }
1059 #endif
1060
1061 - (BOOL)_pauseAnimation:(NSString*)name onNode:(DOMNode *)node atTime:(NSTimeInterval)time
1062 {
1063     Frame* frame = core(self);
1064     if (!frame)
1065         return false;
1066
1067     AnimationController* controller = frame->animation();
1068     if (!controller)
1069         return false;
1070
1071     Node* coreNode = core(node);
1072     if (!coreNode || !coreNode->renderer())
1073         return false;
1074
1075     return controller->pauseAnimationAtTime(coreNode->renderer(), name, time);
1076 }
1077
1078 - (BOOL)_pauseTransitionOfProperty:(NSString*)name onNode:(DOMNode*)node atTime:(NSTimeInterval)time
1079 {
1080     Frame* frame = core(self);
1081     if (!frame)
1082         return false;
1083
1084     AnimationController* controller = frame->animation();
1085     if (!controller)
1086         return false;
1087
1088     Node* coreNode = core(node);
1089     if (!coreNode || !coreNode->renderer())
1090         return false;
1091
1092     return controller->pauseTransitionAtTime(coreNode->renderer(), name, time);
1093 }
1094
1095 // Pause a given SVG animation on the target node at a specific time.
1096 // This method is only intended to be used for testing the SVG animation system.
1097 - (BOOL)_pauseSVGAnimation:(NSString*)elementId onSMILNode:(DOMNode *)node atTime:(NSTimeInterval)time
1098 {
1099     Frame* frame = core(self);
1100     if (!frame)
1101         return false;
1102  
1103     Document* document = frame->document();
1104     if (!document || !document->svgExtensions())
1105         return false;
1106
1107     Node* coreNode = core(node);
1108     if (!coreNode || !SVGSMILElement::isSMILElement(coreNode))
1109         return false;
1110
1111 #if ENABLE(SVG)
1112     return document->accessSVGExtensions()->sampleAnimationAtTime(elementId, static_cast<SVGSMILElement*>(coreNode), time);
1113 #else
1114     return false;
1115 #endif
1116 }
1117
1118 - (unsigned) _numberOfActiveAnimations
1119 {
1120     Frame* frame = core(self);
1121     if (!frame)
1122         return false;
1123
1124     AnimationController* controller = frame->animation();
1125     if (!controller)
1126         return false;
1127
1128     return controller->numberOfActiveAnimations();
1129 }
1130
1131 - (void) _suspendAnimations
1132 {
1133     Frame* frame = core(self);
1134     if (!frame)
1135         return;
1136
1137     AnimationController* controller = frame->animation();
1138     if (!controller)
1139         return;
1140
1141     controller->suspendAnimations(frame->document());
1142 }
1143
1144 - (void) _resumeAnimations
1145 {
1146     Frame* frame = core(self);
1147     if (!frame)
1148         return;
1149
1150     AnimationController* controller = frame->animation();
1151     if (!controller)
1152         return;
1153
1154     controller->resumeAnimations(frame->document());
1155 }
1156
1157 - (void)_replaceSelectionWithFragment:(DOMDocumentFragment *)fragment selectReplacement:(BOOL)selectReplacement smartReplace:(BOOL)smartReplace matchStyle:(BOOL)matchStyle
1158 {
1159     if (_private->coreFrame->selection()->isNone() || !fragment)
1160         return;
1161     
1162     applyCommand(ReplaceSelectionCommand::create(_private->coreFrame->document(), core(fragment), selectReplacement, smartReplace, matchStyle));
1163     _private->coreFrame->revealSelection(ScrollAlignment::alignToEdgeIfNeeded);
1164 }
1165
1166 - (void)_replaceSelectionWithText:(NSString *)text selectReplacement:(BOOL)selectReplacement smartReplace:(BOOL)smartReplace
1167 {   
1168     DOMDocumentFragment* fragment = kit(createFragmentFromText(_private->coreFrame->selection()->toNormalizedRange().get(), text).get());
1169     [self _replaceSelectionWithFragment:fragment selectReplacement:selectReplacement smartReplace:smartReplace matchStyle:YES];
1170 }
1171
1172 - (void)_replaceSelectionWithMarkupString:(NSString *)markupString baseURLString:(NSString *)baseURLString selectReplacement:(BOOL)selectReplacement smartReplace:(BOOL)smartReplace
1173 {
1174     DOMDocumentFragment *fragment = [self _documentFragmentWithMarkupString:markupString baseURLString:baseURLString];
1175     [self _replaceSelectionWithFragment:fragment selectReplacement:selectReplacement smartReplace:smartReplace matchStyle:NO];
1176 }
1177
1178 // Determines whether whitespace needs to be added around aString to preserve proper spacing and
1179 // punctuation when it's inserted into the receiver's text over charRange. Returns by reference
1180 // in beforeString and afterString any whitespace that should be added, unless either or both are
1181 // nil. Both are returned as nil if aString is nil or if smart insertion and deletion are disabled.
1182 - (void)_smartInsertForString:(NSString *)pasteString replacingRange:(DOMRange *)rangeToReplace beforeString:(NSString **)beforeString afterString:(NSString **)afterString
1183 {
1184     // give back nil pointers in case of early returns
1185     if (beforeString)
1186         *beforeString = nil;
1187     if (afterString)
1188         *afterString = nil;
1189         
1190     // inspect destination
1191     Node *startContainer = core([rangeToReplace startContainer]);
1192     Node *endContainer = core([rangeToReplace endContainer]);
1193
1194     Position startPos(startContainer, [rangeToReplace startOffset]);
1195     Position endPos(endContainer, [rangeToReplace endOffset]);
1196
1197     VisiblePosition startVisiblePos = VisiblePosition(startPos, VP_DEFAULT_AFFINITY);
1198     VisiblePosition endVisiblePos = VisiblePosition(endPos, VP_DEFAULT_AFFINITY);
1199     
1200     // this check also ensures startContainer, startPos, endContainer, and endPos are non-null
1201     if (startVisiblePos.isNull() || endVisiblePos.isNull())
1202         return;
1203
1204     bool addLeadingSpace = startPos.leadingWhitespacePosition(VP_DEFAULT_AFFINITY, true).isNull() && !isStartOfParagraph(startVisiblePos);
1205     if (addLeadingSpace)
1206         if (UChar previousChar = startVisiblePos.previous().characterAfter())
1207             addLeadingSpace = !isCharacterSmartReplaceExempt(previousChar, true);
1208     
1209     bool addTrailingSpace = endPos.trailingWhitespacePosition(VP_DEFAULT_AFFINITY, true).isNull() && !isEndOfParagraph(endVisiblePos);
1210     if (addTrailingSpace)
1211         if (UChar thisChar = endVisiblePos.characterAfter())
1212             addTrailingSpace = !isCharacterSmartReplaceExempt(thisChar, false);
1213     
1214     // inspect source
1215     bool hasWhitespaceAtStart = false;
1216     bool hasWhitespaceAtEnd = false;
1217     unsigned pasteLength = [pasteString length];
1218     if (pasteLength > 0) {
1219         NSCharacterSet *whiteSet = [NSCharacterSet whitespaceAndNewlineCharacterSet];
1220         
1221         if ([whiteSet characterIsMember:[pasteString characterAtIndex:0]]) {
1222             hasWhitespaceAtStart = YES;
1223         }
1224         if ([whiteSet characterIsMember:[pasteString characterAtIndex:(pasteLength - 1)]]) {
1225             hasWhitespaceAtEnd = YES;
1226         }
1227     }
1228     
1229     // issue the verdict
1230     if (beforeString && addLeadingSpace && !hasWhitespaceAtStart)
1231         *beforeString = @" ";
1232     if (afterString && addTrailingSpace && !hasWhitespaceAtEnd)
1233         *afterString = @" ";
1234 }
1235
1236 - (NSMutableDictionary *)_cacheabilityDictionary
1237 {
1238     NSMutableDictionary *result = [NSMutableDictionary dictionary];
1239     
1240     FrameLoader* frameLoader = _private->coreFrame->loader();
1241     DocumentLoader* documentLoader = frameLoader->documentLoader();
1242     if (documentLoader && !documentLoader->mainDocumentError().isNull())
1243         [result setObject:(NSError *)documentLoader->mainDocumentError() forKey:WebFrameMainDocumentError];
1244         
1245     if (frameLoader->subframeLoader()->containsPlugins())
1246         [result setObject:[NSNumber numberWithBool:YES] forKey:WebFrameHasPlugins];
1247     
1248     if (DOMWindow* domWindow = _private->coreFrame->domWindow()) {
1249         if (domWindow->hasEventListeners(eventNames().unloadEvent))
1250             [result setObject:[NSNumber numberWithBool:YES] forKey:WebFrameHasUnloadListener];
1251             
1252 #if ENABLE(OFFLINE_WEB_APPLICATIONS)
1253         if (domWindow->optionalApplicationCache())
1254             [result setObject:[NSNumber numberWithBool:YES] forKey:WebFrameUsesApplicationCache];
1255 #endif
1256     }
1257     
1258     if (Document* document = _private->coreFrame->document()) {
1259 #if ENABLE(DATABASE)
1260         if (document->hasOpenDatabases())
1261             [result setObject:[NSNumber numberWithBool:YES] forKey:WebFrameUsesDatabases];
1262 #endif
1263             
1264         if (document->usingGeolocation())
1265             [result setObject:[NSNumber numberWithBool:YES] forKey:WebFrameUsesGeolocation];
1266             
1267         if (!document->canSuspendActiveDOMObjects())
1268             [result setObject:[NSNumber numberWithBool:YES] forKey:WebFrameCanSuspendActiveDOMObjects];
1269     }
1270     
1271     return result;
1272 }
1273
1274 - (BOOL)_allowsFollowingLink:(NSURL *)URL
1275 {
1276     if (!_private->coreFrame)
1277         return YES;
1278     return SecurityOrigin::canDisplay(URL, String(), _private->coreFrame->document());
1279 }
1280
1281 - (NSString *)_stringByEvaluatingJavaScriptFromString:(NSString *)string withGlobalObject:(JSObjectRef)globalObjectRef inScriptWorld:(WebScriptWorld *)world
1282 {
1283     // Start off with some guess at a frame and a global object, we'll try to do better...!
1284     JSDOMWindow* anyWorldGlobalObject = _private->coreFrame->script()->globalObject(mainThreadNormalWorld());
1285
1286     // The global object is probably a shell object? - if so, we know how to use this!
1287     JSC::JSObject* globalObjectObj = toJS(globalObjectRef);
1288     if (!strcmp(globalObjectObj->classInfo()->className, "JSDOMWindowShell"))
1289         anyWorldGlobalObject = static_cast<JSDOMWindowShell*>(globalObjectObj)->window();
1290
1291     // Get the frame frome the global object we've settled on.
1292     Frame* frame = anyWorldGlobalObject->impl()->frame();
1293     ASSERT(frame->document());
1294     JSValue result = frame->script()->executeScriptInWorld(core(world), string, true).jsValue();
1295
1296     if (!frame) // In case the script removed our frame from the page.
1297         return @"";
1298
1299     // This bizarre set of rules matches behavior from WebKit for Safari 2.0.
1300     // If you don't like it, use -[WebScriptObject evaluateWebScript:] or 
1301     // JSEvaluateScript instead, since they have less surprising semantics.
1302     if (!result || !result.isBoolean() && !result.isString() && !result.isNumber())
1303         return @"";
1304
1305     JSLock lock(SilenceAssertionsOnly);
1306     return ustringToString(result.toString(anyWorldGlobalObject->globalExec()));
1307 }
1308
1309 - (JSGlobalContextRef)_globalContextForScriptWorld:(WebScriptWorld *)world
1310 {
1311     Frame* coreFrame = _private->coreFrame;
1312     if (!coreFrame)
1313         return 0;
1314     DOMWrapperWorld* coreWorld = core(world);
1315     if (!coreWorld)
1316         return 0;
1317     return toGlobalRef(coreFrame->script()->globalObject(coreWorld)->globalExec());
1318 }
1319
1320 - (void)setAllowsScrollersToOverlapContent:(BOOL)flag
1321 {
1322     ASSERT([[[self frameView] _scrollView] isKindOfClass:[WebDynamicScrollBarsView class]]);
1323     [(WebDynamicScrollBarsView *)[[self frameView] _scrollView] setAllowsScrollersToOverlapContent:flag];
1324 }
1325
1326 - (void)setAlwaysHideHorizontalScroller:(BOOL)flag
1327 {
1328     ASSERT([[[self frameView] _scrollView] isKindOfClass:[WebDynamicScrollBarsView class]]);
1329     [(WebDynamicScrollBarsView *)[[self frameView] _scrollView] setAlwaysHideHorizontalScroller:flag];
1330 }
1331 - (void)setAlwaysHideVerticalScroller:(BOOL)flag
1332 {
1333     ASSERT([[[self frameView] _scrollView] isKindOfClass:[WebDynamicScrollBarsView class]]);
1334     [(WebDynamicScrollBarsView *)[[self frameView] _scrollView] setAlwaysHideVerticalScroller:flag];
1335 }
1336
1337 - (void)setAccessibleName:(NSString *)name
1338 {
1339 #if HAVE(ACCESSIBILITY)
1340     if (!AXObjectCache::accessibilityEnabled())
1341         return;
1342     
1343     RenderView* root = toRenderView(_private->coreFrame->document()->renderer());
1344     if (!root)
1345         return;
1346     
1347     AccessibilityObject* rootObject = _private->coreFrame->document()->axObjectCache()->getOrCreate(root);
1348     String strName(name);
1349     rootObject->setAccessibleName(strName);
1350 #endif
1351 }
1352
1353 - (NSString*)_layerTreeAsText
1354 {
1355     Frame* coreFrame = _private->coreFrame;
1356     if (!coreFrame)
1357         return @"";
1358
1359     return coreFrame->layerTreeAsText();
1360 }
1361
1362 static Node* spellingNode(Frame* coreFrame)
1363 {
1364     Node* focusedNode = coreFrame->selection()->start().node();
1365     if (!focusedNode || !focusedNode->renderer())
1366         return 0;
1367
1368     for (const RenderObject* renderer = focusedNode->renderer(); renderer; renderer = renderer->childAt(0)) {
1369         if (renderer->isText())
1370             return renderer->node();
1371     }
1372     return 0;
1373 }
1374
1375 - (BOOL)hasSpellingMarker:(int)from length:(int)length
1376 {
1377     Frame* coreFrame = _private->coreFrame;
1378     if (!coreFrame)
1379         return NO;
1380
1381     Node* node = spellingNode(coreFrame);
1382     if (!node)
1383         return NO;
1384
1385     unsigned int startOffset = static_cast<unsigned int>(from);
1386     unsigned int endOffset = static_cast<unsigned int>(from + length);
1387     Vector<DocumentMarker> markers = coreFrame->document()->markers()->markersForNode(node);
1388     for (size_t i = 0; i < markers.size(); ++i) {
1389         DocumentMarker marker = markers[i];
1390         if (marker.startOffset <= startOffset && endOffset <= marker.endOffset && marker.type == DocumentMarker::Spelling)
1391             return YES;
1392     }
1393     return NO;
1394 }
1395
1396 @end
1397
1398 @implementation WebFrame
1399
1400 - (id)init
1401 {
1402     return nil;
1403 }
1404
1405 // Should be deprecated.
1406 - (id)initWithName:(NSString *)name webFrameView:(WebFrameView *)view webView:(WebView *)webView
1407 {
1408     return nil;
1409 }
1410
1411 - (void)dealloc
1412 {
1413     if (_private && _private->includedInWebKitStatistics)
1414         --WebFrameCount;
1415
1416     [_private release];
1417
1418     [super dealloc];
1419 }
1420
1421 - (void)finalize
1422 {
1423     if (_private && _private->includedInWebKitStatistics)
1424         --WebFrameCount;
1425
1426     [super finalize];
1427 }
1428
1429 - (NSString *)name
1430 {
1431     Frame* coreFrame = _private->coreFrame;
1432     if (!coreFrame)
1433         return nil;
1434     return coreFrame->tree()->name();
1435 }
1436
1437 - (WebFrameView *)frameView
1438 {
1439     ASSERT(!getWebView(self) || [getWebView(self) _usesDocumentViews]);
1440     return _private->webFrameView;
1441 }
1442
1443 - (WebView *)webView
1444 {
1445     return getWebView(self);
1446 }
1447
1448 static bool needsMicrosoftMessengerDOMDocumentWorkaround()
1449 {
1450     static bool needsWorkaround = applicationIsMicrosoftMessenger() && [[[NSBundle mainBundle] objectForInfoDictionaryKey:(NSString *)kCFBundleVersionKey] compare:@"7.1" options:NSNumericSearch] == NSOrderedAscending;
1451     return needsWorkaround;
1452 }
1453
1454 - (DOMDocument *)DOMDocument
1455 {
1456     if (needsMicrosoftMessengerDOMDocumentWorkaround() && !pthread_main_np())
1457         return nil;
1458
1459     Frame* coreFrame = _private->coreFrame;
1460     if (!coreFrame)
1461         return nil;
1462     
1463     // FIXME: <rdar://problem/5145841> When loading a custom view/representation 
1464     // into a web frame, the old document can still be around. This makes sure that
1465     // we'll return nil in those cases.
1466     if (![[self _dataSource] _isDocumentHTML]) 
1467         return nil; 
1468
1469     Document* document = coreFrame->document();
1470     
1471     // According to the documentation, we should return nil if the frame doesn't have a document.
1472     // While full-frame images and plugins do have an underlying HTML document, we return nil here to be
1473     // backwards compatible.
1474     if (document && (document->isPluginDocument() || document->isImageDocument()))
1475         return nil;
1476     
1477     return kit(coreFrame->document());
1478 }
1479
1480 - (DOMHTMLElement *)frameElement
1481 {
1482     Frame* coreFrame = _private->coreFrame;
1483     if (!coreFrame)
1484         return nil;
1485     return kit(coreFrame->ownerElement());
1486 }
1487
1488 - (WebDataSource *)provisionalDataSource
1489 {
1490     Frame* coreFrame = _private->coreFrame;
1491     return coreFrame ? dataSource(coreFrame->loader()->provisionalDocumentLoader()) : nil;
1492 }
1493
1494 - (WebDataSource *)dataSource
1495 {
1496     Frame* coreFrame = _private->coreFrame;
1497     return coreFrame && coreFrame->loader()->frameHasLoaded() ? [self _dataSource] : nil;
1498 }
1499
1500 - (void)loadRequest:(NSURLRequest *)request
1501 {
1502     Frame* coreFrame = _private->coreFrame;
1503     if (!coreFrame)
1504         return;
1505     coreFrame->loader()->load(request, false);
1506 }
1507
1508 static NSURL *createUniqueWebDataURL()
1509 {
1510     CFUUIDRef UUIDRef = CFUUIDCreate(kCFAllocatorDefault);
1511     NSString *UUIDString = (NSString *)CFUUIDCreateString(kCFAllocatorDefault, UUIDRef);
1512     CFRelease(UUIDRef);
1513     NSURL *URL = [NSURL URLWithString:[NSString stringWithFormat:@"applewebdata://%@", UUIDString]];
1514     CFRelease(UUIDString);
1515     return URL;
1516 }
1517
1518 - (void)_loadData:(NSData *)data MIMEType:(NSString *)MIMEType textEncodingName:(NSString *)encodingName baseURL:(NSURL *)baseURL unreachableURL:(NSURL *)unreachableURL
1519 {
1520     if (!pthread_main_np())
1521         return [[self _webkit_invokeOnMainThread] _loadData:data MIMEType:MIMEType textEncodingName:encodingName baseURL:baseURL unreachableURL:unreachableURL];
1522     
1523     KURL responseURL;
1524     if (!baseURL) {
1525         baseURL = blankURL();
1526         responseURL = createUniqueWebDataURL();
1527     }
1528     
1529     ResourceRequest request([baseURL absoluteURL]);
1530
1531     // hack because Mail checks for this property to detect data / archive loads
1532     [NSURLProtocol setProperty:@"" forKey:@"WebDataRequest" inRequest:(NSMutableURLRequest *)request.nsURLRequest()];
1533
1534     SubstituteData substituteData(WebCore::SharedBuffer::wrapNSData(data), MIMEType, encodingName, [unreachableURL absoluteURL], responseURL);
1535
1536     _private->coreFrame->loader()->load(request, substituteData, false);
1537 }
1538
1539
1540 - (void)loadData:(NSData *)data MIMEType:(NSString *)MIMEType textEncodingName:(NSString *)encodingName baseURL:(NSURL *)baseURL
1541 {
1542     WebCoreThreadViolationCheckRoundTwo();
1543     
1544     if (!MIMEType)
1545         MIMEType = @"text/html";
1546     [self _loadData:data MIMEType:MIMEType textEncodingName:encodingName baseURL:baseURL unreachableURL:nil];
1547 }
1548
1549 - (void)_loadHTMLString:(NSString *)string baseURL:(NSURL *)baseURL unreachableURL:(NSURL *)unreachableURL
1550 {
1551     NSData *data = [string dataUsingEncoding:NSUTF8StringEncoding];
1552     [self _loadData:data MIMEType:@"text/html" textEncodingName:@"UTF-8" baseURL:baseURL unreachableURL:unreachableURL];
1553 }
1554
1555 - (void)loadHTMLString:(NSString *)string baseURL:(NSURL *)baseURL
1556 {
1557     WebCoreThreadViolationCheckRoundTwo();
1558
1559     [self _loadHTMLString:string baseURL:baseURL unreachableURL:nil];
1560 }
1561
1562 - (void)loadAlternateHTMLString:(NSString *)string baseURL:(NSURL *)baseURL forUnreachableURL:(NSURL *)unreachableURL
1563 {
1564     WebCoreThreadViolationCheckRoundTwo();
1565
1566     [self _loadHTMLString:string baseURL:baseURL unreachableURL:unreachableURL];
1567 }
1568
1569 - (void)loadArchive:(WebArchive *)archive
1570 {
1571     if (LegacyWebArchive* coreArchive = [archive _coreLegacyWebArchive])
1572         _private->coreFrame->loader()->loadArchive(coreArchive);
1573 }
1574
1575 - (void)stopLoading
1576 {
1577     if (!_private->coreFrame)
1578         return;
1579     _private->coreFrame->loader()->stopForUserCancel();
1580 }
1581
1582 - (void)reload
1583 {
1584     if (!WebKitLinkedOnOrAfter(WEBKIT_FIRST_VERSION_WITH_RELOAD_FROM_ORIGIN) && applicationIsSafari())
1585         _private->coreFrame->loader()->reload(GetCurrentKeyModifiers() & shiftKey);
1586     else
1587         _private->coreFrame->loader()->reload(false);
1588 }
1589
1590 - (void)reloadFromOrigin
1591 {
1592     _private->coreFrame->loader()->reload(true);
1593 }
1594
1595 - (WebFrame *)findFrameNamed:(NSString *)name
1596 {
1597     Frame* coreFrame = _private->coreFrame;
1598     if (!coreFrame)
1599         return nil;
1600     return kit(coreFrame->tree()->find(name));
1601 }
1602
1603 - (WebFrame *)parentFrame
1604 {
1605     Frame* coreFrame = _private->coreFrame;
1606     if (!coreFrame)
1607         return nil;
1608     return [[kit(coreFrame->tree()->parent()) retain] autorelease];
1609 }
1610
1611 - (NSArray *)childFrames
1612 {
1613     Frame* coreFrame = _private->coreFrame;
1614     if (!coreFrame)
1615         return [NSArray array];
1616     NSMutableArray *children = [NSMutableArray arrayWithCapacity:coreFrame->tree()->childCount()];
1617     for (Frame* child = coreFrame->tree()->firstChild(); child; child = child->tree()->nextSibling())
1618         [children addObject:kit(child)];
1619     return children;
1620 }
1621
1622 - (WebScriptObject *)windowObject
1623 {
1624     Frame* coreFrame = _private->coreFrame;
1625     if (!coreFrame)
1626         return 0;
1627     return coreFrame->script()->windowScriptObject();
1628 }
1629
1630 - (JSGlobalContextRef)globalContext
1631 {
1632     Frame* coreFrame = _private->coreFrame;
1633     if (!coreFrame)
1634         return 0;
1635     return toGlobalRef(coreFrame->script()->globalObject(mainThreadNormalWorld())->globalExec());
1636 }
1637
1638 @end