/* | |
* Copyright (C) 2006 Apple Computer, Inc. All rights reserved. | |
* | |
* Redistribution and use in source and binary forms, with or without | |
* modification, are permitted provided that the following conditions | |
* are met: | |
* 1. Redistributions of source code must retain the above copyright | |
* notice, this list of conditions and the following disclaimer. | |
* 2. Redistributions in binary form must reproduce the above copyright | |
* notice, this list of conditions and the following disclaimer in the | |
* documentation and/or other materials provided with the distribution. | |
* | |
* THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY | |
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR | |
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR | |
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | |
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR | |
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY | |
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
*/ | |
#import "config.h" | |
#import "Pasteboard.h" | |
#import "CachedResource.h" | |
#import "CharacterNames.h" | |
#import "DOMRangeInternal.h" | |
#import "Document.h" | |
#import "DocumentFragment.h" | |
#import "Editor.h" | |
#import "EditorClient.h" | |
#import "Frame.h" | |
#import "HitTestResult.h" | |
#import "Image.h" | |
#import "KURL.h" | |
#import "LegacyWebArchive.h" | |
#import "LoaderNSURLExtras.h" | |
#import "MIMETypeRegistry.h" | |
#import "RenderImage.h" | |
#import "WebCoreNSStringExtras.h" | |
#import "markup.h" | |
#import <wtf/StdLibExtras.h> | |
#import <wtf/RetainPtr.h> | |
#import <wtf/UnusedParam.h> | |
@interface NSAttributedString (AppKitSecretsIKnowAbout) | |
- (id)_initWithDOMRange:(DOMRange *)domRange; | |
@end | |
namespace WebCore { | |
// FIXME: It's not great to have these both here and in WebKit. | |
NSString *WebArchivePboardType = @"Apple Web Archive pasteboard type"; | |
NSString *WebSmartPastePboardType = @"NeXT smart paste pasteboard type"; | |
NSString *WebURLNamePboardType = @"public.url-name"; | |
NSString *WebURLPboardType = @"public.url"; | |
NSString *WebURLsWithTitlesPboardType = @"WebURLsWithTitlesPboardType"; | |
#ifndef BUILDING_ON_TIGER | |
static NSArray* selectionPasteboardTypes(bool canSmartCopyOrDelete, bool selectionContainsAttachments) | |
{ | |
if (selectionContainsAttachments) { | |
if (canSmartCopyOrDelete) | |
return [NSArray arrayWithObjects:WebSmartPastePboardType, WebArchivePboardType, NSRTFDPboardType, NSRTFPboardType, NSStringPboardType, nil]; | |
else | |
return [NSArray arrayWithObjects:WebArchivePboardType, NSRTFDPboardType, NSRTFPboardType, NSStringPboardType, nil]; | |
} else { // Don't write RTFD to the pasteboard when the copied attributed string has no attachments. | |
if (canSmartCopyOrDelete) | |
return [NSArray arrayWithObjects:WebSmartPastePboardType, WebArchivePboardType, NSRTFPboardType, NSStringPboardType, nil]; | |
else | |
return [NSArray arrayWithObjects:WebArchivePboardType, NSRTFPboardType, NSStringPboardType, nil]; | |
} | |
} | |
#endif | |
static NSArray* writableTypesForURL() | |
{ | |
DEFINE_STATIC_LOCAL(RetainPtr<NSArray>, types, ([[NSArray alloc] initWithObjects: | |
WebURLsWithTitlesPboardType, | |
NSURLPboardType, | |
WebURLPboardType, | |
WebURLNamePboardType, | |
NSStringPboardType, | |
nil])); | |
return types.get(); | |
} | |
static inline NSArray* createWritableTypesForImage() | |
{ | |
NSMutableArray *types = [[NSMutableArray alloc] initWithObjects:NSTIFFPboardType, nil]; | |
[types addObjectsFromArray:writableTypesForURL()]; | |
[types addObject:NSRTFDPboardType]; | |
return types; | |
} | |
static NSArray* writableTypesForImage() | |
{ | |
DEFINE_STATIC_LOCAL(RetainPtr<NSArray>, types, (createWritableTypesForImage())); | |
return types.get(); | |
} | |
Pasteboard* Pasteboard::generalPasteboard() | |
{ | |
static Pasteboard* pasteboard = new Pasteboard([NSPasteboard generalPasteboard]); | |
return pasteboard; | |
} | |
Pasteboard::Pasteboard(NSPasteboard* pboard) | |
: m_pasteboard(pboard) | |
{ | |
} | |
void Pasteboard::clear() | |
{ | |
[m_pasteboard.get() declareTypes:[NSArray array] owner:nil]; | |
} | |
static NSAttributedString *stripAttachmentCharacters(NSAttributedString *string) | |
{ | |
const unichar attachmentCharacter = NSAttachmentCharacter; | |
DEFINE_STATIC_LOCAL(RetainPtr<NSString>, attachmentCharacterString, ([NSString stringWithCharacters:&attachmentCharacter length:1])); | |
NSMutableAttributedString *result = [[string mutableCopy] autorelease]; | |
NSRange attachmentRange = [[result string] rangeOfString:attachmentCharacterString.get()]; | |
while (attachmentRange.location != NSNotFound) { | |
[result replaceCharactersInRange:attachmentRange withString:@""]; | |
attachmentRange = [[result string] rangeOfString:attachmentCharacterString.get()]; | |
} | |
return result; | |
} | |
void Pasteboard::writeSelection(NSPasteboard* pasteboard, Range* selectedRange, bool canSmartCopyOrDelete, Frame* frame) | |
{ | |
if (!WebArchivePboardType) | |
Pasteboard::generalPasteboard(); // Initializes pasteboard types. | |
ASSERT(selectedRange); | |
NSAttributedString *attributedString = [[[NSAttributedString alloc] _initWithDOMRange:kit(selectedRange)] autorelease]; | |
#ifdef BUILDING_ON_TIGER | |
// 4930197: Mail overrides [WebHTMLView pasteboardTypesForSelection] in order to add another type to the pasteboard | |
// after WebKit does. On Tiger we must call this function so that Mail code will be executed, meaning that | |
// we can't call WebCore::Pasteboard's method for setting types. | |
UNUSED_PARAM(canSmartCopyOrDelete); | |
NSArray *types = frame->editor()->client()->pasteboardTypesForSelection(frame); | |
// Don't write RTFD to the pasteboard when the copied attributed string has no attachments. | |
NSMutableArray *mutableTypes = nil; | |
if (![attributedString containsAttachments]) { | |
mutableTypes = [[types mutableCopy] autorelease]; | |
[mutableTypes removeObject:NSRTFDPboardType]; | |
types = mutableTypes; | |
} | |
[pasteboard declareTypes:types owner:nil]; | |
#else | |
NSArray *types = selectionPasteboardTypes(canSmartCopyOrDelete, [attributedString containsAttachments]); | |
[pasteboard declareTypes:types owner:nil]; | |
frame->editor()->client()->didSetSelectionTypesForPasteboard(); | |
#endif | |
// Put HTML on the pasteboard. | |
if ([types containsObject:WebArchivePboardType]) { | |
RefPtr<LegacyWebArchive> archive = LegacyWebArchive::createFromSelection(frame); | |
RetainPtr<CFDataRef> data = archive ? archive->rawDataRepresentation() : 0; | |
[pasteboard setData:(NSData *)data.get() forType:WebArchivePboardType]; | |
} | |
// Put the attributed string on the pasteboard (RTF/RTFD format). | |
if ([types containsObject:NSRTFDPboardType]) { | |
NSData *RTFDData = [attributedString RTFDFromRange:NSMakeRange(0, [attributedString length]) documentAttributes:nil]; | |
[pasteboard setData:RTFDData forType:NSRTFDPboardType]; | |
} | |
if ([types containsObject:NSRTFPboardType]) { | |
if ([attributedString containsAttachments]) | |
attributedString = stripAttachmentCharacters(attributedString); | |
NSData *RTFData = [attributedString RTFFromRange:NSMakeRange(0, [attributedString length]) documentAttributes:nil]; | |
[pasteboard setData:RTFData forType:NSRTFPboardType]; | |
} | |
// Put plain string on the pasteboard. | |
if ([types containsObject:NSStringPboardType]) { | |
// Map to a plain old space because this is better for source code, other browsers do it, | |
// and because HTML forces you to do this any time you want two spaces in a row. | |
String text = frame->displayStringModifiedByEncoding(selectedRange->text()); | |
NSMutableString *s = [[[(NSString*)text copy] autorelease] mutableCopy]; | |
NSString *NonBreakingSpaceString = [NSString stringWithCharacters:&noBreakSpace length:1]; | |
[s replaceOccurrencesOfString:NonBreakingSpaceString withString:@" " options:0 range:NSMakeRange(0, [s length])]; | |
[pasteboard setString:s forType:NSStringPboardType]; | |
[s release]; | |
} | |
if ([types containsObject:WebSmartPastePboardType]) { | |
[pasteboard setData:nil forType:WebSmartPastePboardType]; | |
} | |
} | |
void Pasteboard::writePlainText(NSPasteboard* pasteboard, const String& text) | |
{ | |
NSArray *types = [NSArray arrayWithObject:NSStringPboardType]; | |
[pasteboard declareTypes:types owner:nil]; | |
[pasteboard setString:text forType:NSStringPboardType]; | |
} | |
void Pasteboard::writeSelection(Range* selectedRange, bool canSmartCopyOrDelete, Frame* frame) | |
{ | |
Pasteboard::writeSelection(m_pasteboard.get(), selectedRange, canSmartCopyOrDelete, frame); | |
} | |
void Pasteboard::writePlainText(const String& text) | |
{ | |
if (!WebArchivePboardType) | |
Pasteboard::generalPasteboard(); // Initializes pasteboard types. | |
NSArray *types = [NSArray arrayWithObject:NSStringPboardType]; | |
NSPasteboard *pasteboard = m_pasteboard.get(); | |
[pasteboard declareTypes:types owner:nil]; | |
[pasteboard setString:text forType:NSStringPboardType]; | |
} | |
void Pasteboard::writeURL(NSPasteboard* pasteboard, NSArray* types, const KURL& url, const String& titleStr, Frame* frame) | |
{ | |
if (!WebArchivePboardType) | |
Pasteboard::generalPasteboard(); // Initializes pasteboard types. | |
if (!types) { | |
types = writableTypesForURL(); | |
[pasteboard declareTypes:types owner:nil]; | |
} | |
ASSERT(!url.isEmpty()); | |
NSURL *cocoaURL = url; | |
NSString *userVisibleString = frame->editor()->client()->userVisibleString(cocoaURL); | |
NSString *title = (NSString*)titleStr; | |
if ([title length] == 0) { | |
title = [[cocoaURL path] lastPathComponent]; | |
if ([title length] == 0) | |
title = userVisibleString; | |
} | |
if ([types containsObject:WebURLsWithTitlesPboardType]) | |
[pasteboard setPropertyList:[NSArray arrayWithObjects:[NSArray arrayWithObject:userVisibleString], | |
[NSArray arrayWithObject:(NSString*)titleStr.stripWhiteSpace()], | |
nil] | |
forType:WebURLsWithTitlesPboardType]; | |
if ([types containsObject:NSURLPboardType]) | |
[cocoaURL writeToPasteboard:pasteboard]; | |
if ([types containsObject:WebURLPboardType]) | |
[pasteboard setString:userVisibleString forType:WebURLPboardType]; | |
if ([types containsObject:WebURLNamePboardType]) | |
[pasteboard setString:title forType:WebURLNamePboardType]; | |
if ([types containsObject:NSStringPboardType]) | |
[pasteboard setString:userVisibleString forType:NSStringPboardType]; | |
} | |
void Pasteboard::writeURL(const KURL& url, const String& titleStr, Frame* frame) | |
{ | |
Pasteboard::writeURL(m_pasteboard.get(), nil, url, titleStr, frame); | |
} | |
static NSFileWrapper* fileWrapperForImage(CachedResource* resource, NSURL *url) | |
{ | |
SharedBuffer* coreData = resource->data(); | |
NSData *data = [[[NSData alloc] initWithBytes:coreData->data() length:coreData->size()] autorelease]; | |
NSFileWrapper *wrapper = [[[NSFileWrapper alloc] initRegularFileWithContents:data] autorelease]; | |
String coreMIMEType = resource->response().mimeType(); | |
NSString *MIMEType = nil; | |
if (!coreMIMEType.isNull()) | |
MIMEType = coreMIMEType; | |
[wrapper setPreferredFilename:suggestedFilenameWithMIMEType(url, MIMEType)]; | |
return wrapper; | |
} | |
void Pasteboard::writeFileWrapperAsRTFDAttachment(NSFileWrapper* wrapper) | |
{ | |
NSTextAttachment *attachment = [[NSTextAttachment alloc] initWithFileWrapper:wrapper]; | |
NSAttributedString *string = [NSAttributedString attributedStringWithAttachment:attachment]; | |
[attachment release]; | |
NSData *RTFDData = [string RTFDFromRange:NSMakeRange(0, [string length]) documentAttributes:nil]; | |
[m_pasteboard.get() setData:RTFDData forType:NSRTFDPboardType]; | |
} | |
void Pasteboard::writeImage(Node* node, const KURL& url, const String& title) | |
{ | |
ASSERT(node); | |
Frame* frame = node->document()->frame(); | |
NSURL *cocoaURL = url; | |
ASSERT(cocoaURL); | |
ASSERT(node->renderer() && node->renderer()->isImage()); | |
RenderImage* renderer = toRenderImage(node->renderer()); | |
CachedImage* cachedImage = renderer->cachedImage(); | |
if (!cachedImage || cachedImage->errorOccurred()) | |
return; | |
NSArray* types = writableTypesForImage(); | |
[m_pasteboard.get() declareTypes:types owner:nil]; | |
writeURL(m_pasteboard.get(), types, cocoaURL, nsStringNilIfEmpty(title), frame); | |
Image* image = cachedImage->image(); | |
ASSERT(image); | |
[m_pasteboard.get() setData:[image->getNSImage() TIFFRepresentation] forType:NSTIFFPboardType]; | |
String MIMEType = cachedImage->response().mimeType(); | |
ASSERT(MIMETypeRegistry::isSupportedImageResourceMIMEType(MIMEType)); | |
writeFileWrapperAsRTFDAttachment(fileWrapperForImage(cachedImage, cocoaURL)); | |
} | |
bool Pasteboard::canSmartReplace() | |
{ | |
return [[m_pasteboard.get() types] containsObject:WebSmartPastePboardType]; | |
} | |
String Pasteboard::plainText(Frame* frame) | |
{ | |
NSArray *types = [m_pasteboard.get() types]; | |
if ([types containsObject:NSStringPboardType]) | |
return [[m_pasteboard.get() stringForType:NSStringPboardType] precomposedStringWithCanonicalMapping]; | |
NSAttributedString *attributedString = nil; | |
NSString *string; | |
if ([types containsObject:NSRTFDPboardType]) | |
attributedString = [[NSAttributedString alloc] initWithRTFD:[m_pasteboard.get() dataForType:NSRTFDPboardType] documentAttributes:NULL]; | |
if (attributedString == nil && [types containsObject:NSRTFPboardType]) | |
attributedString = [[NSAttributedString alloc] initWithRTF:[m_pasteboard.get() dataForType:NSRTFPboardType] documentAttributes:NULL]; | |
if (attributedString != nil) { | |
string = [[attributedString string] precomposedStringWithCanonicalMapping]; | |
[attributedString release]; | |
return string; | |
} | |
if ([types containsObject:NSFilenamesPboardType]) { | |
string = [[[m_pasteboard.get() propertyListForType:NSFilenamesPboardType] componentsJoinedByString:@"\n"] precomposedStringWithCanonicalMapping]; | |
if (string != nil) | |
return string; | |
} | |
if (NSURL *url = [NSURL URLFromPasteboard:m_pasteboard.get()]) { | |
// FIXME: using the editorClient to call into webkit, for now, since | |
// calling _web_userVisibleString from WebCore involves migrating a sizable web of | |
// helper code that should either be done in a separate patch or figured out in another way. | |
string = frame->editor()->client()->userVisibleString(url); | |
if ([string length] > 0) | |
return [string precomposedStringWithCanonicalMapping]; | |
} | |
return String(); | |
} | |
PassRefPtr<DocumentFragment> Pasteboard::documentFragment(Frame* frame, PassRefPtr<Range> context, bool allowPlainText, bool& chosePlainText) | |
{ | |
NSArray *types = [m_pasteboard.get() types]; | |
chosePlainText = false; | |
if ([types containsObject:NSHTMLPboardType]) { | |
NSString *HTMLString = [m_pasteboard.get() stringForType:NSHTMLPboardType]; | |
// This is a hack to make Microsoft's HTML pasteboard data work. See 3778785. | |
if ([HTMLString hasPrefix:@"Version:"]) { | |
NSRange range = [HTMLString rangeOfString:@"<html" options:NSCaseInsensitiveSearch]; | |
if (range.location != NSNotFound) { | |
HTMLString = [HTMLString substringFromIndex:range.location]; | |
} | |
} | |
if ([HTMLString length] != 0) { | |
RefPtr<DocumentFragment> fragment = createFragmentFromMarkup(frame->document(), HTMLString, "", FragmentScriptingNotAllowed); | |
if (fragment) | |
return fragment.release(); | |
} | |
} | |
if (allowPlainText && [types containsObject:NSStringPboardType]) { | |
chosePlainText = true; | |
RefPtr<DocumentFragment> fragment = createFragmentFromText(context.get(), [[m_pasteboard.get() stringForType:NSStringPboardType] precomposedStringWithCanonicalMapping]); | |
if (fragment) | |
return fragment.release(); | |
} | |
return 0; | |
} | |
} |