@expo/serve-sim 0.1.53 → 0.2.0

package/src/state.ts

@@ -1,6 +1,6 @@
 import { tmpdir } from "os";
 import { join } from "path";
-import { readdirSync, mkdirSync, writeFileSync, renameSync } from "fs";
+import { readdirSync, mkdirSync, writeFileSync, renameSync, readFileSync, unlinkSync } from "fs";
 import type { StreamSettings } from "./stream-settings";
 export type {
   HttpStreamCodec,
@@ -9,16 +9,13 @@ export type {
   WebRtcStreamCodec,
 } from "./stream-settings";
 
-/** Directory where serve-sim stores runtime state. */
-export const STATE_DIR = join(tmpdir(), "serve-sim");
-
-/** Path to the serve-sim server state file (JSON with pid, port, URLs).
- *  @deprecated Use `stateFileForDevice(udid)` for multi-device support. Kept for backward compat. */
-export const STATE_FILE = join(STATE_DIR, "server.json");
+/** Directory where serve-sim stores runtime state. Override with `SERVE_SIM_STATE_DIR`. */
+export function stateDir(): string {
+  return process.env.SERVE_SIM_STATE_DIR || join(tmpdir(), "serve-sim");
+}
 
-/** Per-device state file: `/tmp/serve-sim/server-{udid}.json` */
 export function stateFileForDevice(udid: string): string {
-  return join(STATE_DIR, `server-${udid}.json`);
+  return join(stateDir(), `server-${udid}.json`);
 }
 
 /** Runtime record for a device streamed in-process by a preview server. */
@@ -30,6 +27,24 @@ export interface ServeSimDeviceState {
   streamUrl: string;
   wsUrl: string;
   streamSettings?: StreamSettings;
+  /** Present only under `--require-token`, so local subcommands can reach the gated socket. */
+  token?: string;
+}
+
+/** `--quiet` startup payload. Carries the session token only when the gate is on. */
+export function previewStartupPayload(
+  states: ServeSimDeviceState[],
+  token?: string,
+): Record<string, unknown> {
+  const view = (s: ServeSimDeviceState) => ({
+    url: s.url,
+    streamUrl: s.streamUrl,
+    wsUrl: s.wsUrl,
+    port: s.port,
+    device: s.device,
+  });
+  const base = states.length === 1 ? view(states[0]!) : { devices: states.map(view) };
+  return token ? { ...base, token } : base;
 }
 
 /**
@@ -65,21 +80,33 @@ export function inProcessServeSimState(
  *  Writes atomically (temp file + rename) so a concurrent reader never observes
  *  a truncated or partially-written file. */
 export function writeServeSimState(state: ServeSimDeviceState): void {
-  mkdirSync(STATE_DIR, { recursive: true });
+  mkdirSync(stateDir(), { recursive: true });
   const file = stateFileForDevice(state.device);
   const tmp = `${file}.${process.pid}.tmp`;
-  // WebRTC state can contain short-lived TURN credentials. Keep the file
-  // readable only by the account running serve-sim.
+  // Holds TURN credentials and, when gated, the session token.
   writeFileSync(tmp, JSON.stringify(state, null, 2), { mode: 0o600 });
   renameSync(tmp, file);
 }
 
+export function clearServeSimState(udid: string, ownerPid: number): void {
+  const file = stateFileForDevice(udid);
+  let state: ServeSimDeviceState;
+  try {
+    state = JSON.parse(readFileSync(file, "utf-8")) as ServeSimDeviceState;
+  } catch {
+    return;
+  }
+  if (state.pid !== ownerPid) return;
+  try { unlinkSync(file); } catch {}
+}
+
 /** List all per-device state files in the state directory. */
 export function listStateFiles(): string[] {
   try {
-    return readdirSync(STATE_DIR)
+    const dir = stateDir();
+    return readdirSync(dir)
       .filter((f) => f.startsWith("server-") && f.endsWith(".json"))
-      .map((f) => join(STATE_DIR, f));
+      .map((f) => join(dir, f));
   } catch {
     return [];
   }