[Attachment Support] Implement SPI for clients to request data for a given attachment
[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/SharedBuffer.h>
35 #import <wtf/BlockPtr.h>
36
37 using namespace WebKit;
38
39 @implementation _WKAttachmentDisplayOptions : NSObject
40
41 - (instancetype)init
42 {
43     if (self = [super init]) {
44         _mode = _WKAttachmentDisplayModeAuto;
45         _expandsImageToMaximumWidth = NO;
46     }
47     return self;
48 }
49
50 @end
51
52 @implementation _WKAttachment
53
54 - (API::Object&)_apiObject
55 {
56     return *_attachment;
57 }
58
59 - (BOOL)isEqual:(id)object
60 {
61     return [object isKindOfClass:[_WKAttachment class]] && [self.uniqueIdentifier isEqual:[(_WKAttachment *)object uniqueIdentifier]];
62 }
63
64 - (void)requestData:(void(^)(NSData *, NSError *))completionHandler
65 {
66     _attachment->requestData([ capturedBlock = makeBlockPtr(completionHandler) ] (RefPtr<WebCore::SharedBuffer> buffer, CallbackBase::Error error) {
67         if (!capturedBlock)
68             return;
69
70         if (buffer && error == CallbackBase::Error::None)
71             capturedBlock(buffer->createNSData().autorelease(), nil);
72         else
73             capturedBlock(nil, [NSError errorWithDomain:WKErrorDomain code:1 userInfo:nil]);
74     });
75 }
76
77 - (NSString *)uniqueIdentifier
78 {
79     return _attachment->identifier();
80 }
81
82 - (NSUInteger)hash
83 {
84     return [self.uniqueIdentifier hash];
85 }
86
87 - (NSString *)description
88 {
89     return [NSString stringWithFormat:@"<%@ id='%@'>", [self class], self.uniqueIdentifier];
90 }
91
92 @end
93
94 #endif // WK_API_ENABLED