beenull/ladybird
1
0
Fork 0
mirror of https://github.com/LadybirdBrowser/ladybird.git synced 2024-11-22 23:50:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

nl: Use the Stream API instead of a FILE *

Browse source
This commit is contained in:
Lucas CHOLLET 2023-06-07 11:11:37 -04:00 committed by Andreas Kling
parent 14826ced68
commit bccd12f297
Notes: sideshowbarker 2024-07-17 01:06:10 +09:00 
Author: https://github.com/LucasChollet
Commit: https://github.com/SerenityOS/serenity/commit/bccd12f297
Pull-request: https://github.com/SerenityOS/serenity/pull/19305
1 changed files with 17 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
Userland/Utilities/nl.cpp
View file

@ -7,10 +7,8 @@
#include <AK/DeprecatedString.h>
#include <AK/Vector.h>
#include <LibCore/ArgsParser.h>
#include <LibCore/File.h>
#include <LibMain/Main.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
enum NumberStyle {
NumberAllLines,
@ -25,7 +23,7 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
StringView separator = " "sv;
int start_number = 1;
int number_width = 6;
Vector<DeprecatedString> files;
Vector<StringView> filenames;
Core::ArgsParser args_parser;
@ -54,28 +52,25 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
args_parser.add_option(separator, "Separator between line numbers and lines", "separator", 's', "string");
args_parser.add_option(start_number, "Initial line number", "startnum", 'v', "number");
args_parser.add_option(number_width, "Number width", "width", 'w', "number");
args_parser.add_positional_argument(files, "Files to process", "file", Core::ArgsParser::Required::No);
args_parser.add_positional_argument(filenames, "Files to process", "file", Core::ArgsParser::Required::No);
args_parser.parse(arguments);
Vector<FILE*> file_pointers;
if (!files.is_empty()) {
for (auto& file : files) {
FILE* file_pointer = fopen(file.characters(), "r");
if (!file_pointer) {
warnln("Failed to open {}: {}", file, strerror(errno));
continue;
}
file_pointers.append(file_pointer);
}
} else {
file_pointers.append(stdin);
}
if (filenames.is_empty())
filenames.append(""sv);
for (auto const filename : filenames) {
auto maybe_file = Core::File::open_file_or_standard_stream(filename, Core::File::OpenMode::Read);
if (maybe_file.is_error()) {
warnln("Failed to open {}: {}", filename, maybe_file.release_error());
continue;
}
auto file = maybe_file.release_value();
for (auto& file_pointer : file_pointers) {
int line_number = start_number - increment; // so the line number can start at 1 when added below
int previous_character = 0;
int next_character = 0;
while ((next_character = fgetc(file_pointer)) != EOF) {
u8 next_character;
for (Bytes bytes = TRY(file->read_some({ &next_character, 1 })); bytes.size() != 0; bytes = TRY(file->read_some(bytes))) {
if (previous_character == 0 || previous_character == '\n') {
if (next_character == '\n' && number_style != NumberAllLines) {
// Skip printing line count on empty lines.
@ -90,7 +85,7 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
putchar(next_character);
previous_character = next_character;
}
fclose(file_pointer);
if (previous_character != '\n')
outln(); // for cases where files have no trailing newline
}