haiku-beta6/src/preferences/locale/LocaleWindow.cpp

/*
 * Copyright 2005-2009, Axel Dörfler, [email protected].
 * All rights reserved. Distributed under the terms of the MIT License.
 */


#include "Locale.h"
#include "LocaleWindow.h"

#include <Alert.h>
#include <Application.h>
#include <Bitmap.h>
#include <Button.h>
#include <Catalog.h>
#include <GroupLayout.h>
#include <LayoutBuilder.h>
#include <ListView.h>
#include <Locale.h>
#include <LocaleRoster.h>
#include <Screen.h>
#include <ScrollView.h>
#include <StringView.h>
#include <TabView.h>

#include <ICUWrapper.h>

#include <unicode/locid.h>
#include <unicode/datefmt.h>

#include "TimeFormatSettingsView.h"


#define TR_CONTEXT "Locale Preflet Window"
#define MAX_DRAG_HEIGHT		200.0
#define ALPHA				170
#define TEXT_OFFSET			5.0


class LanguageListItem: public BStringItem
{
	public:
		LanguageListItem(const char* text, const char* code)
		:
		BStringItem(text),
		fLanguageCode(code)
		{}

		~LanguageListItem() {};

		const inline BString LanguageCode() { return fLanguageCode; }

	private:
		const BString fLanguageCode;
};


class LanguageListView: public BListView
{
	public:
		LanguageListView(const char* name, list_view_type type)
			: BListView(name, type)
		{}

		bool InitiateDrag(BPoint point, int32 index, bool wasSelected);
		void MouseMoved(BPoint where, uint32 transit, const BMessage* msg);
		void MoveItems(BList& items, int32 index);
		void AttachedToWindow()
		{
			BListView::AttachedToWindow();
			ScrollToSelection();
		}

		void MessageReceived (BMessage* message)
		{
			if (message->what == 'DRAG') {
				LanguageListView* list = NULL;
				if (message->FindPointer("list", (void**)&list) == B_OK) {
					if (list == this) {
						int32 count = CountItems();
						if (fDropIndex < 0 || fDropIndex > count)
							fDropIndex = count;

						BList items;
						int32 index;
						for (int32 i = 0; message->FindInt32("index", i, &index)
							 	== B_OK; i++)
							if (BListItem* item = ItemAt(index))
								items.AddItem((void*)item);

						if (items.CountItems() > 0) {
							MoveItems(items, fDropIndex);
						}
						fDropIndex = -1;
					} else {
						int32 count = CountItems();
						if (fDropIndex < 0 || fDropIndex > count)
							fDropIndex = count;

						int32 index;
						for (int32 i = 0; message->FindInt32("index", i, &index)
								== B_OK; i++)
							if (BListItem* item = list->RemoveItem(index)) {
								AddItem(item, fDropIndex);
								fDropIndex++;
							}

						fDropIndex = -1;
					}
				}
				Invoke(new BMessage(kMsgPrefLanguagesChanged));
			} else BListView::MessageReceived(message);
		}
	private:
		int32 fDropIndex;
};


void
LanguageListView::MoveItems(BList& items, int32 index)
{
	DeselectAll();
	// we remove the items while we look at them, the insertion index is
	// decreaded when the items index is lower, so that we insert at the right
	// spot after removal
	BList removedItems;
	int32 count = items.CountItems();
	for (int32 i = 0; i < count; i++) {
		BListItem* item = (BListItem*)items.ItemAt(i);
		int32 removeIndex = IndexOf(item);
		if (RemoveItem(item) && removedItems.AddItem((void*)item)) {
			if (removeIndex < index)
				index--;
		}
		// else ??? -> blow up
	}
	for (int32 i = 0;
		 BListItem* item = (BListItem*)removedItems.ItemAt(i); i++) {
		if (AddItem(item, index)) {
			// after we're done, the newly inserted items will be selected
			Select(index, true);
			// next items will be inserted after this one
			index++;
		} else
			delete item;
	}
}


