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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
/*
* Copyright (c) 2021, the SerenityOS developers.
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <LibCore/ArgsParser.h>
#include <LibCore/File.h>
#include <LibCore/System.h>
#include <LibMain/Main.h>
#include <string.h>
#include <strings.h>
#include <unistd.h>
#define COL1_COLOR "\x1B[32m{}\x1B[0m"
#define COL2_COLOR "\x1B[34m{}\x1B[0m"
#define COL3_COLOR "\x1B[31m{}\x1B[0m"
ErrorOr<int> serenity_main(Main::Arguments arguments)
{
TRY(Core::System::pledge("stdio rpath", nullptr));
String file1_path, file2_path;
bool suppress_col1 { false };
bool suppress_col2 { false };
bool suppress_col3 { false };
bool case_insensitive { false };
bool color { false };
bool no_color { false };
bool print_total { false };
Core::ArgsParser args_parser;
args_parser.set_general_help("Compare two sorted files line by line");
args_parser.add_option(suppress_col1, "Suppress column 1 (lines unique to file1)", nullptr, '1');
args_parser.add_option(suppress_col2, "Suppress column 2 (lines unique to file2)", nullptr, '2');
args_parser.add_option(suppress_col3, "Suppress column 3 (lines common to both files)", nullptr, '3');
args_parser.add_option(case_insensitive, "Use case-insensitive comparison of lines", nullptr, 'i');
args_parser.add_option(color, "Always print colored output", "color", 'c');
args_parser.add_option(no_color, "Do not print colored output", "no-color", 0);
args_parser.add_option(print_total, "Print a summary", "total", 't');
args_parser.add_positional_argument(file1_path, "First file to compare", "file1");
args_parser.add_positional_argument(file2_path, "Second file to compare", "file2");
args_parser.parse(arguments);
if (color && no_color) {
warnln("Cannot specify 'color' and 'no-color' together");
return 1;
}
bool print_color = isatty(STDOUT_FILENO);
if (color)
print_color = true;
else if (no_color)
print_color = false;
if (file1_path == "-" && file2_path == "-") {
warnln("File1 and file2 cannot both be the standard input");
return 1;
}
auto process_file = [](const String& path, auto& file, int file_number) {
if (path == "-") {
file = Core::File::standard_input();
} else {
auto file_or_error = Core::File::open(path, Core::OpenMode::ReadOnly);
if (file_or_error.is_error()) {
warnln("Failed to open file{} '{}': {}", file_number, path, file_or_error.error());
return false;
}
if (Core::File::is_directory(path)) {
warnln("Failed to open file{} '{}': is a directory", file_number, path);
return false;
}
file = file_or_error.value();
}
return true;
};
RefPtr<Core::File> file1, file2;
if (!(process_file(file1_path, file1, 1) && process_file(file2_path, file2, 2)))
return 1;
char tab { '\t' };
size_t tab_count { 0 };
String col1_fmt, col2_fmt, col3_fmt;
if (!suppress_col1)
col1_fmt = String::formatted("{}{}", String::repeated(tab, tab_count++), print_color ? COL1_COLOR : "{}");
if (!suppress_col2)
col2_fmt = String::formatted("{}{}", String::repeated(tab, tab_count++), print_color ? COL2_COLOR : "{}");
if (!suppress_col3)
col3_fmt = String::formatted("{}{}", String::repeated(tab, tab_count++), print_color ? COL3_COLOR : "{}");
auto cmp = [&](const String& str1, const String& str2) {
if (case_insensitive)
return strcasecmp(str1.characters(), str2.characters());
else
return strcmp(str1.characters(), str2.characters());
};
auto process_remaining = [](const String& fmt, auto& file, int& count, bool print) {
while (file->can_read_line()) {
++count;
auto line = file->read_line();
if (print)
outln(fmt, line);
}
};
bool read_file1 { true };
bool read_file2 { true };
int col1_count { 0 };
int col2_count { 0 };
int col3_count { 0 };
String file1_line, file2_line;
while (file1->can_read_line() && file2->can_read_line()) {
if (read_file1)
file1_line = file1->read_line();
if (read_file2)
file2_line = file2->read_line();
int cmp_result = cmp(file1_line, file2_line);
if (!cmp_result) {
++col3_count;
read_file1 = read_file2 = true;
if (!suppress_col3)
outln(col3_fmt, file1_line);
} else if (cmp_result < 0) {
++col1_count;
read_file1 = true;
read_file2 = false;
if (!suppress_col1)
outln(col1_fmt, file1_line);
} else {
++col2_count;
read_file1 = false;
read_file2 = true;
if (!suppress_col2)
outln(col2_fmt, file2_line);
}
}
process_remaining(col1_fmt, file1, col1_count, !suppress_col1);
process_remaining(col2_fmt, file2, col2_count, !suppress_col2);
if (print_total)
outln(print_color ? COL1_COLOR "\t" COL2_COLOR "\t" COL3_COLOR "\ttotal" : "{}\t{}\t{}\ttotal", col1_count, col2_count, col3_count);
return 0;
}
|
