/*
 * Copyright 2012, Michael Lotz, mmlr@mlotz.ch. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 */


#include "AppAccessRequestWindow.h"

#include <Catalog.h>
#include <LayoutBuilder.h>
#include <LayoutUtils.h>
#include <NodeInfo.h>
#include <Roster.h>
#include <SpaceLayoutItem.h>
#include <TextView.h>

#include <new>

#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "AppAccessRequestWindow"


static const uint32 kMessageDisallow = 'btda';
static const uint32 kMessageOnce = 'btao';
static const uint32 kMessageAlways = 'btaa';


AppAccessRequestWindow::AppAccessRequestWindow(const char* keyringName,
	const char* signature, const char* path, const char* accessString,
	bool appIsNew, bool appWasUpdated)
	:
	BWindow(BRect(50, 50, 100, 100), B_TRANSLATE("Application keyring access"),
		B_TITLED_WINDOW, B_NOT_RESIZABLE | B_ASYNCHRONOUS_CONTROLS
			| B_NOT_ZOOMABLE | B_NOT_MINIMIZABLE | B_AUTO_UPDATE_SIZE_LIMITS
			| B_CLOSE_ON_ESCAPE),
	fDoneSem(-1),
	fResult(kMessageDisallow)
{
	fDoneSem = create_sem(0, "application keyring access dialog");
	if (fDoneSem < 0)
		return;


	BBitmap icon = GetIcon(32 * icon_layout_scale());
	fStripeView = new BStripeView(icon);

	float inset = ceilf(be_plain_font->Size() * 0.7);

	BTextView* message = new(std::nothrow) BTextView("Message");
	if (message == NULL)
		return;

	BString details;
	details <<  B_TRANSLATE("The application:\n"
		"%signature% (%path%)\n\n");
	details.ReplaceFirst("%signature%", signature);
	details.ReplaceFirst("%path%", path);

	if (keyringName != NULL) {
		details <<  B_TRANSLATE("requests access to keyring:\n"
			"%keyringName%\n\n");
		details.ReplaceFirst("%keyringName%", keyringName);
	}

	if (accessString != NULL) {
		details <<  B_TRANSLATE("to perform the following action:\n"
			"%accessString%\n\n");
		details.ReplaceFirst("%accessString%", accessString);
	}

	if (appIsNew)
		details <<  B_TRANSLATE("This application hasn't been granted "
		"access before.");
	else if (appWasUpdated) {
		details <<  B_TRANSLATE("This application has been updated since "
		"it was last granted access.");
	} else {
		details <<  B_TRANSLATE("This application doesn't yet have the "
		"required privileges.");
	}

	message->SetText(details);
	message->SetViewUIColor(B_PANEL_BACKGROUND_COLOR);
	rgb_color textColor = ui_color(B_PANEL_TEXT_COLOR);
	message->SetFontAndColor(be_plain_font, B_FONT_ALL, &textColor);
	message->MakeEditable(false);
	message->MakeSelectable(false);
	message->SetWordWrap(true);

	message->SetExplicitMinSize(BSize(message->StringWidth(
		"01234567890123456789012345678901234567890123456789") + inset,
		B_SIZE_UNSET));

	fDisallowButton = new(std::nothrow) BButton(B_TRANSLATE("Disallow"),
		new BMessage(kMessageDisallow));
	fOnceButton = new(std::nothrow) BButton(B_TRANSLATE("Allow once"),
		new BMessage(kMessageOnce));
	fAlwaysButton = new(std::nothrow) BButton(B_TRANSLATE("Allow always"),
		new BMessage(kMessageAlways));

	BLayoutBuilder::Group<>(this, B_HORIZONTAL, B_USE_ITEM_SPACING)
		.Add(fStripeView)
		.AddGroup(B_VERTICAL, 0)
			.SetInsets(0, B_USE_WINDOW_SPACING,
				B_USE_WINDOW_SPACING, B_USE_WINDOW_SPACING)
			.AddGroup(new BGroupView(B_VERTICAL, B_USE_ITEM_SPACING))
				.Add(message)
			.End()
			.AddStrut(B_USE_SMALL_SPACING)
			.AddGroup(new BGroupView(B_HORIZONTAL))
				.Add(fDisallowButton)
				.AddGlue()
				.Add(fOnceButton)
				.Add(fAlwaysButton)
			.End()
		.End()
	.End();
}


AppAccessRequestWindow::~AppAccessRequestWindow()
{
	if (fDoneSem >= 0)
		delete_sem(fDoneSem);
}


bool
AppAccessRequestWindow::QuitRequested()
{
	fResult = kMessageDisallow;
	release_sem(fDoneSem);
	return false;
}


void
AppAccessRequestWindow::MessageReceived(BMessage* message)
{
	switch (message->what) {
		case kMessageDisallow:
		case kMessageOnce:
		case kMessageAlways:
			fResult = message->what;
			release_sem(fDoneSem);
			return;
	}

	BWindow::MessageReceived(message);
}


status_t
AppAccessRequestWindow::RequestAppAccess(bool& allowAlways)
{
	ResizeToPreferred();
	CenterOnScreen();
	Show();

	while (acquire_sem(fDoneSem) == B_INTERRUPTED)
		;

	status_t result;
	switch (fResult) {
		default:
		case kMessageDisallow:
			result = B_NOT_ALLOWED;
			allowAlways = false;
			break;
		case kMessageOnce:
			result = B_OK;
			allowAlways = false;
			break;
		case kMessageAlways:
			result = B_OK;
			allowAlways = true;
			break;
	}

	LockLooper();
	Quit();
	return result;
}


BBitmap
AppAccessRequestWindow::GetIcon(int32 iconSize)
{
	BBitmap icon(BRect(0, 0, iconSize - 1, iconSize - 1), 0, B_RGBA32);
	team_info teamInfo;
	get_team_info(B_CURRENT_TEAM, &teamInfo);
	app_info appInfo;
	be_roster->GetRunningAppInfo(teamInfo.team, &appInfo);
	BNodeInfo::GetTrackerIcon(&appInfo.ref, &icon, icon_size(iconSize));
	return icon;
}