/*
 * Copyright 2002-2005 Haiku Inc.
 * Distributed under the terms of the MIT license
 *
 * Authors:
 *              Mathew Hounsell
 *              Axel Dörfler, axeld@pinc-software.de.
 */


#include <Roster.h>
#include <Path.h>
#include <Entry.h>
#include <List.h>
#include <String.h>

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


static struct option const kLongOptions[] = {
        {"name", no_argument, 0, 'n'},
        {"no-trunc", no_argument, 0, 't'},
        {"help", no_argument, 0, 'h'},
        {NULL}
};

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

static const int32 kNameFieldWidth = 34;

// view modes
static const int32 kStandardMode        = 0x0;
static const int32 kNameOnlyMode        = 0x1;
static const int32 kNoTruncateMode      = 0x2;


void
truncate_string(BString &name, int32 length)
{
        if (name.Length() <= length)
                return;
        if (length < 6)
                length = 6;
        
        int32 beginLength = length / 3 - 1;
        int32 endLength = length - 3 - beginLength;

        BString begin, end;
        name.CopyInto(begin, 0, beginLength); 
        name.CopyInto(end, name.Length() - endLength, endLength);

        name = begin;
        name.Append("...").Append(end);
}


void
output_team(team_id id, int32 mode)
{
        // Get info on team
        app_info info;
        if (be_roster->GetRunningAppInfo(id, &info) != B_OK)
                return;

        // Allocate a entry and get it's path.
        // - works as they are independant (?)
        BEntry entry(&info.ref);
        BPath path(&entry);

        BString name;
        if (mode & kNameOnlyMode)
                name = info.ref.name;
        else
                name = path.Path();

        if ((mode & kNoTruncateMode) == 0)
                truncate_string(name, kNameFieldWidth);

        printf("%6" B_PRId32 " %*s %5" B_PRId32 " %5" B_PRIx32 " (%s)\n", 
                id, (int)kNameFieldWidth, name.String(), 
                info.port, info.flags, info.signature);
}


void
usage(int exitCode)
{
        fprintf(stderr, "usage: %s [-nt]\n"
                "  -n, --name\t\tInstead of the full path, only the name of the teams are written\n"
                "  -t, --no-trunc\tDon't truncate the path name\n"
                "  -h, --help\t\tDisplay this help and exit\n",
                kProgramName);

        exit(exitCode);
}


int
main(int argc, char **argv)
{
        int32 mode = kStandardMode;

        // Don't have an BApplication as it is not needed for what we do

        int c;
        while ((c = getopt_long(argc, argv, "nth", kLongOptions, NULL)) != -1) {
                switch (c) {
                        case 'n':
                                mode |= kNameOnlyMode;
                                break;
                        case 't':
                                mode |= kNoTruncateMode;
                                break;
                        case 0:
                                break;
                        case 'h':
                                usage(0);
                                break;
                        default:
                                usage(1);
                                break;
                }
        }

        // print title line

        printf("  team %*s  port flags signature\n", 
                (int)kNameFieldWidth, mode & kNameOnlyMode ? "name" : "path");

        printf("------ ");
        for (int32 i = 0; i < kNameFieldWidth; i++) 
                putchar('-');
        puts(" ----- ----- ---------");

        // Retrieve the running list.
        BList applicationList;
        be_roster->GetAppList(&applicationList);

        // Iterate through the list.
        for (int32 i = 0; i < applicationList.CountItems(); i++)
                output_team((team_id)(addr_t)applicationList.ItemAt(i), mode);

        return 0;
}