132 lines
3.3 KiB
Objective-C
132 lines
3.3 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 <objc/runtime.h>
|
|
|
|
#import "RCTConvert.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()
|
|
|
|
- (NSArray *)supportedEvents
|
|
{
|
|
return @[@"websocketMessage",
|
|
@"websocketOpen",
|
|
@"websocketFailed",
|
|
@"websocketClosed"];
|
|
}
|
|
|
|
- (void)dealloc
|
|
{
|
|
for (RCTSRWebSocket *socket in _sockets.allValues) {
|
|
socket.delegate = nil;
|
|
[socket close];
|
|
}
|
|
}
|
|
|
|
RCT_EXPORT_METHOD(connect:(NSURL *)URL protocols:(NSArray *)protocols headers:(NSDictionary *)headers socketID:(nonnull NSNumber *)socketID)
|
|
{
|
|
NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:URL];
|
|
[headers enumerateKeysAndObjectsUsingBlock:^(NSString *key, id value, BOOL *stop) {
|
|
[request addValue:[RCTConvert NSString:value] forHTTPHeaderField:key];
|
|
}];
|
|
|
|
RCTSRWebSocket *webSocket = [[RCTSRWebSocket alloc] initWithURLRequest:request protocols:protocols];
|
|
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(sendBinary:(NSString *)base64String socketID:(nonnull NSNumber *)socketID)
|
|
{
|
|
NSData *message = [[NSData alloc] initWithBase64EncodedString:base64String options:0];
|
|
[_sockets[socketID] send:message];
|
|
}
|
|
|
|
RCT_EXPORT_METHOD(ping:(nonnull NSNumber *)socketID)
|
|
{
|
|
[_sockets[socketID] sendPing:NULL];
|
|
}
|
|
|
|
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]];
|
|
[self sendEventWithName:@"websocketMessage" body:@{
|
|
@"data": binary ? [message base64EncodedStringWithOptions:0] : message,
|
|
@"type": binary ? @"binary" : @"text",
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
- (void)webSocketDidOpen:(RCTSRWebSocket *)webSocket
|
|
{
|
|
[self sendEventWithName:@"websocketOpen" body:@{
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
- (void)webSocket:(RCTSRWebSocket *)webSocket didFailWithError:(NSError *)error
|
|
{
|
|
[self sendEventWithName:@"websocketFailed" body:@{
|
|
@"message":error.localizedDescription,
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
- (void)webSocket:(RCTSRWebSocket *)webSocket didCloseWithCode:(NSInteger)code
|
|
reason:(NSString *)reason wasClean:(BOOL)wasClean
|
|
{
|
|
[self sendEventWithName:@"websocketClosed" body:@{
|
|
@"code": @(code),
|
|
@"reason": RCTNullIfNil(reason),
|
|
@"clean": @(wasClean),
|
|
@"id": webSocket.reactTag
|
|
}];
|
|
}
|
|
|
|
@end
|