@northern.tech/store 0.3.3 → 0.4.0

package/dist/sockethook.js

@@ -0,0 +1,139 @@
+import {
+  DEVICE_MESSAGE_PROTOCOLS,
+  DEVICE_MESSAGE_TYPES,
+  TIMEOUTS,
+  apiUrl
+} from "./chunk-7C4SQBEE.js";
+import "./chunk-2DYA3X77.js";
+import "./chunk-ZKAH3APE.js";
+import "./chunk-FAJ3I2UQ.js";
+import "./chunk-OHF3YUXX.js";
+
+// src/sockethook.ts
+import { useCallback, useEffect, useRef, useState } from "react";
+import { blobToString, byteArrayToString } from "@northern.tech/utils/helpers";
+import msgpack5 from "msgpack5";
+import Cookies from "universal-cookie";
+var cookies = new Cookies();
+var MessagePack = msgpack5();
+var useSession = ({ onClose, onHealthCheckFailed, onMessageReceived, onNotify, onOpen, token }) => {
+  const [sessionId, setSessionId] = useState();
+  const healthcheckTimeout = useRef();
+  const socketRef = useRef();
+  const sendMessage = useCallback(({ typ, body, props }) => {
+    if (!socketRef.current) {
+      return;
+    }
+    const proto_header = { proto: DEVICE_MESSAGE_PROTOCOLS.Shell, typ, sid: socketRef.current.sessionId, props };
+    const encodedData = MessagePack.encode({ hdr: proto_header, body });
+    socketRef.current.send(encodedData);
+  }, []);
+  const close = useCallback(() => {
+    if (!socketRef.current || socketRef.current?.readyState !== WebSocket.OPEN) {
+      return;
+    }
+    sendMessage({ typ: DEVICE_MESSAGE_TYPES.Stop, body: {}, props: {} });
+    socketRef.current.close();
+    setSessionId();
+  }, [sendMessage]);
+  const healthcheckFailed = useCallback(() => {
+    onHealthCheckFailed();
+    close();
+  }, [close, onHealthCheckFailed]);
+  const onSocketMessage = useCallback(
+    (event) => blobToString(event.data).then((data) => {
+      const {
+        hdr: { props = {}, proto, sid, typ },
+        body
+      } = MessagePack.decode(data);
+      if (proto !== DEVICE_MESSAGE_PROTOCOLS.Shell) {
+        return;
+      }
+      switch (typ) {
+        case DEVICE_MESSAGE_TYPES.New: {
+          if (props.status == WebSocket.CLOSING) {
+            onNotify(`Error: ${byteArrayToString(body)}`);
+            setSessionId();
+            return close();
+          } else {
+            clearTimeout(healthcheckTimeout.current);
+            healthcheckTimeout.current = setTimeout(healthcheckFailed, 65 * TIMEOUTS.oneSecond);
+            socketRef.current.sessionId = sid;
+            return setSessionId(sid);
+          }
+        }
+        case DEVICE_MESSAGE_TYPES.Shell:
+          return onMessageReceived(body);
+        case DEVICE_MESSAGE_TYPES.Stop:
+          return close();
+        case DEVICE_MESSAGE_TYPES.Ping: {
+          if (healthcheckTimeout.current) {
+            clearTimeout(healthcheckTimeout.current);
+          }
+          sendMessage({ typ: DEVICE_MESSAGE_TYPES.Pong });
+          const timeout = parseInt(props.timeout);
+          if (timeout > 0) {
+            healthcheckTimeout.current = setTimeout(healthcheckFailed, timeout * TIMEOUTS.oneSecond);
+          }
+          return;
+        }
+        default:
+          break;
+      }
+    }),
+    [close, healthcheckFailed, onMessageReceived, onNotify, sendMessage]
+  );
+  const onSocketError = useCallback(
+    (error) => {
+      onNotify(`WebSocket error: ${error.message}`);
+      close();
+      clearTimeout(healthcheckTimeout.current);
+    },
+    [close, onNotify]
+  );
+  const onSocketOpen = useCallback(() => {
+    sendMessage({ typ: DEVICE_MESSAGE_TYPES.New, props: {} });
+    onOpen(true);
+  }, [onOpen, sendMessage]);
+  const onSocketClose = useCallback(
+    (e) => {
+      console.log("closing");
+      onClose(e);
+      close();
+    },
+    [close, onClose]
+  );
+  const connect = useCallback(
+    (deviceId) => {
+      const uri = `wss://${window.location.host}${apiUrl.v1}/deviceconnect/devices/${deviceId}/connect`;
+      setSessionId();
+      cookies.set("JWT", token, { path: "/", secure: true, sameSite: "strict", maxAge: 5 });
+      try {
+        socketRef.current = new WebSocket(uri);
+      } catch (error) {
+        console.log(error);
+      }
+    },
+    [token]
+  );
+  useEffect(() => {
+    if (!socketRef.current) {
+      return;
+    }
+    socketRef.current.addEventListener("close", onSocketClose);
+    socketRef.current.addEventListener("error", onSocketError);
+    socketRef.current.addEventListener("message", onSocketMessage);
+    socketRef.current.addEventListener("open", onSocketOpen);
+    return () => {
+      socketRef.current.removeEventListener("close", onSocketClose);
+      socketRef.current.removeEventListener("error", onSocketError);
+      socketRef.current.removeEventListener("message", onSocketMessage);
+      socketRef.current.removeEventListener("open", onSocketOpen);
+    };
+  }, [onSocketClose, onSocketError, onSocketMessage, onSocketOpen, socketRef.current?.readyState]);
+  return [connect, sendMessage, close, socketRef.current?.readyState ?? WebSocket.CLOSED, sessionId];
+};
+export {
+  useSession
+};
+//# sourceMappingURL=sockethook.js.map

