countdown etc

Author: ds300

index.js

@@ -1,9 +1,9 @@
-const { spawn } = require("child_process")
 const { join } = require("path")
 
 const chalk = require("kleur")
-const { spawnSafeSync } = require("./src/spawnSafeSync")
-const { getDevice } = require("./src/getDevice")
+const { init, adb, adbAsync } = require("./src/adb")
+const { existsSync } = require("fs")
+const { countdown } = require("./src/countdown")
 
 const version = require(join(__dirname, "./package.json")).version
 
@@ -18,31 +18,25 @@ process.stdin.resume()
 process.stdin.setEncoding("utf8")
 
 /**
- * @returns {"1" | "0"}
+ * @returns {boolean}
  */
 function getVisibleTouches() {
-  // @ts-ignore
-  return spawnSafeSync("adb", [
-    "shell",
-    "settings",
-    "get",
-    "system",
-    "show_touches",
-  ]).stdout.toString()
+  return (
+    adb(
+      "shell",
+      "settings",
+      "get",
+      "system",
+      "show_touches",
+    ).stdout.toString() === "1"
+  )
 }
 
 /**
  * @param {boolean} visible
  */
 function setVisibleTouches(visible) {
-  spawnSafeSync("adb", [
-    "shell",
-    "settings",
-    "put",
-    "system",
-    "show_touches",
-    visible ? "1" : "0",
-  ])
+  adb("shell", "settings", "put", "system", "show_touches", visible ? "1" : "0")
 }
 
 const ESCAPE_KEYS = {
@@ -64,12 +58,12 @@ const ENTER_KEYS = {
 
 /**
  * @param {string} outFile
- * @returns {Promise<void>}
+ * @returns {Promise<{escaped: boolean}>}
  */
 function recordVideo(outFile) {
   return new Promise((resolve, reject) => {
     const internalFilePath = `/sdcard/android-screen-recording.mp4`
-    const recordProc = spawn("adb", ["shell", "screenrecord", internalFilePath])
+    const recordProc = adbAsync("shell", "screenrecord", internalFilePath)
     let err = ""
     recordProc.stderr.on("data", (data) => {
       err += data.toString()
@@ -82,13 +76,17 @@ function recordVideo(outFile) {
       }
       process.stdin.removeListener("data", handleKeyPress)
       console.log("\n         ", chalk.green("✔"), "Cut! 🎬", "\n")
-      console.log("Transferring video from phone...\n")
-      await new Promise((r) => setTimeout(r, 2000))
-      spawnSafeSync("adb", ["pull", internalFilePath, outFile])
-      spawnSafeSync("adb", ["shell", "rm", internalFilePath])
-      resolve()
+      if (!escaped) {
+        console.log("Transferring video from phone...\n")
+        await new Promise((r) => setTimeout(r, 2000))
+        adb("pull", internalFilePath, outFile)
+      } else {
+        console.log("Cancelling...")
+      }
+      adb("shell", "rm", internalFilePath)
+      resolve({ escaped })
     })
-    console.log("\n         ", chalk.red("⦿"), "Recording...", "\n")
+    console.log("         ", chalk.red("⦿"), "Recording...", "\n")
     printCallToAction()
 
     // on any data into stdin
@@ -134,24 +132,42 @@ function printCallToAction() {
 }
 
 async function run() {
-  const device = await getDevice()
-  console.log({device})
-  process.exit(1)
-  const outPath = "./recording.mp4"
-  setVisibleTouches(true)
-
   try {
-    await recordVideo(outPath)
-    console.log("That's a wrap!", chalk.bold(outPath))
-    console.log()
+    await init()
+    await countdown()
+
+    const date = new Date()
+    const year = date.getFullYear()
+    const month = date.getMonth() + 1
+    const day = date.getDate()
+
+    let outPath = `./recording.${year}-${month}-${day}.00.mp4`
+    let i = 1
+    while (existsSync(outPath)) {
+      outPath = `./recording.${year}-${month}-${day}.${i
+        .toString()
+        .padStart(2, "0")}.mp4`
+      i++
+    }
+    setVisibleTouches(true)
+
+    try {
+      const result = await recordVideo(outPath)
+      if (!result.escaped) {
+        console.log("That's a wrap!", chalk.bold(outPath))
+      }
+      console.log()
+    } catch (e) {
+      console.error("failed in main loop", e)
+    }
+
+    setVisibleTouches(false)
+
+    process.exit(0)
   } catch (e) {
     console.error(e)
+    process.exit(1)
   }
-
-  setVisibleTouches(false)
-
-  process.exit(0)
 }
 
 run()
-// getVisibleTouches()

src/adb.js

@@ -0,0 +1,15 @@
+const { spawn } = require("child_process")
+const { getDevice } = require("./getDevice")
+const { spawnSafeSync } = require("./spawnSafeSync")
+
+let device = ""
+
+module.exports.adb = (/** @type {string[]} */ ...args) =>
+  spawnSafeSync("adb", ["-s", device, ...args])
+
+module.exports.adbAsync = (/** @type {string[]} */ ...args) =>
+  spawn("adb", ["-s", device, ...args])
+
+module.exports.init = async () => {
+  device = await getDevice()
+}

src/countdown.js

@@ -0,0 +1,26 @@
+const countdownString = "3...2...1..."
+
+module.exports.countdown = async () => {
+  return new Promise((r) => {
+    const startTime = Date.now()
+    function tick() {
+      const now = Date.now()
+      const elapsed = now - startTime
+      const ratio = elapsed / 3000
+      const numCharsOfCountdownStringToUse = Math.floor(
+        countdownString.length * ratio,
+      )
+      const countdownSoFar = countdownString.slice(
+        0,
+        numCharsOfCountdownStringToUse,
+      )
+      process.stdout.write(`    🎬 Get ready! ${countdownSoFar}\r`)
+      if (ratio >= 1) {
+        process.stdout.write("\033[K")
+        clearInterval(interval)
+        r(null)
+      }
+    }
+    const interval = setInterval(tick, 50)
+  })
+}

src/getDevice.js

@@ -1,5 +1,5 @@
 const { spawnSafeSync } = require("./spawnSafeSync")
-const { bgRed, bold, bgBlue, cyan, gray, magenta } = require("kleur")
+const { bgRed, bold, bgBlue, cyan, gray } = require("kleur")
 
 const ERROR = bgRed(" ERROR ")
 
@@ -57,7 +57,7 @@ function parseDeviceLine(line) {
  * @returns {Promise<number>}
  */
 async function selectOption(title, options) {
-  return new Promise((resolve, reject) => {
+  return new Promise((resolve) => {
     console.log()
     console.log("        ", bgBlue(` ${title} `))
     console.log()
@@ -78,6 +78,7 @@ async function selectOption(title, options) {
     const reset = () => {
       process.stdout.write("\r\033[K")
       process.stdout.write("\033[1A\r")
+      process.stdout.write("\033[K")
       process.stdout.write("\033[1A\r")
       for (let i = 0; i < options.length; i++) {
         process.stdout.write("\033[1A\r")
@@ -90,15 +91,22 @@ async function selectOption(title, options) {
      * @param {string} key
      */
     function handleKeyPress(key) {
-      key = key.toString()
       if (key === "\r") {
         // enter!
-        process.stdout.removeListener("data", handleKeyPress)
+        process.stdin.removeListener("data", handleKeyPress)
+
+        reset()
+        process.stdout.write("\033[1A\r")
+        process.stdout.write("\033[K")
+        process.stdout.write("\033[1A\r")
+        process.stdout.write("\033[K")
+
         resolve(selection)
       } else if (key === "\u001b[B") {
         // arrow down!
         selection = Math.min(selection + 1, options.length - 1)
         reset()
+
         draw()
       } else if (key === "\u001b[A") {
         // arrow up!
@@ -109,7 +117,7 @@ async function selectOption(title, options) {
         process.exit(1)
       }
     }
-    process.stdout.on("data", handleKeyPress)
+    process.stdin.on("data", handleKeyPress)
   })
 }
 

src/spawnSafeSync.js

@@ -10,6 +10,16 @@ const defaultOptions = {
 
 const { spawnSync } = require("child_process")
 
+class ConsoleError {
+  /**
+   * @param {import('node:child_process').SpawnSyncReturns<Buffer>} result
+   */
+  constructor(result) {
+    this.result = result
+  }
+}
+module.exports.ConsoleError = ConsoleError
+
 module.exports.spawnSafeSync = (
   /** @type {string} */ command,
   /** @type {string[]} */ args,
@@ -26,7 +36,7 @@ module.exports.spawnSafeSync = (
       }
     }
     if (mergedOptions.throwOnError) {
-      throw result
+      throw new ConsoleError(result)
     }
   }
   return result