mirror of
https://github.com/lemon07r/opencode-kimi-full.git
synced 2026-07-18 08:05:52 +02:00
lemon07r
94 lines
3.7 KiB
TypeScript
94 lines
3.7 KiB
TypeScript
import { test, expect } from "bun:test"
|
|
import fs from "node:fs"
|
|
import os from "node:os"
|
|
import path from "node:path"
|
|
import { KIMI_CLI_VERSION, USER_AGENT } from "../src/constants.ts"
|
|
import { asciiHeaderValue, getDeviceId, kimiDeviceModel, kimiHeaders } from "../src/headers.ts"
|
|
|
|
// Note: getDeviceId() reads/writes ~/.kimi/device_id. That file is shared
|
|
// with kimi-cli on purpose (AGENTS.md rule 2) and the function is
|
|
// idempotent — if the file exists we reuse it, otherwise we create it. The
|
|
// tests therefore use the real HOME: they cannot clobber anything, and
|
|
// mocking `os.homedir` for Node's built-in `os` is fragile (Bun resolves
|
|
// the import binding eagerly).
|
|
|
|
test("kimiHeaders emits exactly the 7 fingerprint keys kimi-cli sends", () => {
|
|
const h = kimiHeaders()
|
|
expect(Object.keys(h).sort()).toEqual(
|
|
[
|
|
"User-Agent",
|
|
"X-Msh-Device-Id",
|
|
"X-Msh-Device-Model",
|
|
"X-Msh-Device-Name",
|
|
"X-Msh-Os-Version",
|
|
"X-Msh-Platform",
|
|
"X-Msh-Version",
|
|
].sort(),
|
|
)
|
|
})
|
|
|
|
test("User-Agent and X-Msh-Version track KIMI_CLI_VERSION (AGENTS.md rule 1)", () => {
|
|
const h = kimiHeaders()
|
|
expect(h["User-Agent"]).toBe(USER_AGENT)
|
|
expect(h["User-Agent"]).toContain(KIMI_CLI_VERSION)
|
|
expect(h["X-Msh-Version"]).toBe(KIMI_CLI_VERSION)
|
|
expect(h["X-Msh-Platform"]).toBe("kimi_cli")
|
|
})
|
|
|
|
test("All header values are ASCII-only (undici rejects non-ASCII)", () => {
|
|
for (const [k, v] of Object.entries(kimiHeaders())) {
|
|
expect(v, `header ${k}`).toMatch(/^[\x20-\x7e]+$/)
|
|
}
|
|
})
|
|
|
|
test("asciiHeaderValue strips non-ASCII bytes like kimi-cli and falls back when empty", () => {
|
|
expect(asciiHeaderValue(" hoste ")).toBe("hoste")
|
|
expect(asciiHeaderValue("hést")).toBe("hst")
|
|
expect(asciiHeaderValue("你好")).toBe("unknown")
|
|
})
|
|
|
|
test("Device id is a 32-char lowercase hex string (kimi-cli UUIDv4 no-dashes format)", () => {
|
|
expect(getDeviceId()).toMatch(/^[0-9a-f]{32}$/)
|
|
})
|
|
|
|
test("Device id is stable across calls and matches ~/.kimi/device_id on disk (AGENTS.md rule 2)", () => {
|
|
const first = getDeviceId()
|
|
const second = getDeviceId()
|
|
expect(second).toBe(first)
|
|
// Mirrors kimi-cli's path; shared by design.
|
|
const onDisk = fs.readFileSync(path.join(os.homedir(), ".kimi", "device_id"), "utf8").trim()
|
|
expect(onDisk).toBe(first)
|
|
})
|
|
|
|
test("Device id is also present and matches in the headers map", () => {
|
|
const h = kimiHeaders()
|
|
expect(h["X-Msh-Device-Id"]).toBe(getDeviceId())
|
|
})
|
|
|
|
// Regression guard: prior to v1.0.3 we were sending `X-Msh-Device-Model =
|
|
// <arch>` and `X-Msh-Os-Version = <type release>`. That didn't match kimi-cli
|
|
// and caused Moonshot to 403 every request from this plugin with
|
|
// "access_terminated_error". Keep this tied to the helper so Linux, macOS,
|
|
// and Windows all stay on the same parity path.
|
|
test("X-Msh-Device-Model matches the kimi-cli parity helper", () => {
|
|
const h = kimiHeaders()
|
|
expect(h["X-Msh-Device-Model"]).toBe(asciiHeaderValue(kimiDeviceModel()))
|
|
})
|
|
|
|
test("kimiDeviceModel mirrors kimi-cli Darwin and Windows special cases", () => {
|
|
expect(kimiDeviceModel({ system: "Darwin", release: "23.6.0", machine: "arm64", macVersion: "15.4.1" })).toBe(
|
|
"macOS 15.4.1 arm64",
|
|
)
|
|
expect(kimiDeviceModel({ system: "Windows_NT", release: "10.0.26100", machine: "x64" })).toBe("Windows 11 x64")
|
|
expect(kimiDeviceModel({ system: "Windows_NT", release: "10.0.19045", machine: "x64" })).toBe("Windows 10 x64")
|
|
expect(kimiDeviceModel({ system: "Linux", release: "6.8.0", machine: "x86_64" })).toBe("Linux 6.8.0 x86_64")
|
|
})
|
|
|
|
test("X-Msh-Os-Version matches os.version() (kernel build string on Linux)", () => {
|
|
const h = kimiHeaders()
|
|
// `os.version()` exists in Node 13.13+ — be lenient if missing.
|
|
if (typeof os.version === "function") {
|
|
expect(h["X-Msh-Os-Version"]).toBe(os.version())
|
|
}
|
|
})
|