ladybird/Userland/Utilities/pkg/main.cpp

/*
 * Copyright (c) 2023, Liav A. <liavalb@hotmail.co.il>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include "AvailablePort.h"
#include "InstalledPort.h"
#include <LibCore/ArgsParser.h>
#include <LibCore/System.h>
#include <LibMain/Main.h>
#include <unistd.h>

static void print_port_details(InstalledPort const& port)
{
    outln("{}, installed as {}, version {}", port.name(), port.type_as_string_view(), port.version());

    if (!port.dependencies().is_empty()) {
        out("    Dependencies:");
        for (auto const& dependency : port.dependencies())
            out(" {}", dependency);
        outln();
    }
}

ErrorOr<int> serenity_main(Main::Arguments arguments)
{
    TRY(Core::System::pledge("stdio recvfd thread unix rpath cpath wpath"));

    TRY(Core::System::unveil("/tmp/session/%sid/portal/request", "rw"));
    TRY(Core::System::unveil("/usr"sv, "c"sv));
    TRY(Core::System::unveil("/usr/Ports"sv, "rwc"sv));
    TRY(Core::System::unveil("/res"sv, "r"sv));
    TRY(Core::System::unveil("/usr/lib"sv, "r"sv));
    TRY(Core::System::unveil(nullptr, nullptr));

    bool verbose = false;
    bool show_all_installed_ports = false;
    bool update_packages_db = false;
    StringView query_package {};

    Core::ArgsParser args_parser;
    args_parser.add_option(show_all_installed_ports, "Show all manually-installed ports", "list-manual-ports", 'l');
    args_parser.add_option(update_packages_db, "Sync/Update ports database", "update-ports-database", 'u');
    args_parser.add_option(query_package, "Query ports database for package name", "query-package", 'q', "Package name to query");
    args_parser.add_option(verbose, "Verbose", "verbose", 'v');
    args_parser.parse(arguments);

    if (!update_packages_db && !show_all_installed_ports && query_package.is_null()) {
        outln("pkg: No action to be performed was specified.");
        return 0;
    }

    HashMap<String, InstalledPort> installed_ports;
    HashMap<String, AvailablePort> available_ports;
    if (show_all_installed_ports || !query_package.is_null()) {
        if (Core::System::access(ports_database, R_OK).is_error()) {
            warnln("pkg: {} isn't accessible, did you install a package in the past?", ports_database);
            return 1;
        }
        installed_ports = TRY(InstalledPort::read_ports_database());
    }

    int return_value = 0;
    if (update_packages_db) {
        if (getuid() != 0) {
            outln("pkg: Requires root to update packages database.");
            return 1;
        }
        return_value = TRY(AvailablePort::update_available_ports_list_file());
    }

    if (!query_package.is_null()) {
        if (Core::System::access("/usr/Ports/AvailablePorts.md"sv, R_OK).is_error()) {
            outln("pkg: Please run this program with -u first!");
            return 0;
        }
        available_ports = TRY(AvailablePort::read_available_ports_list());
    }

    if (show_all_installed_ports) {
        outln("Manually-installed ports:");
        TRY(InstalledPort::for_each_by_type(installed_ports, InstalledPort::Type::Manual, [](auto& port) -> ErrorOr<void> {
            print_port_details(port);
            return {};
        }));
    }

    if (!query_package.is_null()) {
        if (query_package.is_empty()) {
            outln("pkg: Queried package name is empty.");
            return 0;
        }
        return_value = TRY(AvailablePort::query_details_for_package(available_ports, installed_ports, query_package, verbose));
    }

    return return_value;
}