pyledriver/listeners.py

import logging, os, sys, stat
from threading import Thread
from evdev import InputDevice, ecodes
from select import select
from auxilary import freeBusyPath
import stateMachine

logger = logging.getLogger(__name__)

class KeypadListener(Thread):
	def __init__(self, stateMachine, callbackDisarm, callbackArm, soundLib, passwd):
		
		ctrlKeys = { 69: 'NUML', 98: '/', 14: 'BS', 96: 'ENTER'}
		
		volKeys = { 55: '*', 74: '-', 78: '+'}
		
		numKeys = {
			71: '7', 72: '8', 73: '9',
			75: '4', 76: '5', 77: '6',
			79: '1', 80: '2', 81: '3',
			82: '0', 83: '.'
		}
		
		devPath = '/dev/input/by-id/usb-04d9_1203-event-kbd'

		freeBusyPath(devPath, logger)
		
		self._dev = InputDevice(devPath)
		self._dev.grab()
		
		numKeySound = soundLib.soundEffects['numKey']
		ctrlKeySound = soundLib.soundEffects['ctrlKey']
		wrongPassSound = soundLib.soundEffects['wrongPass']
		backspaceSound = soundLib.soundEffects['backspace']
		
		self.resetBuffer()
		
		def getInput():
			while 1:
				r, w, x = select([self._dev], [], [])
				for event in self._dev.read():
					if event.type == 1 and event.value == 1:
						
						# numeral input
						if event.code in numKeys:
							if stateMachine.currentState != stateMachine.states.disarmed:
								self._buf = self._buf + numKeys[event.code]
							numKeySound.play()

						# ctrl input
						elif event.code in ctrlKeys:
							val = ctrlKeys[event.code]
							
							# disarm if correct passwd
							if val=='ENTER':
								if stateMachine.currentState == stateMachine.states.disarmed:
									ctrlKeySound.play()
								else:
									if self._buf == '':
										ctrlKeySound.play()
									elif self._buf == passwd:
										callbackDisarm()
									else:
										self.resetBuffer()
										wrongPassSound.play()
							
							# arm
							elif val == 'NUML':
								callbackArm()
								ctrlKeySound.play()
								
							# delete last char in buffer
							elif val == 'BS':
								self._buf = self._buf[:-1]
								backspaceSound.play()
							
							# reset buffer
							elif val == '/':
								self.resetBuffer()
								backspaceSound.play()
						
						# volume input
						elif event.code in volKeys:
							val = volKeys[event.code]
								
							if val == '+':
								soundLib.changeVolume(10)
								
							elif val == '-':
								soundLib.changeVolume(-10)
								
							elif val == '*':
								soundLib.mute()

							ctrlKeySound.play()
							self._dev.set_led(ecodes.LED_NUML, 0 if soundLib.volume > 0 else 1)

		super().__init__(target=getInput, daemon=True)
		self.start()
		logger.debug('Started keypad device')
		
	# TODO: make timer to clear buffer if user doesn't clear it

	def resetBuffer(self):
		self._buf = ''
		
	def __del__(self):
		try:
			self._dev.ungrab()
			logger.debug('Released keypad device')
		except IOError:
			logger.debug('Failed to release keypad device')
		except AttributeError:
			pass
			
# TODO: this code gets really confused if the pipe is deleted
class PipeListener(Thread):
	def __init__(self, callback, path):
		self._path = path
		
		if os.path.exists(self._path):
			if not stat.S_ISFIFO(os.stat(self._path)[0]):
				os.remove(self._path)
				os.mkfifo(self._path)
		else:
			os.mkfifo(self._path)
		
		os.chmod(self._path, 0o0777)
		
		def listenForSecret():
			while 1:
				with open(self._path, 'r') as f:
					msg = f.readline()[:-1]
					callback(msg, logger)
		
		super().__init__(target=listenForSecret, daemon=True)
		self.start()
		logger.debug('Started pipe listener at path %s', self._path)
		
	def __del__(self):
		if os.path.exists(self._path):
			os.remove(self._path)
		logger.debug('Cleaned up pipe listener at path %s', self._path)