mirror of
https://github.com/status-im/react-native.git
synced 2025-01-17 21:11:45 +00:00
66158ab2bf
Summary:
This allows consumers to subclass and extend `RCTWebSocketModule` and make use of the `RCTSRWebSocketDelegate` methods.
The use case here is to do some pre-processing of WebSocket data before handing it off to JS. Consumers could that in the following way:
```
interface MyWebSocketModule : RCTWebSocketModule
end
implementation MyWebSocketModule
// Don't use RCT_EXPORT_MODULE macro for this so we replace the existing RCTWebSocketModule.
+ (NSString *)moduleName { return @"RCTWebSocketModule"; }
RCT_EXTERN_METHOD(connect:(NSURL *)URL socketID:(nonnull NSNumber *)socketID)
RCT_EXTERN_METHOD(send:(NSString *)message socketID:(nonnull NSNumber *)socketID)
RCT_EXTERN_METHOD(close:(nonnull NSNumber *)socketID)
- (void)webSocket:(RCTSRWebSocket *)webSocket didReceiveMessage:(id)message
{
[super webSocket:webSocket didReceiveMessage:[DoSomethingWith message]];
}
end
```
... and then returning a `MyWebSocketModule` ins
Closes https://github.com/facebook/react-native/pull/5321
Reviewed By: svcscm
Differential Revision: D2832374
Pulled By: nicklockwood
fb-gh-sync-id: f208516b2b2f76276223ffc972871d96afe87e27
109 lines
2.7 KiB
Objective-C
109 lines
2.7 KiB
Objective-C
/**
|
|
* Copyright (c) 2015-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*/
|
|
|
|
#import "RCTWebSocketModule.h"
|
|
|
|
#import "RCTBridge.h"
|
|
#import "RCTEventDispatcher.h"
|
|
#import "RCTUtils.h"
|
|
|
|
@implementation RCTSRWebSocket (React)
|
|
|
|
- (NSNumber *)reactTag
|
|
{
|
|
return objc_getAssociatedObject(self, _cmd);
|
|
}
|
|
|
|
- (void)setReactTag:(NSNumber *)reactTag
|
|
{
|
|
objc_setAssociatedObject(self, @selector(reactTag), reactTag, OBJC_ASSOCIATION_COPY_NONATOMIC);
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation RCTWebSocketModule
|
|
{
|
|
NSMutableDictionary<NSNumber *, RCTSRWebSocket *> *_sockets;
|
|
}
|
|
|
|
RCT_EXPORT_MODULE()
|
|
|
|
@synthesize bridge = _bridge;
|
|
|
|
- (void)dealloc
|
|
{
|
|
for (RCTSRWebSocket *socket in _sockets.allValues) {
|
|
socket.delegate = nil;
|
|
[socket close];
|
|
}
|
|
}
|
|
|
|
RCT_EXPORT_METHOD(connect:(NSURL *)URL socketID:(nonnull NSNumber *)socketID)
|
|
{
|
|
RCTSRWebSocket *webSocket = [[RCTSRWebSocket alloc] initWithURL:URL];
|
|
webSocket.delegate = self;
|
|
webSocket.reactTag = socketID;
|
|
if (!_sockets) {
|
|
_sockets = [NSMutableDictionary new];
|
|
}
|
|
_sockets[socketID] = webSocket;
|
|
[webSocket open];
|
|
}
|
|
|
|
RCT_EXPORT_METHOD(send:(NSString *)message socketID:(nonnull NSNumber *)socketID)
|
|
{
|
|
[_sockets[socketID] send:message];
|
|
}
|
|
|
|
RCT_EXPORT_METHOD(close:(nonnull NSNumber *)socketID)
|
|
{
|
|
[_sockets[socketID] close];
|
|
[_sockets removeObjectForKey:socketID];
|
|
}
|
|
|
|
#pragma mark - RCTSRWebSocketDelegate methods
|
|
|
|
- (void)webSocket:(RCTSRWebSocket *)webSocket didReceiveMessage:(id)message
|
|
{
|
|
BOOL binary = [message isKindOfClass:[NSData class]];
|
|
[_bridge.eventDispatcher sendDeviceEventWithName:@"websocketMessage" body:@{
|
|
@"data": binary ? [message base64EncodedStringWithOptions:0] : message,
|
|
@"type": binary ? @"binary" : @"text",
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
- (void)webSocketDidOpen:(RCTSRWebSocket *)webSocket
|
|
{
|
|
[_bridge.eventDispatcher sendDeviceEventWithName:@"websocketOpen" body:@{
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
- (void)webSocket:(RCTSRWebSocket *)webSocket didFailWithError:(NSError *)error
|
|
{
|
|
[_bridge.eventDispatcher sendDeviceEventWithName:@"websocketFailed" body:@{
|
|
@"message":error.localizedDescription,
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
- (void)webSocket:(RCTSRWebSocket *)webSocket didCloseWithCode:(NSInteger)code
|
|
reason:(NSString *)reason wasClean:(BOOL)wasClean
|
|
{
|
|
[_bridge.eventDispatcher sendDeviceEventWithName:@"websocketClosed" body:@{
|
|
@"code": @(code),
|
|
@"reason": RCTNullIfNil(reason),
|
|
@"clean": @(wasClean),
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
@end
|