import { visibleWidth } from "@earendil-works/pi-tui";
import { beforeAll, describe, expect, it } from "vitest";
import { IPythonCellComponent } from "../src/modes/interactive/components/ipython-cell.js";
import { initTheme } from "../src/modes/interactive/theme/theme.js";
type CellState = ConstructorParameters<typeof IPythonCellComponent>[0];
function foregroundLeftOpen(line: string): boolean {
let fg = false;
for (const match of line.matchAll(/\x1b\[([0-9;]*)m/g)) {
const params = match[1] === "" ? ["0"] : match[1].split(";");
for (let i = 0; i < params.length; i++) {
const code = Number(params[i]);
if (code === 0 || code === 39) {
fg = false;
} else if (code === 38) {
fg = true;
const mode = Number(params[i + 1]);
i += mode === 2 ? 4 : mode === 5 ? 2 : 1;
} else if (code === 48) {
const mode = Number(params[i + 1]);
i += mode === 2 ? 4 : mode === 5 ? 2 : 1;
} else if ((code >= 30 && code <= 37) || (code >= 90 && code <= 97)) {
fg = true;
}
}
}
return fg;
}
const WRAPPING_STATE: CellState = {
code: "import numpy as np\nresult = np.linspace(0, 100, 50)\nprint('the first element of the linspace array is', result[0])",
content: [
{
type: "text",
text: "the first element of the linspace array is 0.0\nsecond line of output that is also fairly long and will wrap on a small terminal",
},
],
details: {
status: "ok",
durationMs: 12,
stdout:
"the first element of the linspace array is 0.0\nsecond line of output that is also fairly long and will wrap on a small terminal",
},
executionStarted: true,
argsComplete: true,
expanded: true,
};
describe("IPythonCellComponent wrapping", () => {
beforeAll(() => {
initTheme("dark");
});
it("never leaves a foreground color open at a wrapped line end", () => {
for (let width = 20; width <= 60; width++) {
const lines = new IPythonCellComponent(WRAPPING_STATE).render(width);
const leaks = lines.filter(foregroundLeftOpen);
expect(leaks, `width=${width} leaked foreground on ${leaks.length} line(s)`).toHaveLength(0);
}
});
it("keeps every wrapped line within the available width", () => {
for (const width of [20, 30, 40, 50]) {
const lines = new IPythonCellComponent(WRAPPING_STATE).render(width);
expect(
lines.every((line) => visibleWidth(line) <= width),
`width=${width}`,
).toBe(true);
}
});
it("renders the same after a resize as a fresh render at the target width", () => {
const resized = new IPythonCellComponent(WRAPPING_STATE);
resized.render(100);
resized.invalidate();
const afterResize = resized.render(34);
const fresh = new IPythonCellComponent(WRAPPING_STATE).render(34);
expect(afterResize).toEqual(fresh);
});
it("leaves non-wrapping (wide) output untouched", () => {
const lines = new IPythonCellComponent(WRAPPING_STATE).render(100);
expect(lines.some(foregroundLeftOpen)).toBe(false);
expect(lines.every((line) => visibleWidth(line) <= 100)).toBe(true);
});
});