/*
 * Copyright 2011, Oliver Tappe <zooey@hirschkaefer.de>
 * Distributed under the terms of the MIT License.
 */


#include "pkgman.h"

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <package/manager/Exceptions.h>

#include "Command.h"


using namespace BPackageKit::BManager::BPrivate;


extern const char* __progname;
const char* kProgramName = __progname;


static const char* const kUsage =
        "Usage: %s <command> <command args>\n"
        "Manages packages and package repositories.\n"
        "\n"
        "Package management commands:\n"
        "%s"
        "Repository management commands:\n"
        "%s"
        "Other commands:\n"
        "%s"
        "Common options:\n"
        "  -h, --help   - Print usage info for a specific command.\n"
;


static BString
get_commands_usage_for_category(const char* category)
{
        BString commandsUsage;
        CommandList commands;
        CommandManager::Default()->GetCommandsForCategory(category, commands);
        for (int32 i = 0; Command* command = commands.ItemAt(i); i++)
                commandsUsage << command->ShortUsage() << '\n';
        return commandsUsage;
}


void
print_usage_and_exit(bool error)
{
        BString packageCommandsUsage
                = get_commands_usage_for_category(COMMAND_CATEGORY_PACKAGES);
        BString repositoryCommandsUsage
                = get_commands_usage_for_category(COMMAND_CATEGORY_REPOSITORIES);
        BString otherCommandsUsage
                = get_commands_usage_for_category(COMMAND_CATEGORY_OTHER);

    fprintf(error ? stderr : stdout, kUsage, kProgramName,
        packageCommandsUsage.String(), repositoryCommandsUsage.String(),
        otherCommandsUsage.String());

    exit(error ? 1 : 0);
}


int
main(int argc, const char* const* argv)
{
        CommandManager::Default()->InitCommands(kProgramName);

        if (argc < 2)
                print_usage_and_exit(true);

        const char* command = argv[1];
        if (strcmp(command, "help") == 0)
                print_usage_and_exit(false);

        CommandList commands;
        CommandManager::Default()->GetCommands(command, commands);
        if (commands.CountItems() != 1)
                print_usage_and_exit(true);

        try {
                return commands.ItemAt(0)->Execute(argc - 1, argv + 1);
        } catch (BNothingToDoException&) {
                fprintf(stderr, "Nothing to do.\n");
                return 0;
        } catch (std::bad_alloc&) {
                fprintf(stderr, "Out of memory!\n");
                return 1;
        } catch (BFatalErrorException& exception) {
                if (!exception.Details().IsEmpty())
                        fprintf(stderr, "%s", exception.Details().String());
                if (exception.Error() == B_OK) {
                        fprintf(stderr, "*** %s\n", exception.Message().String());
                } else {
                        fprintf(stderr, "*** %s: %s\n", exception.Message().String(),
                                strerror(exception.Error()));
                }
                return 1;
        } catch (BAbortedByUserException&) {
                return 0;
        }
}