add types

package.json

@@ -15,7 +15,9 @@
     "node-int64": "^0.4.0"
   },
   "devDependencies": {
+    "@types/crypto-js": "^3.1.43",
     "@types/jest": "^24.0.11",
+    "@types/node-int64": "^0.4.29",
     "html-webpack-plugin": "^3.2.0",
     "jest": "^24.5.0",
     "ts-jest": "^24.0.0",

src/constant.ts

@@ -1,5 +0,0 @@
-// 单个数据包能够传输的最大数据量
-export const MAX_PAYLOAD = 1024 * 1024;
-
-// 版本
-export const VERSION = '1.0';

src/index.ts

@@ -1,12 +1,13 @@
-import * as constant from './constant';
 import { ReadyStateCallback, RequestCallback } from './callback';
 import { Packet } from './packet';
 import { Utils } from './utils';
 
+const MAX_PAYLOAD = 1024 * 1024;
+
 /**
  * Client ws client, 单例模式, 负责维护连接
  */
-export class Client {
+class Client {
   private requestCallback: RequestCallback;
   private requestHeader: string;
   private responseHeader: string;
@@ -18,7 +19,7 @@ export class Client {
   private readyStateCallback: ReadyStateCallback;
 
   constructor(url: string, readyStateCallback: ReadyStateCallback) {
-    this.maxPayload = constant.MAX_PAYLOAD;
+    this.maxPayload = MAX_PAYLOAD;
     this.url = url;
     this.readyStateCallback = readyStateCallback;
 
@@ -36,7 +37,7 @@ export class Client {
     }
 
     let _this = this;
-    this.addMessageListener(0, function (data) {
+    this.addMessageListener(0, function(data) {
       let code = _this.getResponseProperty('code');
       if (typeof code !== 'undefined') {
         let message = _this.getResponseProperty('message');
@@ -93,7 +94,7 @@ export class Client {
     let _this = this;
     let sequence = new Date().getTime();
     let listener = Utils.crc32(operator) + sequence;
-    this.requestCallback[listener] = function (data) {
+    this.requestCallback[listener] = function(data) {
       let code = _this.getResponseProperty('code');
       if (typeof code !== 'undefined') {
         let message = _this.getResponseProperty('message');
@@ -197,16 +198,14 @@ export class Client {
   }
 
   // 创建连接
-  connect() {
-    const url = this.url;
+  connect(): WebSocket {
     const readyStateCallback = this.readyStateCallback;
 
-    let ws = new WebSocket(url);
-    let _this = this;
-
+    let ws = new WebSocket(this.url);
     ws.binaryType = 'blob';
+    let _this = this;
 
-    ws.onopen = function (ev) {
+    ws.onopen = function(ev) {
       _this.reconnectTimes = 0;
       if (
         readyStateCallback.hasOwnProperty('onOpen') &&
@@ -216,7 +215,7 @@ export class Client {
       }
     };
 
-    ws.onclose = function (ev) {
+    ws.onclose = function(ev) {
       _this.reconnect();
       if (
         readyStateCallback.hasOwnProperty('onClose') &&
@@ -226,7 +225,7 @@ export class Client {
       }
     };
 
-    ws.onerror = function (ev) {
+    ws.onerror = function(ev) {
       _this.reconnect();
       if (
         readyStateCallback.hasOwnProperty('onError') &&
@@ -236,16 +235,16 @@ export class Client {
       }
     };
 
-    ws.onmessage = function (ev) {
+    ws.onmessage = function(ev) {
       if (ev.data instanceof Blob) {
         let reader = new FileReader();
         reader.readAsArrayBuffer(ev.data);
-        reader.onload = function () {
+        reader.onload = function() {
           try {
             let packet = new Packet().unPack(this.result);
             let packetLength = packet.headerLength + packet.bodyLength + 20;
-            if (packetLength > constant.MAX_PAYLOAD) {
-              throw new Error('the packet is big than ' + constant.MAX_PAYLOAD);
+            if (packetLength > MAX_PAYLOAD) {
+              throw new Error('the packet is big than ' + MAX_PAYLOAD);
             }
 
             let operator = Number(packet.operator) + Number(packet.sequence);
@@ -285,3 +284,5 @@ export class Client {
     }
   }
 }
+
+export { Client, MAX_PAYLOAD };

src/packet.ts

@@ -1,5 +1,5 @@
 import { Utils } from './utils';
-import * as Int64 from 'node-int64';
+import Int64 = require('node-int64');
 
 export class Packet {
   private key: string = 'b8ca9aa66def05ff3f24919274bb4a66';

src/version.ts

@@ -0,0 +1 @@
+export const VERSION = '1.0';

yarn.lock

@@ -281,6 +281,10 @@
   dependencies:
     "@babel/types" "^7.3.0"
 
+"@types/crypto-js@^3.1.43":
+  version "3.1.43"
+  resolved "https://registry.npm.taobao.org/@types/crypto-js/download/@types/crypto-js-3.1.43.tgz#b859347d6289ba13e347c335a4c9efa63337a748"
+
 "@types/events@*":
   version "3.0.0"
   resolved "https://registry.npm.taobao.org/@types/events/download/@types/events-3.0.0.tgz#2862f3f58a9a7f7c3e78d79f130dd4d71c25c2a7"
@@ -311,6 +315,12 @@
   version "3.0.3"
   resolved "https://registry.npm.taobao.org/@types/minimatch/download/@types/minimatch-3.0.3.tgz#3dca0e3f33b200fc7d1139c0cd96c1268cadfd9d"
 
+"@types/node-int64@^0.4.29":
+  version "0.4.29"
+  resolved "https://registry.npm.taobao.org/@types/node-int64/download/@types/node-int64-0.4.29.tgz#8c7c16a7c1195ae4f8beaa903b0018ac66291d16"
+  dependencies:
+    "@types/node" "*"
+
 "@types/node@*":
   version "11.11.6"
   resolved "http://registry.npm.taobao.org/@types/node/download/@types/node-11.11.6.tgz#df929d1bb2eee5afdda598a41930fe50b43eaa6a"