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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/*
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
* Copyright (c) 2021, Emanuele Torre <torreemanuele6@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/String.h>
#include <AK/Vector.h>
#include <LibCore/ArgsParser.h>
#include <LibCore/System.h>
#include <ctype.h>
#include <stdio.h>
#include <sys/stat.h>
#include <unistd.h>
struct Count {
String name;
bool exists { true };
unsigned lines { 0 };
unsigned characters { 0 };
unsigned words { 0 };
size_t bytes { 0 };
};
bool g_output_line = false;
bool g_output_byte = false;
bool g_output_word = false;
static void wc_out(Count const& count)
{
if (g_output_line)
out("{:7} ", count.lines);
if (g_output_word)
out("{:7} ", count.words);
if (g_output_byte)
out("{:7} ", count.bytes);
outln("{:>14}", count.name);
}
static Count get_count(String const& file_specifier)
{
Count count;
FILE* file_pointer = nullptr;
if (file_specifier == "-") {
count.name = "";
file_pointer = stdin;
} else {
count.name = file_specifier;
if ((file_pointer = fopen(file_specifier.characters(), "r")) == nullptr) {
warnln("wc: unable to open {}", file_specifier);
count.exists = false;
return count;
}
}
bool start_a_new_word = true;
for (int ch = fgetc(file_pointer); ch != EOF; ch = fgetc(file_pointer)) {
count.bytes++;
if (isspace(ch)) {
start_a_new_word = true;
if (ch == '\n')
count.lines++;
} else if (start_a_new_word) {
start_a_new_word = false;
count.words++;
}
}
if (file_pointer != stdin)
fclose(file_pointer);
return count;
}
static Count get_total_count(Vector<Count> const& counts)
{
Count total_count { "total" };
for (auto& count : counts) {
total_count.lines += count.lines;
total_count.words += count.words;
total_count.characters += count.characters;
total_count.bytes += count.bytes;
}
return total_count;
}
ErrorOr<int> serenity_main(Main::Arguments arguments)
{
TRY(Core::System::pledge("stdio rpath"));
Vector<char const*> file_specifiers;
Core::ArgsParser args_parser;
args_parser.add_option(g_output_line, "Output line count", "lines", 'l');
args_parser.add_option(g_output_byte, "Output byte count", "bytes", 'c');
args_parser.add_option(g_output_word, "Output word count", "words", 'w');
args_parser.add_positional_argument(file_specifiers, "File to process", "file", Core::ArgsParser::Required::No);
args_parser.parse(arguments);
if (!g_output_line && !g_output_byte && !g_output_word)
g_output_line = g_output_byte = g_output_word = true;
Vector<Count> counts;
for (auto const& file_specifier : file_specifiers)
counts.append(get_count(file_specifier));
TRY(Core::System::pledge("stdio"));
if (file_specifiers.is_empty())
counts.append(get_count("-"));
else if (file_specifiers.size() > 1)
counts.append(get_total_count(counts));
for (auto const& count : counts) {
if (count.exists)
wc_out(count);
}
return 0;
}
|
