mirror of
https://github.com/zhigang1992/react-native.git
synced 2026-03-30 23:22:41 +08:00
205b5d4732
Summary:This is a follow up of 9b87e6c860.
- Allows custom headers on connection request
- Adds a default `origin` header to Android, just like iOS
**Introduces no breaking changes.**
I was working on something similar and would like to propose a few changes that make the API more consistent across both iOS and Android platforms and brings this closer to [spec](https://tools.ietf.org/html/rfc6455).
I believe aprock first implementation of adding custom `headers` was correct. It makes sense naming this argument `headers` since we have no other general options available, and the current `options` field is being used to pass in a header anyway.
My use case for custom headers was attaching a token to the `Authorization` header on the connection request. I have been testing this by passing a JWT inside the `Authorization` header and verifying it on the server before establishing a connection.
Closes https://github.com/facebook/react-native/pull/6016
Differential Revision: D3040735
Pulled By: nicklockwood
fb-gh-sync-id: f81bd14ccbdba36309b9d4b4850fb66fe4deae11
shipit-source-id: f81bd14ccbdba36309b9d4b4850fb66fe4deae11
115 lines
3.0 KiB
Objective-C
115 lines
3.0 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 "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()
|
|
|
|
@synthesize bridge = _bridge;
|
|
|
|
- (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(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
|