1 /*

     2  * Copyright (C) 2005, 2006, 2007 Apple, Inc.  All rights reserved.

     3  *           (C) 2007 Graham Dennis (graham.dennis@gmail.com)

     4  *           (C) 2007 Eric Seidel <eric@webkit.org>

     5  *

     6  * Redistribution and use in source and binary forms, with or without

     7  * modification, are permitted provided that the following conditions

     8  * are met:

     9  *

    10  * 1.  Redistributions of source code must retain the above copyright

    11  *     notice, this list of conditions and the following disclaimer. 

    12  * 2.  Redistributions in binary form must reproduce the above copyright

    13  *     notice, this list of conditions and the following disclaimer in the

    14  *     documentation and/or other materials provided with the distribution. 

    15  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of

    16  *     its contributors may be used to endorse or promote products derived

    17  *     from this software without specific prior written permission. 

    18  *

    19  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY

    20  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

    21  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE

    22  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY

    23  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

    24  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

    25  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

    26  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

    28  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

    29  */

    30 

    31 #import "config.h"

    32 #import "DumpRenderTreeMac.h"

    33 #import "DumpRenderTreePasteboard.h"

    34 

    35 #import <WebKit/WebTypesInternal.h>

    36 

    37 @interface LocalPasteboard : NSPasteboard

    38 {

    39     NSMutableArray *typesArray;

    40     NSMutableSet *typesSet;

    41     NSMutableDictionary *dataByType;

    42     NSInteger changeCount;

    43 }

    44 @end

    45 

    46 static NSMutableDictionary *localPasteboards;

    47 

    48 @implementation DumpRenderTreePasteboard

    49 

    50 // Return a local pasteboard so we don't disturb the real pasteboards when running tests.

    51 + (NSPasteboard *)_pasteboardWithName:(NSString *)name

    52 {

    53     static int number = 0;

    54     if (!name)

    55         name = [NSString stringWithFormat:@"LocalPasteboard%d", ++number];

    56     if (!localPasteboards)

    57         localPasteboards = [[NSMutableDictionary alloc] init];

    58     LocalPasteboard *pasteboard = [localPasteboards objectForKey:name];

    59     if (pasteboard)

    60         return pasteboard;

    61     pasteboard = [[LocalPasteboard alloc] init];

    62     [localPasteboards setObject:pasteboard forKey:name];

    63     [pasteboard release];

    64     return pasteboard;

    65 }

    66 

    67 + (void)releaseLocalPasteboards

    68 {

    69     [localPasteboards release];

    70     localPasteboards = nil;

    71 }

    72 

    73 // Convenience method for JS so that it doesn't have to try and create a NSArray on the objc side instead

    74 // of the usual WebScriptObject that is passed around

    75 - (NSInteger)declareType:(NSString *)type owner:(id)newOwner

    76 {

    77     return [self declareTypes:[NSArray arrayWithObject:type] owner:newOwner];

    78 }

    79 

    80 @end

    81 

    82 @implementation LocalPasteboard

    83 

    84 + (id)alloc

    85 {

    86     return NSAllocateObject(self, 0, 0);

    87 }

    88 

    89 - (id)init

    90 {

    91     typesArray = [[NSMutableArray alloc] init];

    92     typesSet = [[NSMutableSet alloc] init];

    93     dataByType = [[NSMutableDictionary alloc] init];

    94     return self;

    95 }

    96 

    97 - (void)dealloc

    98 {

    99     [typesArray release];

   100     [typesSet release];

   101     [dataByType release];

   102     [super dealloc];

   103 }

   104 

   105 - (NSString *)name

   106 {

   107     return nil;

   108 }

   109 

   110 - (void)releaseGlobally

   111 {

   112 }

   113 

   114 - (NSInteger)declareTypes:(NSArray *)newTypes owner:(id)newOwner

   115 {

   116     [typesArray removeAllObjects];

   117     [typesSet removeAllObjects];

   118     [dataByType removeAllObjects];

   119     return [self addTypes:newTypes owner:newOwner];

   120 }

   121 

   122 - (NSInteger)addTypes:(NSArray *)newTypes owner:(id)newOwner

   123 {

   124     unsigned count = [newTypes count];

   125     unsigned i;

   126     for (i = 0; i < count; ++i) {

   127         NSString *type = [newTypes objectAtIndex:i];

   128         NSString *setType = [typesSet member:type];

   129         if (!setType) {

   130             setType = [type copy];

   131             [typesArray addObject:setType];

   132             [typesSet addObject:setType];

   133             [setType release];

   134         }

   135         if (newOwner && [newOwner respondsToSelector:@selector(pasteboard:provideDataForType:)])

   136             [newOwner pasteboard:self provideDataForType:setType];

   137     }

   138     return ++changeCount;

   139 }

   140 

   141 - (NSInteger)changeCount

   142 {

   143     return changeCount;

   144 }

   145 

   146 - (NSArray *)types

   147 {

   148     return typesArray;

   149 }

   150 

   151 - (NSString *)availableTypeFromArray:(NSArray *)types

   152 {

   153     unsigned count = [types count];

   154     unsigned i;

   155     for (i = 0; i < count; ++i) {

   156         NSString *type = [types objectAtIndex:i];

   157         NSString *setType = [typesSet member:type];

   158         if (setType)

   159             return setType;

   160     }

   161     return nil;

   162 }

   163 

   164 - (BOOL)setData:(NSData *)data forType:(NSString *)dataType

   165 {

   166     if (data == nil)

   167         data = [NSData data];

   168     if (![typesSet containsObject:dataType])

   169         return NO;

   170     [dataByType setObject:data forKey:dataType];

   171     ++changeCount;

   172     return YES;

   173 }

   174 

   175 - (NSData *)dataForType:(NSString *)dataType

   176 {

   177     return [dataByType objectForKey:dataType];

   178 }

   179 

   180 - (BOOL)setPropertyList:(id)propertyList forType:(NSString *)dataType

   181 {

   182     CFDataRef data = NULL;

   183     if (propertyList)

   184         data = CFPropertyListCreateXMLData(NULL, propertyList);

   185     BOOL result = [self setData:(NSData *)data forType:dataType];

   186     if (data)

   187         CFRelease(data);

   188     return result;

   189 }

   190 

   191 - (BOOL)setString:(NSString *)string forType:(NSString *)dataType

   192 {

   193     CFDataRef data = NULL;

   194     if (string) {

   195         if ([string length] == 0)

   196             data = CFDataCreate(NULL, NULL, 0);

   197         else

   198             data = CFStringCreateExternalRepresentation(NULL, (CFStringRef)string, kCFStringEncodingUTF8, 0);

   199     }

   200     BOOL result = [self setData:(NSData *)data forType:dataType];

   201     if (data)

   202         CFRelease(data);

   203     return result;

   204 }

   205 

   206 @end