2016-04-19 07:42:56 +08:00
|
|
|
const std = @import("std");
|
|
|
|
const io = std.io;
|
2017-02-27 03:35:30 +08:00
|
|
|
const mem = std.mem;
|
2017-04-04 12:17:24 +08:00
|
|
|
const os = std.os;
|
2017-10-31 16:47:55 +08:00
|
|
|
const warn = std.debug.warn;
|
2017-11-11 05:32:37 +08:00
|
|
|
const allocator = std.debug.global_allocator;
|
2016-01-14 09:15:51 +08:00
|
|
|
|
2018-02-01 11:48:40 +08:00
|
|
|
pub fn main() !void {
|
2017-10-11 22:16:13 +08:00
|
|
|
var args_it = os.args();
|
2018-06-10 11:42:14 +08:00
|
|
|
const exe = try unwrapArg(args_it.next(allocator).?);
|
2016-01-16 18:10:15 +08:00
|
|
|
var catted_anything = false;
|
2018-01-08 05:51:46 +08:00
|
|
|
var stdout_file = try io.getStdOut();
|
2017-10-31 16:47:55 +08:00
|
|
|
|
2017-10-11 22:16:13 +08:00
|
|
|
while (args_it.next(allocator)) |arg_or_err| {
|
2018-01-08 05:51:46 +08:00
|
|
|
const arg = try unwrapArg(arg_or_err);
|
2017-02-27 03:35:30 +08:00
|
|
|
if (mem.eql(u8, arg, "-")) {
|
2016-01-16 18:10:15 +08:00
|
|
|
catted_anything = true;
|
2018-01-08 05:51:46 +08:00
|
|
|
var stdin_file = try io.getStdIn();
|
|
|
|
try cat_file(&stdout_file, &stdin_file);
|
2016-01-16 18:10:15 +08:00
|
|
|
} else if (arg[0] == '-') {
|
|
|
|
return usage(exe);
|
|
|
|
} else {
|
2018-08-22 08:28:37 +08:00
|
|
|
var file = os.File.openRead(arg) catch |err| {
|
2017-10-31 16:47:55 +08:00
|
|
|
warn("Unable to open file: {}\n", @errorName(err));
|
2016-01-16 18:10:15 +08:00
|
|
|
return err;
|
2016-01-26 14:56:46 +08:00
|
|
|
};
|
2017-10-31 16:47:55 +08:00
|
|
|
defer file.close();
|
2016-01-14 09:15:51 +08:00
|
|
|
|
2016-01-16 18:10:15 +08:00
|
|
|
catted_anything = true;
|
2018-01-08 05:51:46 +08:00
|
|
|
try cat_file(&stdout_file, &file);
|
2016-01-16 18:10:15 +08:00
|
|
|
}
|
|
|
|
}
|
2016-01-21 09:18:50 +08:00
|
|
|
if (!catted_anything) {
|
2018-01-08 05:51:46 +08:00
|
|
|
var stdin_file = try io.getStdIn();
|
|
|
|
try cat_file(&stdout_file, &stdin_file);
|
2016-01-16 18:10:15 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-02-01 11:48:40 +08:00
|
|
|
fn usage(exe: []const u8) !void {
|
2017-10-31 16:47:55 +08:00
|
|
|
warn("Usage: {} [FILE]...\n", exe);
|
2016-01-24 16:34:48 +08:00
|
|
|
return error.Invalid;
|
2016-01-16 18:10:15 +08:00
|
|
|
}
|
|
|
|
|
2018-05-31 22:56:59 +08:00
|
|
|
fn cat_file(stdout: *os.File, file: *os.File) !void {
|
2016-01-26 14:56:46 +08:00
|
|
|
var buf: [1024 * 4]u8 = undefined;
|
2016-01-16 18:10:15 +08:00
|
|
|
|
|
|
|
while (true) {
|
2018-01-08 06:28:20 +08:00
|
|
|
const bytes_read = file.read(buf[0..]) catch |err| {
|
2017-10-31 16:47:55 +08:00
|
|
|
warn("Unable to read from stream: {}\n", @errorName(err));
|
2016-01-23 18:06:29 +08:00
|
|
|
return err;
|
2016-04-09 04:13:33 +08:00
|
|
|
};
|
2016-01-23 18:06:29 +08:00
|
|
|
|
|
|
|
if (bytes_read == 0) {
|
|
|
|
break;
|
2016-01-16 18:10:15 +08:00
|
|
|
}
|
|
|
|
|
2018-01-08 06:28:20 +08:00
|
|
|
stdout.write(buf[0..bytes_read]) catch |err| {
|
2017-10-31 16:47:55 +08:00
|
|
|
warn("Unable to write to stdout: {}\n", @errorName(err));
|
2016-01-23 18:06:29 +08:00
|
|
|
return err;
|
2016-04-09 04:13:33 +08:00
|
|
|
};
|
2016-01-16 18:10:15 +08:00
|
|
|
}
|
2016-01-14 09:15:51 +08:00
|
|
|
}
|
2017-10-11 22:16:13 +08:00
|
|
|
|
2018-11-13 21:08:37 +08:00
|
|
|
fn unwrapArg(arg: anyerror![]u8) ![]u8 {
|
2018-01-08 06:28:20 +08:00
|
|
|
return arg catch |err| {
|
2017-10-31 16:47:55 +08:00
|
|
|
warn("Unable to parse command line: {}\n", err);
|
2017-10-11 22:16:13 +08:00
|
|
|
return err;
|
|
|
|
};
|
|
|
|
}
|