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


#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>

#include <Errors.h>
#include <StringList.h>

#include <package/Context.h>
#include <package/RefreshRepositoryRequest.h>
#include <package/PackageRoster.h>

#include "Command.h"
#include "DecisionProvider.h"
#include "JobStateListener.h"
#include "pkgman.h"


using namespace BPackageKit;


// TODO: internationalization!


static const char* const kShortUsage =
        "  %command% [<repo-name> ...]\n"
        "    Refreshes all or just the given repositories.\n";

static const char* const kLongUsage =
        "Usage: %program% %command% [<repo-name> ...]\n"
        "Refreshes all or just the given repositories.\n"
        "\n";


DEFINE_COMMAND(RefreshCommand, "refresh", kShortUsage, kLongUsage,
        COMMAND_CATEGORY_REPOSITORIES)


int
RefreshCommand::Execute(int argc, const char* const* argv)
{
        while (true) {
                static struct option sLongOptions[] = {
                        { "help", no_argument, 0, 'h' },
                        { 0, 0, 0, 0 }
                };

                opterr = 0; // don't print errors
                int c = getopt_long(argc, (char**)argv, "hu", sLongOptions, NULL);
                if (c == -1)
                        break;

                switch (c) {
                        case 'h':
                                PrintUsageAndExit(false);
                                break;

                        default:
                                PrintUsageAndExit(true);
                                break;
                }
        }

        // The remaining arguments are repo names.
        const char* const* repoArgs = argv + optind;
        int nameCount = argc - optind;

        BStringList repositoryNames(20);

        BPackageRoster roster;
        if (nameCount == 0) {
                status_t result = roster.GetRepositoryNames(repositoryNames);
                if (result != B_OK)
                        DIE(result, "can't collect repository names");
        } else {
                for (int i = 0; i < nameCount; ++i) {
                        if (!repositoryNames.Add(repoArgs[i]))
                                DIE(B_NO_MEMORY, "can't allocate repository name");
                }
        }

        DecisionProvider decisionProvider;
        JobStateListener listener;
        BContext context(decisionProvider, listener);

        status_t result;
        for (int i = 0; i < repositoryNames.CountStrings(); ++i) {
                const BString& repoName = repositoryNames.StringAt(i);
                BRepositoryConfig repoConfig;
                result = roster.GetRepositoryConfig(repoName, &repoConfig);
                if (result != B_OK) {
                        BPath path;
                        repoConfig.Entry().GetPath(&path);
                        WARN(result, "skipping repository-config '%s'", path.Path());
                        continue;
                }

                BRefreshRepositoryRequest refreshRequest(context, repoConfig);
                result = refreshRequest.Process();
                if (result != B_OK) {
                        DIE(result, "request for refreshing repository \"%s\" failed",
                                repoName.String());
                }
        }

        return 0;
}