| /* |
| * 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 = 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) { |
| // FIXME: FragmentScriptingNotAllowed is a HACK and should |
| // be removed or replaced with an enum with a better name. |
| // FragmentScriptingNotAllowed causes the Parser to remove children |
| // of <script> tags (so javascript doesn't show up in pastes). |
| 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; |
| } |
| |
| } |