Something went wrong. Try again.
My version of https://github.com/vercel-labs/fx
Something went wrong. Try again.
5.3 kB · 144 lines
Zig
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145// Host contract:// - fields: stdout_file, shadow_vt, shadow_vt_alloc, layout,// viewport_top_row, installed pre-fx document state// This module owns the normal frame sink. Exact probes, lifecycle controls,// titles, crash recovery, and noninteractive output stay with their domain// owners.
const std = @import("std");const io_mod = @import("../../core/shared/io.zig");const record_tape = @import("../../core/workspace/record_tape.zig");const types = @import("../../core/shared/types.zig");const terminal_diff = @import("../render_engine/terminal_diff.zig");const vt_emulator = @import("../../core/terminal/engine.zig");
const Allocator = std.mem.Allocator;const Metrics = types.Metrics;
pub fn enableShadowVt(shell: anytype, alloc: Allocator) !void { if (shell.shadow_vt != null) return; const cols: u16 = if (shell.layout.cols > 0) shell.layout.cols else 80; const rows: u16 = if (shell.layout.rows > 0) shell.layout.rows else 24; const grid = try alloc.create(vt_emulator.Grid); errdefer alloc.destroy(grid); grid.* = try vt_emulator.Grid.init(alloc, cols, rows); // The render oracle needs committed cells, not deferred sync state. grid.defer_sync_updates = false; grid.autowrap = false; shell.shadow_vt = grid; shell.shadow_vt_alloc = alloc;}
pub fn disableShadowVt(shell: anytype) void { if (shell.shadow_vt) |grid| { grid.deinit(); if (shell.shadow_vt_alloc) |alloc| alloc.destroy(grid); shell.shadow_vt = null; shell.shadow_vt_alloc = null; }}
pub fn writeFrameBytes(shell: anytype, metrics: *Metrics, bytes: []const u8) terminal_diff.FrameSinkWriteResult { var accepted_bytes: usize = 0; while (accepted_bytes < bytes.len) { const written = shell.stdout_file.writeStreaming( io_mod.getIo(), &.{}, &.{bytes[accepted_bytes..]}, 1, ) catch |err| return recordPartialFrame(metrics, bytes[0..accepted_bytes], err); if (written == 0) return recordPartialFrame(metrics, bytes[0..accepted_bytes], error.NoProgress); accepted_bytes += written; } return recordCompleteFrame(metrics, bytes);}
fn recordPartialFrame(metrics: *Metrics, accepted: []const u8, err: anyerror) terminal_diff.FrameSinkWriteResult { record_tape.recordStdout(accepted); metrics.ansi_bytes += accepted.len; return .{ .partial = .{ .accepted_bytes = accepted.len, .err = err, } };}
fn recordCompleteFrame(metrics: *Metrics, bytes: []const u8) terminal_diff.FrameSinkWriteResult { record_tape.recordStdout(bytes); metrics.ansi_bytes += bytes.len; return .complete;}
pub fn frameSinkFrom(ctx: *anyopaque, write_frame: *const fn (*anyopaque, *Metrics, []const u8) terminal_diff.FrameSinkWriteResult) terminal_diff.FrameSink { return .{ .ctx = ctx, .write_frame = write_frame };}
test "writeFrameBytes writes the frame without feeding shadow" { const FakeShell = struct { stdout_file: std.Io.File, shadow_vt: ?*vt_emulator.Grid = null, };
var tmp = std.testing.tmpDir(.{}); defer tmp.cleanup(); var file = try tmp.dir.createFile(std.testing.io, "frame-bytes.log", .{ .read = true }); defer file.close(io_mod.getIo());
var shadow = try vt_emulator.Grid.init(std.testing.allocator, 8, 4); defer shadow.deinit();
var shell = FakeShell{ .stdout_file = file, .shadow_vt = &shadow, }; var metrics: Metrics = .{};
try std.testing.expect(writeFrameBytes(&shell, &metrics, "abc") == .complete); try std.testing.expectEqual(@as(u21, ' '), shadow.cellAt(1, 1).?.codepoint); try std.testing.expectEqual(@as(usize, 3), metrics.ansi_bytes);}
test "standalone presentation bell is written without changing the shadow grid" { const FakeShell = struct { stdout_file: std.Io.File, shadow_vt: ?*vt_emulator.Grid = null, };
var tmp = std.testing.tmpDir(.{}); defer tmp.cleanup(); var file = try tmp.dir.createFile(std.testing.io, "bell.log", .{ .read = true }); defer file.close(io_mod.getIo());
var shadow = try vt_emulator.Grid.init(std.testing.allocator, 8, 4); defer shadow.deinit(); var shell = FakeShell{ .stdout_file = file, .shadow_vt = &shadow }; var metrics: Metrics = .{};
try std.testing.expect(writeFrameBytes(&shell, &metrics, "\x07") == .complete); var bytes: [1]u8 = undefined; try std.testing.expectEqual(@as(usize, 1), try file.readPositionalAll(io_mod.getIo(), &bytes, 0)); try std.testing.expectEqual(@as(u8, 0x07), bytes[0]); try std.testing.expectEqual(@as(u21, ' '), shadow.cellAt(1, 1).?.codepoint);}
test "enableShadowVt matches fx's no-autowrap terminal mode" { const FakeShell = struct { layout: types.Layout, shadow_vt: ?*vt_emulator.Grid = null, shadow_vt_alloc: ?Allocator = null, };
var shell = FakeShell{ .layout = .{ .rows = 24, .cols = 80, .content_bottom = 21, .divider_top_row = 22, .input_row = 23, .divider_bottom_row = 24, .hint_row = 24, } }; defer disableShadowVt(&shell);
try enableShadowVt(&shell, std.testing.allocator);
try std.testing.expect(!shell.shadow_vt.?.autowrap);}