Unify most of the WebKit Objective-C API sources
[WebKit-https.git] / Source / WebKit / UIProcess / API / Cocoa / WKConnection.mm
1 /*
2  * Copyright (C) 2011 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 "WKConnectionInternal.h"
28
29 #if WK_API_ENABLED
30
31 #import "ObjCObjectGraph.h"
32 #import "WKRetainPtr.h"
33 #import "WKSharedAPICast.h"
34 #import "WKStringCF.h"
35 #import <wtf/RetainPtr.h>
36 #import <wtf/WeakObjCPtr.h>
37 #import <wtf/text/WTFString.h>
38
39 @implementation WKConnection {
40     WeakObjCPtr<id <WKConnectionDelegate>> _delegate;
41 }
42
43 - (void)dealloc
44 {
45     self._connection.~WebConnection();
46
47     [super dealloc];
48 }
49
50 static void didReceiveMessage(WKConnectionRef, WKStringRef messageName, WKTypeRef messageBody, const void* clientInfo)
51 {
52     auto connection = (__bridge WKConnection *)clientInfo;
53     auto delegate = connection->_delegate.get();
54
55     if ([delegate respondsToSelector:@selector(connection:didReceiveMessageWithName:body:)]) {
56         RetainPtr<CFStringRef> nsMessageName = adoptCF(WKStringCopyCFString(kCFAllocatorDefault, messageName));
57         RetainPtr<id> nsMessageBody = static_cast<WebKit::ObjCObjectGraph*>(WebKit::toImpl(messageBody))->rootObject();
58         [delegate connection:connection didReceiveMessageWithName:(__bridge NSString *)nsMessageName.get() body:nsMessageBody.get()];
59     }
60 }
61
62 static void didClose(WKConnectionRef, const void* clientInfo)
63 {
64     auto connection = (__bridge WKConnection *)clientInfo;
65     auto delegate = connection->_delegate.get();
66
67     if ([delegate respondsToSelector:@selector(connectionDidClose:)])
68         [delegate connectionDidClose:connection];
69 }
70
71 static void setUpClient(WKConnection *wrapper, WebKit::WebConnection& connection)
72 {
73     WKConnectionClientV0 client;
74     memset(&client, 0, sizeof(client));
75
76     client.base.version = 0;
77     client.base.clientInfo = (__bridge void*)wrapper;
78     client.didReceiveMessage = didReceiveMessage;
79     client.didClose = didClose;
80
81     connection.initializeConnectionClient(&client.base);
82 }
83
84 - (id <WKConnectionDelegate>)delegate
85 {
86     return _delegate.getAutoreleased();
87 }
88
89 - (void)setDelegate:(id <WKConnectionDelegate>)delegate
90 {
91     _delegate = delegate;
92     if (delegate)
93         setUpClient(self, self._connection);
94     else
95         self._connection.initializeConnectionClient(nullptr);
96 }
97
98 - (void)sendMessageWithName:(NSString *)messageName body:(id)messageBody
99 {
100     RefPtr<WebKit::ObjCObjectGraph> wkMessageBody = WebKit::ObjCObjectGraph::create(messageBody);
101     self._connection.postMessage(messageName, wkMessageBody.get());
102 }
103
104 - (WebKit::WebConnection&)_connection
105 {
106     return static_cast<WebKit::WebConnection&>(API::Object::fromWKObjectExtraSpace(self));
107 }
108
109 #pragma mark WKObject protocol implementation
110
111 - (API::Object&)_apiObject
112 {
113     return API::Object::fromWKObjectExtraSpace(self);
114 }
115
116 @end
117
118 #endif // WK_API_ENABLED