@tokilake/tokiame 0.0.0

package/README.md

@@ -0,0 +1,39 @@
+# `@tokilake/tokiame`
+
+`tokiame` 的 npm 包只做一件事：在安装时下载对应平台的预编译二进制，并把它暴露成 `tokiame` 命令。
+
+## 安装
+
+```bash
+npm install -g @tokilake/tokiame
+```
+
+## 默认配置路径
+
+安装完成后，`tokiame` 默认读取：
+
+```text
+~/.tokilake/tokiame.json
+```
+
+安装脚本还会额外写入：
+
+```text
+~/.tokilake/tokiame.json.example
+```
+
+你可以直接复制一份开始修改：
+
+```bash
+cp ~/.tokilake/tokiame.json.example ~/.tokilake/tokiame.json
+```
+
+你也可以通过下面两种方式覆盖：
+
+- `TOKIAME_CONFIG=/path/to/tokiame.json`
+- `tokiame --config /path/to/tokiame.json`
+
+## 可选环境变量
+
+- `TOKIAME_RELEASE_REPO`: 覆盖 GitHub Releases 仓库，默认 `Tokimorphling/Tokilake`
+- `TOKIAME_RELEASE_TAG`: 覆盖 GitHub Releases tag，默认 `v<package-version>`

package/bin/tokiame.js

@@ -0,0 +1,30 @@
+#!/usr/bin/env node
+
+const { spawn } = require("node:child_process");
+const { existsSync } = require("node:fs");
+const path = require("node:path");
+
+const binaryName = process.platform === "win32" ? "tokiame.exe" : "tokiame";
+const binaryPath = path.join(__dirname, binaryName);
+
+if (!existsSync(binaryPath)) {
+  console.error("tokiame binary is missing. Reinstall @tokilake/tokiame.");
+  process.exit(1);
+}
+
+const child = spawn(binaryPath, process.argv.slice(2), {
+  stdio: "inherit",
+});
+
+child.on("error", (error) => {
+  console.error(error.message);
+  process.exit(1);
+});
+
+child.on("exit", (code, signal) => {
+  if (signal) {
+    process.kill(process.pid, signal);
+    return;
+  }
+  process.exit(code ?? 1);
+});

package/package.json

@@ -0,0 +1,39 @@
+{
+  "name": "@tokilake/tokiame",
+  "version": "0.0.0",
+  "description": "Thin npm installer for the Tokiame worker binary",
+  "license": "Apache-2.0",
+  "homepage": "https://github.com/Tokimorphling/Tokilake",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Tokimorphling/Tokilake.git",
+    "directory": "packaging/npm/tokiame"
+  },
+  "bugs": {
+    "url": "https://github.com/Tokimorphling/Tokilake/issues"
+  },
+  "bin": {
+    "tokiame": "./bin/tokiame.js"
+  },
+  "files": [
+    "bin",
+    "scripts",
+    "README.md",
+    "tokiame.json.example"
+  ],
+  "scripts": {
+    "postinstall": "node ./scripts/install.js"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "os": [
+    "darwin",
+    "linux",
+    "win32"
+  ],
+  "cpu": [
+    "x64",
+    "arm64"
+  ]
+}

package/scripts/install.js

