/*
 * Copyright 2019-2023, Adrien Destugues, pulkomandy@pulkomandy.tk.
 * Distributed under the terms of the MIT License.
 */


#include "AlertWithCheckbox.h"

#include <algorithm>
#include <stdio.h>

#include <Button.h>
#include <Catalog.h>
#include <CheckBox.h>
#include <ControlLook.h>
#include <GroupLayout.h>
#include <LayoutBuilder.h>
#include <Locale.h>
#include <IconUtils.h>
#include <Resources.h>
#include <StripeView.h>
#include <TextView.h>

#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "DebugServer"


enum {
	kButton1Message,
	kButton2Message,
	kButton3Message
};


AlertWithCheckbox::AlertWithCheckbox(const char* title, const char* messageText,
	const char* checkboxLabel, const char* label1, const char* label2, const char* label3)
	:
	BWindow(BRect(0, 0, 100, 50), title,
		B_MODAL_WINDOW_LOOK, B_FLOATING_ALL_WINDOW_FEEL,
		B_CLOSE_ON_ESCAPE | B_NOT_RESIZABLE | B_AUTO_UPDATE_SIZE_LIMITS),
	fBitmap(IconSize(), B_RGBA32),
	fSemaphore(create_sem(0, "AlertWithCheckbox")),
	fAction(0)
{
	BIconUtils::GetSystemIcon("dialog-information", &fBitmap);
	BStripeView *stripeView = new BStripeView(fBitmap);

	BTextView *message = new BTextView("_tv_");
	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->SetText(messageText);
	BRect textRect = message->TextRect();
	textRect.PrintToStream();
	message->SetWordWrap(true);
	message->SetExplicitMaxSize(BSize(B_SIZE_UNSET, B_SIZE_UNSET));
	float width = message->StringWidth("W") * 40;
	if (width < textRect.Width()) {
		message->SetExplicitMinSize(BSize(width, B_SIZE_UNSET));
	}

	fDontAskAgain = new BCheckBox("checkbox", checkboxLabel, NULL);

	BGroupView* buttonGroup = new BGroupView(B_HORIZONTAL);
	BLayoutBuilder::Group<> layoutBuilder(buttonGroup);

	layoutBuilder.AddGlue();

	if (label1) {
		BButton* button = new BButton(label1, label1, new BMessage(kButton1Message));
		button->MakeDefault(true);
		layoutBuilder.Add(button);
	}

	if (label2) {
		BButton* button = new BButton(label2, label2, new BMessage(kButton2Message));
		layoutBuilder.Add(button);
	}

	if (label3) {
		BButton* button = new BButton(label3, label3, new BMessage(kButton3Message));
		layoutBuilder.Add(button);
	}

	BLayoutBuilder::Group<>(this)
		.AddGroup(B_HORIZONTAL)
			.Add(stripeView)
			.AddGroup(B_VERTICAL)
				.SetInsets(B_USE_SMALL_SPACING)
				.Add(message)
				.AddGroup(B_HORIZONTAL, 0)
					.Add(fDontAskAgain)
					.AddGlue()
				.End()
				.Add(buttonGroup)
			.End()
		.End();

	ResizeToPreferred();
	CenterOnScreen();
}


AlertWithCheckbox::~AlertWithCheckbox()
{
	delete_sem(fSemaphore);
}


void
AlertWithCheckbox::MessageReceived(BMessage* message)
{
	switch(message->what) {
		case B_QUIT_REQUESTED:
			release_sem(fSemaphore);
			break;

		case 0:
		case 1:
		case 2:
			fAction = message->what;
			PostMessage(B_QUIT_REQUESTED);
			return;
	}

	BWindow::MessageReceived(message);
}


int32
AlertWithCheckbox::Go(bool& dontAskAgain)
{
	Show();

	// Wait for user to close the window
	acquire_sem(fSemaphore);
	dontAskAgain = fDontAskAgain->Value();
	return fAction;
}


BRect
AlertWithCheckbox::IconSize()
{
	return BRect(BPoint(0, 0), be_control_look->ComposeIconSize(32));
}