one
Version:
One is a new React Framework that makes Vite serve both native and web.
237 lines (236 loc) • 7.23 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.mjs";
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.mjs.map