Funkin/source/ui/Prompt.hx

package ui;

import flixel.FlxSprite;
import flixel.graphics.frames.FlxAtlasFrames;
import flixel.text.FlxText;
import flixel.util.FlxColor;
import ui.AtlasText;
import ui.MenuList;

class Prompt extends flixel.FlxSubState
{
	inline static var MARGIN = 100;

	public var onYes:Void->Void;
	public var onNo:Void->Void;
	public var buttons:TextMenuList;
	public var field:AtlasText;
	public var back:FlxSprite;

	var style:ButtonStyle;

	public function new(text:String, style:ButtonStyle = Ok)
	{
		this.style = style;
		super(0x80000000);

		buttons = new TextMenuList(Horizontal);

		field = new BoldText(text);
		field.scrollFactor.set(0, 0);
	}

	override function create()
	{
		super.create();

		field.y = MARGIN;
		field.screenCenter(X);
		add(field);

		createButtons();
		add(buttons);
	}

	public function createBg(width:Int, height:Int, color = 0xFF808080)
	{
		back = new FlxSprite();
		back.makeGraphic(width, height, color, false, "prompt-bg");
		back.screenCenter(XY);
		add(back);
		members.unshift(members.pop()); // bring to front
	}

	public function createBgFromMargin(margin = MARGIN, color = 0xFF808080)
	{
		createBg(Std.int(FlxG.width - margin * 2), Std.int(FlxG.height - margin * 2), color);
	}

	public function setButtons(style:ButtonStyle)
	{
		if (this.style != style)
		{
			this.style = style;
			createButtons();
		}
	}

	function createButtons()
	{
		// destroy previous buttons
		while (buttons.members.length > 0)
		{
			buttons.remove(buttons.members[0], true).destroy();
		}

		switch (style)
		{
			case Yes_No:
				createButtonsHelper("yes", "no");
			case Ok:
				createButtonsHelper("ok");
			case Custom(yes, no):
				createButtonsHelper(yes, no);
			case None:
				buttons.exists = false;
		};
	}

	function createButtonsHelper(yes:String, ?no:String)
	{
		buttons.exists = true;
		// pass anonymous functions rather than the current callbacks, in case they change later
		var yesButton = buttons.createItem(yes, function() onYes());
		yesButton.screenCenter(X);
		yesButton.y = FlxG.height - yesButton.height - MARGIN;
		yesButton.scrollFactor.set(0, 0);
		if (no != null)
		{
			// place right
			yesButton.x = FlxG.width - yesButton.width - MARGIN;

			var noButton = buttons.createItem(no, function() onNo());
			noButton.x = MARGIN;
			noButton.y = FlxG.height - noButton.height - MARGIN;
			noButton.scrollFactor.set(0, 0);
		}
	}

	public function setText(text:String)
	{
		field.text = text;
		field.screenCenter(X);
	}
}

enum ButtonStyle
{
	Ok;
	Yes_No;
	Custom(yes:String, no:Null<String>); // Todo: more than 2
	None;
}