diff options
| -rw-r--r-- | .gitignore | 2 | ||||
| -rw-r--r-- | build.zig | 42 | ||||
| -rw-r--r-- | src/main.zig | 75 | ||||
| -rw-r--r-- | src/root.zig | 10 | 
4 files changed, 129 insertions, 0 deletions
diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ee7098f --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +zig-out/ +zig-cache/ diff --git a/build.zig b/build.zig new file mode 100644 index 0000000..17d4fb7 --- /dev/null +++ b/build.zig @@ -0,0 +1,42 @@ +const std = @import("std"); + +pub fn build(b: *std.Build) void { +    const target = b.standardTargetOptions(.{}); + +    const optimize = b.standardOptimizeOption(.{ +        .preferred_optimize_mode = std.builtin.OptimizeMode.ReleaseFast, +    }); + +    const exe = b.addExecutable(.{ +        .name = "uf", +        .root_source_file = b.path("src/main.zig"), +        .target = target, +        .optimize = optimize, +        .single_threaded = true, +        .strip = true, +    }); + +    b.installArtifact(exe); + +    const run_cmd = b.addRunArtifact(exe); + +    run_cmd.step.dependOn(b.getInstallStep()); + +    if (b.args) |args| { +        run_cmd.addArgs(args); +    } + +    const run_step = b.step("run", "Run the app"); +    run_step.dependOn(&run_cmd.step); + +    const exe_unit_tests = b.addTest(.{ +        .root_source_file = b.path("src/main.zig"), +        .target = target, +        .optimize = optimize, +    }); + +    const run_exe_unit_tests = b.addRunArtifact(exe_unit_tests); + +    const test_step = b.step("test", "Run unit tests"); +    test_step.dependOn(&run_exe_unit_tests.step); +} 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); +}  | 
