serenity/Userland/wc.cpp

/*
 * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 *    list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <AK/String.h>
#include <AK/Vector.h>
#include <LibCore/ArgsParser.h>

#include <stdio.h>
#include <sys/stat.h>

struct Count {
    String name;
    bool exists = true;
    unsigned int lines = 0;
    unsigned int characters = 0;
    unsigned int words = 0;
    size_t bytes = 0;
};

bool output_line = false;
bool output_byte = false;
bool output_character = false;
bool output_word = false;

void wc_out(Count& count)
{
    if (output_line)
        printf("%7i ", count.lines);
    if (output_word)
        printf("%7i ", count.words);
    if (output_byte)
        printf("%7lu ", count.bytes);
    if (output_character)
        printf("%7i ", count.characters);

    printf("%14s\n", count.name.characters());
}

Count get_count(const String& file_name)
{
    Count count;
    FILE* file_pointer = nullptr;
    if (file_name == "-") {
        count.name = "";
        file_pointer = stdin;
    } else {
        count.name = file_name;
        if ((file_pointer = fopen(file_name.characters(), "r")) == NULL) {
            fprintf(stderr, "wc: unable to open %s\n", file_name.characters());
            count.exists = false;
            return count;
        }
    }
    bool tab_flag = false;
    bool space_flag = false;
    bool line_flag = true;
    int current_character;
    while ((current_character = fgetc(file_pointer)) != EOF) {
        count.characters++;
        if (current_character >= 'A' && current_character <= 'z' && (space_flag || line_flag || tab_flag)) {
            count.words++;
            space_flag = false;
            line_flag = false;
            tab_flag = false;
        }
        switch (current_character) {
        case '\n':
            count.lines++;
            line_flag = true;
            break;
        case ' ':
            space_flag = true;
            break;
        case '\t':
            tab_flag = true;
            break;
        }
    }
    fclose(file_pointer);
    if (file_pointer != stdin) {
        struct stat st;
        stat(file_name.characters(), &st);
        count.bytes = st.st_size;
    }
    return count;
}

Count get_total_count(Vector<Count>& 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;
}

int main(int argc, char** argv)
{
    if (pledge("stdio rpath", nullptr) < 0) {
        perror("pledge");
        return 1;
    }

    Vector<const char*> files;

    Core::ArgsParser args_parser;
    args_parser.add_option(output_line, "Output line count", "lines", 'l');
    args_parser.add_option(output_byte, "Output byte count", "bytes", 'c');
    args_parser.add_option(output_word, "Output word count", "words", 'w');
    args_parser.add_positional_argument(files, "File to process", "file", Core::ArgsParser::Required::No);
    args_parser.parse(argc, argv);

    if (!output_line && !output_byte && !output_word)
        output_line = output_byte = output_word = true;

    Vector<Count> counts;
    for (auto& file : files) {
        Count count = get_count(file);
        counts.append(count);
    }

    if (pledge("stdio", nullptr) < 0) {
        perror("pledge");
        return 1;
    }

    if (files.size() > 1) {
        Count total_count = get_total_count(counts);
        counts.append(total_count);
    }

    if (files.is_empty()) {
        Count count = get_count("-");
        counts.append(count);
    }

    for (auto& count : counts)
        if (count.exists)
            wc_out(count);

    return 0;
}