Prevent mobile browsers from sleeping during transfer

client/package.json

@@ -20,6 +20,7 @@
     "bowser": "^2.11.0",
     "classnames": "^2.3.1",
     "fastest-levenshtein": "^1.0.12",
+    "nosleep.js": "^0.12.0",
     "react": "^18.0.0",
     "react-dom": "^18.0.0",
     "react-dropzone": "^14.2.1",

client/src/app/NoSleep.ts

@@ -0,0 +1,3 @@
+import nosleep from "nosleep.js";
+
+export const NoSleep = new nosleep();

client/src/app/components/Dropzone.tsx

@@ -13,6 +13,7 @@ import classNames from "classnames";
 import React from "react";
 import { FileRejection, useDropzone } from "react-dropzone";
 import { Plus } from "tabler-icons-react";
+import { NoSleep } from "../NoSleep";
 import Link from "./Link";
 
 const useStyles = createStyles((theme) => ({
@@ -96,7 +97,10 @@ export default function Dropzone(props: Props) {
   const button = (
     <Group position="center">
       <Button
-        onClick={open}
+        onClick={() => {
+          NoSleep.enable();
+          open();
+        }}
         className={classNames(classes.dropzoneButton)}
         color="tertiary"
       >

client/src/app/components/providers/WormholeProvider.tsx

@@ -2,6 +2,7 @@ import React, { PropsWithChildren, useEffect, useRef, useState } from "react";
 import { useCodeInput } from "../../hooks/useCodeInput";
 import { useError } from "../../hooks/useError";
 import { useRateLimitedState } from "../../hooks/useRateLimitedState";
+import { NoSleep } from "../../NoSleep";
 import { PROGRESS_BAR_MS_PER_UPDATES } from "../../util/constants";
 import { detectErrorType, ErrorTypes } from "../../util/errors";
 import ClientWorker from "../../wormhole/client_worker";
@@ -181,6 +182,12 @@ export default function WormholeProvider(props: Props) {
     );
   }, []);
 
+  useEffect(() => {
+    if (!fileMeta || done) {
+      NoSleep.disable();
+    }
+  }, [fileMeta, done]);
+
   async function sendFile(
     file: File,
     opts?: TransferOptions

client/src/app/components/screens/receive/ReceiveBeginScreen.tsx

@@ -1,10 +1,11 @@
 import { Modal, Space, Text } from "@mantine/core";
-import React from "react";
+import React, { useEffect } from "react";
 import { useCancelModal } from "../../../hooks/useCancelModal";
 import { useCodeInput } from "../../../hooks/useCodeInput";
 import { useCommonStyles } from "../../../hooks/useCommonStyles";
 import { useError } from "../../../hooks/useError";
 import { useWormhole } from "../../../hooks/useWormhole";
+import { NoSleep } from "../../../NoSleep";
 import { detectErrorType } from "../../../util/errors";
 import CodeInput from "../../CodeInput";
 import Content from "../../Content";
@@ -58,11 +59,18 @@ export default function ReceiveBeginScreen({}: Props) {
   const error = useError();
   const codeInput = useCodeInput();
 
+  useEffect(() => {
+    if (error?.error) {
+      NoSleep.disable();
+    }
+  }, [error]);
+
   return (
     <ReceiveBeginScreenContent
       cancelModalOpen={cancelModal}
       onCancelModalClose={() => setCancelModal(false)}
       onSubmit={(code) => {
+        NoSleep.enable();
         if (wormhole) {
           codeInput?.setSubmitting(true);
           wormhole

client/src/app/components/screens/receive/ReceiveConsentScreen.tsx

@@ -6,6 +6,7 @@ import { useError } from "../../../hooks/useError";
 import { useNavigate } from "../../../hooks/useNavigate";
 import { onTabExit, useTabExitWarning } from "../../../hooks/useTabExitWarning";
 import { useWormhole } from "../../../hooks/useWormhole";
+import { NoSleep } from "../../../NoSleep";
 import { detectErrorType } from "../../../util/errors";
 import Content from "../../Content";
 import FileLabel from "../../FileLabel";
@@ -75,6 +76,7 @@ export default function ReceiveConsentScreen({}: Props) {
     <ReceiveConsentScreenContent
       submitting={submitting}
       onAccept={() => {
+        NoSleep.enable();
         setSubmitting(true);
         wormhole?.fileMeta
           ?.accept()

client/src/app/components/screens/send/SendBeginScreen.tsx

@@ -4,6 +4,7 @@ import { FileRejection } from "react-dropzone";
 import { useCancelModal } from "../../../hooks/useCancelModal";
 import { useCommonStyles } from "../../../hooks/useCommonStyles";
 import { useWormhole } from "../../../hooks/useWormhole";
+import { NoSleep } from "../../../NoSleep";
 import Content from "../../Content";
 import Dropzone from "../../Dropzone";
 import Link from "../../Link";
@@ -139,6 +140,10 @@ export default function SendBeginScreen(props: Props) {
         wormhole?.sendFile(files[0]);
       }}
       onReject={(rejections) => {
+        if (rejections.length) {
+          NoSleep.disable();
+        }
+
         if (rejections.length > 1) {
           setModalState("DIRECTORIES_NOT_SUPPORTED");
         } else if (rejections.length == 1) {

client/src/app/index.development.tsx

@@ -15,6 +15,7 @@ import CodeInputProvider from "./components/providers/CodeInputProvider";
 import ErrorProvider from "./components/providers/ErrorProvider";
 import ThemeProvider from "./components/providers/ThemeProvider";
 import WormholeProvider from "./components/providers/WormholeProvider";
+import "./NoSleep";
 
 Sentry.init({
   dsn: process.env["SENTRY_DSN"],

client/src/app/index.production.tsx

@@ -7,6 +7,7 @@ import CodeInputProvider from "./components/providers/CodeInputProvider";
 import ErrorProvider from "./components/providers/ErrorProvider";
 import ThemeProvider from "./components/providers/ThemeProvider";
 import WormholeProvider from "./components/providers/WormholeProvider";
+import "./NoSleep";
 
 ReactDOM.render(
   <CodeInputProvider>