* 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)
{
}
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);
}
}
