dsh-session-ops 0.1.4 → 0.2.0

package/lib/store/http.js

@@ -22,10 +22,9 @@
  *
  * @module dsh-session-ops/store/http
  */
-
+import { messageOf } from './util.js';
 /** 请求体上限：1 MiB。 */
-const MAX_BODY_BYTES = 1024 * 1024
-
+const MAX_BODY_BYTES = 1024 * 1024;
 /**
  * 写一个 JSON 响应。
  * @param res - Node HTTP 响应。
@@ -37,19 +36,19 @@ const MAX_BODY_BYTES = 1024 * 1024
  * @returns 无。
  */
 export function writeJson(res, status, body, options = {}) {
-  const text = JSON.stringify(body)
-  if (options.contentLength === true) {
-    res.writeHead(status, {
-      'content-type': 'application/json; charset=utf-8',
-      'cache-control': 'no-store',
-      'content-length': Buffer.byteLength(text),
-    })
-  } else {
-    res.writeHead(status, { 'content-type': 'application/json; charset=utf-8', 'cache-control': 'no-store' })
-  }
-  res.end(text)
+    const text = JSON.stringify(body);
+    if (options.contentLength === true) {
+        res.writeHead(status, {
+            'content-type': 'application/json; charset=utf-8',
+            'cache-control': 'no-store',
+            'content-length': Buffer.byteLength(text),
+        });
+    }
+    else {
+        res.writeHead(status, { 'content-type': 'application/json; charset=utf-8', 'cache-control': 'no-store' });
+    }
+    res.end(text);
 }
-
 /**
  * 读一个 JSON 请求体（上限 1 MiB）。
  *
@@ -64,24 +63,28 @@ export function writeJson(res, status, body, options = {}) {
  * @returns 解析出来的载荷对象。
  */
 export async function readJsonBody(req, options = {}) {
-  const requireObject = options.requireObject === true
-  const chunks = []
-  let size = 0
-  for await (const chunk of req) {
-    size += chunk.length
-    if (size > MAX_BODY_BYTES) throw new Error('request body too large')
-    chunks.push(chunk)
-  }
-  if (chunks.length === 0) return {}
-  const buffer = Buffer.concat(chunks).toString('utf8')
-  if (!requireObject) return JSON.parse(buffer)
-  const text = buffer.trim()
-  if (text === '') return {}
-  const parsed = JSON.parse(text)
-  if (parsed === null || typeof parsed !== 'object' || Array.isArray(parsed)) throw new Error('request body must be a JSON object')
-  return parsed
+    const requireObject = options.requireObject === true;
+    const chunks = [];
+    let size = 0;
+    for await (const chunk of req) {
+        size += chunk.length;
+        if (size > MAX_BODY_BYTES)
+            throw new Error('request body too large');
+        chunks.push(chunk);
+    }
+    if (chunks.length === 0)
+        return {};
+    const buffer = Buffer.concat(chunks).toString('utf8');
+    if (!requireObject)
+        return JSON.parse(buffer);
+    const text = buffer.trim();
+    if (text === '')
+        return {};
+    const parsed = JSON.parse(text);
+    if (parsed === null || typeof parsed !== 'object' || Array.isArray(parsed))
+        throw new Error('request body must be a JSON object');
+    return parsed;
 }
