d6f4155c50a3ba17384853bd0d968fcb7558a66a
[WebKit-https.git] / Source / WebCore / platform / mac / DragDataMac.mm
1 /*
2  * Copyright (C) 2007 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  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #import "config.h"
27 #import "DragData.h"
28
29 #if ENABLE(DRAG_SUPPORT)
30 #import "Document.h"
31 #import "DocumentFragment.h"
32 #import "DOMDocumentFragment.h"
33 #import "DOMDocumentFragmentInternal.h"
34 #import "MIMETypeRegistry.h"
35 #import "Pasteboard.h"
36 #import "Range.h"
37
38 namespace WebCore {
39
40 DragData::DragData(DragDataRef data, const IntPoint& clientPosition, const IntPoint& globalPosition, 
41     DragOperation sourceOperationMask, DragApplicationFlags flags)
42     : m_clientPosition(clientPosition)
43     , m_globalPosition(globalPosition)
44     , m_platformDragData(data)
45     , m_draggingSourceOperationMask(sourceOperationMask)
46     , m_applicationFlags(flags)
47     , m_pasteboard([m_platformDragData draggingPasteboard])
48 {
49 }
50
51 DragData::DragData(const String& dragStorageName, const IntPoint& clientPosition, const IntPoint& globalPosition,
52     DragOperation sourceOperationMask, DragApplicationFlags flags)
53     : m_clientPosition(clientPosition)
54     , m_globalPosition(globalPosition)
55     , m_platformDragData(0)
56     , m_draggingSourceOperationMask(sourceOperationMask)
57     , m_applicationFlags(flags)
58     , m_pasteboard([NSPasteboard pasteboardWithName:dragStorageName])
59 {
60 }
61     
62 bool DragData::canSmartReplace() const
63 {
64     //Need to call this so that the various Pasteboard type strings are intialised
65     Pasteboard::generalPasteboard();
66     return [[m_pasteboard.get() types] containsObject:WebSmartPastePboardType];
67 }
68
69 bool DragData::containsColor() const
70 {
71     return [[m_pasteboard.get() types] containsObject:NSColorPboardType];
72 }
73
74 bool DragData::containsFiles() const
75 {
76     return [[m_pasteboard.get() types] containsObject:NSFilenamesPboardType];
77 }
78
79 unsigned DragData::numberOfFiles() const
80 {
81     if (![[m_pasteboard.get() types] containsObject:NSFilenamesPboardType])
82         return 0;
83     return [[m_pasteboard.get() propertyListForType:NSFilenamesPboardType] count];
84 }
85
86 void DragData::asFilenames(Vector<String>& result) const
87 {
88     NSArray *filenames = [m_pasteboard.get() propertyListForType:NSFilenamesPboardType];
89     NSEnumerator *fileEnumerator = [filenames objectEnumerator];
90     
91     while (NSString *filename = [fileEnumerator nextObject])
92         result.append(filename);
93 }
94
95 bool DragData::containsPlainText() const
96 {
97     NSArray *types = [m_pasteboard.get() types];
98     
99     return [types containsObject:NSStringPboardType] 
100         || [types containsObject:NSRTFDPboardType]
101         || [types containsObject:NSRTFPboardType]
102         || [types containsObject:NSFilenamesPboardType]
103         || [NSURL URLFromPasteboard:m_pasteboard.get()];
104 }
105
106 String DragData::asPlainText(Frame *frame) const
107 {
108     Pasteboard pasteboard([m_pasteboard.get() name]);
109     return pasteboard.plainText(frame);
110 }
111
112 Color DragData::asColor() const
113 {
114     NSColor *color = [NSColor colorFromPasteboard:m_pasteboard.get()];
115     
116     // The color may not be in an RGB colorspace. This commonly occurs when a color is 
117     // dragged from the NSColorPanel grayscale picker.
118     if ([[color colorSpace] colorSpaceModel] != NSRGBColorSpaceModel)
119         color = [color colorUsingColorSpaceName:NSCalibratedRGBColorSpace];
120     
121     return makeRGBA((int)([color redComponent] * 255.0 + 0.5), (int)([color greenComponent] * 255.0 + 0.5), 
122                     (int)([color blueComponent] * 255.0 + 0.5), (int)([color alphaComponent] * 255.0 + 0.5));
123 }
124
125 static NSArray *insertablePasteboardTypes()
126 {
127     static NSArray *types = nil;
128     if (!types) {
129         types = [[NSArray alloc] initWithObjects:WebArchivePboardType, NSHTMLPboardType, NSFilenamesPboardType, NSTIFFPboardType, NSPDFPboardType,
130 #ifdef BUILDING_ON_LEOPARD
131                  NSPICTPboardType,
132 #endif
133                  NSURLPboardType, NSRTFDPboardType, NSRTFPboardType, NSStringPboardType, NSColorPboardType, kUTTypePNG, nil];
134         CFRetain(types);
135     }
136     return types;
137 }
138     
139 bool DragData::containsCompatibleContent() const
140 {
141     NSMutableSet *types = [NSMutableSet setWithArray:[m_pasteboard.get() types]];
142     [types intersectSet:[NSSet setWithArray:insertablePasteboardTypes()]];
143     return [types count] != 0;
144 }
145     
146 bool DragData::containsURL(Frame* frame, FilenameConversionPolicy filenamePolicy) const
147 {
148     return !asURL(frame, filenamePolicy).isEmpty();
149 }
150     
151 String DragData::asURL(Frame* frame, FilenameConversionPolicy filenamePolicy, String* title) const
152 {
153     // FIXME: Use filenamePolicy.
154     (void)filenamePolicy;
155
156     if (title) {
157         if (NSString *URLTitleString = [m_pasteboard.get() stringForType:WebURLNamePboardType])
158             *title = URLTitleString;
159     }
160     Pasteboard pasteboard([m_pasteboard.get() name]);
161     return pasteboard.asURL(frame);
162 }
163
164 PassRefPtr<DocumentFragment> DragData::asFragment(Frame* frame, PassRefPtr<Range> range, bool allowPlainText, bool& chosePlainText) const
165 {
166     Pasteboard pasteboard([m_pasteboard.get() name]);
167     
168     return pasteboard.documentFragment(frame, range, allowPlainText, chosePlainText);
169 }
170     
171 } // namespace WebCore
172
173 #endif // ENABLE(DRAG_SUPPORT)