homeio

1import { EventEmitter } from "node:events";
2import { describe, expect, it, vi } from "vitest";
3
4const { spawnMock } = vi.hoisted(() => ({
5  spawnMock: vi.fn(),
6}));
7
8vi.mock("node:child_process", () => ({
9  spawn: spawnMock,
10}));
11
12import {
13  executeTerminalCommand,
14} from "@/lib/server/modules/terminal/service";
15
16function createMockChildProcess() {
17  const processEvents = new EventEmitter();
18  const stdout = new EventEmitter();
19  const stderr = new EventEmitter();
20
21  return Object.assign(processEvents, {
22    stdout,
23    stderr,
24    kill: vi.fn(),
25  });
26}
27
28describe("terminal service", () => {
29  it("returns builtin help output", async () => {
30    const result = await executeTerminalCommand({
31      command: "help",
32    });
33
34    expect(result.exitCode).toBe(0);
35    expect(result.lines[0]?.content).toContain("Supported commands");
36  });
37
38  it("rejects forbidden commands", async () => {
39    await expect(
40      executeTerminalCommand({
41        command: "bash",
42      }),
43    ).rejects.toMatchObject({
44      code: "forbidden_command",
45      statusCode: 403,
46    });
47  });
48
49  it("executes external allowed commands", async () => {
50    spawnMock.mockImplementationOnce(() => {
51      const child = createMockChildProcess();
52      queueMicrotask(() => {
53        child.stdout.emit("data", Buffer.from("hello from command\n"));
54        child.emit("close", 0);
55      });
56      return child;
57    });
58
59    const result = await executeTerminalCommand({
60      command: "echo hello",
61      cwd: process.cwd(),
62    });
63
64    expect(spawnMock).toHaveBeenCalledWith("echo", ["hello"], expect.any(Object));
65    expect(result.lines[0]).toMatchObject({
66      type: "output",
67    });
68    expect(result.lines[0]?.content).toContain("hello from command");
69  });
70
71  it("maps ENOENT errors to command_not_found", async () => {
72    spawnMock.mockImplementationOnce(() => {
73      const child = createMockChildProcess();
74      queueMicrotask(() => {
75        const error = Object.assign(new Error("missing"), {
76          code: "ENOENT",
77        });
78        child.emit("error", error);
79      });
80      return child;
81    });
82
83    await expect(
84      executeTerminalCommand({
85        command: "date",
86      }),
87    ).rejects.toMatchObject({
88      code: "command_not_found",
89      statusCode: 404,
90    });
91  });
92});
93