-
 /**
  * 造一个 `kind: 'prefix'` 路由处理器（`ctx.webServer.register({kind:'prefix', path, handler})` 的 `handler`）。
  *
@@ -98,31 +101,33 @@ export async function readJsonBody(req, options = {}) {
  * @returns `async (req, res) => void`。
  */
 export function createApiHandler(options) {
-  const { prefix, methods, trustRequest, log, readBody = readJsonBody, contentLength = false } = options
-  const reply = (res, status, body) => writeJson(res, status, body, { contentLength })
-  return async (req, res) => {
-    if (typeof trustRequest !== 'function' || !trustRequest(req)) {
-      reply(res, 403, { ok: false, error: { code: 'forbidden', message: 'forbidden' } })
-      return
-    }
-    const url = new URL(req.url ?? '/', 'http://dsh.internal')
-    const method = url.pathname.startsWith(`${prefix}/`) ? url.pathname.slice(prefix.length + 1) : undefined
-    if (method === undefined || method.includes('/')) {
-      reply(res, 404, { ok: false, error: { code: 'not-found', message: 'unknown method' } })
-      return
-    }
-    const handler = methods[method]
-    if (typeof handler !== 'function') {
-      reply(res, 404, { ok: false, error: { code: 'not-found', message: `unknown method ${method}` } })
-      return
-    }
-    try {
-      const payload = req.method === 'GET' ? Object.fromEntries(url.searchParams) : await readBody(req)
-      const value = await handler(payload)
-      reply(res, 200, { ok: true, value })
-    } catch (error) {
-      if (typeof log === 'function') log('warn', `${method} failed: ${String(error?.message ?? error)}`)
-      reply(res, 400, { ok: false, error: { code: 'bad-request', message: String(error?.message ?? error) } })
-    }
-  }
+    const { prefix, methods, trustRequest, log, readBody = readJsonBody, contentLength = false } = options;
+    const reply = (res, status, body) => writeJson(res, status, body, { contentLength });
+    return async (req, res) => {
+        if (typeof trustRequest !== 'function' || !trustRequest(req)) {
+            reply(res, 403, { ok: false, error: { code: 'forbidden', message: 'forbidden' } });
+            return;
+        }
+        const url = new URL(req.url ?? '/', 'http://dsh.internal');
+        const method = url.pathname.startsWith(`${prefix}/`) ? url.pathname.slice(prefix.length + 1) : undefined;
+        if (method === undefined || method.includes('/')) {
+            reply(res, 404, { ok: false, error: { code: 'not-found', message: 'unknown method' } });
+            return;
+        }
+        const handler = methods[method];
+        if (typeof handler !== 'function') {
+            reply(res, 404, { ok: false, error: { code: 'not-found', message: `unknown method ${method}` } });
+            return;
+        }
+        try {
+            const payload = req.method === 'GET' ? Object.fromEntries(url.searchParams) : await readBody(req);
+            const value = await handler(payload);
+            reply(res, 200, { ok: true, value });
+        }
+        catch (error) {
+            if (typeof log === 'function')
+                log('warn', `${method} failed: ${messageOf(error)}`);
+            reply(res, 400, { ok: false, error: { code: 'bad-request', message: messageOf(error) } });
+        }
+    };
 }

package/lib/store/index.js

@@ -23,9 +23,8 @@
  *
  * @module dsh-session-ops/store
  */
-
-export * from './paths.js'
-export * from './frames.js'
-export * from './registry.js'
-export * from './host.js'
-export * from './http.js'
+export * from './paths.js';
+export * from './frames.js';
+export * from './registry.js';
+export * from './host.js';
+export * from './http.js';

package/lib/store/paths.js

@@ -16,58 +16,63 @@
  *
  * @module dsh-session-ops/store/paths
  */
-
-import { readdirSync } from 'node:fs'
-import { join } from 'node:path'
-
+import { readdirSync } from 'node:fs';
+import { join } from 'node:path';
+import { messageOf } from './util.js';
 /** generation 规范名：`session.jsonl[.zstd]` 或 `session.vN.jsonl[.zstd]`。 */
-const GENERATION_NAME = /^session(?:\.v(\d+))?\.jsonl(\.zstd)?$/
-
+const GENERATION_NAME = /^session(?:\.v(\d+))?\.jsonl(\.zstd)?$/;
 /**
  * 把任意字符串编码成一个安全路径段（与会话 id 共用同一套转义）。
  * @param raw - 原始字符串。
  * @returns 单个文件系统安全路径段。
  */
 export function encodeSegment(raw) {
-  if (raw.length === 0) throw new Error('cannot encode an empty path segment')
-  if (raw === '.') return '~002E'
-  if (raw === '..') return '~002E~002E'
-  let out = ''
-  for (let i = 0; i < raw.length; i++) {
-    const code = raw.charCodeAt(i)
-    const ch = String.fromCharCode(code)
-    if (ch !== '~' && /^[A-Za-z0-9._-]$/.test(ch)) out += ch
-    else out += '~' + code.toString(16).toUpperCase().padStart(4, '0')
-  }
-  return out
+    if (raw.length === 0)
+        throw new Error('cannot encode an empty path segment');
+    if (raw === '.')
+        return '~002E';
+    if (raw === '..')
+        return '~002E~002E';
+    let out = '';
+    for (let i = 0; i < raw.length; i++) {
+        const code = raw.charCodeAt(i);
+        const ch = String.fromCharCode(code);
+        if (ch !== '~' && /^[A-Za-z0-9._-]$/.test(ch))
+            out += ch;
+        else
+            out += '~' + code.toString(16).toUpperCase().padStart(4, '0');
+    }
+    return out;
 }
