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); +}  | 
