'use strict';

Object.defineProperty(exports, "__esModule", {

  value: true

});

var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();

var _class, _temp;

function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }

var _socketTask = new WeakMap();

var WebSocket = (_temp = _class = function () {

  // TODO 更新 binaryType

  // The connection is in the process of closing.

  // The connection is not yet open.

  function WebSocket(url) {

    var _this = this;

    var protocols = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];

    _classCallCheck(this, WebSocket);

    this.binaryType = '';

    this.bufferedAmount = 0;

    this.extensions = '';

    this.onclose = null;

    this.onerror = null;

    this.onmessage = null;

    this.onopen = null;

    this.protocol = '';

    this.readyState = 3;

    if (typeof url !== 'string' || !/(^ws:\/\/)|(^wss:\/\/)/.test(url)) {

      throw new TypeError('Failed to construct \'WebSocket\': The URL \'' + url + '\' is invalid');

    }

    this.url = url;

    this.readyState = WebSocket.CONNECTING;

    var socketTask = wx.connectSocket({

      url: url,

      protocols: Array.isArray(protocols) ? protocols : [protocols],

      tcpNoDelay: true

    });

    _socketTask.set(this, socketTask);

    socketTask.onClose(function (res) {

      _this.readyState = WebSocket.CLOSED;

      if (typeof _this.onclose === 'function') {

        _this.onclose(res);

      }

    });

    socketTask.onMessage(function (res) {

      if (typeof _this.onmessage === 'function') {

        _this.onmessage(res);

      }

    });

    socketTask.onOpen(function () {

      _this.readyState = WebSocket.OPEN;

      if (typeof _this.onopen === 'function') {

        _this.onopen();

      }

    });

    socketTask.onError(function (res) {

      if (typeof _this.onerror === 'function') {

        _this.onerror(new Error(res.errMsg));

      }

    });

    return this;

  } // TODO 小程序内目前获取不到，实际上需要根据服务器选择的 sub-protocol 返回

  // TODO 更新 bufferedAmount

  // The connection is closed or couldn't be opened.

  // The connection is open and ready to communicate.