package/dist/sockethook.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/sockethook.ts"],"sourcesContent":["// Copyright 2022 Northern.tech AS\n//\n//    Licensed under the Apache License, Version 2.0 (the \"License\");\n//    you may not use this file except in compliance with the License.\n//    You may obtain a copy of the License at\n//\n//        http://www.apache.org/licenses/LICENSE-2.0\n//\n//    Unless required by applicable law or agreed to in writing, software\n//    distributed under the License is distributed on an \"AS IS\" BASIS,\n//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n//    See the License for the specific language governing permissions and\n//    limitations under the License.\n//@ts-nocheck\nimport { useCallback, useEffect, useRef, useState } from 'react';\n\nimport { blobToString, byteArrayToString } from '@northern.tech/utils/helpers';\nimport msgpack5 from 'msgpack5';\nimport Cookies from 'universal-cookie';\n\nimport { DEVICE_MESSAGE_PROTOCOLS as MessageProtocols, DEVICE_MESSAGE_TYPES as MessageTypes, TIMEOUTS, apiUrl } from './constants';\n\nconst cookies = new Cookies();\n\nconst MessagePack = msgpack5();\n\nexport const useSession = ({ onClose, onHealthCheckFailed, onMessageReceived, onNotify, onOpen, token }) => {\n  const [sessionId, setSessionId] = useState();\n  const healthcheckTimeout = useRef();\n  const socketRef = useRef();\n  const sendMessage = useCallback(({ typ, body, props }) => {\n    if (!socketRef.current) {\n      return;\n    }\n    const proto_header = { proto: MessageProtocols.Shell, typ, sid: socketRef.current.sessionId, props };\n    const encodedData = MessagePack.encode({ hdr: proto_header, body });\n    socketRef.current.send(encodedData);\n  }, []);\n\n  const close = useCallback(() => {\n    if (!socketRef.current || socketRef.current?.readyState !== WebSocket.OPEN) {\n      return;\n    }\n    sendMessage({ typ: MessageTypes.Stop, body: {}, props: {} });\n    socketRef.current.close();\n    setSessionId();\n  }, [sendMessage]);\n\n  const healthcheckFailed = useCallback(() => {\n    onHealthCheckFailed();\n    close();\n  }, [close, onHealthCheckFailed]);\n\n  const onSocketMessage = useCallback(\n    event =>\n      blobToString(event.data).then(data => {\n        const {\n          hdr: { props = {}, proto, sid, typ },\n          body\n        } = MessagePack.decode(data);\n        if (proto !== MessageProtocols.Shell) {\n          return;\n        }\n        switch (typ) {\n          case MessageTypes.New: {\n            if (props.status == WebSocket.CLOSING) {\n              onNotify(`Error: ${byteArrayToString(body)}`);\n              setSessionId();\n              return close();\n            } else {\n              clearTimeout(healthcheckTimeout.current);\n              healthcheckTimeout.current = setTimeout(healthcheckFailed, 65 * TIMEOUTS.oneSecond);\n              socketRef.current.sessionId = sid;\n              return setSessionId(sid);\n            }\n          }\n          case MessageTypes.Shell:\n            return onMessageReceived(body);\n          case MessageTypes.Stop:\n            return close();\n          case MessageTypes.Ping: {\n            if (healthcheckTimeout.current) {\n              clearTimeout(healthcheckTimeout.current);\n            }\n            sendMessage({ typ: MessageTypes.Pong });\n            //\n            const timeout = parseInt(props.timeout);\n            if (timeout > 0) {\n              healthcheckTimeout.current = setTimeout(healthcheckFailed, timeout * TIMEOUTS.oneSecond);\n            }\n            return;\n          }\n          default:\n            break;\n        }\n      }),\n    [close, healthcheckFailed, onMessageReceived, onNotify, sendMessage]\n  );\n\n  const onSocketError = useCallback(\n    error => {\n      onNotify(`WebSocket error: ${error.message}`);\n      close();\n      clearTimeout(healthcheckTimeout.current);\n    },\n    [close, onNotify]\n  );\n\n  const onSocketOpen = useCallback(() => {\n    sendMessage({ typ: MessageTypes.New, props: {} });\n    onOpen(true);\n  }, [onOpen, sendMessage]);\n\n  const onSocketClose = useCallback(\n    e => {\n      console.log('closing');\n      onClose(e);\n      close();\n    },\n    [close, onClose]\n  );\n\n  const connect = useCallback(\n    deviceId => {\n      const uri = `wss://${window.location.host}${apiUrl.v1}/deviceconnect/devices/${deviceId}/connect`;\n      setSessionId();\n      cookies.set('JWT', token, { path: '/', secure: true, sameSite: 'strict', maxAge: 5 });\n      try {\n        socketRef.current = new WebSocket(uri);\n      } catch (error) {\n        console.log(error);\n      }\n    },\n    [token]\n  );\n\n  useEffect(() => {\n    if (!socketRef.current) {\n      return;\n    }\n\n    socketRef.current.addEventListener('close', onSocketClose);\n    socketRef.current.addEventListener('error', onSocketError);\n    socketRef.current.addEventListener('message', onSocketMessage);\n    socketRef.current.addEventListener('open', onSocketOpen);\n\n    return () => {\n      socketRef.current.removeEventListener('close', onSocketClose);\n      socketRef.current.removeEventListener('error', onSocketError);\n      socketRef.current.removeEventListener('message', onSocketMessage);\n      socketRef.current.removeEventListener('open', onSocketOpen);\n    };\n  }, [onSocketClose, onSocketError, onSocketMessage, onSocketOpen, socketRef.current?.readyState]);\n\n  return [connect, sendMessage, close, socketRef.current?.readyState ?? WebSocket.CLOSED, sessionId];\n};\n"],"mappings":";;;;;;;;;;;;AAcA,SAAS,aAAa,WAAW,QAAQ,gBAAgB;AAEzD,SAAS,cAAc,yBAAyB;AAChD,OAAO,cAAc;AACrB,OAAO,aAAa;AAIpB,IAAM,UAAU,IAAI,QAAQ;AAE5B,IAAM,cAAc,SAAS;AAEtB,IAAM,aAAa,CAAC,EAAE,SAAS,qBAAqB,mBAAmB,UAAU,QAAQ,MAAM,MAAM;AAC1G,QAAM,CAAC,WAAW,YAAY,IAAI,SAAS;AAC3C,QAAM,qBAAqB,OAAO;AAClC,QAAM,YAAY,OAAO;AACzB,QAAM,cAAc,YAAY,CAAC,EAAE,KAAK,MAAM,MAAM,MAAM;AACxD,QAAI,CAAC,UAAU,SAAS;AACtB;AAAA,IACF;AACA,UAAM,eAAe,EAAE,OAAO,yBAAiB,OAAO,KAAK,KAAK,UAAU,QAAQ,WAAW,MAAM;AACnG,UAAM,cAAc,YAAY,OAAO,EAAE,KAAK,cAAc,KAAK,CAAC;AAClE,cAAU,QAAQ,KAAK,WAAW;AAAA,EACpC,GAAG,CAAC,CAAC;AAEL,QAAM,QAAQ,YAAY,MAAM;AAC9B,QAAI,CAAC,UAAU,WAAW,UAAU,SAAS,eAAe,UAAU,MAAM;AAC1E;AAAA,IACF;AACA,gBAAY,EAAE,KAAK,qBAAa,MAAM,MAAM,CAAC,GAAG,OAAO,CAAC,EAAE,CAAC;AAC3D,cAAU,QAAQ,MAAM;AACxB,iBAAa;AAAA,EACf,GAAG,CAAC,WAAW,CAAC;AAEhB,QAAM,oBAAoB,YAAY,MAAM;AAC1C,wBAAoB;AACpB,UAAM;AAAA,EACR,GAAG,CAAC,OAAO,mBAAmB,CAAC;AAE/B,QAAM,kBAAkB;AAAA,IACtB,WACE,aAAa,MAAM,IAAI,EAAE,KAAK,UAAQ;AACpC,YAAM;AAAA,QACJ,KAAK,EAAE,QAAQ,CAAC,GAAG,OAAO,KAAK,IAAI;AAAA,QACnC;AAAA,MACF,IAAI,YAAY,OAAO,IAAI;AAC3B,UAAI,UAAU,yBAAiB,OAAO;AACpC;AAAA,MACF;AACA,cAAQ,KAAK;AAAA,QACX,KAAK,qBAAa,KAAK;AACrB,cAAI,MAAM,UAAU,UAAU,SAAS;AACrC,qBAAS,UAAU,kBAAkB,IAAI,CAAC,EAAE;AAC5C,yBAAa;AACb,mBAAO,MAAM;AAAA,UACf,OAAO;AACL,yBAAa,mBAAmB,OAAO;AACvC,+BAAmB,UAAU,WAAW,mBAAmB,KAAK,SAAS,SAAS;AAClF,sBAAU,QAAQ,YAAY;AAC9B,mBAAO,aAAa,GAAG;AAAA,UACzB;AAAA,QACF;AAAA,QACA,KAAK,qBAAa;AAChB,iBAAO,kBAAkB,IAAI;AAAA,QAC/B,KAAK,qBAAa;AAChB,iBAAO,MAAM;AAAA,QACf,KAAK,qBAAa,MAAM;AACtB,cAAI,mBAAmB,SAAS;AAC9B,yBAAa,mBAAmB,OAAO;AAAA,UACzC;AACA,sBAAY,EAAE,KAAK,qBAAa,KAAK,CAAC;AAEtC,gBAAM,UAAU,SAAS,MAAM,OAAO;AACtC,cAAI,UAAU,GAAG;AACf,+BAAmB,UAAU,WAAW,mBAAmB,UAAU,SAAS,SAAS;AAAA,UACzF;AACA;AAAA,QACF;AAAA,QACA;AACE;AAAA,MACJ;AAAA,IACF,CAAC;AAAA,IACH,CAAC,OAAO,mBAAmB,mBAAmB,UAAU,WAAW;AAAA,EACrE;AAEA,QAAM,gBAAgB;AAAA,IACpB,WAAS;AACP,eAAS,oBAAoB,MAAM,OAAO,EAAE;AAC5C,YAAM;AACN,mBAAa,mBAAmB,OAAO;AAAA,IACzC;AAAA,IACA,CAAC,OAAO,QAAQ;AAAA,EAClB;AAEA,QAAM,eAAe,YAAY,MAAM;AACrC,gBAAY,EAAE,KAAK,qBAAa,KAAK,OAAO,CAAC,EAAE,CAAC;AAChD,WAAO,IAAI;AAAA,EACb,GAAG,CAAC,QAAQ,WAAW,CAAC;AAExB,QAAM,gBAAgB;AAAA,IACpB,OAAK;AACH,cAAQ,IAAI,SAAS;AACrB,cAAQ,CAAC;AACT,YAAM;AAAA,IACR;AAAA,IACA,CAAC,OAAO,OAAO;AAAA,EACjB;AAEA,QAAM,UAAU;AAAA,IACd,cAAY;AACV,YAAM,MAAM,SAAS,OAAO,SAAS,IAAI,GAAG,OAAO,EAAE,0BAA0B,QAAQ;AACvF,mBAAa;AACb,cAAQ,IAAI,OAAO,OAAO,EAAE,MAAM,KAAK,QAAQ,MAAM,UAAU,UAAU,QAAQ,EAAE,CAAC;AACpF,UAAI;AACF,kBAAU,UAAU,IAAI,UAAU,GAAG;AAAA,MACvC,SAAS,OAAO;AACd,gBAAQ,IAAI,KAAK;AAAA,MACnB;AAAA,IACF;AAAA,IACA,CAAC,KAAK;AAAA,EACR;AAEA,YAAU,MAAM;AACd,QAAI,CAAC,UAAU,SAAS;AACtB;AAAA,IACF;AAEA,cAAU,QAAQ,iBAAiB,SAAS,aAAa;AACzD,cAAU,QAAQ,iBAAiB,SAAS,aAAa;AACzD,cAAU,QAAQ,iBAAiB,WAAW,eAAe;AAC7D,cAAU,QAAQ,iBAAiB,QAAQ,YAAY;AAEvD,WAAO,MAAM;AACX,gBAAU,QAAQ,oBAAoB,SAAS,aAAa;AAC5D,gBAAU,QAAQ,oBAAoB,SAAS,aAAa;AAC5D,gBAAU,QAAQ,oBAAoB,WAAW,eAAe;AAChE,gBAAU,QAAQ,oBAAoB,QAAQ,YAAY;AAAA,IAC5D;AAAA,EACF,GAAG,CAAC,eAAe,eAAe,iBAAiB,cAAc,UAAU,SAAS,UAAU,CAAC;AAE/F,SAAO,CAAC,SAAS,aAAa,OAAO,UAAU,SAAS,cAAc,UAAU,QAAQ,SAAS;AACnG;","names":[]}