Expand DragController to provide more information about the dragging session
[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());
109     return pasteboard.plainText(frame);
110 }
111
112 Color DragData::asColor() const
113 {
114     NSColor *color = [NSColor colorFromPasteboard:m_pasteboard.get()];
115     return makeRGBA((int)([color redComponent] * 255.0 + 0.5), (int)([color greenComponent] * 255.0 + 0.5), 
116                     (int)([color blueComponent] * 255.0 + 0.5), (int)([color alphaComponent] * 255.0 + 0.5));
117 }
118
119 static NSArray *insertablePasteboardTypes()
120 {
121     static NSArray *types = nil;
122     if (!types) {
123         types = [[NSArray alloc] initWithObjects:WebArchivePboardType, NSHTMLPboardType, NSFilenamesPboardType, NSTIFFPboardType, NSPDFPboardType,
124 #ifdef BUILDING_ON_LEOPARD
125                  NSPICTPboardType,
126 #endif
127                  NSURLPboardType, NSRTFDPboardType, NSRTFPboardType, NSStringPboardType, NSColorPboardType, kUTTypePNG, nil];
128         CFRetain(types);
129     }
130     return types;
131 }
132     
133 bool DragData::containsCompatibleContent() const
134 {
135     NSMutableSet *types = [NSMutableSet setWithArray:[m_pasteboard.get() types]];
136     [types intersectSet:[NSSet setWithArray:insertablePasteboardTypes()]];
137     return [types count] != 0;
138 }
139     
140 bool DragData::containsURL(Frame* frame, FilenameConversionPolicy filenamePolicy) const
141 {
142     return !asURL(frame, filenamePolicy).isEmpty();
143 }
144     
145 String DragData::asURL(Frame* frame, FilenameConversionPolicy filenamePolicy, String* title) const
146 {
147     // FIXME: Use filenamePolicy.
148     (void)filenamePolicy;
149
150     if (title) {
151         if (NSString *URLTitleString = [m_pasteboard.get() stringForType:WebURLNamePboardType])
152             *title = URLTitleString;
153     }
154     Pasteboard pasteboard(m_pasteboard.get());
155     return pasteboard.asURL(frame);
156 }
157
158 PassRefPtr<DocumentFragment> DragData::asFragment(Frame* frame, PassRefPtr<Range> range, bool allowPlainText, bool& chosePlainText) const
159 {
160     Pasteboard pasteboard(m_pasteboard.get());
161     
162     return pasteboard.documentFragment(frame, range, allowPlainText, chosePlainText);
163 }
164     
165 } // namespace WebCore
166
167 #endif // ENABLE(DRAG_SUPPORT)