package clui

import (
	term "github.com/nsf/termbox-go"
)

type ConfirmationDialog struct {
	view    View
	parent  *Composer
	result  int
	onClose func()
}

func NewConfirmationDialog(c *Composer, title, question string, buttons []string, defaultButton int) *ConfirmationDialog {
	dlg := new(ConfirmationDialog)

	if len(buttons) == 0 {
		buttons = []string{"OK"}
	}

	cw, ch := term.Size()

	dlg.parent = c
	dlg.view = c.CreateView(cw/2-12, ch/2-8, 20, 10, title)
	dlg.view.SetModal(true)
	dlg.view.SetPack(Vertical)
	NewFrame(dlg.view, dlg.view, 1, 1, BorderNone, DoNotScale)

	fbtn := NewFrame(dlg.view, dlg.view, 1, 1, BorderNone, 1)
	NewFrame(dlg.view, fbtn, 1, 1, BorderNone, DoNotScale)
	lb := NewLabel(dlg.view, fbtn, 10, 3, question, 1)
	NewFrame(dlg.view, fbtn, 1, 1, BorderNone, DoNotScale)
	lb.SetMultiline(true)

	NewFrame(dlg.view, dlg.view, 1, 1, BorderNone, DoNotScale)
	frm1 := NewFrame(dlg.view, dlg.view, 16, 4, BorderNone, DoNotScale)
	NewFrame(dlg.view, frm1, 1, 1, BorderNone, 1)

	bText := buttons[0]
	btn1 := NewButton(dlg.view, frm1, AutoSize, AutoSize, bText, DoNotScale)
	btn1.OnClick(func(ev Event) {
		dlg.result = DialogButton1
		c.DestroyView(dlg.view)
		if dlg.onClose != nil {
			go dlg.onClose()
		}
	})
	if defaultButton == DialogButton1 {
		dlg.view.ActivateControl(btn1)
	}
	var btn2, btn3 *Button

	if len(buttons) > 1 {
		NewFrame(dlg.view, frm1, 1, 1, BorderNone, 1)
		btn2 = NewButton(dlg.view, frm1, AutoSize, AutoSize, buttons[1], DoNotScale)
		btn2.OnClick(func(ev Event) {
			dlg.result = DialogButton2
			c.DestroyView(dlg.view)
			if dlg.onClose != nil {
				go dlg.onClose()
			}
		})
		if defaultButton == DialogButton2 {
			dlg.view.ActivateControl(btn2)
		}
	}
	if len(buttons) > 2 {
		NewFrame(dlg.view, frm1, 1, 1, BorderNone, 1)
		btn3 = NewButton(dlg.view, frm1, AutoSize, AutoSize, buttons[2], DoNotScale)
		btn3.OnClick(func(ev Event) {
			dlg.result = DialogButton3
			c.DestroyView(dlg.view)
			if dlg.onClose != nil {
				go dlg.onClose()
			}
		})
		if defaultButton == DialogButton3 {
			dlg.view.ActivateControl(btn3)
		}
	}

	NewFrame(dlg.view, frm1, 1, 1, BorderNone, 1)

	dlg.view.OnClose(func(ev Event) {
		if dlg.result == DialogAlive {
			dlg.result = DialogClosed
			c.DestroyView(dlg.view)
			if dlg.onClose != nil {
				go dlg.onClose()
			}
		}
	})

	return dlg
}

func (d *ConfirmationDialog) OnClose(fn func()) {
	d.onClose = fn
}

func (d *ConfirmationDialog) Result() int {
	return d.result
}