ToorChat/toorchat.py

#!/usr/bin/env python

import sys
from rflib import *
import rflib.chipcon_nic as rfnic
import atexit
from libtoorchat import *
from os import system
import curses
import webbrowser
import time
from threading import Thread

def find_message_in_website(message, visual):
	found = False
	for item in visual.website_buffer:
		if message.index == item.index:
			found = True
	if not found:
		visual.website_buffer.append(message)

def thread_run(visual):
	''' This is our function for the thread '''
	#Thread should run until exit
	while not visual.exit:
		try:
			msg, timestamp = visual.badge.RFrecv()
			toor_message = ToorChatProtocol.parse_message(msg)
			if toor_message != None:
				if toor_message.type == ToorChatProtocol.get_chat_type():
					visual.message_queue.append(toor_message)
				if toor_message.type == ToorChatProtocol.get_web_request_type():
					#If we are registered as a server, lets type to make that request
					if visual.server:
						visual.request_xid = toor_message.xid
						ToorChatProtocol.get_web_messages(toor_message.data, visual)
				if toor_message.type == ToorChatProtocol.get_web_response_type():
					# lets see if its the response were looking for
					if toor_message.xid == visual.request_xid:
						find_message_in_website(toor_message, visual)
					if len(visual.website_buffer) == int(toor_message.last)+1:
						#sort messages
						newlist = sorted(visual.website_buffer, key=lambda x: int(x.index), reverse=False)
						try:
							os.remove('temp.html')
						except Exception:
							#There is a chance this file might now exist
							pass
						temp_file = open('temp.html', 'w')
						total = ""
						for item in newlist:
							total += item.data
						temp_file.write(total)
						temp_file.close()
						visual.website_buffer = []
						#Render website
						webbrowser.get('firefox').open_new('temp.html')

		except ChipconUsbTimeoutException:
			pass

class Visualizer():

	def __init__(self):
		self.screen = curses.initscr()
		self.screen.nodelay(1)
		self.screen.scrollok(1)
		self.badge = RfCat(idx=0)
		self.badge.setModeRX()
		self.protocol = ToorChatProtocol(self.badge)
		self.message_queue = []
		self.website_buffer = []
		self.user = None
		self.channel = None
		self.frequency = None
		self.request_xid = None
		#This when set to True will kill the thread
		self.exit = False
		self.server = False

	def set_server(self):
		self.server = True

	def start(self):
		self.__start_recv_thread__()
		self.__run__()

	def __start_recv_thread__(self):
		'''This spins off a thread to deal with recving information from the rf device '''
		self.thread = Thread(target=thread_run, args=(self,))
		self.thread.start()

	def __run__(self):
		try:
			self.screen_max_y, self.screen_max_x = self.screen.getmaxyx()
			self.__draw_frame__()
			self.screen.refresh()
			self.last_message_index = 0
			self.screen.setscrreg(3, self.screen_max_y - 2)
			while True:
				self.screen_max_y, self.screen_max_x = self.screen.getmaxyx()
				self.screen.addstr(0, 1, "[ENTER] Message [U] Username [C] Channel [F] Frequency [W] Load Webpage")
				self.__add_message_to_screen__()
				entry = self.screen.getch()
				if entry == curses.KEY_RESIZE:
					self.__draw_frame__(False)
				if entry == 10:
					self.screen.nodelay(0)
					user_input = self.screen.getstr(1, 1, 60)
					old_message = self.protocol.send_chat_message(user_input, self.user)
					self.message_queue.append(old_message)
					self.screen.nodelay(1)
					self.screen.addstr(1,1," "*(self.screen_max_x-3))
				if entry == ord('u'):
					self.screen.nodelay(0)
					user_input = self.screen.getstr(1, 1, 60)
					self.user = user_input[:USER_NAME_SIZE]
					self.screen.nodelay(1)
					self.screen.addstr(1,1," "*(self.screen_max_x-3))
				if entry == ord('c'):
					self.screen.nodelay(0)
					user_input = self.screen.getstr(1, 1, 60)
					self.channel = self.protocol.change_channel(user_input)
					self.screen.nodelay(1)
					self.screen.addstr(1,1," "*(self.screen_max_x-3))
				if entry == ord('f'):
					self.screen.nodelay(0)
					user_input = self.screen.getstr(1, 1, 60)
					self.frequency = self.protocol.change_frequency(user_input)
					self.screen.nodelay(1)
					self.screen.addstr(1,1," "*(self.screen_max_x-3))
				if entry == ord('w'):
					self.screen.nodelay(0)
					user_input = self.screen.getstr(1, 1, 60)
					self.request_xid = ToorMessage.get_random_xid()
					self.website_buffer = []
					request = self.protocol.send_web_request(user_input, self.request_xid)
					self.screen.nodelay(1)
					self.screen.addstr(1,1," "*(self.screen_max_x-3))

		except KeyboardInterrupt:
			self.exit = True
			self.stop()

	def __add_message_to_screen__(self):
		if len(self.message_queue) > 0:
			message = self.message_queue[len(self.message_queue)-1]
			message_string = str(message.user)+": "+ str(message.data)
			self.screen.addstr(self.last_message_index+3,1, message_string + " "*(self.screen_max_x-(2+len(message_string))))
			self.last_message_index +=1
			self.message_queue.pop()
			if self.last_message_index > self.screen_max_y-5:
				self.screen.scroll()
				self.screen.border(0)
				self.last_message_index -= 1
				
	def __draw_frame__(self, clear = True):
		if clear:
			self.screen.clear()
		self.screen.border(0)
		self.screen.addstr(2,1,"_"*(self.screen_max_x-2))

	def stop(self):
		curses.endwin()

if __name__ == '__main__':
	visual = Visualizer()
	if len(sys.argv) > 1:
		if sys.argv[1] == "-s":
			print "SET server"
			visual.set_server()
	visual.start()