bool
LanguageListView::InitiateDrag(BPoint point, int32 index, bool)
{
	bool success = false;
	BListItem* item = ItemAt(CurrentSelection(0));
	if (!item) {
		// workarround a timing problem
		Select(index);
		item = ItemAt(index);
	}
	if (item) {
		// create drag message
		BMessage msg('DRAG');
		msg.AddPointer("list",(void*)(this));
		int32 index;
		for (int32 i = 0; (index = CurrentSelection(i)) >= 0; i++)
			msg.AddInt32("index", index);
		// figure out drag rect
		float width = Bounds().Width();
		BRect dragRect(0.0, 0.0, width, -1.0);
		// figure out, how many items fit into our bitmap
		int32 numItems;
		bool fade = false;
		for (numItems = 0; BListItem* item = ItemAt(CurrentSelection(numItems));
				numItems++) {
			dragRect.bottom += ceilf(item->Height()) + 1.0;
			if (dragRect.Height() > MAX_DRAG_HEIGHT) {
				fade = true;
				dragRect.bottom = MAX_DRAG_HEIGHT;
				numItems++;
				break;
			}
		}
		BBitmap* dragBitmap = new BBitmap(dragRect, B_RGB32, true);
		if (dragBitmap && dragBitmap->IsValid()) {
			if (BView* v = new BView(dragBitmap->Bounds(), "helper",
									 B_FOLLOW_NONE, B_WILL_DRAW)) {
				dragBitmap->AddChild(v);
				dragBitmap->Lock();
				BRect itemBounds(dragRect) ;
				itemBounds.bottom = 0.0;
				// let all selected items, that fit into our drag_bitmap, draw
				for (int32 i = 0; i < numItems; i++) {
					int32 index = CurrentSelection(i);
					LanguageListItem* item
						= static_cast<LanguageListItem*>(ItemAt(index));
					itemBounds.bottom = itemBounds.top + ceilf(item->Height());
					if (itemBounds.bottom > dragRect.bottom)
						itemBounds.bottom = dragRect.bottom;
					item->DrawItem(v, itemBounds);
					itemBounds.top = itemBounds.bottom + 1.0;
				}
				// make a black frame arround the edge
				v->SetHighColor(0, 0, 0, 255);
				v->StrokeRect(v->Bounds());
				v->Sync();
	
				uint8* bits = (uint8*)dragBitmap->Bits();
				int32 height = (int32)dragBitmap->Bounds().Height() + 1;
				int32 width = (int32)dragBitmap->Bounds().Width() + 1;
				int32 bpr = dragBitmap->BytesPerRow();
	
				if (fade) {
					for (int32 y = 0; y < height - ALPHA / 2;
							y++, bits += bpr) {
						uint8* line = bits + 3;
						for (uint8* end = line + 4 * width; line < end;
								line += 4)
							*line = ALPHA;
					}
					for (int32 y = height - ALPHA / 2; y < height;
							y++, bits += bpr) {
						uint8* line = bits + 3;
						for (uint8* end = line + 4 * width; line < end;
								line += 4)
							*line = (height - y) << 1;
					}
				} else {
					for (int32 y = 0; y < height; y++, bits += bpr) {
						uint8* line = bits + 3;
						for (uint8* end = line + 4 * width; line < end;
								line += 4)
							*line = ALPHA;
					}
				}
				dragBitmap->Unlock();
			}
		} else {
			delete dragBitmap;
			dragBitmap = NULL;
		}
		if (dragBitmap)
			DragMessage(&msg, dragBitmap, B_OP_ALPHA, BPoint(0.0, 0.0));
		else
			DragMessage(&msg, dragRect.OffsetToCopy(point), this);

		success = true;
	}
	return success;
}


void
LanguageListView::MouseMoved(BPoint where, uint32 transit, const BMessage* msg)
{
	if (msg && (msg->what == 'DRAG')) {
		switch (transit) {
			case B_ENTERED_VIEW:
			case B_INSIDE_VIEW: {
				// set drop target through virtual function
				// offset where by half of item height
				BRect r = ItemFrame(0);
				where.y += r.Height() / 2.0;
	
				int32 index = IndexOf(where);
				if (index < 0)
					index = CountItems();
				if (fDropIndex != index) {
					fDropIndex = index;
					if (fDropIndex >= 0) {
						int32 count = CountItems();
						if (fDropIndex == count) {
							BRect r;
							if (ItemAt(count - 1)) {
								r = ItemFrame(count - 1);
								r.top = r.bottom;
								r.bottom = r.top + 1.0;
							} else {
								r = Bounds();
								r.bottom--;
									// compensate for scrollbars moved slightly
									// out of window
							}
						} else {
							BRect r = ItemFrame(fDropIndex);
							r.top--;
							r.bottom = r.top + 1.0;
						}
					}
				}
				break;
			}
		}
	} else {
		BListView::MouseMoved(where, transit, msg);
	}
}


