one
Version:
One is a new React Framework that makes Vite serve both native and web.
236 lines (235 loc) • 7.21 kB
JavaScript
import * as net from "node:net";
import * as fs from "node:fs";
import * as path from "node:path";
import * as os from "node:os";
import {
registerServer,
unregisterServer,
getAllServers,
getAllRoutes,
setRoute,
clearRoute,
findServerById,
touchServer,
getLastActiveServer
} from "./registry";
const SOCKET_DIR = path.join(os.homedir(), ".one"), SOCKET_PATH = path.join(SOCKET_DIR, "daemon.sock"), SERVERS_FILE = path.join(SOCKET_DIR, "servers.json");
function getSocketPath() {
return SOCKET_PATH;
}
function getServersFilePath() {
return SERVERS_FILE;
}
function ensureSocketDir() {
fs.existsSync(SOCKET_DIR) || fs.mkdirSync(SOCKET_DIR, { recursive: !0 });
}
function writeServerFile(server) {
ensureSocketDir();
const filtered = readServerFiles().filter((s) => s.root !== server.root);
filtered.push(server), fs.writeFileSync(SERVERS_FILE, JSON.stringify(filtered, null, 2));
}
function removeServerFile(root) {
const filtered = readServerFiles().filter((s) => s.root !== root);
fs.writeFileSync(SERVERS_FILE, JSON.stringify(filtered, null, 2));
}
function readServerFiles() {
try {
if (fs.existsSync(SERVERS_FILE))
return JSON.parse(fs.readFileSync(SERVERS_FILE, "utf-8"));
} catch {
}
return [];
}
function cleanupSocket() {
try {
fs.existsSync(SOCKET_PATH) && fs.unlinkSync(SOCKET_PATH);
} catch {
}
}
function createIPCServer(state, onServerRegistered, onServerUnregistered) {
ensureSocketDir(), cleanupSocket();
const server = net.createServer((socket) => {
let buffer = "";
socket.on("data", (data) => {
buffer += data.toString();
const lines = buffer.split(`
`);
buffer = lines.pop() || "";
for (const line of lines)
if (line.trim())
try {
const message = JSON.parse(line), response = handleMessage(
state,
message,
onServerRegistered,
onServerUnregistered
);
socket.write(JSON.stringify(response) + `
`);
} catch (err) {
const errResponse = {
type: "error",
message: err instanceof Error ? err.message : "Unknown error"
};
socket.write(JSON.stringify(errResponse) + `
`);
}
}), socket.on("error", (err) => {
console.error("[daemon] IPC socket error:", err.message);
});
});
return server.listen(SOCKET_PATH, () => {
fs.chmodSync(SOCKET_PATH, 384);
}), server;
}
function handleMessage(state, message, onServerRegistered, onServerUnregistered) {
switch (message.type) {
case "register": {
const registration = registerServer(state, {
port: message.port,
bundleId: message.bundleId,
root: message.root
});
return onServerRegistered?.(registration.id), { type: "registered", id: registration.id };
}
case "unregister":
return unregisterServer(state, message.id), onServerUnregistered?.(message.id), { type: "unregistered" };
case "route":
return findServerById(state, message.serverId) ? (setRoute(state, message.bundleId, message.serverId), { type: "routed" }) : { type: "error", message: `Server not found: ${message.serverId}` };
case "route-clear":
return clearRoute(state, message.bundleId), { type: "routed" };
case "status":
return {
type: "status",
servers: getAllServers(state),
routes: getAllRoutes(state)
};
case "ping":
return { type: "pong" };
case "touch":
return touchServer(state, message.id) ? { type: "touched" } : { type: "error", message: `Server not found: ${message.id}` };
case "get-last-active":
return { type: "last-active", server: getLastActiveServer(state) };
default:
return { type: "error", message: "Unknown message type" };
}
}
async function isDaemonRunning() {
return new Promise((resolve) => {
if (!fs.existsSync(SOCKET_PATH)) {
resolve(!1);
return;
}
const client = net.connect(SOCKET_PATH), timeout = setTimeout(() => {
client.destroy(), resolve(!1);
}, 1e3);
client.on("connect", () => {
clearTimeout(timeout), client.write(JSON.stringify({ type: "ping" }) + `
`);
}), client.on("data", (data) => {
clearTimeout(timeout);
try {
const response = JSON.parse(data.toString().trim());
resolve(response.type === "pong");
} catch {
resolve(!1);
}
client.destroy();
}), client.on("error", () => {
clearTimeout(timeout), resolve(!1);
});
});
}
async function sendIPCMessage(message) {
return new Promise((resolve, reject) => {
const client = net.connect(SOCKET_PATH);
let buffer = "";
const timeout = setTimeout(() => {
client.destroy(), reject(new Error("IPC timeout"));
}, 5e3);
client.on("connect", () => {
client.write(JSON.stringify(message) + `
`);
}), client.on("data", (data) => {
buffer += data.toString();
const lines = buffer.split(`
`);
buffer = lines.pop() || "";
for (const line of lines)
if (line.trim())
try {
clearTimeout(timeout);
const response = JSON.parse(line);
client.destroy(), resolve(response);
return;
} catch {
}
}), client.on("error", (err) => {
clearTimeout(timeout), reject(err);
});
});
}
async function registerWithDaemon(opts) {
const response = await sendIPCMessage({
type: "register",
...opts
});
if (response.type === "registered")
return response.id;
throw response.type === "error" ? new Error(response.message) : new Error("Unexpected response from daemon");
}
async function unregisterFromDaemon(id) {
await sendIPCMessage({ type: "unregister", id });
}
async function getDaemonStatus() {
const response = await sendIPCMessage({ type: "status" });
if (response.type === "status")
return {
servers: response.servers,
routes: response.routes
};
throw new Error("Failed to get daemon status");
}
async function setDaemonRoute(bundleId, serverId) {
const response = await sendIPCMessage({
type: "route",
bundleId,
serverId
});
if (response.type === "error")
throw new Error(response.message);
}
async function clearDaemonRoute(bundleId) {
await sendIPCMessage({ type: "route-clear", bundleId });
}
async function touchDaemonServer(id) {
const response = await sendIPCMessage({ type: "touch", id });
if (response.type === "error")
throw new Error(response.message);
}
async function getLastActiveDaemonServer() {
const response = await sendIPCMessage({ type: "get-last-active" });
if (response.type === "last-active")
return response.server;
throw new Error("Failed to get last active server");
}
export {
cleanupSocket,
clearDaemonRoute,
createIPCServer,
ensureSocketDir,
getDaemonStatus,
getLastActiveDaemonServer,
getServersFilePath,
getSocketPath,
isDaemonRunning,
readServerFiles,
registerWithDaemon,
removeServerFile,
sendIPCMessage,
setDaemonRoute,
touchDaemonServer,
unregisterFromDaemon,
writeServerFile
};
//# sourceMappingURL=ipc.js.map