/*
 * Copyright 2000, Georges-Edouard Berenger. All rights reserved.
 * Distributed under the terms of the MIT License.
 */


#include "QuitMenu.h"
#include "IconMenuItem.h"
#include "ProcessController.h"

#include <Roster.h>
#include <Window.h>
#include <stdio.h>


class QuitMenuItem : public IconMenuItem {
        public:
                                QuitMenuItem(team_id team, BBitmap* icon, const char* title,
                                        BMessage* m, bool purge = false);
                team_id Team() { return fTeam; }

        private:
                team_id fTeam;
};


QuitMenuItem::QuitMenuItem(team_id team, BBitmap* icon, const char* title,
        BMessage* m, bool purge)
        :
        IconMenuItem(icon, title, m, true, purge), fTeam(team)
{
}


//      #pragma mark -


QuitMenu::QuitMenu(const char* title, info_pack* infos, int infosCount)
        : BMenu(title),
        fInfos(infos),
        fInfosCount(infosCount),
        fMe(NULL)
{
        SetTargetForItems(gPCView);
}


void
QuitMenu::AttachedToWindow()
{
        if (!fMe)
                fMe = new BMessenger(this);

        be_roster->StartWatching(*fMe, B_REQUEST_LAUNCHED | B_REQUEST_QUIT);
        BList apps;
        team_id tmid;
        be_roster->GetAppList(&apps);
        for (int t = CountItems() - 1; t >= 0; t--) {
                QuitMenuItem* item = (QuitMenuItem*)ItemAt(t);
                bool found = false;
                for (int a = 0; !found && (tmid = (team_id)(addr_t)apps.ItemAt(a)) != 0; a++)
                        if (item->Team() == tmid)
                                found = true;
                if (!found)
                        RemoveItem(t);
        }
        for (int a = 0; (tmid = (team_id)(addr_t) apps.ItemAt(a)) != 0; a++) {
                AddTeam(tmid);
        }

        BMenu::AttachedToWindow();
}


void
QuitMenu::DetachedFromWindow()
{
        BMenu::DetachedFromWindow();
        be_roster->StopWatching(*fMe);
        delete fMe;
        fMe = NULL;
}


void
QuitMenu::AddTeam(team_id tmid)
{
        int     t = 0;
        QuitMenuItem* item;
        while ((item = (QuitMenuItem*) ItemAt(t++)) != NULL) {
                if (item->Team() == tmid)
                        return;
        }

        t = 0;
        while (t < fInfosCount && tmid != fInfos[t].team_info.team) {
                t++;
        }

        BMessage* message = new BMessage ('QtTm');
        message->AddInt32 ("team", tmid);
        item = NULL;
        if (t < fInfosCount)
                item = new QuitMenuItem(tmid, fInfos[t].team_icon, fInfos[t].team_name,
                        message);
        else {
                info_pack infos;
                if (get_team_info(tmid, &infos.team_info) == B_OK
                        && get_team_name_and_icon(infos, true)) {
                        item = new QuitMenuItem(tmid, infos.team_icon, infos.team_name,
                                message, true);
                }
        }
        if (item) {
                item->SetTarget(gPCView);
                AddItem(item);
        } else
                delete message;
}


void
QuitMenu::MessageReceived(BMessage *msg)
{
        switch (msg->what) {
                case B_SOME_APP_LAUNCHED:
                {
                        int32 tmid;
                        if (msg->FindInt32("be:team", &tmid) == B_OK)
                                AddTeam(tmid);
                        break;
                }
                case B_SOME_APP_QUIT:
                {
                        int32 tmid;
                        if (msg->FindInt32("be:team", &tmid) == B_OK) {
                                QuitMenuItem* item;
                                int t = 0;
                                while ((item = (QuitMenuItem*) ItemAt(t++)) != NULL) {
                                        if (item->Team() == tmid) {
                                                delete RemoveItem(--t);
                                                return;
                                        }
                                }
                        }
                        break;
                }

                default:
                        BMenu::MessageReceived(msg);
        }
}