aoc-2024/utils.zig

const std = @import("std");

// pub const LineReader = struct {
//     const FileReader = std.fs.File.Reader;
//     const BufferedReader = std.io.BufferedReader(4096, FileReader);
//     const Reader = std.io.Reader(*BufferedReader, std.fs.File.Reader.Error, BufferedReader.read);

//     alloc: std.mem.Allocator,

//     file: std.fs.File,

//     line_buf: std.ArrayList(u8),

//     buf_reader: *BufferedReader,
//     reader: Reader,

//     pub fn init(alloc: std.mem.Allocator, filename: []const u8) !LineReader {
//         const file = try std.fs.cwd().openFile(filename, .{});

//         const file_reader = file.reader();

//         const buf_reader = try alloc.create(BufferedReader);
//         buf_reader.* = std.io.bufferedReader(file_reader);

//         const reader: Reader = Reader{ .context = buf_reader };

//         const line_buf = std.ArrayList(u8).init(alloc);

//         return .{
//             .alloc = alloc,
//             .file = file,
//             .line_buf = line_buf,
//             .buf_reader = buf_reader,
//             .reader = reader,
//         };
//     }

//     pub fn deinit(self: *LineReader) void {
//         self.file.close();

//         self.line_buf.deinit();

//         self.alloc.destroy(self.buf_reader);
//     }

//     pub fn next(self: *LineReader) !?[]u8 {
//         self.line_buf.clearRetainingCapacity();

//         self.reader.streamUntilDelimiter(self.line_buf.writer(), '\n', 4096) catch |err| switch (err) {
//             error.EndOfStream => return null,
//             else => return err,
//         };

//         return self.line_buf.items;
//     }
// };

pub const FileReader = struct {
    const BufferedReader = std.io.BufferedReader(4096, std.fs.File.Reader);
    const Reader = std.io.Reader(*BufferedReader, std.fs.File.Reader.Error, BufferedReader.read);

    alloc: std.mem.Allocator,

    file: std.fs.File,

    buf_reader_ptr: *BufferedReader,

    pub fn init(alloc: std.mem.Allocator, filename: []const u8) !FileReader {
        const file = try std.fs.cwd().openFile(filename, .{});

        const file_reader = file.reader();

        const buf_reader_ptr = try alloc.create(BufferedReader);
        buf_reader_ptr.* = std.io.bufferedReader(file_reader);

        return .{
            .alloc = alloc,
            .file = file,
            .buf_reader_ptr = buf_reader_ptr,
        };
    }

    pub fn deinit(self: FileReader) void {
        self.file.close();

        self.alloc.destroy(self.buf_reader_ptr);
    }

    pub fn reader(self: FileReader) Reader {
        return Reader{ .context = self.buf_reader_ptr };
    }
};

pub fn LineReader(comptime ReaderType: type) type {
    return struct {
        const Self = @This();

        reader: ReaderType,

        line_buf: std.ArrayList(u8),

        pub fn init(alloc: std.mem.Allocator, reader: ReaderType) Self {
            const line_buf = std.ArrayList(u8).init(alloc);

            return .{
                .reader = reader,
                .line_buf = line_buf,
            };
        }

        pub fn deinit(self: *Self) void {
            self.line_buf.deinit();
        }

        pub fn next(self: *Self) !?[]u8 {
            self.line_buf.clearRetainingCapacity();

            self.reader.streamUntilDelimiter(self.line_buf.writer(), '\n', 4096) catch |err| switch (err) {
                error.EndOfStream => if (self.line_buf.items.len == 0) {
                    // the first time we get an EndOfStream we return the content of the line_buf,
                    // the second time we're finished
                    return null;
                },
                else => return err,
            };

            return self.line_buf.items;
        }
    };
}

pub fn lineReader(alloc: std.mem.Allocator, reader: anytype) LineReader(@TypeOf(reader)) {
    return LineReader(@TypeOf(reader)).init(alloc, reader);
}

pub fn NumberParser(comptime T: type) type {
    return struct {
        token_it: std.mem.TokenIterator(u8, .scalar),

        pub fn next(self: *@This()) !?T {
            if (self.token_it.next()) |tok| {
                return try std.fmt.parseUnsigned(T, tok, 10);
            }

            return null;
        }
    };
}

pub fn numberParser(comptime T: type, input: []const u8) NumberParser(T) {
    return NumberParser(T){ .token_it = std.mem.tokenizeScalar(u8, input, ' ') };
}