from rox import g
import gobject

refleak_bug_workaround = []

class Arrow(g.EventBox):
	def __init__(self, arrow_type, callback, data, time = 50):
		g.EventBox.__init__(self)
		self.add_events(g.gdk.BUTTON_PRESS_MASK)

		self.data = data
		self.callback = callback
		self.arrow_type = arrow_type
		self.time = time

		self.arrow = g.Arrow(arrow_type, g.SHADOW_OUT)

		self.add(self.arrow)
		self.button = 0
		
		self.connect('button-press-event', self.button_press)
		self.connect('button-release-event', self.button_release)

		self.connect('enter-notify-event',
			lambda w, e: self.set_state(g.STATE_PRELIGHT))
		self.connect('leave-notify-event',
			lambda w, e: self.set_state(g.STATE_NORMAL))

		self.connect('destroy', lambda w: refleak_bug_workaround.remove(self))
		refleak_bug_workaround.append(self)
	
	def button_press(self, box, event):
		if self.button != 0 or event.type != g.gdk.BUTTON_PRESS:
			return
		self.button = event.button
		#g.grab_add(self)
		self.arrow.set(self.arrow_type, g.SHADOW_IN)
		self.cb = gobject.timeout_add(self.time * 2 + 300, self.incr)
		self.incr(resched = 0)
	
	def button_release(self, box, event):
		if self.button != event.button:
			return
		self.button = 0
		#g.grab_remove(self)
		self.arrow.set(self.arrow_type, g.SHADOW_OUT)
		gobject.source_remove(self.cb)
	
	def incr(self, resched = 1):
		if resched:
			gobject.source_remove(self.cb)
			self.cb = gobject.timeout_add(self.time, self.incr)
		self.callback(self.data)
		return 0