one/dist/esm/daemon/server.native.js

One is a new React Framework that makes Vite serve both native and web.

import * as http from "http";
import * as fs from "fs";
import * as os from "os";
import * as path from "path";
import { createRegistry, findServersByBundleId, findServerById, getAllServers, getRoute, setRoute, clearRoute, touchServer, pruneDeadServers, checkServerAlive, registerServer } from "./registry.native.js";
import { createIPCServer, getSocketPath, cleanupSocket, readServerFiles } from "./ipc.native.js";
import { proxyHttpRequest, proxyWebSocket } from "./proxy.native.js";
import { getBootedSimulators, resolvePendingPicker } from "./picker.native.js";
import colors from "picocolors";
function _type_of(obj) {
  "@swc/helpers - typeof";

  return obj && typeof Symbol < "u" && obj.constructor === Symbol ? "symbol" : typeof obj;
}
var debugLogPath = path.join(os.homedir(), ".one", "daemon-debug.log");
function debugLog(msg) {
  fs.appendFileSync(debugLogPath, `${(/* @__PURE__ */new Date()).toISOString()} ${msg}
`);
}
var serverAppNames = /* @__PURE__ */new Map();
async function getAppNameForServer(root) {
  var cached = serverAppNames.get(root);
  if (cached) return cached;
  try {
    var appJsonPath = path.join(root, "app.json");
    if (fs.existsSync(appJsonPath)) {
      var _content_expo,
        content = JSON.parse(fs.readFileSync(appJsonPath, "utf-8")),
        name = ((_content_expo = content.expo) === null || _content_expo === void 0 ? void 0 : _content_expo.name) || content.name;
      if (name) return serverAppNames.set(root, name), name;
    }
    var appConfigTsPath = path.join(root, "app.config.ts");
    if (fs.existsSync(appConfigTsPath)) {
      var content1 = fs.readFileSync(appConfigTsPath, "utf-8"),
        nameMatch = content1.match(/name:\s*['"]([^'"]+)['"]/);
      if (nameMatch) {
        var name1 = nameMatch[1];
        return serverAppNames.set(root, name1), name1;
      }
    }
    var appConfigJsPath = path.join(root, "app.config.js");
    if (fs.existsSync(appConfigJsPath)) {
      var content2 = fs.readFileSync(appConfigJsPath, "utf-8"),
        nameMatch1 = content2.match(/name:\s*['"]([^'"]+)['"]/);
      if (nameMatch1) {
        var name2 = nameMatch1[1];
        return serverAppNames.set(root, name2), name2;
      }
    }
    var dirName = path.basename(root);
    if (dirName) return serverAppNames.set(root, dirName), dirName;
  } catch (err) {
    debugLog(`Failed to get app name for ${root}: ${err}`);
  }
  return null;
}
var clientMappings = /* @__PURE__ */new Map(),
  MAPPING_TTL_MS = 36e5;
