/*-
 * Copyright (c) 2008 Juan Romero Pardines.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

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

#include "xbps_api.h"

typedef struct repository_info {
	const char *index_version;
	const char *location_local;
	const char *location_remote;
	size_t total_pkgs;
} repo_info_t;

static bool pkgindex_getinfo(prop_dictionary_t, repo_info_t *);
static void usage(void);

static void
usage(void)
{
	printf("Usage: xbps-bin [action] [arguments]\n\n"
	" Available actions:\n"
        "    repo-add, repo-list, show\n"
	" Action arguments:\n"
	"    repo-add\t[<URI>]\n"
	"    repo-list\t[none]\n"
	"    show\t[<pkgname>]\n"
	"\n"
	" Examples:\n"
	"    $ xbps-bin repo-list\n"
	"    $ xbps-bin repo-add /path/to/directory\n"
	"    $ xbps-bin repo-add http://www.location.org/xbps-repo\n"
	"    $ xbps-bin show klibc\n");
	exit(1);
}

static bool
pkgindex_getinfo(prop_dictionary_t dict, repo_info_t *ri)
{
	if (dict == NULL || ri == NULL)
		return false;

	if (!prop_dictionary_get_cstring_nocopy(dict,
	    "pkgindex-version", &ri->index_version))
		return false;

	if (!prop_dictionary_get_cstring_nocopy(dict,
	    "location-local", &ri->location_local))
		return false;

	/* This one is optional, thus don't panic */
	prop_dictionary_get_cstring_nocopy(dict, "location-remote",
	    &ri->location_remote);

	if (!prop_dictionary_get_uint64(dict, "total-pkgs",
	    &ri->total_pkgs))
		return false;

	/* Reject empty repositories, how could this happen? :-) */
	if (ri->total_pkgs <= 0)
		return false;

	return true;
}

int
main(int argc, char **argv)
{
	prop_dictionary_t dict;
	prop_array_t array;
	repo_info_t *rinfo = NULL;
	char pkgindex[PATH_MAX], *tmp;

	if (argc < 2)
		usage();

	if (strcmp(argv[1], "repo-add") == 0) {
		/* Adds a new repository to the pool. */
		if (argc != 3)
			usage();

		tmp = strncpy(pkgindex, argv[2], sizeof(pkgindex));
		if (sizeof(*tmp) >= sizeof(pkgindex))
			exit(ENAMETOOLONG);

		/* Append trailing slash if needed */
		if (pkgindex[strlen(pkgindex) - 1] != '/')
			strncat(pkgindex, "/", sizeof(pkgindex));

		tmp = strncat(pkgindex, XBPS_PKGINDEX, sizeof(pkgindex));
		if (sizeof(*tmp) >= sizeof(pkgindex))
			exit(ENAMETOOLONG);

		dict = prop_dictionary_internalize_from_file(pkgindex);
		if (dict == NULL) {
			printf("Directory %s does not contain any "
			    "xbps pkgindex file.\n", argv[2]);
			exit(EINVAL);
		}

		rinfo = malloc(sizeof(*rinfo));
		if (rinfo == NULL)
			exit(ENOMEM);

		if (!pkgindex_getinfo(dict, rinfo)) {
			printf("'%s' is incomplete.\n", pkgindex);
			exit(EINVAL);
		}

		if (!xbps_register_repository((const char *)&pkgindex)) {
			printf("ERROR: couldn't register repository (%s)\n",
			    strerror(errno));
			exit(EINVAL);
		}

		printf("Added repository at %s (%s) with %zu packages.\n",
		       rinfo->location_local, rinfo->index_version,
		       rinfo->total_pkgs);
		free(rinfo);

	} else if (strcmp(argv[1], "repo-list") == 0) {
		/* Lists all repositories registered in pool. */
		if (argc != 2)
			usage();

		dict =
		    prop_dictionary_internalize_from_file(XBPS_REPOLIST_PATH);
		if (dict == NULL) {
			printf("cannot find repository list file: %s\n",
			    strerror(errno));
			exit(EINVAL);
		}

		array = prop_dictionary_get(dict, "repository-list");
		if (array)
			xbps_list_strings_in_array(array);

	} else if (strcmp(argv[1], "show") == 0) {
		/* Shows info about a binary package. */
		if (argc != 3)
			usage();

	} else {
		usage();
	}

	exit(0);
}