@@ -0,0 +1,150 @@
+"use strict";
+
+const crypto = require("node:crypto");
+const fs = require("node:fs");
+const fsp = require("node:fs/promises");
+const https = require("node:https");
+const os = require("node:os");
+const path = require("node:path");
+const { spawnSync } = require("node:child_process");
+
+const packageJSON = require("../package.json");
+
+const releaseRepo = process.env.TOKIAME_RELEASE_REPO || "Tokimorphling/Tokilake";
+const releaseTag = process.env.TOKIAME_RELEASE_TAG || `v${packageJSON.version}`;
+
+function getTarget() {
+  const platform = process.platform;
+  const arch = process.arch;
+
+  if (platform === "darwin" && arch === "x64") {
+    return { assetOS: "darwin", assetArch: "amd64", archiveExt: "tar.gz", checksumOS: "darwin", binaryName: "tokiame" };
+  }
+  if (platform === "darwin" && arch === "arm64") {
+    return { assetOS: "darwin", assetArch: "arm64", archiveExt: "tar.gz", checksumOS: "darwin", binaryName: "tokiame" };
+  }
+  if (platform === "linux" && arch === "x64") {
+    return { assetOS: "linux", assetArch: "amd64", archiveExt: "tar.gz", checksumOS: "linux", binaryName: "tokiame" };
+  }
+  if (platform === "linux" && arch === "arm64") {
+    return { assetOS: "linux", assetArch: "arm64", archiveExt: "tar.gz", checksumOS: "linux", binaryName: "tokiame" };
+  }
+  if (platform === "win32" && arch === "x64") {
+    return { assetOS: "windows", assetArch: "amd64", archiveExt: "zip", checksumOS: "windows", binaryName: "tokiame.exe" };
+  }
+  throw new Error(`unsupported platform: ${platform}/${arch}`);
+}
+
+function getAssetURL(fileName) {
+  return `https://github.com/${releaseRepo}/releases/download/${releaseTag}/${fileName}`;
+}
+
+function download(url) {
+  return new Promise((resolve, reject) => {
+    const request = https.get(url, { headers: { "User-Agent": "@tokilake/tokiame" } }, (response) => {
+      if (response.statusCode && response.statusCode >= 300 && response.statusCode < 400 && response.headers.location) {
+        response.resume();
+        download(response.headers.location).then(resolve).catch(reject);
+        return;
+      }
+      if (response.statusCode !== 200) {
+        response.resume();
+        reject(new Error(`download failed: ${url} (${response.statusCode})`));
+        return;
+      }
+
+      const chunks = [];
+      response.on("data", (chunk) => chunks.push(chunk));
+      response.on("end", () => resolve(Buffer.concat(chunks)));
+    });
+    request.on("error", reject);
+  });
+}
+
+function verifyChecksum(buffer, checksumText, assetName) {
+  const digest = crypto.createHash("sha256").update(buffer).digest("hex");
+  const expectedLine = checksumText
+    .split(/\r?\n/)
+    .map((line) => line.trim())
+    .find((line) => line.endsWith(` ${assetName}`));
+
+  if (!expectedLine) {
+    throw new Error(`checksum entry not found for ${assetName}`);
+  }
+
+  const expected = expectedLine.split(/\s+/)[0];
+  if (expected !== digest) {
+    throw new Error(`checksum mismatch for ${assetName}`);
+  }
+}
+
+function extractArchive(archivePath, extractDir, archiveExt) {
+  if (archiveExt === "tar.gz") {
+    const result = spawnSync("tar", ["-xzf", archivePath, "-C", extractDir], { stdio: "inherit" });
+    if (result.status !== 0) {
+      throw new Error("failed to extract tar.gz archive");
+    }
+    return;
+  }
+
+  if (archiveExt === "zip") {
+    const command = `Expand-Archive -LiteralPath '${archivePath.replace(/'/g, "''")}' -DestinationPath '${extractDir.replace(/'/g, "''")}' -Force`;
+    const result = spawnSync("powershell", ["-NoProfile", "-NonInteractive", "-Command", command], { stdio: "inherit" });
+    if (result.status !== 0) {
+      throw new Error("failed to extract zip archive");
+    }
+    return;
+  }
+
+  throw new Error(`unsupported archive format: ${archiveExt}`);
+}
+
+async function main() {
+  const target = getTarget();
+  const assetName = `tokiame_${packageJSON.version}_${target.assetOS}_${target.assetArch}.${target.archiveExt}`;
+  const checksumName = `SHA256SUMS-${target.checksumOS}.txt`;
+
+  const archiveBuffer = await download(getAssetURL(assetName));
+  const checksumBuffer = await download(getAssetURL(checksumName));
+
+  verifyChecksum(archiveBuffer, checksumBuffer.toString("utf8"), assetName);
+
+  const tempDir = await fsp.mkdtemp(path.join(os.tmpdir(), "tokiame-install-"));
+  const archivePath = path.join(tempDir, assetName);
+  const extractDir = path.join(tempDir, "extract");
+
+  await fsp.mkdir(extractDir, { recursive: true });
+  await fsp.writeFile(archivePath, archiveBuffer);
+  extractArchive(archivePath, extractDir, target.archiveExt);
+
+  const sourceBinaryPath = path.join(extractDir, target.binaryName);
+  const destinationDir = path.join(__dirname, "..", "bin");
+  const destinationPath = path.join(destinationDir, target.binaryName);
+
+  if (!fs.existsSync(sourceBinaryPath)) {
+    throw new Error(`extracted binary not found: ${sourceBinaryPath}`);
+  }
+
+  await fsp.mkdir(destinationDir, { recursive: true });
+  await fsp.copyFile(sourceBinaryPath, destinationPath);
+
+  if (process.platform !== "win32") {
+    await fsp.chmod(destinationPath, 0o755);
+  }
+
+  const homeDir = os.homedir();
+  if (homeDir) {
+    const configDir = path.join(homeDir, ".tokilake");
+    const exampleSource = path.join(__dirname, "..", "tokiame.json.example");
+    const exampleDestination = path.join(configDir, "tokiame.json.example");
+    await fsp.mkdir(configDir, { recursive: true });
+    if (fs.existsSync(exampleSource) && !fs.existsSync(exampleDestination)) {
+      await fsp.copyFile(exampleSource, exampleDestination);
+    }
+  }
+}
+
+main().catch((error) => {
+  console.error(`[tokiame] ${error.message}`);
+  process.exit(1);
+});

package/tokiame.json.example

@@ -0,0 +1,16 @@
+{
+  "gateway_url": "wss://api.example.com/api/tokilake/connect",
+  "token": "sk-your-user-token",
+  "namespace": "demo-worker",
+  "node_name": "gpu-01",
+  "group": "default",
+  "backend_type": "openai",
+  "heartbeat_interval_seconds": 15,
+  "reconnect_delay_seconds": 5,
+  "model_targets": {
+    "gpt-4o-mini": {
+      "url": "http://127.0.0.1:8000/v1",
+      "mapped_name": "gpt-4o-mini"
+    }
+  }
+}