@1agh/maude 0.45.1 → 0.46.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/apps/studio/acp/bridge.ts +628 -74
- package/apps/studio/acp/index.ts +172 -23
- package/apps/studio/acp/probe.ts +31 -7
- package/apps/studio/acp/transcript.ts +91 -5
- package/apps/studio/annotations-snap.ts +50 -10
- package/apps/studio/api.ts +344 -14
- package/apps/studio/artboard-guides-overlay.tsx +270 -0
- package/apps/studio/bin/_import-asset.mjs +35 -5
- package/apps/studio/bin/_pdf-playwright.mjs +35 -4
- package/apps/studio/bin/_png-playwright.mjs +38 -4
- package/apps/studio/bin/_pw-launch.mjs +52 -0
- package/apps/studio/bin/_pw-launch.test.mjs +90 -0
- package/apps/studio/bin/_smart-frames.mjs +419 -0
- package/apps/studio/bin/_smart-frames.test.mjs +140 -0
- package/apps/studio/bin/smart-frames.sh +30 -0
- package/apps/studio/canvas-edit.ts +302 -3
- package/apps/studio/canvas-lib.tsx +203 -11
- package/apps/studio/canvas-shell.tsx +167 -28
- package/apps/studio/client/app.jsx +678 -176
- package/apps/studio/client/panels/CapabilityBar.jsx +101 -0
- package/apps/studio/client/panels/ChatPanel.jsx +846 -133
- package/apps/studio/client/panels/ElicitationPrompt.jsx +429 -0
- package/apps/studio/client/panels/PermissionPrompt.jsx +119 -0
- package/apps/studio/client/panels/ReadinessList.jsx +17 -3
- package/apps/studio/client/panels/SettingsPanel.jsx +211 -0
- package/apps/studio/client/panels/ToolGroup.jsx +79 -0
- package/apps/studio/client/panels/acp-capabilities.js +82 -0
- package/apps/studio/client/panels/acp-elicitation.js +194 -0
- package/apps/studio/client/panels/acp-runtime.js +248 -9
- package/apps/studio/client/panels/acp-usage.js +65 -0
- package/apps/studio/client/panels/transcript-view.js +36 -0
- package/apps/studio/client/styles/6-acp-chat.css +689 -11
- package/apps/studio/dist/client.bundle.js +1738 -1736
- package/apps/studio/dist/comment-mount.js +2 -2
- package/apps/studio/dist/styles.css +1 -1
- package/apps/studio/dom-selection.ts +20 -0
- package/apps/studio/export-dialog.tsx +138 -18
- package/apps/studio/exporters/pdf.ts +332 -18
- package/apps/studio/exporters/png.ts +45 -3
- package/apps/studio/footage/schema.ts +17 -1
- package/apps/studio/generation/gemma-models.ts +224 -0
- package/apps/studio/generation/prefs.ts +43 -0
- package/apps/studio/generation/whisper-models.test.ts +28 -1
- package/apps/studio/generation/whisper-models.ts +22 -7
- package/apps/studio/http.ts +246 -9
- package/apps/studio/print/marks.ts +113 -0
- package/apps/studio/print/units.ts +269 -0
- package/apps/studio/print-overlay-content.tsx +132 -0
- package/apps/studio/test/acp-bridge.test.ts +11 -6
- package/apps/studio/test/acp-capabilities.test.ts +123 -0
- package/apps/studio/test/acp-caps-bridge.test.ts +274 -0
- package/apps/studio/test/acp-elicitation-bridge.test.ts +475 -0
- package/apps/studio/test/acp-elicitation.test.ts +251 -0
- package/apps/studio/test/acp-mode-banner.test.ts +45 -0
- package/apps/studio/test/acp-permission-prompt.test.ts +77 -0
- package/apps/studio/test/acp-permission.test.ts +262 -0
- package/apps/studio/test/acp-session-allowed-tools.test.ts +91 -0
- package/apps/studio/test/acp-toolgroup.test.ts +76 -0
- package/apps/studio/test/acp-transcript-view.test.ts +71 -0
- package/apps/studio/test/acp-transcript.test.ts +75 -2
- package/apps/studio/test/acp-usage-bridge.test.ts +136 -0
- package/apps/studio/test/acp-usage.test.ts +143 -0
- package/apps/studio/test/annotations-snap.test.ts +56 -0
- package/apps/studio/test/artboard-guides-overlay.test.tsx +152 -0
- package/apps/studio/test/artboard-kinds.test.tsx +83 -0
- package/apps/studio/test/artboard-selection-attrs.test.ts +69 -0
- package/apps/studio/test/canvas-meta-api.test.ts +167 -0
- package/apps/studio/test/element-structural-edit.test.ts +250 -0
- package/apps/studio/test/exporters/png.test.ts +49 -1
- package/apps/studio/test/fixtures/mock-acp-agent-caps.mjs +158 -0
- package/apps/studio/test/fixtures/mock-acp-agent-elicit-flood.mjs +46 -0
- package/apps/studio/test/fixtures/mock-acp-agent-elicit-url.mjs +42 -0
- package/apps/studio/test/fixtures/mock-acp-agent-elicit.mjs +63 -0
- package/apps/studio/test/fixtures/mock-acp-agent-permission-flood.mjs +51 -0
- package/apps/studio/test/fixtures/mock-acp-agent-permission.mjs +50 -0
- package/apps/studio/test/fixtures/mock-acp-agent-usage.mjs +56 -0
- package/apps/studio/test/import-asset.test.ts +31 -3
- package/apps/studio/test/pdf-print-boxes.test.ts +272 -0
- package/apps/studio/test/print-marks.test.ts +113 -0
- package/apps/studio/test/print-units.test.ts +173 -0
- package/apps/studio/test/use-snap-guides.test.ts +81 -0
- package/apps/studio/use-chrome-visibility.tsx +19 -0
- package/apps/studio/use-element-resize.tsx +21 -2
- package/apps/studio/use-snap-guides.tsx +73 -5
- package/apps/studio/use-spacing-handles.tsx +9 -5
- package/apps/studio/whats-new.json +62 -0
- package/cli/commands/design.mjs +116 -3
- package/cli/lib/pkg-root.mjs +42 -10
- package/cli/lib/pkg-root.test.mjs +33 -1
- package/package.json +11 -9
- package/plugins/design/dependencies.json +35 -0
|
@@ -12,6 +12,7 @@ import { tmpdir } from 'node:os';
|
|
|
12
12
|
import path from 'node:path';
|
|
13
13
|
|
|
14
14
|
import JSZip from 'jszip';
|
|
15
|
+
import { CSS_DPI } from '../print/units.ts';
|
|
15
16
|
import { exportShimPath, runShim } from './_runtime.ts';
|
|
16
17
|
import {
|
|
17
18
|
canvasShellUrl,
|
|
@@ -27,7 +28,8 @@ import type { Target } from './scope.ts';
|
|
|
27
28
|
const PNG_PLAYWRIGHT = exportShimPath('_png-playwright.mjs');
|
|
28
29
|
|
|
29
30
|
interface CaptureOptions {
|
|
30
|
-
|
|
31
|
+
/** Fully-resolved Playwright deviceScaleFactor — see resolveDeviceScale. */
|
|
32
|
+
scale?: number;
|
|
31
33
|
timeoutSec?: number;
|
|
32
34
|
}
|
|
33
35
|
|
|
@@ -42,6 +44,44 @@ export function clampScale(raw: unknown): 1 | 2 | 3 {
|
|
|
42
44
|
return 2;
|
|
43
45
|
}
|
|
44
46
|
|
|
47
|
+
/** feature-2-print-artboards T4 — supported DPI presets (Adobe-conventional
|
|
48
|
+
* print ladder). Exported for unit coverage. */
|
|
49
|
+
export const DPI_PRESETS = [96, 150, 300, 600] as const;
|
|
50
|
+
export type DpiPreset = (typeof DPI_PRESETS)[number];
|
|
51
|
+
|
|
52
|
+
/** Coerce an arbitrary `options.dpi` to the nearest supported preset, or
|
|
53
|
+
* `undefined` when absent/invalid (caller then falls back to `scale`). */
|
|
54
|
+
export function clampDpi(raw: unknown): DpiPreset | undefined {
|
|
55
|
+
if (raw === undefined || raw === null) return undefined;
|
|
56
|
+
const n = Number(raw);
|
|
57
|
+
if (!Number.isFinite(n)) return undefined;
|
|
58
|
+
let best: DpiPreset = DPI_PRESETS[0];
|
|
59
|
+
let bestDelta = Math.abs(n - best);
|
|
60
|
+
for (const preset of DPI_PRESETS) {
|
|
61
|
+
const delta = Math.abs(n - preset);
|
|
62
|
+
if (delta < bestDelta) {
|
|
63
|
+
best = preset;
|
|
64
|
+
bestDelta = delta;
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
return best;
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
/**
|
|
71
|
+
* The Playwright `deviceScaleFactor` for a capture — `dpi` (physical print
|
|
72
|
+
* resolution) wins when present, else the legacy 1×/2×/3× UI preset.
|
|
73
|
+
* `dpi/96` because authoring is at CSS px @96dpi (print/units.ts CSS_DPI) —
|
|
74
|
+
* this is the ONE place outside that module allowed to divide by 96 (a
|
|
75
|
+
* device-scale-factor derivation, not an mm→px conversion, so it's exempt
|
|
76
|
+
* from the T1 single-source `25.4` lint guard, but stays in lockstep with
|
|
77
|
+
* CSS_DPI by importing it rather than hardcoding 96 again).
|
|
78
|
+
*/
|
|
79
|
+
export function resolveDeviceScale(options: ExportOptions): number {
|
|
80
|
+
const dpi = clampDpi(options.dpi);
|
|
81
|
+
if (dpi !== undefined) return dpi / CSS_DPI;
|
|
82
|
+
return clampScale(options.scale);
|
|
83
|
+
}
|
|
84
|
+
|
|
45
85
|
async function captureElement(
|
|
46
86
|
target: Extract<Target, { kind: 'element' }>,
|
|
47
87
|
ctx: ExportContext,
|
|
@@ -112,8 +152,10 @@ export async function run(
|
|
|
112
152
|
const captureOpts: CaptureOptions = {
|
|
113
153
|
// Default 2× — a single-scale PNG was uselessly small (item 1). The dialog
|
|
114
154
|
// sends an explicit scale; this default covers direct API / curl callers.
|
|
115
|
-
//
|
|
116
|
-
|
|
155
|
+
// feature-2-print-artboards T4 — `options.dpi` (96/150/300/600) wins over
|
|
156
|
+
// the legacy 1–3 preset when present; the shim re-clamps deviceScaleFactor
|
|
157
|
+
// ≤ 8 and guards oversized output (see _png-playwright.mjs).
|
|
158
|
+
scale: resolveDeviceScale(options),
|
|
117
159
|
timeoutSec: (options.timeoutSec as number | undefined) ?? 8,
|
|
118
160
|
};
|
|
119
161
|
|
|
@@ -150,6 +150,10 @@ export interface FootageAnalysis {
|
|
|
150
150
|
summary?: string;
|
|
151
151
|
/** Cross-cutting tags ("rebrand", "exterior", "people"). */
|
|
152
152
|
tags?: string[];
|
|
153
|
+
/** Optional audio/speech note — gist + language of what is said, filled ONLY when
|
|
154
|
+
* an orchestrator (e.g. /design:video-analyze) hands the analyst a transcript.
|
|
155
|
+
* Absent for the default vision-only path (e.g. /design:reel). */
|
|
156
|
+
speech?: string;
|
|
153
157
|
}
|
|
154
158
|
|
|
155
159
|
/** An optional graphic overlay laid over a beat (title card / lower-third / logo). */
|
|
@@ -437,7 +441,18 @@ export function validateFootageAnalysis(input: unknown): ValidationResult {
|
|
|
437
441
|
assertKeys(
|
|
438
442
|
errors,
|
|
439
443
|
input,
|
|
440
|
-
[
|
|
444
|
+
[
|
|
445
|
+
'version',
|
|
446
|
+
'asset',
|
|
447
|
+
'durationSec',
|
|
448
|
+
'width',
|
|
449
|
+
'height',
|
|
450
|
+
'keyframes',
|
|
451
|
+
'shots',
|
|
452
|
+
'summary',
|
|
453
|
+
'tags',
|
|
454
|
+
'speech',
|
|
455
|
+
],
|
|
441
456
|
'root'
|
|
442
457
|
);
|
|
443
458
|
if ('version' in input && input.version != null && typeof input.version !== 'number')
|
|
@@ -449,6 +464,7 @@ export function validateFootageAnalysis(input: unknown): ValidationResult {
|
|
|
449
464
|
intGe(errors, input, 'keyframes', 0, 'root');
|
|
450
465
|
str(errors, input, 'summary', 4000, 'root');
|
|
451
466
|
strArray(errors, input, 'tags', 64, 'root');
|
|
467
|
+
str(errors, input, 'speech', 4000, 'root');
|
|
452
468
|
|
|
453
469
|
const dur = typeof input.durationSec === 'number' ? input.durationSec : Infinity;
|
|
454
470
|
if ('shots' in input && input.shots != null) {
|
|
@@ -0,0 +1,224 @@
|
|
|
1
|
+
// generation/gemma-models.ts — managed local Gemma-4 MLX models for the `gemma`
|
|
2
|
+
// tier of `maude design smart-frames` (feature-scene-aware-keyframes). The
|
|
3
|
+
// "one-click Gemma scout" story, mirroring whisper-models.ts — but with three
|
|
4
|
+
// real differences from whisper:
|
|
5
|
+
//
|
|
6
|
+
// 1. Apple-Silicon only. The scout runs through mlx-vlm (a Python package); the
|
|
7
|
+
// model is useless without it. So the download is GATED on `mlxVlmAvailable`.
|
|
8
|
+
// 2. Multi-file HF snapshot, not a single .bin. mlx-vlm resolves a model from the
|
|
9
|
+
// standard HuggingFace hub cache, so we download THERE (via huggingface_hub,
|
|
10
|
+
// a transitive dep of mlx-vlm) rather than a Maude-private cache — otherwise
|
|
11
|
+
// mlx-vlm wouldn't find it. "downloaded" = the snapshot is in the HF cache.
|
|
12
|
+
// 3. The RUNTIME (mlx-vlm itself) is a manual `pip install` the app can't do for
|
|
13
|
+
// you — the Settings card says so and only the MODEL half is one click.
|
|
14
|
+
//
|
|
15
|
+
// This module owns the registry + availability probes + the HF-cache-aware
|
|
16
|
+
// downloaded check + resolve. The download SPAWN lives here (it shells the mlx
|
|
17
|
+
// Python's huggingface_hub); the http route wraps it with progress state, next to
|
|
18
|
+
// the other provider egress.
|
|
19
|
+
|
|
20
|
+
import { spawn, spawnSync } from 'node:child_process';
|
|
21
|
+
import { existsSync, readdirSync, statSync } from 'node:fs';
|
|
22
|
+
import { homedir } from 'node:os';
|
|
23
|
+
import { join } from 'node:path';
|
|
24
|
+
|
|
25
|
+
export interface GemmaModelDescriptor {
|
|
26
|
+
/** Stable id used by the pref + route. */
|
|
27
|
+
id: string;
|
|
28
|
+
/** The mlx-community HF repo (also the `--model` mlx-vlm resolves). */
|
|
29
|
+
repo: string;
|
|
30
|
+
/** Pinned commit SHA — the download fetches THIS revision, never floating `main`,
|
|
31
|
+
* so a poisoned-main / namespace-reuse push can't land arbitrary model files
|
|
32
|
+
* (DDR-183 supply-chain finding; the whisper baseline pins exact file URLs). */
|
|
33
|
+
revision: string;
|
|
34
|
+
label: string;
|
|
35
|
+
/** Approximate on-disk size (consent copy). */
|
|
36
|
+
sizeMB: number;
|
|
37
|
+
note: string;
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
// Curated allowlist of Gemma-4 MLX scout models. `repo` + `revision` are the frozen
|
|
41
|
+
// download target (never user input). e4b is the default (better beats); e2b is the
|
|
42
|
+
// small, fast option for lower-RAM Macs. Revisions pinned 2026-07-16.
|
|
43
|
+
export const GEMMA_MODELS: readonly GemmaModelDescriptor[] = [
|
|
44
|
+
{
|
|
45
|
+
id: 'gemma-4-e4b-it-4bit',
|
|
46
|
+
repo: 'mlx-community/gemma-4-e4b-it-4bit',
|
|
47
|
+
revision: '475b9088d29754a3379866cf5aeb6b41acd313c2',
|
|
48
|
+
label: 'Gemma 4 E4B (4-bit)',
|
|
49
|
+
sizeMB: 3300,
|
|
50
|
+
note: 'The recommended scout — best semantic beats. ~3.3 GB. Needs an Apple-Silicon Mac + mlx-vlm.',
|
|
51
|
+
},
|
|
52
|
+
{
|
|
53
|
+
id: 'gemma-4-e2b-it-4bit',
|
|
54
|
+
repo: 'mlx-community/gemma-4-e2b-it-4bit',
|
|
55
|
+
revision: '238767527555cb75a05732a84dff5d6ba0dd6809',
|
|
56
|
+
label: 'Gemma 4 E2B (4-bit)',
|
|
57
|
+
sizeMB: 1900,
|
|
58
|
+
note: 'Smaller/faster, coarser beats. ~1.9 GB. Good for 16 GB Macs.',
|
|
59
|
+
},
|
|
60
|
+
];
|
|
61
|
+
|
|
62
|
+
export function getGemmaModel(id: unknown): GemmaModelDescriptor | null {
|
|
63
|
+
return GEMMA_MODELS.find((m) => m.id === id) ?? null;
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
/** The HuggingFace hub cache dir mlx-vlm resolves models from (HF_HOME-aware). */
|
|
67
|
+
export function hfHubDir(): string {
|
|
68
|
+
const hfHome = process.env.HF_HOME;
|
|
69
|
+
if (hfHome && hfHome.length > 0) return join(hfHome, 'hub');
|
|
70
|
+
const base = process.env.XDG_CACHE_HOME || join(homedir(), '.cache');
|
|
71
|
+
return join(base, 'huggingface', 'hub');
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
/** HF snapshot dir name for a repo: `models--<org>--<name>`. */
|
|
75
|
+
function repoCacheDir(repo: string): string {
|
|
76
|
+
return join(hfHubDir(), `models--${repo.replace('/', '--')}`);
|
|
77
|
+
}
|
|
78
|
+
|
|
79
|
+
/** A model counts as downloaded when its HF snapshot dir has a non-empty snapshot. */
|
|
80
|
+
export function gemmaModelDownloaded(m: GemmaModelDescriptor): boolean {
|
|
81
|
+
const snaps = join(repoCacheDir(m.repo), 'snapshots');
|
|
82
|
+
try {
|
|
83
|
+
if (!existsSync(snaps)) return false;
|
|
84
|
+
for (const rev of readdirSync(snaps)) {
|
|
85
|
+
const revDir = join(snaps, rev);
|
|
86
|
+
try {
|
|
87
|
+
if (statSync(revDir).isDirectory() && readdirSync(revDir).length > 0) return true;
|
|
88
|
+
} catch {
|
|
89
|
+
/* skip */
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
} catch {
|
|
93
|
+
/* not downloaded */
|
|
94
|
+
}
|
|
95
|
+
return false;
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
export interface GemmaModelStatus extends GemmaModelDescriptor {
|
|
99
|
+
downloaded: boolean;
|
|
100
|
+
}
|
|
101
|
+
|
|
102
|
+
export function listGemmaModels(): GemmaModelStatus[] {
|
|
103
|
+
return GEMMA_MODELS.map((m) => ({ ...m, downloaded: gemmaModelDownloaded(m) }));
|
|
104
|
+
}
|
|
105
|
+
|
|
106
|
+
/** Resolve a Python that can `import mlx_vlm` (honors $MAUDE_MLX_PYTHON), or null. */
|
|
107
|
+
export function resolveMlxPython(): string | null {
|
|
108
|
+
const candidates = [process.env.MAUDE_MLX_PYTHON, 'python3', 'python'].filter(
|
|
109
|
+
Boolean
|
|
110
|
+
) as string[];
|
|
111
|
+
for (const py of candidates) {
|
|
112
|
+
const r = spawnSync(py, ['-c', 'import mlx_vlm'], { stdio: 'ignore' });
|
|
113
|
+
if (r.status === 0) return py;
|
|
114
|
+
}
|
|
115
|
+
return null;
|
|
116
|
+
}
|
|
117
|
+
|
|
118
|
+
// Availability probes spawn subprocesses (importing mlx_vlm can take hundreds of ms
|
|
119
|
+
// to seconds). The GET /_api/generate/keyframe-model route is un-CSRF-gated (correct
|
|
120
|
+
// for a read), so a cross-origin drive-by could hammer it and stall the single-
|
|
121
|
+
// threaded Bun event loop on synchronous spawns (security-auditor DDR-183 finding).
|
|
122
|
+
// Memoize with a short TTL: DoS-bounded to at most one probe per PROBE_TTL_MS
|
|
123
|
+
// regardless of request rate, while still picking up a mid-session `pip install` /
|
|
124
|
+
// PATH change within the TTL.
|
|
125
|
+
const PROBE_TTL_MS = 30_000;
|
|
126
|
+
const probeCache = new Map<string, { at: number; value: boolean }>();
|
|
127
|
+
|
|
128
|
+
function cachedProbe(key: string, compute: () => boolean): boolean {
|
|
129
|
+
const now = Date.now();
|
|
130
|
+
const hit = probeCache.get(key);
|
|
131
|
+
if (hit && now - hit.at < PROBE_TTL_MS) return hit.value;
|
|
132
|
+
const value = compute();
|
|
133
|
+
probeCache.set(key, { at: now, value });
|
|
134
|
+
return value;
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
export function mlxVlmAvailable(): boolean {
|
|
138
|
+
return cachedProbe('mlx', () => resolveMlxPython() !== null);
|
|
139
|
+
}
|
|
140
|
+
|
|
141
|
+
export function ffmpegAvailable(): boolean {
|
|
142
|
+
return cachedProbe('ffmpeg', () => {
|
|
143
|
+
const finder = process.platform === 'win32' ? 'where' : 'which';
|
|
144
|
+
return (
|
|
145
|
+
spawnSync(finder, ['ffmpeg'], { stdio: 'ignore' }).status === 0 &&
|
|
146
|
+
spawnSync(finder, ['ffprobe'], { stdio: 'ignore' }).status === 0
|
|
147
|
+
);
|
|
148
|
+
});
|
|
149
|
+
}
|
|
150
|
+
|
|
151
|
+
/**
|
|
152
|
+
* Download a Gemma model into the HF hub cache via huggingface_hub (a transitive
|
|
153
|
+
* dep of mlx-vlm, so it's present whenever the scout can actually run). We delegate
|
|
154
|
+
* to Python rather than reimplement multi-file HF snapshot download + LFS/Xet
|
|
155
|
+
* redirect handling in TS — the repo id is frozen (never user input), the runtime
|
|
156
|
+
* is pinned to the mlx Python, and the route is loopback + same-origin only.
|
|
157
|
+
* Progress is coarse (tqdm on stderr → a heartbeat), which the route surfaces as an
|
|
158
|
+
* in-flight state. Rejects if mlx-vlm/Python is absent (you couldn't run the model
|
|
159
|
+
* anyway) or the process exits non-zero.
|
|
160
|
+
*/
|
|
161
|
+
export function downloadGemmaModel(
|
|
162
|
+
id: string,
|
|
163
|
+
onProgress: (received: number, total: number) => void,
|
|
164
|
+
signal?: AbortSignal
|
|
165
|
+
): Promise<void> {
|
|
166
|
+
const m = getGemmaModel(id);
|
|
167
|
+
if (!m) return Promise.reject(new Error(`unknown gemma model: ${id}`));
|
|
168
|
+
const py = resolveMlxPython();
|
|
169
|
+
if (!py)
|
|
170
|
+
return Promise.reject(
|
|
171
|
+
new Error(
|
|
172
|
+
'mlx-vlm not installed — the Gemma scout needs an Apple-Silicon Mac + `pip install mlx-vlm`.'
|
|
173
|
+
)
|
|
174
|
+
);
|
|
175
|
+
|
|
176
|
+
// Refuse a steered HF endpoint (mirror/redirect) — the whisper path pins the host;
|
|
177
|
+
// huggingface_hub otherwise honors HF_ENDPOINT/HF_HUB_ENDPOINT. Keep the pull on
|
|
178
|
+
// the canonical hub (DDR-183 supply-chain finding).
|
|
179
|
+
const endpoint = process.env.HF_ENDPOINT || process.env.HF_HUB_ENDPOINT;
|
|
180
|
+
if (endpoint && !/^https:\/\/huggingface\.co\/?$/.test(endpoint))
|
|
181
|
+
return Promise.reject(new Error(`refusing model download: HF endpoint steered to ${endpoint}`));
|
|
182
|
+
|
|
183
|
+
const total = m.sizeMB * 1024 * 1024;
|
|
184
|
+
return new Promise((resolve, reject) => {
|
|
185
|
+
// snapshot_download(repo, revision=<pinned sha>) — repo AND revision are from the
|
|
186
|
+
// frozen registry (never user input), passed as argv (not interpolated into -c).
|
|
187
|
+
const child = spawn(
|
|
188
|
+
py,
|
|
189
|
+
[
|
|
190
|
+
'-c',
|
|
191
|
+
'import sys; from huggingface_hub import snapshot_download; snapshot_download(sys.argv[1], revision=sys.argv[2])',
|
|
192
|
+
m.repo,
|
|
193
|
+
m.revision,
|
|
194
|
+
],
|
|
195
|
+
{
|
|
196
|
+
stdio: ['ignore', 'ignore', 'pipe'],
|
|
197
|
+
env: { ...process.env, HF_HUB_DISABLE_TELEMETRY: '1' },
|
|
198
|
+
}
|
|
199
|
+
);
|
|
200
|
+
let received = 0;
|
|
201
|
+
const onAbort = () => child.kill('SIGTERM');
|
|
202
|
+
signal?.addEventListener('abort', onAbort, { once: true });
|
|
203
|
+
// tqdm writes percentage lines to stderr; use them as a coarse heartbeat.
|
|
204
|
+
child.stderr?.on('data', (buf: Buffer) => {
|
|
205
|
+
const s = buf.toString();
|
|
206
|
+
const pct = /(\d{1,3})%/.exec(s);
|
|
207
|
+
if (pct) received = Math.min(total, Math.round((Number(pct[1]) / 100) * total));
|
|
208
|
+
onProgress(received, total);
|
|
209
|
+
});
|
|
210
|
+
child.on('error', (err) => {
|
|
211
|
+
signal?.removeEventListener('abort', onAbort);
|
|
212
|
+
reject(err);
|
|
213
|
+
});
|
|
214
|
+
child.on('close', (code) => {
|
|
215
|
+
signal?.removeEventListener('abort', onAbort);
|
|
216
|
+
if (code === 0 && gemmaModelDownloaded(m)) {
|
|
217
|
+
onProgress(total, total);
|
|
218
|
+
resolve();
|
|
219
|
+
} else {
|
|
220
|
+
reject(new Error(`gemma model download failed (exit ${code})`));
|
|
221
|
+
}
|
|
222
|
+
});
|
|
223
|
+
});
|
|
224
|
+
}
|
|
@@ -20,6 +20,16 @@ export function isTranscriptionProvider(v: unknown): v is TranscriptionProvider
|
|
|
20
20
|
return typeof v === 'string' && (TRANSCRIPTION_PROVIDERS as readonly string[]).includes(v);
|
|
21
21
|
}
|
|
22
22
|
|
|
23
|
+
/** Scene-aware keyframe engines (feature-scene-aware-keyframes) — the tier the
|
|
24
|
+
* `gemma` scout → `ffmpeg` scene-detect → `blind` Chromium fallback ladder runs
|
|
25
|
+
* in. `auto` self-detects installed deps. */
|
|
26
|
+
export const KEYFRAME_ENGINES = ['auto', 'gemma', 'ffmpeg', 'blind'] as const;
|
|
27
|
+
export type KeyframeEngine = (typeof KEYFRAME_ENGINES)[number];
|
|
28
|
+
|
|
29
|
+
export function isKeyframeEngine(v: unknown): v is KeyframeEngine {
|
|
30
|
+
return typeof v === 'string' && (KEYFRAME_ENGINES as readonly string[]).includes(v);
|
|
31
|
+
}
|
|
32
|
+
|
|
23
33
|
function configPath(repoRoot: string): string {
|
|
24
34
|
return join(repoRoot, '.design', 'config.json');
|
|
25
35
|
}
|
|
@@ -80,3 +90,36 @@ export async function writeTranscriptionProvider(
|
|
|
80
90
|
await Bun.write(configPath(repoRoot), `${JSON.stringify(next, null, 2)}\n`);
|
|
81
91
|
return true;
|
|
82
92
|
}
|
|
93
|
+
|
|
94
|
+
/** The current scene-aware keyframe engine preference, or 'auto' (the default). */
|
|
95
|
+
export function readKeyframeEngine(repoRoot: string): KeyframeEngine {
|
|
96
|
+
const cfg = readConfig(repoRoot);
|
|
97
|
+
const gen = cfg.generation as Record<string, unknown> | undefined;
|
|
98
|
+
const kf = gen?.keyframes as Record<string, unknown> | undefined;
|
|
99
|
+
return isKeyframeEngine(kf?.engine) ? (kf?.engine as KeyframeEngine) : 'auto';
|
|
100
|
+
}
|
|
101
|
+
|
|
102
|
+
/**
|
|
103
|
+
* Persist the keyframe-engine choice into `.design/config.json`, preserving every
|
|
104
|
+
* other field (same additive-merge + fail-closed-on-corrupt discipline as the
|
|
105
|
+
* transcription writer).
|
|
106
|
+
*/
|
|
107
|
+
export async function writeKeyframeEngine(repoRoot: string, engine: string): Promise<boolean> {
|
|
108
|
+
if (!isKeyframeEngine(engine)) throw new Error(`invalid keyframe engine: ${engine}`);
|
|
109
|
+
const p = configPath(repoRoot);
|
|
110
|
+
if (existsSync(p)) {
|
|
111
|
+
try {
|
|
112
|
+
JSON.parse(readFileSync(p, 'utf8'));
|
|
113
|
+
} catch {
|
|
114
|
+
throw new Error(
|
|
115
|
+
'.design/config.json is present but not valid JSON — fix it before changing generation prefs (refusing to overwrite it)'
|
|
116
|
+
);
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
const cfg = readConfig(repoRoot);
|
|
120
|
+
const gen = (cfg.generation as Record<string, unknown>) ?? {};
|
|
121
|
+
const keyframes = (gen.keyframes as Record<string, unknown>) ?? {};
|
|
122
|
+
const next = { ...cfg, generation: { ...gen, keyframes: { ...keyframes, engine } } };
|
|
123
|
+
await Bun.write(configPath(repoRoot), `${JSON.stringify(next, null, 2)}\n`);
|
|
124
|
+
return true;
|
|
125
|
+
}
|
|
@@ -125,7 +125,9 @@ describe('downloadWhisperModel', () => {
|
|
|
125
125
|
headers: new Headers(),
|
|
126
126
|
body: new Response(new Uint8Array(1)).body,
|
|
127
127
|
})) as unknown as typeof fetch;
|
|
128
|
-
await expect(downloadWhisperModel('tiny', () => {})).rejects.toThrow(
|
|
128
|
+
await expect(downloadWhisperModel('tiny', () => {})).rejects.toThrow(
|
|
129
|
+
/off the allowed HF hosts/
|
|
130
|
+
);
|
|
129
131
|
expect(existsSync(join(whisperModelsDir(), 'ggml-tiny.bin'))).toBe(false);
|
|
130
132
|
});
|
|
131
133
|
|
|
@@ -140,6 +142,31 @@ describe('downloadWhisperModel', () => {
|
|
|
140
142
|
const path = await downloadWhisperModel('tiny', () => {});
|
|
141
143
|
expect(existsSync(path)).toBe(true);
|
|
142
144
|
});
|
|
145
|
+
|
|
146
|
+
test("accepts a redirect that lands on HF's Xet CDN (xethub.hf.co)", async () => {
|
|
147
|
+
const bytes = new Uint8Array(512);
|
|
148
|
+
globalThis.fetch = (async () => ({
|
|
149
|
+
url: 'https://cas-bridge.xethub.hf.co/repos/x/ggml-tiny.bin',
|
|
150
|
+
ok: true,
|
|
151
|
+
headers: new Headers({ 'content-length': String(bytes.byteLength) }),
|
|
152
|
+
body: new Response(bytes).body,
|
|
153
|
+
})) as unknown as typeof fetch;
|
|
154
|
+
const path = await downloadWhisperModel('tiny', () => {});
|
|
155
|
+
expect(existsSync(path)).toBe(true);
|
|
156
|
+
});
|
|
157
|
+
|
|
158
|
+
test('rejects a lookalike host that merely contains an allowed apex', async () => {
|
|
159
|
+
globalThis.fetch = (async () => ({
|
|
160
|
+
url: 'https://xethub.hf.co.evil.example/ggml-tiny.bin',
|
|
161
|
+
ok: true,
|
|
162
|
+
headers: new Headers(),
|
|
163
|
+
body: new Response(new Uint8Array(1)).body,
|
|
164
|
+
})) as unknown as typeof fetch;
|
|
165
|
+
await expect(downloadWhisperModel('tiny', () => {})).rejects.toThrow(
|
|
166
|
+
/off the allowed HF hosts/
|
|
167
|
+
);
|
|
168
|
+
expect(existsSync(join(whisperModelsDir(), 'ggml-tiny.bin'))).toBe(false);
|
|
169
|
+
});
|
|
143
170
|
});
|
|
144
171
|
|
|
145
172
|
describe('registry integrity', () => {
|
|
@@ -81,6 +81,18 @@ export const WHISPER_MODELS: readonly WhisperModelDescriptor[] = [
|
|
|
81
81
|
const MODEL_HOST = 'https://huggingface.co';
|
|
82
82
|
const MODEL_REPO_PATH = '/ggerganov/whisper.cpp/resolve/main';
|
|
83
83
|
|
|
84
|
+
// Apex domains a model-download redirect is allowed to land on. `huggingface.co`
|
|
85
|
+
// is the classic Git-LFS CDN; `xethub.hf.co` is HF's newer Xet storage backend,
|
|
86
|
+
// which several repos (incl. ggerganov/whisper.cpp) have been migrated to — its
|
|
87
|
+
// edge nodes (e.g. `cas-bridge.xethub.hf.co`) are a different apex entirely, not
|
|
88
|
+
// a subdomain of huggingface.co. Anchored subdomain match only (never
|
|
89
|
+
// `includes()`), so `xethub.hf.co.evil.com` / `evilxethub.hf.co` still reject.
|
|
90
|
+
const ALLOWED_REDIRECT_APEXES = ['huggingface.co', 'xethub.hf.co'] as const;
|
|
91
|
+
|
|
92
|
+
function isAllowedRedirectHost(hostname: string): boolean {
|
|
93
|
+
return ALLOWED_REDIRECT_APEXES.some((apex) => hostname === apex || hostname.endsWith(`.${apex}`));
|
|
94
|
+
}
|
|
95
|
+
|
|
84
96
|
/** The fixed, non-interpolated download URL for a model (SSRF-safe — the file
|
|
85
97
|
* comes from the frozen registry above, never from a request). */
|
|
86
98
|
export function whisperModelUrl(m: WhisperModelDescriptor): string {
|
|
@@ -163,7 +175,9 @@ export function resolveWhisperModel(preferId?: string): string | null {
|
|
|
163
175
|
* (Task 2.7). Egress discipline: the URL is derived from the FROZEN registry
|
|
164
176
|
* (initial host is always huggingface.co, asserted https + fixed host); the
|
|
165
177
|
* redirect the HF blob store issues is followed but the FINAL hop is re-asserted
|
|
166
|
-
* https + a `*.huggingface.co`
|
|
178
|
+
* https + a host on `ALLOWED_REDIRECT_APEXES` (`*.huggingface.co` for the
|
|
179
|
+
* classic LFS CDN, `*.xethub.hf.co` for HF's newer Xet storage backend —
|
|
180
|
+
* ethical-hacker Finding 2 plus the Xet-migration follow-up); the body is
|
|
167
181
|
* size-capped per-chunk to the model's expected size (+20% margin) before it can
|
|
168
182
|
* fill the disk; written to a `.part` temp and atomically renamed so a
|
|
169
183
|
* partial/aborted download never looks complete. Pass an AbortSignal (a timeout)
|
|
@@ -190,15 +204,16 @@ export async function downloadWhisperModel(
|
|
|
190
204
|
const tmpPath = `${finalPath}.part`;
|
|
191
205
|
|
|
192
206
|
const cap = Math.ceil(m.sizeMB * 1.2) * 1024 * 1024; // expected size + 20% margin
|
|
193
|
-
// huggingface.co 302-redirects model blobs to its own LFS CDN, so we must
|
|
194
|
-
// follow — but re-assert the FINAL hop is still https + a
|
|
195
|
-
// (ethical-hacker Finding 2): an open-redirect/on-path
|
|
196
|
-
// on an arbitrary host. A wall-clock timeout is
|
|
207
|
+
// huggingface.co 302-redirects model blobs to its own LFS/Xet CDN, so we must
|
|
208
|
+
// follow — but re-assert the FINAL hop is still https + a host on
|
|
209
|
+
// ALLOWED_REDIRECT_APEXES (ethical-hacker Finding 2): an open-redirect/on-path
|
|
210
|
+
// must not land the fetch on an arbitrary host. A wall-clock timeout is
|
|
211
|
+
// layered by the caller's signal.
|
|
197
212
|
const res = await fetch(url, { signal, redirect: 'follow' });
|
|
198
213
|
try {
|
|
199
214
|
const finalUrl = new URL(res.url || url);
|
|
200
|
-
if (finalUrl.protocol !== 'https:' ||
|
|
201
|
-
throw new Error(`model download redirected off
|
|
215
|
+
if (finalUrl.protocol !== 'https:' || !isAllowedRedirectHost(finalUrl.hostname))
|
|
216
|
+
throw new Error(`model download redirected off the allowed HF hosts (${finalUrl.hostname})`);
|
|
202
217
|
} catch (err) {
|
|
203
218
|
if (err instanceof Error && err.message.startsWith('model download redirected')) throw err;
|
|
204
219
|
throw new Error('model download resolved to an invalid URL');
|