haiku.git

/*
 * 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 <SupportDefs.h>
#include <Url.h>

#include <package/AddRepositoryRequest.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-base-url>\n"
	"    Adds the repository with the given <repo-base-URL>.\n";

static const char* const kLongUsage =
	"Usage: %program% %command% <repo-URL> [<repo-URL> ...]\n"
	"Adds one or more repositories by downloading them from the given URL(s).\n"
	"\n";


DEFINE_COMMAND(AddRepoCommand, "add-repo", kShortUsage, kLongUsage,
	COMMAND_CATEGORY_REPOSITORIES)


int
AddRepoCommand::Execute(int argc, const char* const* argv)
{
	bool asUserRepository = false;

	while (true) {
		static struct option sLongOptions[] = {
			{ "help", no_argument, 0, 'h' },
			{ "user", no_argument, 0, 'u' },
			{ 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;

			case 'u':
				asUserRepository = true;
				break;

			default:
				PrintUsageAndExit(true);
				break;
		}
	}

	// The remaining arguments are repo URLs, i. e. at least one more argument.
	if (argc < optind + 1)
		PrintUsageAndExit(true);

	const char* const* repoURLs = argv + optind;
	int urlCount = argc - optind;

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

	status_t result;
	for (int i = 0; i < urlCount; ++i) {
		// Test if a valid URL has been supplied before attempting to add
		BUrl repoURL(repoURLs[i], true);
		if (!repoURL.IsValid()) {
			result = B_BAD_VALUE;
			DIE(result, "request for adding repository \"%s\" failed",
				repoURLs[i]);
		}
		AddRepositoryRequest addRequest(context, repoURLs[i], asUserRepository);
		result = addRequest.Process(true);
		if (result != B_OK) {
			if (result != B_CANCELED) {
				DIE(result, "request for adding repository \"%s\" failed",
					repoURLs[i]);
			}
			return 1;
		}

		// now refresh the repo-cache of the new repository
		BString repoName = addRequest.RepositoryName();
		BPackageRoster roster;
		BRepositoryConfig repoConfig;
		roster.GetRepositoryConfig(repoName, &repoConfig);

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

	return 0;
}