import textwrap
from pygame import Rect
from pygame.locals import *
from widget import Widget
from controls import Label, TextField, Button, Row, Column

class Modal(object):

	enter_response = True
	cancel_response = False

	def ok(self):
		self.dismiss(True)
	
	def cancel(self):
		self.dismiss(False)


class Dialog(Modal, Widget):

	click_outside_response = None

	def __init__(self, client = None, responses = None,
			default = 0, cancel = -1, **kwds):
		Widget.__init__(self, **kwds)
		if client or responses:
			rows = []
			w1 = 0
			w2 = 0
			if client:
				rows.append(client)
				w1 = client.width
			if responses:
				buttons = Row([
					Button(text, action = lambda t=text: self.dismiss(t))
						for text in responses])
				rows.append(buttons)
				w2 = buttons.width
			if w1 < w2:
				a = 'l'
			else:
				a = 'r'
			contents = Column(rows, align = a)
			m = self.margin
			contents.topleft = (m, m)
			self.add(contents)
			self.shrink_wrap()
		if responses and default is not None:
			self.enter_response = responses[default]
		if responses and cancel is not None:
			self.cancel_response = responses[cancel]
	
	def mouse_down(self, e):
		if not e in self:
			response = self.click_outside_response
			if response is not None:
				self.dismiss(response)
	

def wrapped_label(text, wrap_width, **kwds):
	paras = text.split("\n\n")
	text = "\n".join([textwrap.fill(para, wrap_width) for para in paras])
	return Label(text, **kwds)

def alert(mess, wrap_width = 60, **kwds):
	box = Dialog(**kwds)
	d = box.margin
	lb = wrapped_label(mess, wrap_width)
	lb.topleft = (d, d)
	box.add(lb)
	box.shrink_wrap()
	return box.present()

def alert(mess, **kwds):
	ask(mess, ["OK"], **kwds)

def ask(mess, responses = ["OK", "Cancel"], default = 0, cancel = -1,
		wrap_width = 60, **kwds):
	box = Dialog(**kwds)
	d = box.margin
	lb = wrapped_label(mess, wrap_width)
	lb.topleft = (d, d)
	buts = []
	for caption in responses:
		but = Button(caption, action = lambda x = caption: box.dismiss(x))
		buts.append(but)
	brow = Row(buts, spacing = d)
	lb.width = max(lb.width, brow.width)
	col = Column([lb, brow], spacing = d, align ='r')
	col.topleft = (d, d)
	if default is not None:
		box.enter_response = responses[default]
	else:
		box.enter_response = None
	if cancel is not None:
		box.cancel_response = responses[cancel]
	else:
		box.cancel_response = None
	box.add(col)
	box.shrink_wrap()
	return box.present()

def input_text(prompt, width, initial = None, **kwds):
	box = Dialog(**kwds)
	d = box.margin
	def ok():
		box.dismiss(True)
	def cancel():
		box.dismiss(False)
	lb = Label(prompt)
	lb.topleft = (d, d)
	tf = TextField(width)
	if initial:
		tf.set_text(initial)
	tf.enter_action = ok
	tf.escape_action = cancel
	tf.top = lb.top
	tf.left = lb.right + 5
	box.add(lb)
	box.add(tf)
	tf.focus()
	box.shrink_wrap()
	if box.present():
		return tf.get_text()
	else:
		return None
