Implement WebSocket module for Android. Fixes #2837

Summary: The JavaScript code for Android is same as the iOS counterpart, I just added few new lines and used arrow functions instead of binding `this`.
Closes https://github.com/facebook/react-native/pull/2839

Reviewed By: @​svcscm, @vjeux

Differential Revision: D2498703

Pulled By: @mkonicek

fb-gh-sync-id: 3fe958dd5af0efba00df07515f8e33b5d87eb05b

Libraries/WebSocket/RCTWebSocketModule.m

@@ -0,0 +1,117 @@
+/**
+ * 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 "RCTSRWebSocket.h"
+#import "RCTSparseArray.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
+
+@interface RCTWebSocketModule () <RCTSRWebSocketDelegate>
+
+@end
+
+@implementation RCTWebSocketModule
+{
+    RCTSparseArray *_sockets;
+}
+
+RCT_EXPORT_MODULE()
+
+@synthesize bridge = _bridge;
+
+- (instancetype)init
+{
+  if ((self = [super init])) {
+    _sockets = [RCTSparseArray new];
+  }
+  return self;
+}
+
+- (void)dealloc
+{
+  for (RCTSRWebSocket *socket in _sockets.allObjects) {
+    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;
+  _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[socketID] = nil;
+}
+
+#pragma mark - RCTSRWebSocketDelegate methods
+
+- (void)webSocket:(RCTSRWebSocket *)webSocket didReceiveMessage:(id)message
+{
+  [_bridge.eventDispatcher sendDeviceEventWithName:@"websocketMessage" body:@{
+    @"data": message,
+    @"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