tiny.coz.profile_file

1const std = @import("std");2const sys = @import("sys");34pub const Writer = struct {5    file: std.Io.File = undefined,6    buffer: [8192]u8 = undefined,7    writer: std.Io.File.Writer = undefined,8    opened: bool = false,910    pub fn open(self: *Writer, path: []const u8) !void {11        if (self.opened) return error.ProfileFileAlreadyOpen;12        self.file = try sys.fs.createFile(path, .{ .truncate = true });13        errdefer self.file.close(sys.fs.debugIo());14        self.writer = self.file.writer(sys.fs.debugIo(), &self.buffer);15        self.opened = true;16    }1718    pub fn interface(self: *Writer) ?*std.Io.Writer {19        if (!self.opened) return null;20        return &self.writer.interface;21    }2223    pub fn finish(self: *Writer) !void {24        if (!self.opened) return;25        defer {26            self.file.close(sys.fs.debugIo());27            self.opened = false;28        }29        try self.writer.flush();30    }3132    pub fn close(self: *Writer) void {33        self.finish() catch {};34    }35};3637test "profile file owns a buffered writer in place" {38    const allocator = std.testing.allocator;39    var tmp = std.testing.tmpDir(.{});40    defer tmp.cleanup();4142    const root = try tmp.parent_dir.realPathFileAlloc(sys.fs.debugIo(), tmp.sub_path[0..], allocator);43    defer allocator.free(root);44    const path = try std.fs.path.join(allocator, &.{ root, "coz.jsonl" });45    defer allocator.free(path);4647    var file: Writer = .{};48    try file.open(path);49    try file.interface().?.writeAll("one\n");50    try file.finish();5152    const contents = try sys.fs.readFileAlloc(allocator, path, 128);53    defer allocator.free(contents);54    try std.testing.expectEqualStrings("one\n", contents);55    try std.testing.expect(file.interface() == null);56    file.close();57}5859test "profile file rejects double open" {60    const allocator = std.testing.allocator;61    var tmp = std.testing.tmpDir(.{});62    defer tmp.cleanup();6364    const root = try tmp.parent_dir.realPathFileAlloc(sys.fs.debugIo(), tmp.sub_path[0..], allocator);65    defer allocator.free(root);66    const path = try std.fs.path.join(allocator, &.{ root, "coz.jsonl" });67    defer allocator.free(path);6869    var file: Writer = .{};70    try file.open(path);71    defer file.close();7273    try std.testing.expectError(error.ProfileFileAlreadyOpen, file.open(path));74}

1pub const profile_file = @import("file.zig");