diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/main.zig | 75 | ||||
-rw-r--r-- | src/root.zig | 10 |
2 files changed, 85 insertions, 0 deletions
diff --git a/src/main.zig b/src/main.zig new file mode 100644 index 0000000..c65a36b --- /dev/null +++ b/src/main.zig @@ -0,0 +1,75 @@ +const std = @import("std"); +const File = std.fs.File; +const Writer = std.fs.File.Writer; + +pub fn main() !void { + const in = std.io.getStdIn(); + const out = std.io.getStdOut(); + Uf.init().streamCopy(in, out) catch |err| { + try std.io.getStdErr().writer().print("Error {}", .{err}); + }; +} + +const buf_size = 1_000_000; + +const Uf = struct { + const Self = @This(); + + var out_buf: [buf_size]u8 = undefined; + var out_index: u32 = 0; + + fn init() *const Self { + return &Self{}; + } + + fn flush(_: *const Self, out: Writer) !void { + _ = try out.write(out_buf[0..out_index]); + out_index = 0; + } + + fn writeChar(self: *const Self, out: Writer, c: u8) !void { + if (out_index >= buf_size) { + try self.flush(out); + } + out_buf[out_index] = c; + out_index += 1; + } + + fn streamCopy(self: *const Self, in: File, out: File) !void { + const reader = in.reader(); + const writer = out.writer(); + + var hit = false; + var second = false; + var in_buf: [buf_size]u8 = undefined; + while (true) { + const size = try reader.read(&in_buf); + + for (in_buf[0..size]) |c| { + if (hit and c == '\n') { + if (second == false) { + second = true; + } else { + continue; + } + } else if (hit and c != '\n') { + if (second) { + try self.writeChar(writer, '\n'); + } + hit = false; + second = false; + } else if (c == '\n') { + hit = true; + continue; + } + + try self.writeChar(writer, c); + } + + if (size != in_buf.len) { + try self.flush(writer); + return; + } + } + } +}; diff --git a/src/root.zig b/src/root.zig new file mode 100644 index 0000000..ecfeade --- /dev/null +++ b/src/root.zig @@ -0,0 +1,10 @@ +const std = @import("std"); +const testing = std.testing; + +export fn add(a: i32, b: i32) i32 { + return a + b; +} + +test "basic add functionality" { + try testing.expect(add(3, 7) == 10); +} |