220 行
7.2 KiB
Zig
220 行
7.2 KiB
Zig
const std = @import("std");
|
|
const fs = std.fs;
|
|
const io = std.io;
|
|
|
|
const version = @import("version.zig");
|
|
|
|
const Option = struct {
|
|
isbyte: bool,
|
|
isline: bool,
|
|
ischar: bool,
|
|
isword: bool,
|
|
isnhed: bool,
|
|
};
|
|
|
|
fn help() !void {
|
|
const stdof = io.getStdOut().writer();
|
|
var bw = io.bufferedWriter(stdof);
|
|
const stdout = bw.writer();
|
|
|
|
try stdout.print("076 coreutils\n", .{});
|
|
try stdout.print("使用法: wc [オプション]... [ファイル]...\n", .{});
|
|
try stdout.print("ファイル (複数可) の内容を結合して標準出力に出力します。\n", .{});
|
|
try stdout.print("FILE が複数指定された場合は行数の合計も表示します。\n", .{});
|
|
try stdout.print("FILE が指定されないか、 FILE が - の場合、 標準入力から読み込みます。\n", .{});
|
|
try stdout.print("単語は、空白類文字で区切られる、長さが 0 でない表示可能文字の列です。\n\n", .{});
|
|
try stdout.print("ファイルの指定がない場合や FILE が - の場合, 標準入力から読み込みを行います。\n\n", .{});
|
|
try stdout.print("下記のオプションを使って、何を数えて表示するかを選択できます。\n", .{});
|
|
try stdout.print("表示は常に次の順です: 改行数、単語数、文字数、バイト数。\n", .{});
|
|
try stdout.print("-c バイト数を表示する\n", .{});
|
|
try stdout.print("-m 文字数を表示する\n", .{});
|
|
try stdout.print("-n ヘッダーを非表示にする\n", .{});
|
|
try stdout.print("-l 改行の数を表示する\n", .{});
|
|
try stdout.print("-w 単語数を表示する\n", .{});
|
|
try stdout.print("-h ヘルプを表示\n", .{});
|
|
try stdout.print("-v バージョンを表示\n", .{});
|
|
|
|
try bw.flush();
|
|
}
|
|
|
|
fn cuntchar(reader: anytype) !struct {
|
|
byte_cunt: usize,
|
|
line_cunt: usize,
|
|
char_cunt: usize,
|
|
word_cunt: usize,
|
|
} {
|
|
var byte_cunt: usize = 0;
|
|
var line_cunt: usize = 0;
|
|
var char_cunt: usize = 0;
|
|
var word_cunt: usize = 0;
|
|
|
|
// -l -c -m
|
|
var buf: [2048]u8 = undefined;
|
|
while (true) {
|
|
const lne = try reader.read(buf[0..]);
|
|
if (lne == 0) {
|
|
break;
|
|
}
|
|
|
|
const chr = try std.unicode.utf8CountCodepoints(buf[0..lne]);
|
|
for (buf[0..lne]) |char| {
|
|
if (char == '\n') {
|
|
line_cunt += 1;
|
|
}
|
|
}
|
|
char_cunt += chr;
|
|
byte_cunt += lne;
|
|
|
|
// -w
|
|
var insideWord: bool = false;
|
|
for (buf[0..lne]) |char| {
|
|
if (std.ascii.isWhitespace(char)) {
|
|
insideWord = false;
|
|
} else if (!insideWord) {
|
|
insideWord = true;
|
|
word_cunt += 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
return .{
|
|
.byte_cunt = byte_cunt,
|
|
.line_cunt = line_cunt,
|
|
.char_cunt = char_cunt,
|
|
.word_cunt = word_cunt,
|
|
};
|
|
}
|
|
|
|
pub fn main() !void {
|
|
var gpa = std.heap.GeneralPurposeAllocator(.{}){};
|
|
defer _ = gpa.deinit();
|
|
const alloc = gpa.allocator();
|
|
|
|
var opt = std.ArrayList(u8).init(alloc);
|
|
defer opt.deinit();
|
|
|
|
var fname = std.ArrayList([]const u8).init(alloc);
|
|
defer fname.deinit();
|
|
|
|
const args = try std.process.argsAlloc(alloc);
|
|
defer std.process.argsFree(alloc, args);
|
|
|
|
for (args, 0..) |arg, i| {
|
|
if (i == 0) continue;
|
|
var m: [1]u8 = [_]u8{'-'};
|
|
if (std.mem.eql(u8, arg[0..1], m[0..])) {
|
|
for (arg, 0..) |a, j| {
|
|
if (j == 0) continue;
|
|
try opt.append(a);
|
|
}
|
|
} else {
|
|
try fname.append(arg);
|
|
}
|
|
}
|
|
|
|
var option = Option{
|
|
.isbyte = false,
|
|
.isline = false,
|
|
.ischar = false,
|
|
.isword = false,
|
|
.isnhed = false,
|
|
};
|
|
|
|
var byte_cuntt: usize = 0;
|
|
var line_cuntt: usize = 0;
|
|
var char_cuntt: usize = 0;
|
|
var word_cuntt: usize = 0;
|
|
|
|
for (opt.items) |i| {
|
|
if (i == 'h') {
|
|
try help();
|
|
return;
|
|
}
|
|
if (i == 'v') {
|
|
try version.ver("wc");
|
|
return;
|
|
}
|
|
if (i == 'c') option.isbyte = true;
|
|
if (i == 'l') option.isline = true;
|
|
if (i == 'm') option.ischar = true;
|
|
if (i == 'n') option.isnhed = true;
|
|
if (i == 'w') option.isword = true;
|
|
}
|
|
|
|
const stdof = io.getStdOut().writer();
|
|
var bw = io.bufferedWriter(stdof);
|
|
const stdout = bw.writer();
|
|
|
|
if (!option.isnhed) {
|
|
if (option.isline) {
|
|
try stdout.print("改行数\tファル名\n", .{});
|
|
} else if (option.isword) {
|
|
try stdout.print("単語数\tファル名\n", .{});
|
|
} else if (option.ischar) {
|
|
try stdout.print("文字数\tファル名\n", .{});
|
|
} else if (option.isbyte) {
|
|
try stdout.print("バイト数\tファル名\n", .{});
|
|
} else {
|
|
try stdout.print("改行数\t単語数\t文字数\tバイト数\tファル名\n", .{});
|
|
}
|
|
}
|
|
|
|
const stdin = io.getStdIn().reader();
|
|
if (fname.items.len == 0) {
|
|
const cnt = try cuntchar(stdin);
|
|
|
|
if (option.isline) {
|
|
try stdout.print("{d}\n", .{cnt.line_cunt});
|
|
} else if (option.isword) {
|
|
try stdout.print("{d}\n", .{cnt.word_cunt});
|
|
} else if (option.ischar) {
|
|
try stdout.print("{d}\n", .{cnt.char_cunt});
|
|
} else if (option.isbyte) {
|
|
try stdout.print("{d}\n", .{cnt.byte_cunt});
|
|
} else {
|
|
try stdout.print("{d}\t{d}\t{d}\t{d}\n", .{ cnt.line_cunt, cnt.word_cunt, cnt.char_cunt, cnt.byte_cunt });
|
|
}
|
|
} else {
|
|
for (fname.items, 0..) |item, i| {
|
|
const file = try fs.cwd().openFile(item, .{});
|
|
defer file.close();
|
|
const fr = file.reader();
|
|
|
|
const cnt = try cuntchar(fr);
|
|
|
|
line_cuntt += cnt.line_cunt;
|
|
char_cuntt += cnt.char_cunt;
|
|
byte_cuntt += cnt.byte_cunt;
|
|
word_cuntt += cnt.word_cunt;
|
|
|
|
if (option.isline) {
|
|
try stdout.print("{d}\n", .{cnt.line_cunt});
|
|
} else if (option.isword) {
|
|
try stdout.print("{d}\n", .{cnt.word_cunt});
|
|
} else if (option.ischar) {
|
|
try stdout.print("{d}\n", .{cnt.char_cunt});
|
|
} else if (option.isbyte) {
|
|
try stdout.print("{d}\n", .{cnt.byte_cunt});
|
|
} else {
|
|
try stdout.print("{d}\t{d}\t{d}\t{d}\n", .{ cnt.line_cunt, cnt.word_cunt, cnt.char_cunt, cnt.byte_cunt });
|
|
}
|
|
|
|
if (i > 0) {
|
|
if (option.isline) {
|
|
try stdout.print("{d}\t合計\n", .{line_cuntt});
|
|
} else if (option.isword) {
|
|
try stdout.print("{d}\t合計\n", .{word_cuntt});
|
|
} else if (option.ischar) {
|
|
try stdout.print("{d}\t合計\n", .{char_cuntt});
|
|
} else if (option.isbyte) {
|
|
try stdout.print("{d}\t\t合計\n", .{byte_cuntt});
|
|
} else {
|
|
try stdout.print("{d}\t{d}\t{d}\t{d}\t\t合計\n", .{ line_cuntt, word_cuntt, char_cuntt, byte_cuntt });
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
try bw.flush();
|
|
}
|