/*
 * Copyright (c) 2001-2014 Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Daniel Reinhold, danielre@users.sf.net
 *		John Scipione, jscipione@gmail.com
 */

/*!	Lists image info for all currently running teams. */


#include <ctype.h>
#include <image.h>
#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <OS.h>


static status_t
list_images_for_team_by_id(team_id id)
{
	image_info imageInfo;
	int32 cookie = 0;
	team_info teamInfo;
	char* header;
	char* format;
	int i;
	status_t result = get_team_info(id, &teamInfo);
	if (id != 1 && result < B_OK)
		return result;

	i = asprintf(&header, "   ID   %*s   %*s  Seq#      Init# Name",
		sizeof(uintptr_t) * 2, "Text", sizeof(uintptr_t) * 2, "Data");
	if (i == -1)
		return B_NO_MEMORY;

	i = asprintf(&format, "%%5" B_PRId32 " 0x%%0%" B_PRIu32 PRIxPTR
		" 0x%%0%" B_PRIu32 PRIxPTR "  %%4" B_PRId32 " %%10" B_PRIu32 " %%s\n",
		sizeof(uintptr_t) * 2, sizeof(uintptr_t) * 2);
	if (i == -1) {
		free(header);
		return B_NO_MEMORY;
	}

	if (id == 1)
		printf("\nKERNEL TEAM:\n");
	else
		printf("\nTEAM %4" B_PRId32 " (%s):\n", id, teamInfo.args);

	puts(header);
	for (i = 0; i < 80; i++)
		putchar('-');

	printf("\n");
	while ((result = get_next_image_info(id, &cookie, &imageInfo)) == B_OK) {
		printf(format, imageInfo.id, imageInfo.text, imageInfo.data,
			imageInfo.sequence, imageInfo.init_order, imageInfo.name);
	}

	free(header);
	free(format);

	if (result != B_ENTRY_NOT_FOUND && result != EINVAL) {
		printf("get images failed: %s\n", strerror(result));
		return result;
	}

	return B_OK;
}


static void
list_images_for_team(const char* arg)
{
	int32 cookie = 0;
	team_info info;
	status_t result;

	if (atoi(arg) > 0 && list_images_for_team_by_id(atoi(arg)) == B_OK)
		return;

	/* search for the team by name */

	while (get_next_team_info(&cookie, &info) >= B_OK) {
		if (strstr(info.args, arg)) {
			result = list_images_for_team_by_id(info.team);
			if (result != B_OK) {
				printf("\nCould not retrieve information about team %"
					B_PRId32 ": %s\n", info.team, strerror(result));
			}
		}
	}
}


int
main(int argc, char** argv)
{
	int32 cookie = 0;
	team_info info;
	status_t result;

	if (argc == 1) {
		/* list for all teams */
		while (get_next_team_info(&cookie, &info) >= B_OK) {
			result = list_images_for_team_by_id(info.team);
			if (result != B_OK) {
				printf("\nCould not retrieve information about team %"
					B_PRId32 ": %s\n", info.team, strerror(result));
			}
		}
	} else {
		/* list for each team_id on the command line */
		while (--argc > 0 && ++argv != NULL)
			list_images_for_team(*argv);
	}

	return 0;
}