LocaleWindow::LocaleWindow()
	:
	BWindow(BRect(0, 0, 0, 0), "Locale", B_TITLED_WINDOW, B_NOT_RESIZABLE
		| B_NOT_ZOOMABLE | B_ASYNCHRONOUS_CONTROLS | B_AUTO_UPDATE_SIZE_LIMITS
		| B_QUIT_ON_WINDOW_CLOSE)
{
	BCountry* defaultCountry;
	be_locale_roster->GetDefaultCountry(&defaultCountry);

	SetLayout(new BGroupLayout(B_HORIZONTAL));

	BButton* button = new BButton(TR("Defaults"), new BMessage(kMsgDefaults));
	fRevertButton = new BButton(TR("Revert"), new BMessage(kMsgRevert));
	fRevertButton->SetEnabled(false);

	BTabView* tabView = new BTabView("tabview");

	BView* languageTab = new BView(TR("Language"), B_WILL_DRAW);
	languageTab->SetLayout(new BGroupLayout(B_VERTICAL, 0));
	
	{
		// first list: available languages
		fLanguageListView = new LanguageListView("available",
			B_MULTIPLE_SELECTION_LIST);
		BScrollView* scrollView = new BScrollView("scroller", fLanguageListView,
			B_WILL_DRAW | B_FRAME_EVENTS, false, true);

		fLanguageListView->SetInvocationMessage(new BMessage(kMsgLangInvoked));

		// Fill the language list from the LocaleRoster data
		BMessage installedLanguages;
		if (be_locale_roster->GetInstalledLanguages(&installedLanguages)
				== B_OK) {

			BString currentLanguage;
			for (int i = 0; installedLanguages.FindString("langs",
					i, &currentLanguage) == B_OK; i++) {

				// Now get an human-readable, loacalized name for each language
				// TODO: sort them using collators.
				Locale currentLocale
					= Locale::createFromName(currentLanguage.String());
				UnicodeString languageFullName;
				BString str;
				BStringByteSink bbs(&str);
				currentLocale.getDisplayName(currentLocale, languageFullName);
				languageFullName.toUTF8(bbs);
				LanguageListItem* si
					= new LanguageListItem(str, currentLanguage.String());
				fLanguageListView->AddItem(si);
			}

		} else {
			BAlert* myAlert = new BAlert("Error",
				TR("Unable to find the available languages! You can't use this "
					"preflet!"),
				TR("Ok"), NULL, NULL,
				B_WIDTH_AS_USUAL, B_OFFSET_SPACING, B_STOP_ALERT);
			myAlert->Go();
		}

		// Second list: active languages
		fPreferredListView = new LanguageListView("preferred",
			B_MULTIPLE_SELECTION_LIST);
		BScrollView* scrollViewEnabled = new BScrollView("scroller",
			fPreferredListView, B_WILL_DRAW | B_FRAME_EVENTS, false, true);

		fPreferredListView
			->SetInvocationMessage(new BMessage(kMsgPrefLangInvoked));

		// get the preferred languages from the Settings. Move them here from
		// the other list.
		BMessage msg;
		be_locale_roster->GetPreferredLanguages(&msg);
		BString langCode;
		for (int index = 0; msg.FindString("language", index, &langCode)
					== B_OK;
				index++) {
			for (int listPos = 0; LanguageListItem* lli
					= static_cast<LanguageListItem*>
						(fLanguageListView->ItemAt(listPos));
					listPos++) {
				if (langCode == lli->LanguageCode()) {
					fPreferredListView->AddItem(lli);
					fLanguageListView->RemoveItem(lli);
				}
			}
		}

		languageTab->AddChild(BLayoutBuilder::Group<>(B_HORIZONTAL, 10)
			.AddGroup(B_VERTICAL, 10)
				.Add(new BStringView("", TR("Available languages")))
				.Add(scrollView)
				.End()
			.AddGroup(B_VERTICAL, 10)
				.Add(new BStringView("", TR("Preferred languages")))
				.Add(scrollViewEnabled)
				.End()
			.View());
	}

	BView* countryTab = new BView(TR("Country"), B_WILL_DRAW);
	countryTab->SetLayout(new BGroupLayout(B_VERTICAL, 0));

	{
		BListView* listView = new BListView("country", B_SINGLE_SELECTION_LIST);
		BScrollView* scrollView = new BScrollView("scroller",
			listView, B_WILL_DRAW | B_FRAME_EVENTS, false, true);
		listView->SetSelectionMessage(new BMessage(kMsgCountrySelection));
		
		// get all available countries from ICU
		// Use DateFormat::getAvailableLocale so we get only the one we can
		// use. Maybe check the NumberFormat one and see if there is more.
		int32_t localeCount;
		const Locale* currentLocale
			= Locale::getAvailableLocales(localeCount);

		for (int index = 0; index < localeCount; index++)
		{
			UnicodeString countryFullName;
			BString str;
			BStringByteSink bbs(&str);
			currentLocale[index].getDisplayName(countryFullName);
			countryFullName.toUTF8(bbs);
			LanguageListItem* si
				= new LanguageListItem(str, currentLocale[index].getName());
			listView->AddItem(si);
			if (strcmp(currentLocale[index].getName(),
					defaultCountry->Code()) == 0)
				listView->Select(listView->CountItems() - 1);
		}

		// TODO: find a real solution intead of this hack
		listView->SetExplicitMinSize(BSize(300, B_SIZE_UNSET));

		fFormatView = new FormatView(defaultCountry);

		countryTab->AddChild(BLayoutBuilder::Group<>(B_HORIZONTAL, 5)
			.AddGroup(B_VERTICAL, 3)
				.Add(scrollView)
				.End()
			.Add(fFormatView)
			.View()
		);

		listView->ScrollToSelection();
	}

	tabView->AddTab(languageTab);
	tabView->AddTab(countryTab);

	BLayoutBuilder::Group<>(this)
		.AddGroup(B_VERTICAL, 3)
			.Add(tabView)
			.AddGroup(B_HORIZONTAL, 3)
				.Add(button)
				.Add(fRevertButton)
				.AddGlue()
				.End()
			.SetInsets(5, 5, 5, 5)
			.End();

	CenterOnScreen();
}


