creevey 0.8.0-beta.0 → 0.8.0-beta.1

package/lib/cjs/client/shared/creeveyClientApi.js

@@ -9,95 +9,68 @@ var _types = require("../../types");
 
 var _helpers = require("./helpers");
 
-function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
+async function initCreeveyClientApi() {
+  let clientApiResolver = _types.noop;
+  const updateListeners = new Set();
+  let statusRequest = null;
+  let statusResolver = _types.noop;
+  const ws = new WebSocket("ws://".concat((0, _helpers.getConnectionUrl)()));
 
-function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
+  function send(request) {
+    ws.send(JSON.stringify(request));
+  }
 
-function initCreeveyClientApi() {
-  return _initCreeveyClientApi.apply(this, arguments);
-}
+  ws.addEventListener('open', () => {
+    clientApiResolver({
+      start(ids) {
+        send({
+          type: 'start',
+          payload: ids
+        });
+      },
 
-function _initCreeveyClientApi() {
-  _initCreeveyClientApi = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee() {
-    var clientApiResolver, updateListeners, statusRequest, statusResolver, ws, send;
-    return regeneratorRuntime.wrap(function _callee$(_context) {
-      while (1) {
-        switch (_context.prev = _context.next) {
-          case 0:
-            send = function _send(request) {
-              ws.send(JSON.stringify(request));
-            };
+      stop() {
+        send({
+          type: 'stop'
+        });
+      },
 
-            clientApiResolver = _types.noop;
-            updateListeners = new Set();
-            statusRequest = null;
-            statusResolver = _types.noop;
-            ws = new WebSocket("ws://".concat((0, _helpers.getConnectionUrl)()));
-            ws.addEventListener('open', function () {
-              clientApiResolver({
-                start: function start(ids) {
-                  send({
-                    type: 'start',
-                    payload: ids
-                  });
-                },
-                stop: function stop() {
-                  send({
-                    type: 'stop'
-                  });
-                },
-                approve: function approve(id, retry, image) {
-                  send({
-                    type: 'approve',
-                    payload: {
-                      id: id,
-                      retry: retry,
-                      image: image
-                    }
-                  });
-                },
-                onUpdate: function onUpdate(fn) {
-                  updateListeners.add(fn);
-                  return function () {
-                    return updateListeners.delete(fn);
-                  };
-                },
+      approve(id, retry, image) {
+        send({
+          type: 'approve',
+          payload: {
+            id,
+            retry,
+            image
+          }
+        });
+      },
 
-                get status() {
-                  if (statusRequest) return statusRequest;
-                  send({
-                    type: 'status'
-                  });
-                  return statusRequest = new Promise(function (resolve) {
-                    return statusResolver = resolve;
-                  });
-                }
+      onUpdate(fn) {
+        updateListeners.add(fn);
+        return () => updateListeners.delete(fn);
+      },
 
-              });
-            });
-            ws.addEventListener('message', function (message) {
-              var data = JSON.parse(message.data);
-              if (data.type == 'update') updateListeners.forEach(function (fn) {
-                return fn(data.payload);
-              });
+      get status() {
+        if (statusRequest) return statusRequest;
+        send({
+          type: 'status'
+        });
+        return statusRequest = new Promise(resolve => statusResolver = resolve);
+      }
 
-              if (data.type == 'status') {
-                statusResolver(data.payload);
-                statusResolver = _types.noop;
-                statusRequest = null;
-              }
-            }); // TODO Reconnect
+    });
+  });
+  ws.addEventListener('message', message => {
+    const data = JSON.parse(message.data);
+    if (data.type == 'update') updateListeners.forEach(fn => fn(data.payload));
 
-            return _context.abrupt("return", new Promise(function (resolve) {
-              return clientApiResolver = resolve;
-            }));
+    if (data.type == 'status') {
+      statusResolver(data.payload);
+      statusResolver = _types.noop;
+      statusRequest = null;
+    }
+  }); // TODO Reconnect
 
-          case 9:
-          case "end":
-            return _context.stop();
-        }
-      }
-    }, _callee);
-  }));
-  return _initCreeveyClientApi.apply(this, arguments);
+  return new Promise(resolve => clientApiResolver = resolve);
 }