blob: 9ea980aecca4a67ece8caa95359f8356c50acc9f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
const std = @import("std");
const fs = std.fs;
const mem = std.mem;
const warn = std.log.warn;
const fatal = std.process.fatal;
pub fn main() !void {
var arena_instance = std.heap.ArenaAllocator.init(std.heap.page_allocator);
defer arena_instance.deinit();
const arena = arena_instance.allocator();
var threaded: std.Io.Threaded = .init(arena);
defer threaded.deinit();
const io = threaded.io();
const args = try std.process.argsAlloc(arena);
const exe = args[0];
var catted_anything = false;
var stdout_buffer: [4096]u8 = undefined;
var stdout_writer = fs.File.stdout().writerStreaming(&stdout_buffer);
const stdout = &stdout_writer.interface;
var stdin_reader = fs.File.stdin().readerStreaming(io, &.{});
const cwd = fs.cwd();
for (args[1..]) |arg| {
if (mem.eql(u8, arg, "-")) {
catted_anything = true;
_ = try stdout.sendFileAll(&stdin_reader, .unlimited);
try stdout.flush();
} else if (mem.startsWith(u8, arg, "-")) {
return usage(exe);
} else {
const file = cwd.openFile(arg, .{}) catch |err| fatal("unable to open file: {t}\n", .{err});
defer file.close();
catted_anything = true;
var file_reader = file.reader(io, &.{});
_ = try stdout.sendFileAll(&file_reader, .unlimited);
try stdout.flush();
}
}
if (!catted_anything) {
_ = try stdout.sendFileAll(&stdin_reader, .unlimited);
try stdout.flush();
}
}
fn usage(exe: []const u8) !void {
warn("Usage: {s} [FILE]...\n", .{exe});
return error.Invalid;
}
|
