@mmnto/cli 1.10.0 → 1.10.1

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.
Files changed (35) hide show
  1. package/LICENSE +190 -190
  2. package/dist/commands/install-hooks.d.ts.map +1 -1
  3. package/dist/commands/install-hooks.js +28 -19
  4. package/dist/commands/install-hooks.js.map +1 -1
  5. package/dist/commands/install-hooks.test.js +4 -7
  6. package/dist/commands/install-hooks.test.js.map +1 -1
  7. package/dist/commands/review-alias.test.js +9 -15
  8. package/dist/commands/review-alias.test.js.map +1 -1
  9. package/dist/exemptions/__tests__/exemption-engine.test.js +9 -0
  10. package/dist/exemptions/__tests__/exemption-engine.test.js.map +1 -1
  11. package/dist/exemptions/exemption-engine.js +1 -1
  12. package/dist/exemptions/exemption-engine.js.map +1 -1
  13. package/dist/utils.test.js +7 -1
  14. package/dist/utils.test.js.map +1 -1
  15. package/package.json +4 -3
  16. package/dist/commands/anchor.d.ts +0 -2
  17. package/dist/commands/anchor.d.ts.map +0 -1
  18. package/dist/commands/anchor.js +0 -84
  19. package/dist/commands/anchor.js.map +0 -1
  20. package/dist/commands/config-cmd.d.ts +0 -4
  21. package/dist/commands/config-cmd.d.ts.map +0 -1
  22. package/dist/commands/config-cmd.js +0 -60
  23. package/dist/commands/config-cmd.js.map +0 -1
  24. package/dist/commands/learn.d.ts +0 -26
  25. package/dist/commands/learn.d.ts.map +0 -1
  26. package/dist/commands/learn.js +0 -325
  27. package/dist/commands/learn.js.map +0 -1
  28. package/dist/commands/learn.test.d.ts +0 -2
  29. package/dist/commands/learn.test.d.ts.map +0 -1
  30. package/dist/commands/learn.test.js +0 -169
  31. package/dist/commands/learn.test.js.map +0 -1
  32. package/dist/orchestrator.test.d.ts +0 -2
  33. package/dist/orchestrator.test.d.ts.map +0 -1
  34. package/dist/orchestrator.test.js +0 -130
  35. package/dist/orchestrator.test.js.map +0 -1
