mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-11 13:36:13 +03:00
uniq: Improve POSIX compliance
Adds -d, -c, -i, -f and -s options. It differs a little bit from other implementations, but it does the trick and corresponds to POSIX's description
This commit is contained in:
parent
1c5f6003d7
commit
847577bea2
Notes:
sideshowbarker
2024-07-17 07:19:48 +09:00
Author: https://github.com/demostanis Commit: https://github.com/SerenityOS/serenity/commit/847577bea2 Pull-request: https://github.com/SerenityOS/serenity/pull/14674 Reviewed-by: https://github.com/ADKaster Reviewed-by: https://github.com/AtkinsSJ Reviewed-by: https://github.com/MacDue ✅ Reviewed-by: https://github.com/linusg Reviewed-by: https://github.com/timschumi
@ -4,76 +4,111 @@
|
|||||||
* SPDX-License-Identifier: BSD-2-Clause
|
* SPDX-License-Identifier: BSD-2-Clause
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
#include <AK/CharacterTypes.h>
|
||||||
#include <AK/RefPtr.h>
|
#include <AK/RefPtr.h>
|
||||||
|
#include <AK/StringView.h>
|
||||||
#include <LibCore/ArgsParser.h>
|
#include <LibCore/ArgsParser.h>
|
||||||
|
#include <LibCore/Stream.h>
|
||||||
#include <LibCore/System.h>
|
#include <LibCore/System.h>
|
||||||
#include <errno.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
|
|
||||||
struct linebuf {
|
static ErrorOr<void> write_line_content(StringView line, size_t count, bool duplicates_only, bool print_count, Core::Stream::File& outfile)
|
||||||
char* buf = NULL;
|
|
||||||
size_t len = 0;
|
|
||||||
};
|
|
||||||
|
|
||||||
static FILE* get_stream(char const* filepath, char const* perms)
|
|
||||||
{
|
{
|
||||||
FILE* ret;
|
if (duplicates_only && count <= 1)
|
||||||
|
return {};
|
||||||
|
|
||||||
if (filepath == nullptr) {
|
if (print_count)
|
||||||
if (perms[0] == 'r')
|
TRY(outfile.write(String::formatted("{} {}\n", count, line).bytes()));
|
||||||
return stdin;
|
else
|
||||||
return stdout;
|
TRY(outfile.write(String::formatted("{}\n", line).bytes()));
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
static StringView skip(StringView line, unsigned char_skip_count, unsigned field_skip_count)
|
||||||
|
{
|
||||||
|
line = line.trim("\n"sv);
|
||||||
|
if (field_skip_count) {
|
||||||
|
bool in_field = false;
|
||||||
|
int field_index = 0;
|
||||||
|
unsigned current_field = 0;
|
||||||
|
for (size_t i = 0; i < line.length(); i++) {
|
||||||
|
char c = line[i];
|
||||||
|
if (is_ascii_space(c)) {
|
||||||
|
in_field = false;
|
||||||
|
field_index = i;
|
||||||
|
if (++current_field > field_skip_count)
|
||||||
|
break;
|
||||||
|
} else if (!in_field) {
|
||||||
|
in_field = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
line = line.substring_view(field_index);
|
||||||
}
|
}
|
||||||
|
char_skip_count = min(char_skip_count, line.length());
|
||||||
ret = fopen(filepath, perms);
|
return line.substring_view(char_skip_count);
|
||||||
if (ret == nullptr) {
|
|
||||||
perror("fopen");
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
||||||
{
|
{
|
||||||
TRY(Core::System::pledge("stdio rpath wpath cpath"));
|
TRY(Core::System::pledge("stdio rpath wpath cpath"));
|
||||||
|
|
||||||
char const* inpath = nullptr;
|
StringView inpath;
|
||||||
char const* outpath = nullptr;
|
StringView outpath;
|
||||||
|
bool duplicates_only = false;
|
||||||
|
bool unique_only = false;
|
||||||
|
bool ignore_case = false;
|
||||||
|
bool print_count = false;
|
||||||
|
unsigned skip_chars = 0;
|
||||||
|
unsigned skip_fields = 0;
|
||||||
|
|
||||||
Core::ArgsParser args_parser;
|
Core::ArgsParser args_parser;
|
||||||
|
args_parser.add_option(duplicates_only, "Only print duplicated lines", "repeated", 'd');
|
||||||
|
args_parser.add_option(unique_only, "Only print unique lines (default)", "unique", 'u');
|
||||||
|
args_parser.add_option(ignore_case, "Ignore case when comparing lines", "ignore-case", 'i');
|
||||||
|
args_parser.add_option(print_count, "Prefix each line by its number of occurrences", "count", 'c');
|
||||||
|
args_parser.add_option(skip_chars, "Skip N chars", "skip-chars", 's', "N");
|
||||||
|
args_parser.add_option(skip_fields, "Skip N fields", "skip-fields", 'f', "N");
|
||||||
args_parser.add_positional_argument(inpath, "Input file", "input", Core::ArgsParser::Required::No);
|
args_parser.add_positional_argument(inpath, "Input file", "input", Core::ArgsParser::Required::No);
|
||||||
args_parser.add_positional_argument(outpath, "Output file", "output", Core::ArgsParser::Required::No);
|
args_parser.add_positional_argument(outpath, "Output file", "output", Core::ArgsParser::Required::No);
|
||||||
args_parser.parse(arguments);
|
args_parser.parse(arguments);
|
||||||
|
|
||||||
FILE* infile = get_stream(inpath, "r");
|
if (!unique_only && !duplicates_only) {
|
||||||
FILE* outfile = get_stream(outpath, "w");
|
unique_only = true;
|
||||||
|
} else if (unique_only && duplicates_only) {
|
||||||
struct linebuf buffers[2];
|
// Printing duplicated and unique lines shouldn't print anything
|
||||||
struct linebuf* previous = &(buffers[0]);
|
return 0;
|
||||||
struct linebuf* current = &(buffers[1]);
|
|
||||||
bool first_run = true;
|
|
||||||
for (;;) {
|
|
||||||
errno = 0;
|
|
||||||
ssize_t rc = getline(&(current->buf), &(current->len), infile);
|
|
||||||
if (rc < 0 && errno != 0) {
|
|
||||||
perror("getline");
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
if (rc < 0)
|
|
||||||
break;
|
|
||||||
if (!first_run && strcmp(current->buf, previous->buf) == 0)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
fputs(current->buf, outfile);
|
|
||||||
swap(current, previous);
|
|
||||||
first_run = false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fclose(infile);
|
auto infile = TRY(Core::Stream::BufferedFile::create(TRY(Core::Stream::File::open_file_or_standard_stream(inpath, Core::Stream::OpenMode::Read))));
|
||||||
fclose(outfile);
|
auto outfile = TRY(Core::Stream::File::open_file_or_standard_stream(outpath, Core::Stream::OpenMode::Write));
|
||||||
|
|
||||||
|
size_t count = 0;
|
||||||
|
ByteBuffer previous_buf = TRY(ByteBuffer::create_uninitialized(1024));
|
||||||
|
ByteBuffer current_buf = TRY(ByteBuffer::create_uninitialized(1024));
|
||||||
|
|
||||||
|
StringView previous = TRY(infile->read_line(previous_buf));
|
||||||
|
StringView previous_to_compare = skip(previous, skip_chars, skip_fields);
|
||||||
|
|
||||||
|
while (TRY(infile->can_read_line())) {
|
||||||
|
// FIXME: The buffer does not automatically resize,
|
||||||
|
// and this will return EMSGSIZE if the read line
|
||||||
|
// is more than 1024 bytes.
|
||||||
|
StringView current = TRY(infile->read_line(current_buf));
|
||||||
|
|
||||||
|
StringView current_to_compare = skip(current, skip_chars, skip_fields);
|
||||||
|
bool lines_equal = ignore_case ? current_to_compare.equals_ignoring_case(previous_to_compare) : current_to_compare == previous_to_compare;
|
||||||
|
if (!lines_equal) {
|
||||||
|
TRY(write_line_content(previous, count, duplicates_only, print_count, *outfile));
|
||||||
|
count = 1;
|
||||||
|
} else {
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
swap(current_to_compare, previous_to_compare);
|
||||||
|
swap(current_buf, previous_buf);
|
||||||
|
swap(current, previous);
|
||||||
|
}
|
||||||
|
|
||||||
|
TRY(write_line_content(previous, count, duplicates_only, print_count, *outfile));
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user