ladybird/Userland/Utilities/tree.cpp
Sam Atkins f0aba519c3 Utilities: Read positional arguments as Strings not char*s
This is a pretty trivial change so they're all batched together.
2022-04-11 21:09:42 +02:00

137 lines
3.9 KiB
C++

/*
* Copyright (c) 2020, Stijn De Ridder <stijn.deridder@hotmail.com>
* Copyright (c) 2022, the SerenityOS developers.
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/LexicalPath.h>
#include <AK/QuickSort.h>
#include <AK/String.h>
#include <AK/StringBuilder.h>
#include <AK/Vector.h>
#include <LibCore/ArgsParser.h>
#include <LibCore/DirIterator.h>
#include <LibCore/System.h>
#include <LibMain/Main.h>
#include <errno.h>
#include <limits.h>
#include <stdio.h>
#include <sys/stat.h>
#include <unistd.h>
static bool flag_show_hidden_files = false;
static bool flag_show_only_directories = false;
static int max_depth = INT_MAX;
static int g_directories_seen = 0;
static int g_files_seen = 0;
static void print_directory_tree(String const& root_path, int depth, String const& indent_string)
{
if (depth > 0) {
String root_indent_string;
if (depth > 1) {
root_indent_string = indent_string.substring(0, (depth - 1) * 4);
} else {
root_indent_string = "";
}
out("{}|-- ", root_indent_string);
}
String root_dir_name = LexicalPath::basename(root_path);
out("\033[34;1m{}\033[0m\n", root_dir_name);
if (depth >= max_depth) {
return;
}
Core::DirIterator di(root_path, flag_show_hidden_files ? Core::DirIterator::SkipParentAndBaseDir : Core::DirIterator::SkipDots);
if (di.has_error()) {
warnln("{}: {}", root_path, di.error_string());
return;
}
Vector<String> names;
while (di.has_next()) {
String name = di.next_path();
if (di.has_error()) {
warnln("{}: {}", root_path, di.error_string());
continue;
}
names.append(name);
}
quick_sort(names);
for (size_t i = 0; i < names.size(); i++) {
String name = names[i];
StringBuilder builder;
builder.append(root_path);
if (!root_path.ends_with('/')) {
builder.append('/');
}
builder.append(name);
String full_path = builder.to_string();
struct stat st;
int rc = lstat(full_path.characters(), &st);
if (rc == -1) {
warnln("lstat({}) failed: {}", full_path, strerror(errno));
continue;
}
if (S_ISDIR(st.st_mode)) {
g_directories_seen++;
bool at_last_entry = i == names.size() - 1;
String new_indent_string;
if (at_last_entry) {
new_indent_string = String::formatted("{} ", indent_string);
} else {
new_indent_string = String::formatted("{}| ", indent_string);
}
print_directory_tree(full_path.characters(), depth + 1, new_indent_string);
} else if (!flag_show_only_directories) {
g_files_seen++;
outln("{}|-- {}", indent_string, name);
}
}
}
ErrorOr<int> serenity_main(Main::Arguments arguments)
{
TRY(Core::System::pledge("stdio rpath tty"));
Vector<String> directories;
Core::ArgsParser args_parser;
args_parser.add_option(flag_show_hidden_files, "Show hidden files", "all", 'a');
args_parser.add_option(flag_show_only_directories, "Show only directories", "only-directories", 'd');
args_parser.add_option(max_depth, "Maximum depth of the tree", "maximum-depth", 'L', "level");
args_parser.add_positional_argument(directories, "Directories to print", "directories", Core::ArgsParser::Required::No);
args_parser.parse(arguments);
if (max_depth < 1) {
warnln("{}: Invalid level, must be greater than 0.", arguments.argv[0]);
return 1;
}
if (directories.is_empty()) {
print_directory_tree(".", 0, "");
puts("");
} else {
for (auto const& directory : directories) {
print_directory_tree(directory, 0, "");
puts("");
}
}
outln("{} directories, {} files", g_directories_seen, g_files_seen);
return 0;
}