WebKit:
[WebKit-https.git] / WebKitTools / DumpRenderTree / DumpRenderTree.m
1 /*
2  * Copyright (C) 2005, 2006 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/DOMExtensions.h>
30 #import <WebKit/DOMRange.h>
31 #import <WebKit/WebBackForwardList.h>
32 #import <WebKit/WebCoreStatistics.h>
33 #import <WebKit/WebDataSource.h>
34 #import <WebKit/WebEditingDelegate.h>
35 #import <WebKit/WebFrameView.h>
36 #import <WebKit/WebPreferences.h>
37 #import <WebKit/WebView.h>
38 #import <WebKit/WebHTMLViewPrivate.h>
39
40 #import <Carbon/Carbon.h>                           // for GetCurrentEventTime()
41 #import <ApplicationServices/ApplicationServices.h> // for CMSetDefaultProfileBySpace
42 #import <objc/objc-runtime.h>                       // for class_poseAs
43
44 #define COMMON_DIGEST_FOR_OPENSSL
45 #import <CommonCrypto/CommonDigest.h>               // for MD5 functions
46
47 #import <getopt.h>
48
49 #import "TextInputController.h"
50 #import "NavigationController.h"
51 #import "AppleScriptController.h"
52
53 @interface DumpRenderTreeWindow : NSWindow
54 @end
55
56 @interface DumpRenderTreeDraggingInfo : NSObject <NSDraggingInfo>
57 {
58 @private
59     NSSize offset;
60     NSImage *draggedImage;
61     NSPasteboard *draggingPasteboard;
62     id draggingSource;
63 }
64 - (id)initWithImage:(NSImage *)image offset:(NSSize)offset pasteboard:(NSPasteboard *)pasteboard source:(id)source; 
65 - (NSWindow *)draggingDestinationWindow;
66 - (NSDragOperation)draggingSourceOperationMask;
67 - (NSPoint)draggingLocation;
68 - (NSPoint)draggedImageLocation;
69 - (NSImage *)draggedImage;
70 - (NSPasteboard *)draggingPasteboard;
71 - (id)draggingSource;
72 - (int)draggingSequenceNumber;
73 - (void)slideDraggedImageTo:(NSPoint)screenPoint;
74 - (NSArray *)namesOfPromisedFilesDroppedAtDestination:(NSURL *)dropDestination;
75 @end
76
77 @interface DumpRenderTreePasteboard : NSPasteboard
78 @end
79
80 @interface WaitUntilDoneDelegate : NSObject
81 @end
82
83 @interface EditingDelegate : NSObject
84 @end
85
86 @interface LayoutTestController : NSObject
87 @end
88
89 @interface EventSendingController : NSObject
90 {
91     BOOL down;
92     int clickCount;
93     NSTimeInterval lastClick;
94 }
95 @end
96
97 static void dumpRenderTree(const char *filename);
98 static NSString *md5HashStringForBitmap(NSBitmapImageRep *bitmap);
99
100 static volatile BOOL done;
101 static WebFrame *frame;
102 static NavigationController *navigationController;
103 static BOOL readyToDump;
104 static BOOL waitToDump;
105 static BOOL dumpAsText;
106 static BOOL dumpTitleChanges;
107 static int dumpPixels = NO;
108 static int dumpTree = YES;
109 static BOOL printSeparators;
110 static NSString *currentTest = nil;
111 static NSPasteboard *localPasteboard;
112 static BOOL windowIsKey = YES;
113 static NSPoint lastMousePosition;
114 static DumpRenderTreeDraggingInfo *draggingInfo;
115
116 static CMProfileRef currentColorProfile = 0;
117 static void restoreColorSpace(int ignored)
118 {
119     if (currentColorProfile) {
120         int error = CMSetDefaultProfileByUse(cmDisplayUse, currentColorProfile);
121         if (error)
122             fprintf(stderr, "Failed to retore previous color profile!  You may need to open System Preferences : Displays : Color and manually restore your color settings.  (Error: %i)", error);
123         currentColorProfile = 0;
124     }
125 }
126
127 static void setDefaultColorProfileToRGB(void)
128 {
129     CMProfileRef genericProfile = [[NSColorSpace genericRGBColorSpace] colorSyncProfile];
130     CMProfileRef previousProfile;
131     int error = CMGetDefaultProfileByUse(cmDisplayUse, &previousProfile);
132     if (error) {
133         fprintf(stderr, "Failed to get current color profile.  I will not be able to restore your current profile, thus I'm not changing it.  Many pixel tests may fail as a result.  (Error: %i)\n", error);
134         return;
135     }
136     if (previousProfile == genericProfile)
137         return;
138     CFStringRef previousProfileName;
139     CFStringRef genericProfileName;
140     CMCopyProfileDescriptionString(previousProfile, &previousProfileName);
141     CMCopyProfileDescriptionString(genericProfile, &genericProfileName);
142     
143     fprintf(stderr, "\n\nWARNING: Temporarily changing your system color profile from \"%s\" to \"%s\".\n",
144         CFStringGetCStringPtr(previousProfileName, kCFStringEncodingMacRoman),
145         CFStringGetCStringPtr(genericProfileName, kCFStringEncodingMacRoman));
146     fprintf(stderr, "This allows the WebKit pixel-based regression tests to have consistent color values across all machines.\n");
147     fprintf(stderr, "The colors on your screen will change for the duration of the testing.\n\n");
148     
149     if ((error = CMSetDefaultProfileByUse(cmDisplayUse, genericProfile)))
150         fprintf(stderr, "Failed to set color profile to \"%s\"! Many pixel tests will fail as a result.  (Error: %i)",
151             CFStringGetCStringPtr(genericProfileName, kCFStringEncodingMacRoman), error);
152     else {
153         currentColorProfile = previousProfile;
154         signal(SIGINT, restoreColorSpace);
155         signal(SIGHUP, restoreColorSpace);
156         signal(SIGTERM, restoreColorSpace);
157     }
158     CFRelease(genericProfileName);
159     CFRelease(previousProfileName);
160 }
161
162 int main(int argc, const char *argv[])
163 {
164     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
165     
166     [NSApplication sharedApplication];
167
168     class_poseAs(objc_getClass("DumpRenderTreePasteboard"), objc_getClass("NSPasteboard"));
169     class_poseAs(objc_getClass("DumpRenderTreeWindow"), objc_getClass("NSWindow"));
170     
171     struct option options[] = {
172         {"pixel-tests", no_argument, &dumpPixels, YES},
173         {"tree", no_argument, &dumpTree, YES},
174         {"notree", no_argument, &dumpTree, NO},
175         {NULL, 0, NULL, 0}
176     };
177
178     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
179     [defaults setObject:@"DoubleMax" forKey:@"AppleScrollBarVariant"];
180     [defaults setInteger:4 forKey:@"AppleAntiAliasingThreshold"];
181     // 2 is the "Medium" font smoothing mode
182     [defaults setInteger:2 forKey:@"AppleFontSmoothing"];
183
184     [defaults setInteger:1 forKey:@"AppleAquaColorVariant"];
185     [defaults setObject:@"0.709800 0.835300 1.000000" forKey:@"AppleHighlightColor"];
186     [defaults setObject:@"0.500000 0.500000 0.500000" forKey:@"AppleOtherHighlightColor"];
187     
188     [defaults setObject:[NSArray arrayWithObject:@"en"] forKey:@"AppleLanguages"];
189     
190     WebPreferences *preferences = [WebPreferences standardPreferences];
191     
192     [preferences setStandardFontFamily:@"Times"];
193     [preferences setFixedFontFamily:@"Courier"];
194     [preferences setSerifFontFamily:@"Times"];
195     [preferences setSansSerifFontFamily:@"Helvetica"];
196     [preferences setCursiveFontFamily:@"Apple Chancery"];
197     [preferences setFantasyFontFamily:@"Papyrus"];
198     [preferences setDefaultFontSize:16];
199     [preferences setDefaultFixedFontSize:13];
200     [preferences setMinimumFontSize:9];
201     [preferences setJavaEnabled:NO];
202
203     int option;
204     while ((option = getopt_long(argc, (char * const *)argv, "", options, NULL)) != -1)
205         switch (option) {
206             case '?':   // unknown or ambiguous option
207             case ':':   // missing argument
208                 exit(1);
209                 break;
210         }
211     
212     if ([[[NSFontManager sharedFontManager] availableMembersOfFontFamily:@"Ahem"] count] == 0) {
213         fprintf(stderr, "\nAhem font is not available. This special simple font is used to construct certain types of predictable tests.\n\nTo run regression tests, please get it from <http://webkit.opendarwin.org/quality/Ahem.ttf>.\n");
214         exit(1);
215     }
216     
217     if (dumpPixels)
218         setDefaultColorProfileToRGB();
219     
220     localPasteboard = [NSPasteboard pasteboardWithUniqueName];
221     navigationController = [[NavigationController alloc] init];
222
223     NSRect rect = NSMakeRect(0, 0, 800, 600);
224     
225     WebView *webView = [[WebView alloc] initWithFrame:rect];
226     WaitUntilDoneDelegate *delegate = [[WaitUntilDoneDelegate alloc] init];
227     EditingDelegate *editingDelegate = [[EditingDelegate alloc] init];
228     [webView setFrameLoadDelegate:delegate];
229     [webView setEditingDelegate:editingDelegate];
230     [webView setUIDelegate:delegate];
231     frame = [webView mainFrame];
232
233     // The back/forward cache is causing problems due to layouts during transition from one page to another.
234     // So, turn it off for now, but we might want to turn it back on some day.
235     [[webView backForwardList] setPageCacheSize:0];
236
237     // To make things like certain NSViews, dragging, and plug-ins work, put the WebView a window, but put it off-screen so you don't see it.
238     NSRect windowRect = NSOffsetRect(rect, -10000, -10000);
239     NSWindow *window = [[NSWindow alloc] initWithContentRect:windowRect styleMask:NSBorderlessWindowMask backing:NSBackingStoreBuffered defer:YES];
240     [[window contentView] addSubview:webView];
241     [window orderBack:nil];
242
243     [webView setContinuousSpellCheckingEnabled:YES];
244     
245     // For reasons that are not entirely clear, the following pair of calls makes WebView handle its
246     // dynamic scrollbars properly. Without it, every frame will always have scrollbars.
247     NSBitmapImageRep *imageRep = [webView bitmapImageRepForCachingDisplayInRect:[webView bounds]];
248     [webView cacheDisplayInRect:[webView bounds] toBitmapImageRep:imageRep];
249     
250     if (argc == optind+1 && strcmp(argv[optind], "-") == 0) {
251         char filenameBuffer[2048];
252         printSeparators = YES;
253         while (fgets(filenameBuffer, sizeof(filenameBuffer), stdin)) {
254             char *newLineCharacter = strchr(filenameBuffer, '\n');
255             if (newLineCharacter)
256                 *newLineCharacter = '\0';
257             
258             if (strlen(filenameBuffer) == 0)
259                 continue;
260                 
261             dumpRenderTree(filenameBuffer);
262             fflush(stdout);
263         }
264     } else {
265         printSeparators = (optind < argc-1 || (dumpPixels && dumpTree));
266         for (int i = optind; i != argc; ++i) {
267             dumpRenderTree(argv[i]);
268         }
269     }
270     
271     [webView setFrameLoadDelegate:nil];
272     [webView setEditingDelegate:nil];
273     [webView setUIDelegate:nil];
274     frame = nil;
275
276     // Work around problem where registering drag types leaves an outstanding
277     // "perform selector" on the window, which retains the window. It's a bit
278     // inelegant and perhaps dangerous to just blow them all away, but in practice
279     // it probably won't cause any trouble (and this is just a test tool, after all).
280     [NSObject cancelPreviousPerformRequestsWithTarget:window];
281     
282     [window release];
283     [webView release];
284     [delegate release];
285     [editingDelegate release];
286
287     [localPasteboard releaseGlobally];
288     localPasteboard = nil;
289     
290     [navigationController release];
291     navigationController = nil;
292     
293     if (dumpPixels)
294         restoreColorSpace(0);
295     
296     [pool release];
297
298     return 0;
299 }
300
301 static void dump(void)
302 {
303     NSString *result = nil;
304     if (dumpTree) {
305         if (dumpAsText) {
306             DOMElement *documentElement = [[frame DOMDocument] documentElement];
307             if ([documentElement isKindOfClass:[DOMHTMLElement class]])
308                 result = [[(DOMHTMLElement *)documentElement innerText] stringByAppendingString:@"\n"];
309         } else {
310             bool isSVGW3CTest = ([currentTest rangeOfString:@"svg/W3C-SVG-1.1"].length);
311             if (isSVGW3CTest)
312                 [[frame webView] setFrameSize:NSMakeSize(480, 360)];
313             else 
314                 [[frame webView] setFrameSize:NSMakeSize(800, 600)];
315             result = [frame renderTreeAsExternalRepresentation];
316         }
317         
318         if (!result)
319             printf("ERROR: nil result from %s", dumpAsText ? "[documentElement innerText]" : "[frame renderTreeAsExternalRepresentation]");
320         else
321             fputs([result UTF8String], stdout);
322         
323         if (printSeparators)
324             puts("#EOF");
325     }
326     
327     if (dumpPixels) {
328         if (!dumpAsText) {
329             // FIXME: It's unfortunate that we hardcode the file naming scheme here.
330             // At one time, the perl script had all the knowledge about file layout.
331             // Some day we should restore that setup by passing in more parameters to this tool.
332
333             NSString *baseTestPath = [currentTest stringByDeletingPathExtension];
334             NSString *baselineHashPath = [baseTestPath stringByAppendingString:@"-expected.checksum"];
335             NSString *baselineHash = [NSString stringWithContentsOfFile:baselineHashPath encoding:NSUTF8StringEncoding error:nil];
336             NSString *baselineImagePath = [baseTestPath stringByAppendingString:@"-expected.png"];
337             
338             // grab a bitmap from the view
339             WebView *view = [frame webView];
340             NSBitmapImageRep *imageRep = [view bitmapImageRepForCachingDisplayInRect:[view frame]];
341             [view cacheDisplayInRect:[view frame] toBitmapImageRep:imageRep];
342             
343             // has the actual hash to compare to the expected image's hash
344             NSString *actualHash = md5HashStringForBitmap(imageRep);
345             printf("\nActualHash: %s\n", [actualHash UTF8String]);
346             printf("BaselineHash: %s\n", [baselineHash UTF8String]);
347             
348             // if the hashes don't match, send image back to stdout for diff comparision
349             if (![baselineHash isEqualToString:actualHash] || access([baselineImagePath fileSystemRepresentation], F_OK) != 0) {            
350                 NSData *imageData = [imageRep representationUsingType:NSPNGFileType properties:nil];
351                 printf("Content-length: %d\n", [imageData length]);
352                 fwrite([imageData bytes], 1, [imageData length], stdout);
353             }
354         }
355
356         printf("#EOF\n");
357     }
358
359     done = YES;
360 }
361
362 @implementation WaitUntilDoneDelegate
363
364 - (void)webView:(WebView *)c locationChangeDone:(NSError *)error forDataSource:(WebDataSource *)dataSource
365 {
366     if ([dataSource webFrame] == frame) {
367         if (waitToDump)
368             readyToDump = YES;
369         else
370             dump();
371     }
372 }
373
374 - (void)webView:(WebView *)sender didCommitLoadForFrame:(WebFrame *)f
375 {
376     if (frame == f)
377         readyToDump = NO;
378
379     windowIsKey = YES;
380     NSView *documentView = [[frame frameView] documentView];
381     [[[frame webView] window] makeFirstResponder:documentView];
382     if ([documentView isKindOfClass:[WebHTMLView class]])
383         [(WebHTMLView *)documentView _updateFocusState];
384 }
385
386 - (void)webView:(WebView *)sender didFailProvisionalLoadWithError:(NSError *)error forFrame:(WebFrame *)frame
387 {
388     [self webView:sender locationChangeDone:error forDataSource:[frame provisionalDataSource]];
389 }
390
391 - (void)webView:(WebView *)sender didFinishLoadForFrame:(WebFrame *)frame
392 {
393     [self webView:sender locationChangeDone:nil forDataSource:[frame dataSource]];
394     [navigationController webView:sender didFinishLoadForFrame:frame];
395 }
396
397 - (void)webView:(WebView *)sender didFailLoadWithError:(NSError *)error forFrame:(WebFrame *)frame
398 {
399     [self webView:sender locationChangeDone:error forDataSource:[frame dataSource]];
400 }
401
402 - (void)webView:(WebView *)sender windowScriptObjectAvailable:(WebScriptObject *)obj 
403
404     LayoutTestController *ltc = [[LayoutTestController alloc] init];
405     [obj setValue:ltc forKey:@"layoutTestController"];
406     [ltc release];
407     EventSendingController *esc = [[EventSendingController alloc] init];
408     [obj setValue:esc forKey:@"eventSender"];
409     [esc release];
410     TextInputController *tic = [[TextInputController alloc] initWithWebView:sender];
411     [obj setValue:tic forKey:@"textInputController"];
412     [tic release];
413     AppleScriptController *asc = [[AppleScriptController alloc] initWithWebView:sender];
414     [obj setValue:asc forKey:@"appleScriptController"];
415     [asc release];
416     [obj setValue:navigationController forKey:@"navigationController"];
417 }
418
419 - (void)webView:(WebView *)sender runJavaScriptAlertPanelWithMessage:(NSString *)message
420 {
421     printf("ALERT: %s\n", [message UTF8String]);
422 }
423
424 - (void)webView:(WebView *)sender didReceiveTitle:(NSString *)title forFrame:(WebFrame *)frame
425 {
426     if (dumpTitleChanges)
427         printf("TITLE CHANGED: %s\n", [title UTF8String]);
428 }
429
430 - (void)webView:(WebView *)sender dragImage:(NSImage *)anImage at:(NSPoint)viewLocation offset:(NSSize)initialOffset event:(NSEvent *)event pasteboard:(NSPasteboard *)pboard source:(id)sourceObj slideBack:(BOOL)slideFlag forView:(NSView *)view
431 {
432     // A new drag was started before the old one ended.  Probably shouldn't happen.
433     if (draggingInfo) {
434         [[draggingInfo draggingSource] draggedImage:[draggingInfo draggedImage] endedAt:lastMousePosition operation:NSDragOperationNone];
435         [draggingInfo release];
436     }
437     draggingInfo = [[DumpRenderTreeDraggingInfo alloc] initWithImage:anImage offset:initialOffset pasteboard:pboard source:sourceObj];
438 }
439
440 - (void)webViewFocus:(WebView *)webView
441 {
442     windowIsKey = YES;
443     NSView *documentView = [[frame frameView] documentView];
444     if ([documentView isKindOfClass:[WebHTMLView class]])
445         [(WebHTMLView *)documentView _updateFocusState];
446 }
447
448 @end
449
450 @interface DOMNode (dumpPath)
451 - (NSString *)dumpPath;
452 @end
453
454 @implementation DOMNode (dumpPath)
455 - (NSString *)dumpPath
456 {
457     DOMNode *parent = [self parentNode];
458     NSString *str = [NSString stringWithFormat:@"%@", [self nodeName]];
459     if (parent != nil) {
460         str = [str stringByAppendingString:@" > "];
461         str = [str stringByAppendingString:[parent dumpPath]];
462     }
463     return str;
464 }
465 @end
466
467 @interface DOMRange (dump)
468 - (NSString *)dump;
469 @end
470
471 @implementation DOMRange (dump)
472 - (NSString *)dump
473 {
474     return [NSString stringWithFormat:@"range from %ld of %@ to %ld of %@", [self startOffset], [[self startContainer] dumpPath], [self endOffset], [[self endContainer] dumpPath]];
475 }
476 @end
477
478
479 @implementation EditingDelegate
480
481 - (BOOL)webView:(WebView *)webView shouldBeginEditingInDOMRange:(DOMRange *)range
482 {
483     printf("EDITING DELEGATE: shouldBeginEditingInDOMRange:%s\n", [[range dump] UTF8String]);
484     return YES;
485 }
486
487 - (BOOL)webView:(WebView *)webView shouldEndEditingInDOMRange:(DOMRange *)range
488 {
489     printf("EDITING DELEGATE: shouldEndEditingInDOMRange:%s\n", [[range dump] UTF8String]);
490     return YES;
491 }
492
493 - (BOOL)webView:(WebView *)webView shouldInsertNode:(DOMNode *)node replacingDOMRange:(DOMRange *)range givenAction:(WebViewInsertAction)action
494 {
495     static const char *insertactionstring[] = {
496         "WebViewInsertActionTyped",
497         "WebViewInsertActionPasted",
498         "WebViewInsertActionDropped",
499     };
500
501     printf("EDITING DELEGATE: shouldInsertNode:%s replacingDOMRange:%s givenAction:%s\n", [[node dumpPath] UTF8String], [[range dump] UTF8String], insertactionstring[action]);
502     return YES;
503 }
504
505 - (BOOL)webView:(WebView *)webView shouldInsertText:(NSString *)text replacingDOMRange:(DOMRange *)range givenAction:(WebViewInsertAction)action
506 {
507     static const char *insertactionstring[] = {
508         "WebViewInsertActionTyped",
509         "WebViewInsertActionPasted",
510         "WebViewInsertActionDropped",
511     };
512
513     printf("EDITING DELEGATE: shouldInsertText:%s replacingDOMRange:%s givenAction:%s\n", [[text description] UTF8String], [[range dump] UTF8String], insertactionstring[action]);
514     return YES;
515 }
516
517 - (BOOL)webView:(WebView *)webView shouldDeleteDOMRange:(DOMRange *)range
518 {
519     printf("EDITING DELEGATE: shouldDeleteDOMRange:%s\n", [[range dump] UTF8String]);
520     return YES;
521 }
522
523 - (BOOL)webView:(WebView *)webView shouldChangeSelectedDOMRange:(DOMRange *)currentRange toDOMRange:(DOMRange *)proposedRange affinity:(NSSelectionAffinity)selectionAffinity stillSelecting:(BOOL)flag
524 {
525     static const char *affinitystring[] = {
526         "NSSelectionAffinityUpstream",
527         "NSSelectionAffinityDownstream"
528     };
529     static const char *boolstring[] = {
530         "FALSE",
531         "TRUE"
532     };
533
534     printf("EDITING DELEGATE: shouldChangeSelectedDOMRange:%s toDOMRange:%s affinity:%s stillSelecting:%s\n", [[currentRange dump] UTF8String], [[proposedRange dump] UTF8String], affinitystring[selectionAffinity], boolstring[flag]);
535     return YES;
536 }
537
538 - (BOOL)webView:(WebView *)webView shouldApplyStyle:(DOMCSSStyleDeclaration *)style toElementsInDOMRange:(DOMRange *)range
539 {
540     printf("EDITING DELEGATE: shouldApplyStyle:%s toElementsInDOMRange:%s\n", [[style description] UTF8String], [[range dump] UTF8String]);
541     return YES;
542 }
543
544 - (BOOL)webView:(WebView *)webView shouldChangeTypingStyle:(DOMCSSStyleDeclaration *)currentStyle toStyle:(DOMCSSStyleDeclaration *)proposedStyle
545 {
546     printf("EDITING DELEGATE: shouldChangeTypingStyle:%s toStyle:%s\n", [[currentStyle description] UTF8String], [[proposedStyle description] UTF8String]);
547     return YES;
548 }
549
550 - (void)webViewDidBeginEditing:(NSNotification *)notification
551 {
552     printf("EDITING DELEGATE: webViewDidBeginEditing:%s\n", [[notification name] UTF8String]);
553 }
554
555 - (void)webViewDidChange:(NSNotification *)notification
556 {
557     printf("EDITING DELEGATE: webViewDidChange:%s\n", [[notification name] UTF8String]);
558 }
559
560 - (void)webViewDidEndEditing:(NSNotification *)notification
561 {
562     printf("EDITING DELEGATE: webViewDidEndEditing:%s\n", [[notification name] UTF8String]);
563 }
564
565 - (void)webViewDidChangeTypingStyle:(NSNotification *)notification
566 {
567     printf("EDITING DELEGATE: webViewDidChangeTypingStyle:%s\n", [[notification name] UTF8String]);
568 }
569
570 - (void)webViewDidChangeSelection:(NSNotification *)notification
571 {
572     if (!done)
573         printf("EDITING DELEGATE: webViewDidChangeSelection:%s\n", [[notification name] UTF8String]);
574 }
575
576 @end
577
578 @implementation LayoutTestController
579
580 + (BOOL)isSelectorExcludedFromWebScript:(SEL)aSelector
581 {
582     if (aSelector == @selector(waitUntilDone)
583             || aSelector == @selector(notifyDone)
584             || aSelector == @selector(dumpAsText)
585             || aSelector == @selector(dumpTitleChanges)
586             || aSelector == @selector(setWindowIsKey:)
587             || aSelector == @selector(setMainFrameIsFirstResponder:))
588         return NO;
589     return YES;
590 }
591
592 + (NSString *)webScriptNameForSelector:(SEL)aSelector
593 {
594     if (aSelector == @selector(setWindowIsKey:))
595         return @"setWindowIsKey";
596     if (aSelector == @selector(setMainFrameIsFirstResponder:))
597         return @"setMainFrameIsFirstResponder";
598     return nil;
599 }
600
601 - (void)waitUntilDone 
602 {
603     waitToDump = YES;
604 }
605
606 - (void)notifyDone
607 {
608     if (waitToDump && readyToDump)
609         dump();
610     waitToDump = NO;
611 }
612
613 - (void)dumpAsText
614 {
615     dumpAsText = YES;
616 }
617
618 - (void)dumpTitleChanges
619 {
620     dumpTitleChanges = YES;
621 }
622
623 - (void)setWindowIsKey:(BOOL)flag
624 {
625     windowIsKey = flag;
626     NSView *documentView = [[frame frameView] documentView];
627     if ([documentView isKindOfClass:[WebHTMLView class]])
628         [(WebHTMLView *)documentView _updateFocusState];
629 }
630
631 - (void)setMainFrameIsFirstResponder:(BOOL)flag
632 {
633     NSView *documentView = [[frame frameView] documentView];
634     
635     NSResponder *firstResponder = flag ? documentView : nil;
636     [[[frame webView] window] makeFirstResponder:firstResponder];
637         
638     if ([documentView isKindOfClass:[WebHTMLView class]])
639         [(WebHTMLView *)documentView _updateFocusState];
640 }
641
642 - (id)invokeUndefinedMethodFromWebScript:(NSString *)name withArguments:(NSArray *)args
643 {
644     return nil;
645 }
646
647 @end
648
649 @implementation EventSendingController
650
651 + (BOOL)isSelectorExcludedFromWebScript:(SEL)aSelector
652 {
653     if (aSelector == @selector(mouseDown)
654             || aSelector == @selector(mouseUp)
655             || aSelector == @selector(mouseMoveToX:Y:))
656         return NO;
657     return YES;
658 }
659
660 + (NSString *)webScriptNameForSelector:(SEL)aSelector
661 {
662     if(aSelector == @selector(mouseMoveToX:Y:))
663         return @"mouseMoveTo";
664     return nil;
665 }
666
667 - (id)init
668 {
669     lastMousePosition = NSMakePoint(0, 0);
670     down = NO;
671     clickCount = 0;
672     lastClick = 0;
673     return self;
674 }
675
676 - (void)mouseDown
677 {
678     [[[frame frameView] documentView] layout];
679     if (GetCurrentEventTime() - lastClick >= 1)
680         clickCount = 1;
681     else
682         clickCount++;
683     NSEvent *event = [NSEvent mouseEventWithType:NSLeftMouseDown location:lastMousePosition modifierFlags:nil timestamp:GetCurrentEventTime() windowNumber:0 context:[NSGraphicsContext currentContext] eventNumber:nil clickCount:clickCount pressure:nil];
684
685     NSView *subView = [[frame webView] hitTest:[event locationInWindow]];
686     if (subView) {
687         [subView mouseDown:event];
688         down = YES;
689     }
690 }
691
692 - (void)mouseUp
693 {
694     [[[frame frameView] documentView] layout];
695     NSEvent *event = [NSEvent mouseEventWithType:NSLeftMouseUp location:lastMousePosition modifierFlags:nil timestamp:GetCurrentEventTime() windowNumber:0 context:[NSGraphicsContext currentContext] eventNumber:nil clickCount:clickCount pressure:nil];
696
697     NSView *subView = [[frame webView] hitTest:[event locationInWindow]];
698     if (subView) {
699         [subView mouseUp:event];
700         down = NO;
701         lastClick = [event timestamp];
702         if (draggingInfo) {
703             WebView *webView = [frame webView];
704             
705             NSDragOperation dragOperation = [webView draggingUpdated:draggingInfo];
706             
707             [[draggingInfo draggingSource] draggedImage:[draggingInfo draggedImage] endedAt:lastMousePosition operation:dragOperation];
708             if (dragOperation != NSDragOperationNone)
709                 [webView performDragOperation:draggingInfo];
710             [draggingInfo release];
711             draggingInfo = nil;
712         }
713     }
714 }
715
716 - (void)mouseMoveToX:(int)x Y:(int)y
717 {
718     lastMousePosition = NSMakePoint(x, [[frame webView] frame].size.height - y);
719     NSEvent *event = [NSEvent mouseEventWithType:(down ? NSLeftMouseDragged : NSMouseMoved) location:lastMousePosition modifierFlags:nil timestamp:GetCurrentEventTime() windowNumber:0 context:[NSGraphicsContext currentContext] eventNumber:nil clickCount:(down ? clickCount : 0) pressure:nil];
720
721     NSView *subView = [[frame webView] hitTest:[event locationInWindow]];
722     if (subView) {
723         if (down) {
724             [subView mouseDragged:event];
725             [[draggingInfo draggingSource] draggedImage:[draggingInfo draggedImage] movedTo:lastMousePosition];
726             [[frame webView] draggingUpdated:draggingInfo];
727         }
728         else
729             [subView mouseMoved:event];
730     }
731 }
732
733 @end
734
735 static void dumpRenderTree(const char *filename)
736 {
737     CFStringRef filenameString = CFStringCreateWithCString(NULL, filename, kCFStringEncodingUTF8);
738     if (!filenameString) {
739         fprintf(stderr, "can't parse filename as UTF-8\n");
740         return;
741     }
742
743     CFURLRef URL = CFURLCreateWithFileSystemPath(NULL, filenameString, kCFURLPOSIXPathStyle, FALSE);
744     if (!URL) {
745         CFRelease(filenameString);
746         fprintf(stderr, "can't turn %s into a CFURL\n", filename);
747         return;
748     }
749
750     done = NO;
751     readyToDump = NO;
752     waitToDump = NO;
753     dumpAsText = NO;
754     dumpTitleChanges = NO;
755     if (currentTest != nil)
756         CFRelease(currentTest);
757     currentTest = (NSString *)filenameString;
758
759     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
760     [frame loadRequest:[NSURLRequest requestWithURL:(NSURL *)URL]];
761     CFRelease(URL);
762     [pool release];
763     while (!done) {
764         pool = [[NSAutoreleasePool alloc] init];
765         [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode beforeDate:[NSDate distantPast]];
766         [pool release];
767     }
768     pool = [[NSAutoreleasePool alloc] init];
769     [[frame webView] setSelectedDOMRange:nil affinity:NSSelectionAffinityDownstream];
770     if (draggingInfo)
771         [draggingInfo release];
772     draggingInfo = nil;
773     [pool release];
774 }
775
776 /* Hashes a bitmap and returns a text string for comparison and saving to a file */
777 static NSString *md5HashStringForBitmap(NSBitmapImageRep *bitmap)
778 {
779     MD5_CTX md5Context;
780     unsigned char hash[16];
781
782     MD5_Init(&md5Context);
783     MD5_Update(&md5Context, [bitmap bitmapData], [bitmap bytesPerPlane]);
784     MD5_Final(hash, &md5Context);
785     
786     char hex[33] = "";
787     for (int i = 0; i < 16; i++) {
788        snprintf(hex, 33, "%s%02x", hex, hash[i]);
789     }
790
791     return [NSString stringWithUTF8String:hex];
792 }
793
794 @implementation DumpRenderTreePasteboard
795
796 // Return a local pasteboard so we don't disturb the real pasteboard when running tests.
797 + (NSPasteboard *)generalPasteboard
798 {
799     return localPasteboard;
800 }
801
802 @end
803
804 @implementation DumpRenderTreeWindow
805
806 - (BOOL)isKeyWindow
807 {
808     return windowIsKey;
809 }
810
811 @end
812
813 @implementation DumpRenderTreeDraggingInfo
814
815 - (id)initWithImage:(NSImage *)anImage offset:(NSSize)o pasteboard:(NSPasteboard *)pboard source:(id)source
816 {
817     draggedImage = [anImage retain];
818     draggingPasteboard = [pboard retain];
819     draggingSource = [source retain];
820     offset = o;
821     
822     return [super init];
823 }
824
825 - (void)dealloc
826 {
827     [draggedImage release];
828     [draggingPasteboard release];
829     [draggingSource release];
830     [super dealloc];
831 }
832
833 - (NSWindow *)draggingDestinationWindow 
834 {
835     return [[frame webView] window];
836 }
837
838 - (NSDragOperation)draggingSourceOperationMask 
839 {
840     return [draggingSource draggingSourceOperationMaskForLocal:YES];
841 }
842
843 - (NSPoint)draggingLocation
844
845     return lastMousePosition; 
846 }
847
848 - (NSPoint)draggedImageLocation 
849 {
850     return NSMakePoint(lastMousePosition.x + offset.width, lastMousePosition.y + offset.height);
851 }
852
853 - (NSImage *)draggedImage
854 {
855     return draggedImage;
856 }
857
858 - (NSPasteboard *)draggingPasteboard
859 {
860     return draggingPasteboard;
861 }
862
863 - (id)draggingSource
864 {
865     return draggingSource;
866 }
867
868 - (int)draggingSequenceNumber
869 {
870     NSLog(@"DumpRenderTree doesn't support draggingSequenceNumber");
871     return 0;
872 }
873
874 - (void)slideDraggedImageTo:(NSPoint)screenPoint
875 {
876     NSLog(@"DumpRenderTree doesn't support slideDraggedImageTo:");
877 }
878
879 - (NSArray *)namesOfPromisedFilesDroppedAtDestination:(NSURL *)dropDestination
880 {
881     NSLog(@"DumpRenderTree doesn't support namesOfPromisedFilesDroppedAtDestination:");
882     return nil;
883 }
884
885 @end
886