[Cocoa] Exception (fileType 'dyn.agq8u' is not a valid UTI) raised when dragging...
[WebKit-https.git] / Source / WebKit / UIProcess / API / Cocoa / _WKAttachment.mm
1 /*
2  * Copyright (C) 2017 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 INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "config.h"
27 #import "_WKAttachment.h"
28
29 #if WK_API_ENABLED
30
31 #import "APIAttachment.h"
32 #import "WKErrorPrivate.h"
33 #import "_WKAttachmentInternal.h"
34 #import <WebCore/AttachmentTypes.h>
35 #import <WebCore/MIMETypeRegistry.h>
36 #import <WebCore/SharedBuffer.h>
37 #import <wtf/BlockPtr.h>
38
39 #if PLATFORM(IOS)
40 #import <MobileCoreServices/MobileCoreServices.h>
41 #endif
42
43 using namespace WebKit;
44
45 static const NSInteger UnspecifiedAttachmentErrorCode = 1;
46 static const NSInteger InvalidAttachmentErrorCode = 2;
47
48 @implementation _WKAttachmentDisplayOptions : NSObject
49
50 - (WebCore::AttachmentDisplayOptions)coreDisplayOptions
51 {
52     return { };
53 }
54
55 @end
56
57 @implementation _WKAttachmentInfo {
58     RetainPtr<NSFileWrapper> _fileWrapper;
59     RetainPtr<NSString> _mimeType;
60     RetainPtr<NSString> _utiType;
61     RetainPtr<NSString> _filePath;
62 }
63
64 - (instancetype)initWithFileWrapper:(NSFileWrapper *)fileWrapper filePath:(NSString *)filePath mimeType:(NSString *)mimeType utiType:(NSString *)utiType
65 {
66     if (!(self = [super init]))
67         return nil;
68
69     _fileWrapper = fileWrapper;
70     _filePath = filePath;
71     _mimeType = mimeType;
72     _utiType = utiType;
73     return self;
74 }
75
76 - (NSData *)data
77 {
78     if (![_fileWrapper isRegularFile]) {
79         // FIXME: Handle attachments backed by NSFileWrappers that represent directories.
80         return nil;
81     }
82
83     return [_fileWrapper regularFileContents];
84 }
85
86 - (NSString *)name
87 {
88     if ([_fileWrapper filename].length)
89         return [_fileWrapper filename];
90
91     return [_fileWrapper preferredFilename];
92 }
93
94 - (NSString *)filePath
95 {
96     return _filePath.get();
97 }
98
99 - (NSFileWrapper *)fileWrapper
100 {
101     return _fileWrapper.get();
102 }
103
104 - (NSString *)contentType
105 {
106     if ([_mimeType length])
107         return _mimeType.get();
108
109     return _utiType.get();
110 }
111
112 @end
113
114 @implementation _WKAttachment
115
116 - (API::Object&)_apiObject
117 {
118     return *_attachment;
119 }
120
121 - (_WKAttachmentInfo *)info
122 {
123     if (!_attachment->isValid())
124         return nil;
125
126     return [[[_WKAttachmentInfo alloc] initWithFileWrapper:_attachment->fileWrapper() filePath:_attachment->filePath() mimeType:_attachment->mimeType() utiType:_attachment->utiType()] autorelease];
127 }
128
129 - (void)requestInfo:(void(^)(_WKAttachmentInfo *, NSError *))completionHandler
130 {
131     completionHandler(self.info, nil);
132 }
133
134 - (void)setDisplayOptions:(_WKAttachmentDisplayOptions *)options completion:(void(^)(NSError *))completionHandler
135 {
136     if (!_attachment->isValid()) {
137         completionHandler([NSError errorWithDomain:WKErrorDomain code:InvalidAttachmentErrorCode userInfo:nil]);
138         return;
139     }
140
141     auto coreOptions = options ? options.coreDisplayOptions : WebCore::AttachmentDisplayOptions { };
142     _attachment->setDisplayOptions(coreOptions, [capturedBlock = makeBlockPtr(completionHandler)] (CallbackBase::Error error) {
143         if (!capturedBlock)
144             return;
145
146         if (error == CallbackBase::Error::None)
147             capturedBlock(nil);
148         else
149             capturedBlock([NSError errorWithDomain:WKErrorDomain code:UnspecifiedAttachmentErrorCode userInfo:nil]);
150     });
151 }
152
153 - (void)setFileWrapper:(NSFileWrapper *)fileWrapper contentType:(NSString *)contentType completion:(void (^)(NSError *))completionHandler
154 {
155     if (!_attachment->isValid()) {
156         completionHandler([NSError errorWithDomain:WKErrorDomain code:InvalidAttachmentErrorCode userInfo:nil]);
157         return;
158     }
159
160     // This file path member is only populated when the attachment is generated upon dropping files. When data is specified via NSFileWrapper
161     // from the SPI client, the corresponding file path of the data is unknown, if it even exists at all.
162     _attachment->setFilePath({ });
163     _attachment->setFileWrapperAndUpdateContentType(fileWrapper, contentType);
164     _attachment->updateAttributes([capturedBlock = makeBlockPtr(completionHandler)] (auto error) {
165         if (!capturedBlock)
166             return;
167
168         if (error == CallbackBase::Error::None)
169             capturedBlock(nil);
170         else
171             capturedBlock([NSError errorWithDomain:WKErrorDomain code:UnspecifiedAttachmentErrorCode userInfo:nil]);
172     });
173 }
174
175 - (void)setData:(NSData *)data newContentType:(NSString *)newContentType newFilename:(NSString *)newFilename completion:(void(^)(NSError *))completionHandler
176 {
177     auto fileWrapper = adoptNS([[NSFileWrapper alloc] initRegularFileWithContents:data]);
178     if (newFilename)
179         [fileWrapper setPreferredFilename:newFilename];
180     [self setFileWrapper:fileWrapper.get() contentType:newContentType completion:completionHandler];
181 }
182
183 - (NSString *)uniqueIdentifier
184 {
185     return _attachment->identifier();
186 }
187
188 - (NSString *)description
189 {
190     return [NSString stringWithFormat:@"<%@ %p id='%@'>", [self class], self, self.uniqueIdentifier];
191 }
192
193 - (BOOL)isConnected
194 {
195     return _attachment->insertionState() == API::Attachment::InsertionState::Inserted;
196 }
197
198 @end
199
200 #endif // WK_API_ENABLED