mirror of
https://github.com/garrytan/gstack.git
synced 2026-05-17 01:31:26 +08:00
feat(browse-skills): atomic write helper for /skillify (D3)
stageSkill writes a candidate skill into ~/.gstack/.tmp/skillify-<spawnId>/ with restrictive perms. commitSkill does an atomic fs.renameSync into the final tier path with realpath/lstat discipline (refuses symlinked staging dirs, refuses to clobber existing skills). discardStaged is the cleanup path for test failures and approval rejections, idempotent and bounded to the per-spawn wrapper. validateSkillName enforces lowercase/digits/ dashes only, no path-escape characters. Implements the D3 contract from the v1.19.0.0 plan review: never a half-written skill on disk. Test fail or approval reject = rm -rf the temp dir, no tombstone for never-approved skills. Closes Codex finding #5 (atomic skill packaging) for Phase 2a. 34 unit assertions covering: stage validation, file-path escape rejection, permission check, atomic rename, clobber refusal, symlink refusal, project tier unresolved, idempotent discard, end-to-end happy + simulated test failure + approval reject paths. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
215
browse/src/browser-skill-write.ts
Normal file
215
browse/src/browser-skill-write.ts
Normal file
@@ -0,0 +1,215 @@
|
||||
/**
|
||||
* Atomic-write helper for agent-authored browser-skills (D3 from Phase 2 plan).
|
||||
*
|
||||
* /skillify stages a candidate skill into ~/.gstack/.tmp/skillify-<spawnId>/,
|
||||
* runs $B skill test against it, and only renames the directory into its final
|
||||
* tier path on success + user approval. On failure or rejection, the staged
|
||||
* directory is removed entirely — no half-written skill ever appears in
|
||||
* $B skill list, no tombstone for something the user never approved.
|
||||
*
|
||||
* stageSkill — write all files into the staging dir, return its path
|
||||
* commitSkill — atomic rename into the final tier path; refuses to clobber
|
||||
* discardStaged — rm -rf the staged dir (called on test fail or reject)
|
||||
*
|
||||
* Symlink discipline: lstat() the staging dir before rename to refuse moves
|
||||
* through symlinks; realpath() the final tier root to ensure the destination
|
||||
* lands inside the expected directory tree.
|
||||
*/
|
||||
|
||||
import * as fs from 'fs';
|
||||
import * as path from 'path';
|
||||
import * as os from 'os';
|
||||
import { isPathWithin } from './platform';
|
||||
import type { TierPaths } from './browser-skills';
|
||||
import { defaultTierPaths } from './browser-skills';
|
||||
|
||||
// ─── Naming validation ──────────────────────────────────────────
|
||||
|
||||
/**
|
||||
* Skill names must be safe directory names: lowercase letters, digits, dashes.
|
||||
* Starts with a letter, no consecutive dashes, no trailing dash, ≤64 chars.
|
||||
* Rejects '..', leading dots, slashes, anything that could escape the tier dir.
|
||||
*/
|
||||
const SKILL_NAME_PATTERN = /^[a-z][a-z0-9]*(-[a-z0-9]+)*$/;
|
||||
|
||||
export function validateSkillName(name: string): void {
|
||||
if (!name) throw new Error('Skill name is empty.');
|
||||
if (name.length > 64) throw new Error(`Skill name too long (${name.length} > 64).`);
|
||||
if (!SKILL_NAME_PATTERN.test(name)) {
|
||||
throw new Error(
|
||||
`Invalid skill name "${name}". Must be lowercase letters/digits/dashes, ` +
|
||||
`start with a letter, no leading/trailing/consecutive dashes.`,
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
// ─── Staging ────────────────────────────────────────────────────
|
||||
|
||||
export interface StageSkillOptions {
|
||||
name: string;
|
||||
/** Map of relative path → contents. Path may contain '/' for nested dirs. */
|
||||
files: Map<string, string | Buffer>;
|
||||
/** Optional override (tests pass synthetic spawn ids). */
|
||||
spawnId?: string;
|
||||
/** Optional override (tests pass a fake tmp root). */
|
||||
tmpRoot?: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* Stage a skill into the staging tree:
|
||||
* <tmpRoot>/.gstack/.tmp/skillify-<spawnId>/<name>/
|
||||
*
|
||||
* The leaf <name> directory is what gets renamed during commit. The wrapper
|
||||
* skillify-<spawnId>/ is per-spawn so concurrent /skillify invocations don't
|
||||
* collide. Returns the absolute path to the staged skill dir (ending in <name>).
|
||||
*/
|
||||
export function stageSkill(opts: StageSkillOptions): string {
|
||||
validateSkillName(opts.name);
|
||||
if (opts.files.size === 0) {
|
||||
throw new Error('stageSkill: files map is empty.');
|
||||
}
|
||||
|
||||
const spawnId = opts.spawnId ?? generateSpawnId();
|
||||
const tmpRoot = opts.tmpRoot ?? path.join(os.homedir(), '.gstack', '.tmp');
|
||||
const wrapperDir = path.join(tmpRoot, `skillify-${spawnId}`);
|
||||
const stagedDir = path.join(wrapperDir, opts.name);
|
||||
|
||||
fs.mkdirSync(wrapperDir, { recursive: true, mode: 0o700 });
|
||||
fs.mkdirSync(stagedDir, { recursive: true, mode: 0o700 });
|
||||
|
||||
for (const [relPath, contents] of opts.files) {
|
||||
if (relPath.startsWith('/') || relPath.includes('..')) {
|
||||
// Defense in depth: validateSkillName above bounds the leaf, but a
|
||||
// bad relPath in files could still write outside the staged dir.
|
||||
throw new Error(`Invalid file path in stageSkill: "${relPath}".`);
|
||||
}
|
||||
const filePath = path.join(stagedDir, relPath);
|
||||
const fileDir = path.dirname(filePath);
|
||||
fs.mkdirSync(fileDir, { recursive: true });
|
||||
fs.writeFileSync(filePath, contents);
|
||||
}
|
||||
|
||||
return stagedDir;
|
||||
}
|
||||
|
||||
// ─── Commit (atomic rename) ─────────────────────────────────────
|
||||
|
||||
export interface CommitSkillOptions {
|
||||
name: string;
|
||||
tier: 'project' | 'global';
|
||||
stagedDir: string;
|
||||
/** Optional override (tests pass synthetic tier paths). */
|
||||
tiers?: TierPaths;
|
||||
}
|
||||
|
||||
/**
|
||||
* Atomically move the staged skill into its final tier path. Refuses to
|
||||
* clobber an existing skill at the same path — the agent's approval gate
|
||||
* MUST surface name collisions before calling this.
|
||||
*
|
||||
* Returns the absolute path of the committed skill dir.
|
||||
*
|
||||
* Throws when:
|
||||
* - tier path is unresolved (project tier with no project root)
|
||||
* - destination already exists
|
||||
* - staged dir is a symlink (refuses to follow)
|
||||
* - resolved destination escapes the tier root (defense in depth)
|
||||
*/
|
||||
export function commitSkill(opts: CommitSkillOptions): string {
|
||||
validateSkillName(opts.name);
|
||||
|
||||
const tiers = opts.tiers ?? defaultTierPaths();
|
||||
const tierRoot = opts.tier === 'project' ? tiers.project : tiers.global;
|
||||
if (!tierRoot) {
|
||||
throw new Error(`commitSkill: tier "${opts.tier}" has no resolved path.`);
|
||||
}
|
||||
|
||||
// Refuse to follow a symlinked staging dir — caller should hand us the path
|
||||
// returned by stageSkill, which is always a real directory.
|
||||
let stagedStat: fs.Stats;
|
||||
try {
|
||||
stagedStat = fs.lstatSync(opts.stagedDir);
|
||||
} catch (err: any) {
|
||||
throw new Error(`commitSkill: staged dir "${opts.stagedDir}" not accessible: ${err.code ?? err.message}`);
|
||||
}
|
||||
if (stagedStat.isSymbolicLink()) {
|
||||
throw new Error(`commitSkill: staged dir "${opts.stagedDir}" is a symlink — refusing to commit.`);
|
||||
}
|
||||
if (!stagedStat.isDirectory()) {
|
||||
throw new Error(`commitSkill: staged path "${opts.stagedDir}" is not a directory.`);
|
||||
}
|
||||
|
||||
// Ensure the tier root exists, then resolve its real path so the final
|
||||
// destination check defends against tierRoot itself being a symlink.
|
||||
fs.mkdirSync(tierRoot, { recursive: true, mode: 0o755 });
|
||||
const realTierRoot = fs.realpathSync(tierRoot);
|
||||
|
||||
const dest = path.join(realTierRoot, opts.name);
|
||||
if (!isPathWithin(dest, realTierRoot)) {
|
||||
// Should be impossible after validateSkillName, but defense in depth.
|
||||
throw new Error(`commitSkill: destination "${dest}" escapes tier root.`);
|
||||
}
|
||||
|
||||
// Refuse to clobber. Both regular dirs and symlinks count.
|
||||
let destExists = false;
|
||||
try {
|
||||
fs.lstatSync(dest);
|
||||
destExists = true;
|
||||
} catch (err: any) {
|
||||
if (err.code !== 'ENOENT') throw err;
|
||||
}
|
||||
if (destExists) {
|
||||
throw new Error(
|
||||
`commitSkill: a skill named "${opts.name}" already exists at ${dest}. ` +
|
||||
`Pick a different name or remove the existing skill first ` +
|
||||
`($B skill rm ${opts.name}${opts.tier === 'global' ? ' --global' : ''}).`,
|
||||
);
|
||||
}
|
||||
|
||||
fs.renameSync(opts.stagedDir, dest);
|
||||
return dest;
|
||||
}
|
||||
|
||||
// ─── Discard (cleanup on failure or reject) ─────────────────────
|
||||
|
||||
/**
|
||||
* Remove the staged skill directory and its per-spawn wrapper. Called on
|
||||
* test failure (step 8 of /skillify) or approval rejection (step 9).
|
||||
*
|
||||
* Idempotent: missing dirs are not an error. Best-effort: failures are
|
||||
* swallowed (cleanup is fire-and-forget, not load-bearing).
|
||||
*/
|
||||
export function discardStaged(stagedDir: string): void {
|
||||
// Remove the leaf skill dir first, then the wrapper skillify-<spawnId>/.
|
||||
// If the wrapper was the only thing inside it, this tidies up that too.
|
||||
try {
|
||||
fs.rmSync(stagedDir, { recursive: true, force: true });
|
||||
} catch {
|
||||
// best effort
|
||||
}
|
||||
const wrapperDir = path.dirname(stagedDir);
|
||||
if (path.basename(wrapperDir).startsWith('skillify-')) {
|
||||
try {
|
||||
// Only remove the wrapper if it's now empty — concurrent /skillify
|
||||
// invocations get their own wrappers, but if a buggy caller passed
|
||||
// a stagedDir not under a skillify-<id> wrapper we should not nuke
|
||||
// an unrelated parent.
|
||||
const remaining = fs.readdirSync(wrapperDir);
|
||||
if (remaining.length === 0) {
|
||||
fs.rmdirSync(wrapperDir);
|
||||
}
|
||||
} catch {
|
||||
// best effort
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ─── Spawn id ───────────────────────────────────────────────────
|
||||
|
||||
/** Per-spawn id matching the format used by skill-token.ts. */
|
||||
function generateSpawnId(): string {
|
||||
// 8 random hex chars + millis suffix — collision risk negligible across
|
||||
// concurrent /skillify invocations on a single machine.
|
||||
const rand = Math.floor(Math.random() * 0xffffffff).toString(16).padStart(8, '0');
|
||||
return `${rand}-${Date.now().toString(36)}`;
|
||||
}
|
||||
Reference in New Issue
Block a user