#include <CtrlLib/CtrlLib.h>

using namespace Upp;

class MyMenuBar : public MenuBar
{
	public:
	typedef MyMenuBar CLASSNAME;
	MyMenuBar() {
		menustyle = MenuBar::StyleDefault();
		menustyle.menutext = Color(198, 198, 0);
		menustyle.popupframe = menustyle.popupbody = menustyle.popupiconbar = Black();
		SetStyle(menustyle);
	}

// Two MenuBar::Execute overrides to force MenuBar run my styled bar
	static void Execute(Ctrl *owner, Callback1<Bar&> proc, Point p) {
		MyMenuBar bar;
		proc(bar);
		bar.Execute(owner, p);
	}
	
	void Execute(Ctrl * owner, Point p) {
		MenuBar::Execute(owner, p);
	}

	MenuBar::Style	menustyle;
};

class MyTreeCtrl : public TreeCtrl
{
	public:
	typedef MyTreeCtrl CLASSNAME;
	
	MyTreeCtrl() {}

// override circumventing regular TreeCtrl behavior, so that it uses my styled bar instead of std MenuBar
	virtual void RightDown(Point p, dword flags)
	{
		TreeCtrl::RightDown(p, flags);
		if(MyWhenBar)
			MyMenuBar::Execute(GetActiveCtrl(), MyWhenBar, GetMousePos());
	}
	
	Callback1<Bar&>		MyWhenBar;
};

class MyApp : public TopWindow
{
	public:
	typedef MyApp CLASSNAME;

	MyApp() {
		tree.LeftPos(10, 300).TopPos(10, 200);
		TreeCtrl::Node node;
		for(int i = 0; i < 20; i++) {
			node.Set(Format("item %d", i));
			tree.Add(0, node);
		}
		tree.OpenDeep(0);
// notice here I use my WhenBar callback
		tree.MyWhenBar = THISBACK(HandleTreeBar);
		Add(tree);
	}
	
	void HandleTreeBar(Bar& bar){
		bar.Add("foo", THISBACK(ActionSink));
		bar.Add("bar", THISBACK(ActionSink));
	}
	
	void ActionSink()
	{
		PromptOK("Upp");
	}
	
	MyTreeCtrl		tree;
};

GUI_APP_MAIN
{
	MyApp().Run();
}