setInterval(function () {
  var now = Date.now(),
    cleaned = 0,
    _iteratorNormalCompletion = !0,
    _didIteratorError = !1,
    _iteratorError = void 0;
  try {
    for (var _iterator = clientMappings[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
      var [key, info] = _step.value;
      now - info.lastUsed > MAPPING_TTL_MS && (clientMappings.delete(key), cleaned++);
    }
  } catch (err) {
    _didIteratorError = !0, _iteratorError = err;
  } finally {
    try {
      !_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
    } finally {
      if (_didIteratorError) throw _iteratorError;
    }
  }
  cleaned > 0 && debugLog(`Cleaned ${cleaned} stale mappings`);
}, 3e4);
function getRouteIdFromCookies(req) {
  var cookieHeader = req.headers.cookie;
  if (!cookieHeader) return null;
  var match = cookieHeader.match(/one-route-id=([^;]+)/);
  return match ? match[1] : null;
}
var pendingMappings = /* @__PURE__ */new Map();
function setPendingMapping(serverId, simulatorUdid) {
  pendingMappings.set(serverId, simulatorUdid), debugLog(`Pending mapping: next request to server ${serverId} will map to sim ${simulatorUdid}`);
}
function clearMappingsForSimulator(simulatorUdid) {
  var count = 0,
    _iteratorNormalCompletion = !0,
    _didIteratorError = !1,
    _iteratorError = void 0;
  try {
    for (var _iterator = clientMappings[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
      var [key, info] = _step.value;
      info.simulatorUdid === simulatorUdid && (clientMappings.delete(key), count++);
    }
  } catch (err) {
    _didIteratorError = !0, _iteratorError = err;
  } finally {
    try {
      !_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
    } finally {
      if (_didIteratorError) throw _iteratorError;
    }
  }
  debugLog(`Cleared ${count} mappings for simulator ${simulatorUdid}`);
}
function clearAllMappings() {
  var count = clientMappings.size;
  clientMappings.clear(), debugLog(`Cleared all ${count} client mappings`);
}
function getSimulatorMappings() {
  var result = /* @__PURE__ */new Map(),
    _iteratorNormalCompletion = !0,
    _didIteratorError = !1,
    _iteratorError = void 0;
  try {
    for (var _iterator = clientMappings[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
      var [_key, info] = _step.value;
      info.simulatorUdid && result.set(info.simulatorUdid, info.serverId);
    }
  } catch (err) {
    _didIteratorError = !0, _iteratorError = err;
  } finally {
    try {
      !_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
    } finally {
      if (_didIteratorError) throw _iteratorError;
    }
  }
  return result;
}
function setSimulatorMapping(simulatorUdid, serverId) {
  var key = `tui:${simulatorUdid}`;
  clientMappings.set(key, {
    serverId,
    simulatorUdid,
    matchedBy: "tui",
    lastUsed: Date.now()
  }), debugLog(`TUI set mapping: sim=${simulatorUdid} -> server=${serverId}`);
}
async function matchUserAgentToServer(headers, servers) {
  var userAgent = headers["user-agent"];
  if (!userAgent || typeof userAgent != "string") return null;
  var uaAppName = userAgent.split("/")[0];
  if (!uaAppName) return null;
  debugLog(`Trying to match user-agent app "${uaAppName}" to servers`);
  var _iteratorNormalCompletion = !0,
    _didIteratorError = !1,
    _iteratorError = void 0;
  try {
    for (var _iterator = servers[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
      var server = _step.value,
        appName = await getAppNameForServer(server.root);
      if (appName) {
        var normalizedUa = uaAppName.toLowerCase().replace(/\s+/g, ""),
          normalizedApp = appName.toLowerCase().replace(/\s+/g, "");
        if (debugLog(`  Comparing "${normalizedUa}" to server app "${normalizedApp}"`), normalizedUa === normalizedApp || normalizedUa.includes(normalizedApp) || normalizedApp.includes(normalizedUa)) return debugLog(`  Matched! ${uaAppName} -> ${server.root}`), server;
      }
    }
  } catch (err) {
    _didIteratorError = !0, _iteratorError = err;
  } finally {
    try {
      !_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
    } finally {
      if (_didIteratorError) throw _iteratorError;
    }
  }
  return null;
}
function isGenericExpoAgent(ua) {
  return ua.startsWith("Expo/") || ua.startsWith("Exponent/");
}
function extractAppNameFromUA(ua) {
  var firstPart = ua.split(" ")[0] || "",
    appName = firstPart.split("/")[0];
  return appName || null;
}
var recentConnections = /* @__PURE__ */new Map(),
  CONNECTION_MEMORY_MS = 5e3;
function getPrimaryIdentifier(headers) {
  var userAgent = headers["user-agent"] || "";
  if (!isGenericExpoAgent(userAgent)) {
    var appName = extractAppNameFromUA(userAgent);
    if (appName) return `app:${appName}`;
  }
  var easClientId = headers["eas-client-id"];
  return easClientId && typeof easClientId == "string" ? `eas:${easClientId}` : userAgent ? `ua:${userAgent}` : null;
}
function lookupClient(headers) {
  var identifier = getPrimaryIdentifier(headers);
  if (!identifier) return {
    info: null,
    identifier: null
  };
  var info = clientMappings.get(identifier);
  return info && (info.lastUsed = Date.now()), {
    info: info || null,
    identifier
  };
}
function saveClientMapping(identifier, serverId, simulatorUdid, matchedBy) {
  clientMappings.set(identifier, {
    serverId,
    simulatorUdid,
    matchedBy,
    lastUsed: Date.now()
  }), debugLog(`Saved mapping: ${identifier} -> server=${serverId}, sim=${simulatorUdid || "unknown"}, via=${matchedBy}`);
}
var DEFAULT_PORT = 8081,
  routeModeOverride = null;
function setRouteMode(mode) {
  routeModeOverride = mode;
}
var activeDaemonState = null;
async function inferSimulator(clientInfo) {
  var _unmappedSims_, _simulators_;
  if (clientInfo?.simulatorUdid) return clientInfo.simulatorUdid;
  var simulators = await getBootedSimulators(),
    existingMappings = getSimulatorMappings(),
    unmappedSims = simulators.filter(function (s) {
      return !existingMappings.has(s.udid);
    });
  return ((_unmappedSims_ = unmappedSims[0]) === null || _unmappedSims_ === void 0 ? void 0 : _unmappedSims_.udid) || ((_simulators_ = simulators[0]) === null || _simulators_ === void 0 ? void 0 : _simulators_.udid);
}
async function resolveServer(state, headers, servers, bundleId) {
  var {
    info: clientInfo,
    identifier
  } = lookupClient(headers);
  debugLog(`resolveServer: identifier=${identifier}, clientInfo=${JSON.stringify(clientInfo)}`);
  var learnMapping = async function (server4, matchedBy) {
    if (identifier && !clientInfo?.simulatorUdid) {
      var simUdid = await inferSimulator(clientInfo);
      if (simUdid) return saveClientMapping(identifier, server4.id, simUdid, matchedBy), !0;
    }
    return !1;
  };
  if (servers.length === 1) {
    var server = servers[0],
      learned = await learnMapping(server, "auto");
    return debugLog(`Single server: ${server.root}`), {
      server,
      learned
    };
  }
  if (pendingMappings.size > 0 && identifier) {
    var _iteratorNormalCompletion = !0,
      _didIteratorError = !1,
      _iteratorError = void 0;
    try {
      for (var _loop = function () {
          var [serverId, simUdid] = _step.value,
            server4 = findServerById(state, serverId);
          if (server4 && servers.some(function (s) {
            return s.id === serverId;
          })) return debugLog(`TUI pending mapping: ${server4.root}, sim=${simUdid}`), saveClientMapping(identifier, serverId, simUdid, "tui"), pendingMappings.delete(serverId), {
            v: {
              server: server4,
              learned: !0
            }
          };
        }, _iterator = pendingMappings[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
        var _ret = _loop();
        if (_type_of(_ret) === "object") return _ret.v;
      }
    } catch (err) {
      _didIteratorError = !0, _iteratorError = err;
    } finally {
      try {
        !_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
      } finally {
        if (_didIteratorError) throw _iteratorError;
      }
    }
  }
  if (clientInfo?.simulatorUdid) {
    var simRoute = getRoute(state, `sim:${clientInfo.simulatorUdid}`);
    if (simRoute) {
      var server1 = findServerById(state, simRoute.serverId);
      if (server1) return debugLog(`TUI cable route: sim=${clientInfo.simulatorUdid} -> ${server1.root}`), {
        server: server1,
        learned: !1
      };
    }
  }
  if (clientInfo?.serverId) {
    var server2 = findServerById(state, clientInfo.serverId);
    if (server2) return debugLog(`Cached mapping: ${identifier} -> ${server2.root}`), {
      server: server2,
      learned: !1
    };
  }
  var userAgent = headers["user-agent"] || "";
  if (!isGenericExpoAgent(userAgent)) {
    var matchedServer = await matchUserAgentToServer(headers, servers);
    if (matchedServer) return debugLog(`UA match: ${extractAppNameFromUA(userAgent)} -> ${matchedServer.root}`), await learnMapping(matchedServer, "user-agent"), {
      server: matchedServer,
      learned: !0
    };
  }
  var routeKey = bundleId || "default",
    fallbackRoute = getRoute(state, bundleId || "") || getRoute(state, "default");
  if (fallbackRoute) {
    var server3 = findServerById(state, fallbackRoute.serverId);
    if (server3) return debugLog(`Fallback route: ${server3.root}`), await learnMapping(server3, "auto"), {
      server: server3,
      learned: !0
    };
  }
  var mostRecent = [...servers].sort(function (a, b) {
    return b.registeredAt - a.registeredAt;
  })[0];
  return debugLog(`Most recent fallback: ${mostRecent.root}`), setRoute(state, routeKey, mostRecent.id), await learnMapping(mostRecent, "auto"), {
    server: mostRecent,
    learned: !0
  };
}
function proxyAndTouch(req, res, server) {
  var pendingSimId = pendingMappings.get(server.id);
  if (pendingSimId) {
    var identifier = getPrimaryIdentifier(req.headers);
    identifier && (saveClientMapping(identifier, server.id, pendingSimId, "tui"), pendingMappings.delete(server.id));
  }
  activeDaemonState && touchServer(activeDaemonState, server.id), proxyHttpRequest(req, res, server);
}
async function startDaemon() {
  var options = arguments.length > 0 && arguments[0] !== void 0 ? arguments[0] : {},
    port = options.port || DEFAULT_PORT,
    host = options.host || "0.0.0.0",
    quiet = options.quiet || !1,
    log = quiet ? function () {
      for (var _len = arguments.length, _args = new Array(_len), _key = 0; _key < _len; _key++) _args[_key] = arguments[_key];
    } : console.log,
    state = createRegistry();
  activeDaemonState = state;
  var persistedServers = readServerFiles(),
    _iteratorNormalCompletion = !0,
    _didIteratorError = !1,
    _iteratorError = void 0;
  try {
    for (var _iterator = persistedServers[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
      var ps = _step.value,
        alive = await checkServerAlive({
          port: ps.port
        });
      alive && (registerServer(state, {
        port: ps.port,
        bundleId: ps.bundleId,
        root: ps.root
      }), log(colors.cyan(`[daemon] Recovered server: ${ps.bundleId} \u2192 :${ps.port}`)));
    }
  } catch (err) {
    _didIteratorError = !0, _iteratorError = err;
  } finally {
    try {
      !_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
    } finally {
      if (_didIteratorError) throw _iteratorError;
    }
  }
  var ipcServer = createIPCServer(state, function (id) {
      var server = findServerById(state, id);
      if (server) {
        var shortRoot = server.root.replace(process.env.HOME || "", "~");
        log(colors.green(`[daemon] Server registered: ${server.bundleId} \u2192 :${server.port} (${shortRoot})`));
      }
    }, function (id) {
      log(colors.yellow(`[daemon] Server unregistered: ${id}`));
    }),
    httpServer = http.createServer(async function (req, res) {
      var _req_url, _req_socket;
      if (debugLog(`${req.method} ${req.url}`), !((_req_url = req.url) === null || _req_url === void 0) && _req_url.startsWith("/__daemon")) {
        await handleDaemonEndpoint(req, res, state);
        return;
      }
      var url = new URL(req.url || "/", `http://${req.headers.host}`),
        bundleId = url.searchParams.get("app"),
        servers = bundleId ? findServersByBundleId(state, bundleId) : getAllServers(state);
      if (servers.length === 0) {
        res.writeHead(404), res.end(bundleId ? `No server for app: ${bundleId}` : "No servers registered");
        return;
      }
      var {
        server
      } = await resolveServer(state, req.headers, servers, bundleId);
      res.setHeader("Set-Cookie", `one-route-id=${server.id}; Path=/; Max-Age=3600`);
      var remotePort = (_req_socket = req.socket) === null || _req_socket === void 0 ? void 0 : _req_socket.remotePort;
      remotePort && (recentConnections.set(remotePort, {
        serverId: server.id,
        timestamp: Date.now()
      }), debugLog(`HTTP: port ${remotePort} -> ${server.root}`)), proxyAndTouch(req, res, server);
    });
  httpServer.on("upgrade", async function (req, rawSocket, head) {
    var socket = rawSocket,
      url = new URL(req.url || "/", `http://${req.headers.host}`),
      bundleId = url.searchParams.get("app"),
      servers = bundleId ? findServersByBundleId(state, bundleId) : getAllServers(state);
    if (servers.length === 0) {
      socket.end(`HTTP/1.1 404 Not Found\r
\r
`);
      return;
    }
    var server,
      routeIdFromCookie = getRouteIdFromCookies(req);
    if (routeIdFromCookie && (server = findServerById(state, routeIdFromCookie), server && servers.some(function (s) {
      return s.id === server.id;
    }) ? debugLog(`WebSocket: cookie route -> ${server.root}`) : server = void 0), !server) {
      var _req_socket,
        remotePort = (_req_socket = req.socket) === null || _req_socket === void 0 ? void 0 : _req_socket.remotePort;
      if (remotePort) {
        var recent = recentConnections.get(remotePort);
        recent && Date.now() - recent.timestamp < CONNECTION_MEMORY_MS && (server = findServerById(state, recent.serverId), server && debugLog(`WebSocket: port ${remotePort} matched to ${server.root}`));
      }
    }
    if (!server) {
      var result = await resolveServer(state, req.headers, servers, bundleId);
      server = result.server, debugLog(`WebSocket: fallback -> ${server.root}`);
    }
    touchServer(state, server.id), proxyWebSocket(req, socket, head, server);
  }), httpServer.listen(port, host, function () {
    log(colors.cyan(`
\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550`)), log(colors.cyan("  one daemon")), log(colors.cyan("\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550")), log(`
  Listening on ${colors.green(`http://${host === "0.0.0.0" ? "localhost" : host}:${port}`)}`), log(`  IPC socket:  ${colors.dim(getSocketPath())}`), log(""), log(colors.dim("  Waiting for dev servers to register...")), log(colors.dim("  Run 'one dev' in your project directories")), log("");
  });
  var HEALTH_CHECK_INTERVAL = 5e3,
    healthCheckInterval = setInterval(async function () {
      var prunedCount = await pruneDeadServers(state, function (server) {
        log(colors.yellow(`[daemon] Pruned dead server: ${server.bundleId} (port ${server.port})`));
      });
      prunedCount > 0 && log(colors.dim(`[daemon] Pruned ${prunedCount} dead server(s)`));
    }, HEALTH_CHECK_INTERVAL),
    shutdown = function () {
      log(colors.yellow(`
[daemon] Shutting down...`)), clearInterval(healthCheckInterval), httpServer.close(), ipcServer.close(), cleanupSocket(), process.exit(0);
    };
  return process.on("SIGINT", shutdown), process.on("SIGTERM", shutdown), {
    httpServer,
    ipcServer,
    state,
    shutdown,
    healthCheckInterval
  };
}
async function handleDaemonEndpoint(req, res, state) {
  var url = new URL(req.url || "/", `http://${req.headers.host}`);
  if (url.pathname === "/__daemon/status") {
    var servers = getAllServers(state),
      simulators = await getBootedSimulators(),
      simMappings = getSimulatorMappings(),
      simulatorRoutes = {},
      _iteratorNormalCompletion = !0,
      _didIteratorError = !1,
      _iteratorError = void 0;
    try {
      for (var _iterator = simMappings[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
        var [udid, serverId] = _step.value;
        simulatorRoutes[udid] = serverId;
      }
    } catch (err) {
      _didIteratorError = !0, _iteratorError = err;
    } finally {
      try {
        !_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
      } finally {
        if (_didIteratorError) throw _iteratorError;
      }
    }
    res.writeHead(200, {
      "Content-Type": "application/json"
    }), res.end(JSON.stringify({
      servers: servers.map(function (s) {
        return {
          id: s.id,
          port: s.port,
          bundleId: s.bundleId,
          root: s.root
        };
      }),
      simulators,
      simulatorRoutes,
      routeMode: routeModeOverride || "most-recent"
    }, null, 2));
    return;
  }
  if (url.pathname === "/__daemon/route" && req.method === "POST") {
    var bundleId = url.searchParams.get("bundleId"),
      serverId1 = url.searchParams.get("serverId");
    if (!bundleId || !serverId1) {
      res.writeHead(400), res.end("Missing bundleId or serverId");
      return;
    }
    var server = findServerById(state, serverId1);
    if (!server) {
      res.writeHead(404), res.end("Server not found");
      return;
    }
    setRoute(state, bundleId, serverId1), resolvePendingPicker(bundleId, serverId1), res.writeHead(200), res.end("Route set");
    return;
  }
  if (url.pathname === "/__daemon/simulator-route" && req.method === "POST") {
    var simulatorUdid = url.searchParams.get("simulatorUdid"),
      serverId2 = url.searchParams.get("serverId");
    if (!simulatorUdid || !serverId2) {
      res.writeHead(400), res.end("Missing simulatorUdid or serverId");
      return;
    }
    var server1 = findServerById(state, serverId2);
    if (!server1) {
      res.writeHead(404), res.end("Server not found");
      return;
    }
    setSimulatorMapping(simulatorUdid, serverId2), setPendingMapping(serverId2, simulatorUdid), setRoute(state, `sim:${simulatorUdid}`, serverId2), res.writeHead(200, {
      "Content-Type": "application/json"
    }), res.end(JSON.stringify({
      ok: !0
    }));
    return;
  }
  if (url.pathname === "/__daemon/simulator-route" && req.method === "DELETE") {
    var simulatorUdid1 = url.searchParams.get("simulatorUdid");
    if (!simulatorUdid1) {
      res.writeHead(400), res.end("Missing simulatorUdid");
      return;
    }
    clearMappingsForSimulator(simulatorUdid1), clearRoute(state, `sim:${simulatorUdid1}`), res.writeHead(200, {
      "Content-Type": "application/json"
    }), res.end(JSON.stringify({
      ok: !0
    }));
    return;
  }
  res.writeHead(404), res.end("Not found");
}
export { clearAllMappings, clearMappingsForSimulator, getSimulatorMappings, setPendingMapping, setRouteMode, setSimulatorMapping, startDaemon };
//# sourceMappingURL=server.native.js.map