@@ -1,169 +0,0 @@
1
- import * as fs from 'node:fs';
2
- import * as os from 'node:os';
3
- import * as path from 'node:path';
4
- import { Readable, Writable } from 'node:stream';
5
- import { afterEach, beforeEach, describe, expect, it } from 'vitest';
6
- import { appendLessons, confirmLessons, parseLessons } from './learn.js';
7
- // ─── parseLessons ───────────────────────────────────────
8
- describe('parseLessons', () => {
9
- it('extracts a single lesson', () => {
10
- const output = `---LESSON---
11
- Tags: git, cli, trap
12
- Always check for ENOENT separately from other errors.
13
- ---END---`;
14
- const lessons = parseLessons(output);
15
- expect(lessons).toHaveLength(1);
16
- expect(lessons[0]).toEqual({
17
- tags: ['git', 'cli', 'trap'],
18
- text: 'Always check for ENOENT separately from other errors.',
19
- });
20
- });
21
- it('extracts multiple lessons', () => {
22
- const output = `---LESSON---
23
- Tags: adapter, DRY
24
- Extract shared fetch logic into a helper immediately.
25
- ---END---
26
-
27
- ---LESSON---
28
- Tags: security, input
29
- Sanitize all user input before writing to files.
30
- ---END---`;
31
- const lessons = parseLessons(output);
32
- expect(lessons).toHaveLength(2);
33
- expect(lessons[0].tags).toEqual(['adapter', 'DRY']);
34
- expect(lessons[1].tags).toEqual(['security', 'input']);
35
- });
36
- it('returns empty array for NONE', () => {
37
- expect(parseLessons('NONE')).toEqual([]);
38
- });
39
- it('returns empty array for NONE with whitespace', () => {
40
- expect(parseLessons(' NONE ')).toEqual([]);
41
- });
42
- it('handles multi-line lesson text', () => {
43
- const output = `---LESSON---
44
- Tags: architecture
45
- First line of the lesson.
46
- Second line with more detail.
47
- ---END---`;
48
- const lessons = parseLessons(output);
49
- expect(lessons).toHaveLength(1);
50
- expect(lessons[0].text).toBe('First line of the lesson.\nSecond line with more detail.');
51
- });
52
- it('strips empty tags', () => {
53
- const output = `---LESSON---
54
- Tags: git, , cli,
55
- A lesson about git.
56
- ---END---`;
57
- const lessons = parseLessons(output);
58
- expect(lessons[0].tags).toEqual(['git', 'cli']);
59
- });
60
- it('skips lessons with empty text', () => {
61
- const output = `---LESSON---
62
- Tags: empty
63
- ---END---`;
64
- const lessons = parseLessons(output);
65
- expect(lessons).toEqual([]);
66
- });
67
- });
68
- // sanitize tests are in utils.test.ts (sanitize now lives in utils.ts)
69
- // ─── appendLessons ──────────────────────────────────────
70
- describe('appendLessons', () => {
71
- let tmpDir;
72
- let lessonsPath;
73
- beforeEach(() => {
74
- tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'totem-learn-'));
75
- lessonsPath = path.join(tmpDir, '.totem', 'lessons.md');
76
- });
77
- afterEach(() => {
78
- fs.rmSync(tmpDir, { recursive: true, force: true });
79
- });
80
- it('creates directory and file if they do not exist', () => {
81
- appendLessons([{ tags: ['test'], text: 'A test lesson.' }], lessonsPath);
82
- expect(fs.existsSync(lessonsPath)).toBe(true);
83
- const content = fs.readFileSync(lessonsPath, 'utf-8');
84
- expect(content).toContain('**Tags:** test');
85
- expect(content).toContain('A test lesson.');
86
- });
87
- it('appends to existing file', () => {
88
- fs.mkdirSync(path.dirname(lessonsPath), { recursive: true });
89
- fs.writeFileSync(lessonsPath, '# Existing content\n');
90
- appendLessons([{ tags: ['new'], text: 'New lesson.' }], lessonsPath);
91
- const content = fs.readFileSync(lessonsPath, 'utf-8');
92
- expect(content).toContain('# Existing content');
93
- expect(content).toContain('New lesson.');
94
- });
95
- it('writes multiple lessons', () => {
96
- appendLessons([
97
- { tags: ['a', 'b'], text: 'First.' },
98
- { tags: ['c'], text: 'Second.' },
99
- ], lessonsPath);
100
- const content = fs.readFileSync(lessonsPath, 'utf-8');
101
- expect(content).toContain('**Tags:** a, b');
102
- expect(content).toContain('First.');
103
- expect(content).toContain('**Tags:** c');
104
- expect(content).toContain('Second.');
105
- });
106
- it('includes ISO timestamp in heading', () => {
107
- appendLessons([{ tags: ['test'], text: 'Timestamped.' }], lessonsPath);
108
- const content = fs.readFileSync(lessonsPath, 'utf-8');
109
- // Match ISO 8601 pattern in heading
110
- expect(content).toMatch(/## Lesson — \d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}/);
111
- });
112
- });
113
- // ─── confirmLessons ─────────────────────────────────────
114
- function makeInput(response) {
115
- const input = new Readable({ read() { } });
116
- // Push response after a tick so readline can consume it
117
- setImmediate(() => {
118
- input.push(response + '\n');
119
- input.push(null);
120
- });
121
- return input;
122
- }
123
- const nullOutput = new Writable({
124
- write(_chunk, _enc, cb) {
125
- cb();
126
- },
127
- });
128
- describe('confirmLessons', () => {
129
- it('returns true when --yes is set', async () => {
130
- const result = await confirmLessons(3, { yes: true, isTTY: false });
131
- expect(result).toBe(true);
132
- });
133
- it('throws in non-TTY without --yes', async () => {
134
- await expect(confirmLessons(3, { isTTY: false })).rejects.toThrow('[Totem Error] Refusing to write lessons in non-interactive mode. Use --yes to bypass confirmation.');
135
- });
136
- it('returns true when user confirms with empty input (default Y)', async () => {
137
- const result = await confirmLessons(2, {
138
- isTTY: true,
139
- input: makeInput(''),
140
- output: nullOutput,
141
- });
142
- expect(result).toBe(true);
143
- });
144
- it('returns true when user types y', async () => {
145
- const result = await confirmLessons(2, {
146
- isTTY: true,
147
- input: makeInput('y'),
148
- output: nullOutput,
149
- });
150
- expect(result).toBe(true);
151
- });
152
- it('returns false when user types n', async () => {
153
- const result = await confirmLessons(2, {
154
- isTTY: true,
155
- input: makeInput('n'),
156
- output: nullOutput,
157
- });
158
- expect(result).toBe(false);
159
- });
160
- it('returns false when user types N', async () => {
161
- const result = await confirmLessons(1, {
162
- isTTY: true,
163
- input: makeInput('N'),
164
- output: nullOutput,
165
- });
166
- expect(result).toBe(false);
167
- });
168
- });
169
- //# sourceMappingURL=learn.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"learn.test.js","sourceRoot":"","sources":["../../src/commands/learn.test.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEjD,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAErE,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAEzE,2DAA2D;AAE3D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,MAAM,GAAG;;;UAGT,CAAC;QACP,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC;YAC5B,IAAI,EAAE,uDAAuD;SAC9D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,MAAM,GAAG;;;;;;;;UAQT,CAAC;QACP,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,MAAM,GAAG;;;;UAIT,CAAC;QACP,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;IAC5F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC3B,MAAM,MAAM,GAAG;;;UAGT,CAAC;QACP,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,MAAM,GAAG;;UAET,CAAC;QACP,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,uEAAuE;AAEvE,2DAA2D;AAE3D,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,MAAc,CAAC;IACnB,IAAI,WAAmB,CAAC;IAExB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;QAChE,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC;QACzE,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,EAAE,CAAC,aAAa,CAAC,WAAW,EAAE,sBAAsB,CAAC,CAAC;QAEtD,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC;QACrE,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAChD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,aAAa,CACX;YACE,EAAE,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;YACpC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE;SACjC,EACD,WAAW,CACZ,CAAC;QACF,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACtD,oCAAoC;QACpC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,iDAAiD,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,2DAA2D;AAE3D,SAAS,SAAS,CAAC,QAAgB;IACjC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,IAAI,KAAI,CAAC,EAAE,CAAC,CAAC;IAC1C,wDAAwD;IACxD,YAAY,CAAC,GAAG,EAAE;QAChB,KAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnB,CAAC,CAAC,CAAC;IACH,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,GAAG,IAAI,QAAQ,CAAC;IAC9B,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE;QACpB,EAAE,EAAE,CAAC;IACP,CAAC;CACF,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,CAAC,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACpE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,oGAAoG,CACrG,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,CAAC,EAAE;YACrC,KAAK,EAAE,IAAI;YACX,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC;YACpB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,CAAC,EAAE;YACrC,KAAK,EAAE,IAAI;YACX,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC;YACrB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,CAAC,EAAE;YACrC,KAAK,EAAE,IAAI;YACX,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC;YACrB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,CAAC,EAAE;YACrC,KAAK,EAAE,IAAI;YACX,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC;YACrB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -1,2 +0,0 @@
1
- export {};
2
- //# sourceMappingURL=orchestrator.test.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"orchestrator.test.d.ts","sourceRoot":"","sources":["../src/orchestrator.test.ts"],"names":[],"mappings":""}
@@ -1,130 +0,0 @@
1
- import { EventEmitter } from 'node:events';
2
- import * as fs from 'node:fs';
3
- import * as os from 'node:os';
4
- import * as path from 'node:path';
5
- import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
6
- import { invokeShellOrchestrator } from './utils.js';
7
- function createMockChild() {
8
- const child = new EventEmitter();
9
- child.stdout = new EventEmitter();
10
- child.stderr = new EventEmitter();
11
- child.kill = vi.fn();
12
- return child;
13
- }
14
- let mockChild;
15
- vi.mock('node:child_process', () => ({
16
- spawn: vi.fn(() => mockChild),
17
- }));
18
- const { spawn } = await import('node:child_process');
19
- const mockedSpawn = vi.mocked(spawn);
20
- // ─── Tests ───────────────────────────────────────────
21
- describe('invokeShellOrchestrator', () => {
22
- let tmpDir;
23
- const totemDir = '.totem';
24
- beforeEach(() => {
25
- tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'totem-orch-'));
26
- vi.spyOn(console, 'error').mockImplementation(() => { });
27
- mockChild = createMockChild();
28
- mockedSpawn.mockClear();
29
- });
30
- afterEach(() => {
31
- fs.rmSync(tmpDir, { recursive: true, force: true });
32
- vi.restoreAllMocks();
33
- });
34
- /** Emit stdout data and close with success */
35
- function emitSuccess(data) {
36
- process.nextTick(() => {
37
- mockChild.stdout.emit('data', Buffer.from(data));
38
- mockChild.emit('close', 0);
39
- });
40
- }
41
- /** Emit close with non-zero exit code and optional stderr */
42
- function emitFailure(code, stderr = '') {
43
- process.nextTick(() => {
44
- if (stderr)
45
- mockChild.stderr.emit('data', Buffer.from(stderr));
46
- mockChild.emit('close', code);
47
- });
48
- }
49
- it('returns raw content when output is not Gemini JSON', async () => {
50
- emitSuccess(' The answer is 42. ');
51
- const result = await invokeShellOrchestrator('prompt text', 'echo {file}', 'test-model', tmpDir, 'Test', totemDir);
52
- expect(result.content).toBe('The answer is 42.');
53
- expect(result.inputTokens).toBeNull();
54
- expect(result.outputTokens).toBeNull();
55
- expect(result.durationMs).toBeGreaterThanOrEqual(0);
56
- });
57
- it('parses Gemini JSON output and returns structured result', async () => {
58
- const geminiOutput = JSON.stringify({
59
- response: 'Gemini says hello.',
60
- stats: {
61
- models: {
62
- 'gemini-2.5-pro': {
63
- tokens: { input: 100, candidates: 50 },
64
- api: { totalLatencyMs: 2000 },
65
- },
66
- },
67
- },
68
- });
69
- emitSuccess(geminiOutput);
70
- const result = await invokeShellOrchestrator('prompt', 'gemini -e none < {file}', 'gemini-2.5-pro', tmpDir, 'Test', totemDir);
71
- expect(result.content).toBe('Gemini says hello.');
72
- expect(result.inputTokens).toBe(100);
73
- expect(result.outputTokens).toBe(50);
74
- expect(result.durationMs).toBe(2000);
75
- });
76
- it('substitutes {file} and {model} in command', async () => {
77
- emitSuccess('ok');
78
- await invokeShellOrchestrator('prompt', 'llm --model {model} < {file}', 'my-model', tmpDir, 'Test', totemDir);
79
- const cmd = mockedSpawn.mock.calls[0][0];
80
- expect(cmd).toContain('my-model');
81
- expect(cmd).not.toContain('{model}');
82
- expect(cmd).not.toContain('{file}');
83
- });
84
- it('writes prompt to temp file and cleans up after', async () => {
85
- emitSuccess('result');
86
- await invokeShellOrchestrator('my prompt content', 'cat {file}', 'model', tmpDir, 'Test', totemDir);
87
- // Temp file should be cleaned up
88
- const tempDir = path.join(tmpDir, totemDir, 'temp');
89
- if (fs.existsSync(tempDir)) {
90
- const files = fs.readdirSync(tempDir).filter((f) => f.startsWith('totem-test-'));
91
- expect(files).toHaveLength(0);
92
- }
93
- });
94
- it('throws QuotaError for quota-related failures', async () => {
95
- emitFailure(1, '429 Too Many Requests quota exceeded');
96
- try {
97
- await invokeShellOrchestrator('prompt', 'cmd', 'model', tmpDir, 'Test', totemDir);
98
- expect.fail('Should have thrown');
99
- }
100
- catch (err) {
101
- expect(err.name).toBe('QuotaError');
102
- }
103
- });
104
- it('throws generic error for non-zero exit code', async () => {
105
- emitFailure(1, 'something went wrong');
106
- await expect(invokeShellOrchestrator('prompt', 'cmd', 'model', tmpDir, 'Test', totemDir)).rejects.toThrow('[Totem Error] Shell orchestrator command failed');
107
- });
108
- it('throws error on spawn error event', async () => {
109
- process.nextTick(() => {
110
- mockChild.emit('error', new Error('command not found'));
111
- });
112
- await expect(invokeShellOrchestrator('prompt', 'cmd', 'model', tmpDir, 'Test', totemDir)).rejects.toThrow('command not found');
113
- });
114
- it('throws descriptive error for timeout', async () => {
115
- vi.useFakeTimers();
116
- // Simulate kill → close (rejection now happens in the close handler)
117
- mockChild.kill = vi.fn(() => {
118
- process.nextTick(() => mockChild.emit('close', null));
119
- });
120
- // Capture the rejection before advancing timers
121
- const promise = invokeShellOrchestrator('prompt', 'cmd', 'model', tmpDir, 'Test', totemDir).catch((err) => err);
122
- await vi.advanceTimersByTimeAsync(180_001);
123
- const err = await promise;
124
- expect(err).toBeInstanceOf(Error);
125
- expect(err.message).toContain('timed out after 180s');
126
- expect(mockChild.kill).toHaveBeenCalled();
127
- vi.useRealTimers();
128
- });
129
- });
130
- //# sourceMappingURL=orchestrator.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"orchestrator.test.js","sourceRoot":"","sources":["../src/orchestrator.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAUrD,SAAS,eAAe;IACtB,MAAM,KAAK,GAAG,IAAI,YAAY,EAAe,CAAC;IAC9C,KAAK,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IAClC,KAAK,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IAClC,KAAK,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACrB,OAAO,KAAK,CAAC;AACf,CAAC;AAED,IAAI,SAAoB,CAAC;AAEzB,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;CAC9B,CAAC,CAAC,CAAC;AAEJ,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;AACrD,MAAM,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAErC,wDAAwD;AAExD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,MAAc,CAAC;IACnB,MAAM,QAAQ,GAAG,QAAQ,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QAC/D,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACxD,SAAS,GAAG,eAAe,EAAE,CAAC;QAC9B,WAAW,CAAC,SAAS,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,8CAA8C;IAC9C,SAAS,WAAW,CAAC,IAAY;QAC/B,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YACpB,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACjD,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,6DAA6D;IAC7D,SAAS,WAAW,CAAC,IAAY,EAAE,MAAM,GAAG,EAAE;QAC5C,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YACpB,IAAI,MAAM;gBAAE,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/D,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,WAAW,CAAC,uBAAuB,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAC1C,aAAa,EACb,aAAa,EACb,YAAY,EACZ,MAAM,EACN,MAAM,EACN,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;YAClC,QAAQ,EAAE,oBAAoB;YAC9B,KAAK,EAAE;gBACL,MAAM,EAAE;oBACN,gBAAgB,EAAE;wBAChB,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE;wBACtC,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE;qBAC9B;iBACF;aACF;SACF,CAAC,CAAC;QACH,WAAW,CAAC,YAAY,CAAC,CAAC;QAC1B,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAC1C,QAAQ,EACR,yBAAyB,EACzB,gBAAgB,EAChB,MAAM,EACN,MAAM,EACN,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,WAAW,CAAC,IAAI,CAAC,CAAC;QAClB,MAAM,uBAAuB,CAC3B,QAAQ,EACR,8BAA8B,EAC9B,UAAU,EACV,MAAM,EACN,MAAM,EACN,QAAQ,CACT,CAAC;QACF,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAW,CAAC;QACpD,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,WAAW,CAAC,QAAQ,CAAC,CAAC;QACtB,MAAM,uBAAuB,CAC3B,mBAAmB,EACnB,YAAY,EACZ,OAAO,EACP,MAAM,EACN,MAAM,EACN,QAAQ,CACT,CAAC;QACF,iCAAiC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QACpD,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC;YACjF,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,WAAW,CAAC,CAAC,EAAE,sCAAsC,CAAC,CAAC;QACvD,IAAI,CAAC;YACH,MAAM,uBAAuB,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;YAClF,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAE,GAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC;QACvC,MAAM,MAAM,CACV,uBAAuB,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAC5E,CAAC,OAAO,CAAC,OAAO,CAAC,iDAAiD,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YACpB,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QACH,MAAM,MAAM,CACV,uBAAuB,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAC5E,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,qEAAqE;QACrE,SAAS,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;YAC1B,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,gDAAgD;QAChD,MAAM,OAAO,GAAG,uBAAuB,CACrC,QAAQ,EACR,KAAK,EACL,OAAO,EACP,MAAM,EACN,MAAM,EACN,QAAQ,CACT,CAAC,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAE7B,MAAM,EAAE,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;QAE3C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC;QAC1B,MAAM,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjE,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAE1C,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}