-
 /**
  * 项目目录键：分隔符折叠成 `-`，不安全码位 `~XXXX`，截断到 251 字符。
  * @param cwd - 会话的项目目录。
  * @returns 单个文件系统安全的项目目录名。
  */
 export function projectKey(cwd) {
-  if (cwd.length === 0) throw new Error('cannot encode an empty project path')
-  let readable = ''
-  let separatorRun = false
-  for (let i = 0; i < cwd.length; i++) {
-    const code = cwd.charCodeAt(i)
-    const ch = String.fromCharCode(code)
-    if (ch === '/' || ch === '\\' || ch === ':') {
-      if (!separatorRun) readable += '-'
-      separatorRun = true
-    } else if (ch !== '~' && /^[A-Za-z0-9._-]$/.test(ch)) {
-      readable += ch
-      separatorRun = false
-    } else {
-      readable += '~' + code.toString(16).toUpperCase().padStart(4, '0')
-      separatorRun = false
+    if (cwd.length === 0)
+        throw new Error('cannot encode an empty project path');
+    let readable = '';
+    let separatorRun = false;
+    for (let i = 0; i < cwd.length; i++) {
+        const code = cwd.charCodeAt(i);
+        const ch = String.fromCharCode(code);
+        if (ch === '/' || ch === '\\' || ch === ':') {
+            if (!separatorRun)
+                readable += '-';
+            separatorRun = true;
+        }
+        else if (ch !== '~' && /^[A-Za-z0-9._-]$/.test(ch)) {
+            readable += ch;
+            separatorRun = false;
+        }
+        else {
+            readable += '~' + code.toString(16).toUpperCase().padStart(4, '0');
+            separatorRun = false;
+        }
     }
-  }
-  return `--${(readable.replace(/^-+/, '') || 'root').slice(0, 251)}--`
+    return `--${(readable.replace(/^-+/, '') || 'root').slice(0, 251)}--`;
 }
-
 /**
  * 把 `sessions/` 下的项目目录键还原成路径。
  *
@@ -78,15 +83,16 @@ export function projectKey(cwd) {
  * @returns 反解出的路径。
  */
 export function decodeProjectKey(key) {
-  let s = key
-  if (s.startsWith('--')) s = s.slice(2)
-  if (s.endsWith('--')) s = s.slice(0, -2)
-  s = s.replace(/~([0-9a-fA-F]{4})/g, (_, hex) => String.fromCharCode(parseInt(hex, 16)))
-  s = s.replace(/-/g, '\\')
-  // 盘符的冒号在目录键里没有单独转义（`C:\a` → `--C-a--`），补回来。
-  return /^[A-Za-z]\\/.test(s) ? `${s[0]}:${s.slice(1)}` : s
+    let s = key;
+    if (s.startsWith('--'))
+        s = s.slice(2);
+    if (s.endsWith('--'))
+        s = s.slice(0, -2);
+    s = s.replace(/~([0-9a-fA-F]{4})/g, (_, hex) => String.fromCharCode(parseInt(hex, 16)));
+    s = s.replace(/-/g, '\\');
+    // 盘符的冒号在目录键里没有单独转义（`C:\a` → `--C-a--`），补回来。
+    return /^[A-Za-z]\\/.test(s) ? `${s[0]}:${s.slice(1)}` : s;
 }
-
 /**
  * **宽松**的会话日志文件名判定：`.jsonl.zstd` 或 `.jsonl` 结尾就算。
  *
@@ -97,20 +103,19 @@ export function decodeProjectKey(key) {
  * @returns 是否像一份会话日志。
  */
 export function isSessionLogFile(name) {
-  return name.endsWith('.jsonl.zstd') || name.endsWith('.jsonl')
+    return name.endsWith('.jsonl.zstd') || name.endsWith('.jsonl');
 }
-
 /**
  * 目录里是不是规范 generation 名。
  * @param name - 目录项名。
  * @returns 版本号（v0 为 0）与是否压缩；不是规范名时 `undefined`。
  */
 export function parseGenerationName(name) {
-  const match = GENERATION_NAME.exec(name)
-  if (match === null) return undefined
-  return { version: match[1] === undefined ? 0 : Number(match[1]), compressed: match[2] === '.zstd' }
+    const match = GENERATION_NAME.exec(name);
+    if (match === null)
+        return undefined;
+    return { version: match[1] === undefined ? 0 : Number(match[1]), compressed: match[2] === '.zstd' };
 }
-
 /**
  * 选出会话目录里版本最高的规范 generation 文件。
  *
@@ -121,43 +126,41 @@ export function parseGenerationName(name) {
  *   （错误文案被两个插件的用户可见报告引用，改动前先看测试）。
  */
 export function findGeneration(dir) {
-  let entries
-  try {
-    entries = readdirSync(dir, { withFileTypes: true })
-  } catch (error) {
-    return { error: `cannot list session directory ${dir}: ${String(error?.message ?? error)}` }
-  }
-  const generations = []
-  const others = []
-  for (const entry of entries) {
-    const parsed = parseGenerationName(entry.name)
-    if (parsed === undefined) {
-      others.push(entry.name)
-      continue
+    let entries;
+    try {
+        entries = readdirSync(dir, { withFileTypes: true });
+    }
+    catch (error) {
+        return { error: `cannot list session directory ${dir}: ${messageOf(error)}` };
+    }
+    const generations = [];
+    const others = [];
+    for (const entry of entries) {
+        const parsed = parseGenerationName(entry.name);
+        if (parsed === undefined) {
+            others.push(entry.name);
+            continue;
+        }
+        generations.push({ name: entry.name, version: parsed.version, compressed: parsed.compressed });
     }
-    generations.push({ name: entry.name, version: parsed.version, compressed: parsed.compressed })
-  }
-  if (generations.length === 0) return { error: `no canonical session log in ${dir}`, others }
-  generations.sort((left, right) => right.version - left.version)
-  return { selected: generations[0], all: generations, others }
+    if (generations.length === 0)
+        return { error: `no canonical session log in ${dir}`, others };
+    generations.sort((left, right) => right.version - left.version);
+    return { selected: generations[0], all: generations, others };
 }
-
 /** 物理编码后缀。 */
 export function compressionSuffix(compression) {
-  return compression === 'zstd' ? '.zstd' : ''
+    return compression === 'zstd' ? '.zstd' : '';
 }
-
 /** 一个 generation 的规范文件名（v0 不带 `vN` 段）。 */
 export function generationLogFilename(version, compression) {
-  return `${version === 0 ? 'session' : `session.v${version}`}.jsonl${compressionSuffix(compression)}`
+    return `${version === 0 ? 'session' : `session.v${version}`}.jsonl${compressionSuffix(compression)}`;
 }
-
 /** 会话目录：`<root>/<projectKey>/<encodeSegment(id)>`。 */
 export function sessionDir(root, cwd, id) {
-  return join(join(root, cwd === undefined ? '_no-cwd' : projectKey(cwd)), encodeSegment(id))
+    return join(join(root, cwd === undefined ? '_no-cwd' : projectKey(cwd)), encodeSegment(id));
 }
-
 /** 一个 generation 的绝对日志路径。 */
 export function logPath(root, cwd, id, version, compression) {
-  return join(sessionDir(root, cwd, id), generationLogFilename(version, compression))
+    return join(sessionDir(root, cwd, id), generationLogFilename(version, compression));
 }