void
LocaleWindow::MessageReceived(BMessage* message)
{
	switch (message->what) {
		case kMsgDefaults:
			// TODO
			break;

		case kMsgRevert:
			// TODO
			break;

		case kMsgPrefLanguagesChanged:
		{
				BMessage update(kMsgSettingsChanged);
				int index = 0;
				while (index < fPreferredListView->CountItems()) {
					update.AddString("language", static_cast<LanguageListItem*>
						(fPreferredListView->ItemAt(index))->LanguageCode());
					index++;
				}
				be_app_messenger.SendMessage(&update);
			break;
		}

		case kMsgCountrySelection:
		{
			// Country selection changed.
			// Get the new selected country from the ListView and send it to the
			// main app event handler.
			void* ptr;
			message->FindPointer("source", &ptr);
			BListView* countryList = static_cast<BListView*>(ptr);
			LanguageListItem* lli = static_cast<LanguageListItem*>
				(countryList->ItemAt(countryList->CurrentSelection()));
			BMessage* newMessage = new BMessage(kMsgSettingsChanged);
			newMessage->AddString("country",lli->LanguageCode());
			be_app_messenger.SendMessage(newMessage);

			BCountry* country = new BCountry(lli->LanguageCode());
			fFormatView->SetCountry(country);
			break;
		}

		case kMsgLangInvoked:
		{
			int32 index = 0;
			if (message->FindInt32("index", &index) == B_OK) {
				LanguageListItem* listItem
					= static_cast<LanguageListItem*>
						(fLanguageListView->RemoveItem(index));
				fPreferredListView->AddItem(listItem);
				fPreferredListView
					->Invoke(new BMessage(kMsgPrefLanguagesChanged));
			}
			break;
		}

		case kMsgPrefLangInvoked:
		{
			if (fPreferredListView->CountItems() == 1)
				break;

			int32 index = 0;
			if (message->FindInt32("index", &index) == B_OK) {
				LanguageListItem* listItem
					= static_cast<LanguageListItem*>
						(fPreferredListView->RemoveItem(index));
				fLanguageListView->AddItem(listItem);
				fPreferredListView
					->Invoke(new BMessage(kMsgPrefLanguagesChanged));
			}
			break;
		}

		default:
			BWindow::MessageReceived(message);
			break;
	}
}


void
LocaleWindow::FrameMoved(BPoint newPosition)
{
	BMessage update(kMsgSettingsChanged);
	update.AddPoint("window_location", newPosition);
	be_app_messenger.SendMessage(&update);
}