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

Userland: Implement -c [characters] option for head

Browse source
This commit is contained in:
Conrad Pankoff 2019-06-08 23:30:42 +10:00 committed by Andreas Kling
parent 842bf96e2c
commit 7b04c7dc48
Notes: sideshowbarker 2024-07-19 13:40:24 +09:00 
Author: https://github.com/deoxxa
Commit: https://github.com/SerenityOS/serenity/commit/7b04c7dc48a
Pull-request: https://github.com/SerenityOS/serenity/pull/214
1 changed files with 76 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

89
Userland/head.cpp
View file

@ -1,27 +1,53 @@
#include <AK/StdLibExtras.h>
#include <LibCore/CArgsParser.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
int head(const String& filename, bool print_filename, int line_count);
int head(const String& filename, bool print_filename, int line_count, int char_count);
int main(int argc, char** argv)
{
CArgsParser args_parser("head");
args_parser.add_arg("n", "lines", "Number of lines to print (default 10)");
args_parser.add_arg("c", "characters", "Number of characters to print");
args_parser.add_arg("q", "Never print filenames");
args_parser.add_arg("v", "Always print filenames");
CArgsParserResult args = args_parser.parse(argc, (const char**)argv);
int line_count = 10;
int line_count = 0;
if (args.is_present("n")) {
line_count = strtol(args.get("n").characters(), NULL, 10);
if (errno) {
args_parser.print_usage();
return -1;
}
if (!line_count) {
args_parser.print_usage();
return -1;
}
}
int char_count = 0;
if (args.is_present("c")) {
char_count = strtol(args.get("c").characters(), NULL, 10);
if (errno) {
args_parser.print_usage();
return -1;
}
if (!char_count) {
args_parser.print_usage();
return -1;
}
}
if (line_count == 0 && char_count == 0) {
line_count = 10;
}
Vector<String> files = args.get_single_values();
@ -35,13 +61,13 @@ int main(int argc, char** argv)
}
if (files.is_empty()) {
return head("", print_filenames, line_count);
return head("", print_filenames, line_count, char_count);
}
int rc = 0;
for (auto& file : files) {
if (head(file, print_filenames, line_count) != 0) {
if (head(file, print_filenames, line_count, char_count) != 0) {
rc = 1;
}
}
@ -49,7 +75,7 @@ int main(int argc, char** argv)
return rc;
}
int head(const String& filename, bool print_filename, int line_count)
int head(const String& filename, bool print_filename, int line_count, int char_count)
{
bool is_stdin = false;
FILE* fp = nullptr;
@ -73,15 +99,52 @@ int head(const String& filename, bool print_filename, int line_count)
}
}
for (int line = 0; line < line_count; ++line) {
char buffer[BUFSIZ];
auto* str = fgets(buffer, sizeof(buffer), fp);
if (!str)
break;
if (line_count) {
for (int line = 0; line < line_count; ++line) {
char buffer[BUFSIZ];
auto* str = fgets(buffer, sizeof(buffer), fp);
if (!str)
break;
// specifically use fputs rather than puts, because fputs doesn't add
// its own newline.
fputs(str, stdout);
// specifically use fputs rather than puts, because fputs doesn't add
// its own newline.
fputs(str, stdout);
}
} else if (char_count) {
char buffer[BUFSIZ];
while (char_count) {
int nread = fread(buffer, 1, min(BUFSIZ, char_count), fp);
if (nread > 0) {
int ncomplete = 0;
while (ncomplete < nread) {
int nwrote = fwrite(&buffer[ncomplete], 1, nread - ncomplete, stdout);
if (nwrote > 0)
ncomplete += nwrote;
if (feof(stdout)) {
fprintf(stderr, "unexpected eof writing to stdout\n");
return 1;
}
if (ferror(stdout)) {
fprintf(stderr, "error writing to stdout\n");
return 1;
}
}
}
char_count -= nread;
if (feof(fp))
break;
if (ferror(fp)) {
fprintf(stderr, "error reading input\n");
break;
}
}
}
fclose(fp);