#!/usr/bin/python
# vim:fileencoding=utf-8
#
# Hacked by Phillip Berndt, www.pberndt.com
#
import gtk
import glib
import sys
import serial
import time

# Functions for PDU en-/decoding
def pduDecode(text):
	pad = lambda number: number + "0" * (8 - len(number))
	data = "".join([ pad("".join(reversed(bin(ord(c))[2:]))) for c in text.decode("hex") ])
	output = ""
	while len(data) >= 7:
		output += chr(int("".join(reversed(data[:7])), 2))
		data = data[7:]
	return output

def pduDecodeSMS(text):
	text = text[2 + int(text[:2], 16) * 2:] # Skip SMSC
	text = text[2:] # Skip info
	#text = text[2:] # Skip reference
	text = text[2 + (int(text[:2], 16) + int(text[:2], 16) % 2) + 2:] # Skip telephone number
	text = text[2:] # Skip protocol
	text = text[2:] # Skip data coding scheme
	text = text[14:] # Skip date
	text = text[2:] # Skip content length
	return pduDecode(text)

def pduEncode(text):
	pad = lambda number: number + "0" * (7 - len(number))
	binrep = ""
	while text:
		binrep += pad("".join(reversed(bin(ord(text[0]) & ~(1<<7))[2:])))
		text = text[1:]
	output = ""
	while len(binrep) >= 8:
		output += chr(int("".join(reversed(binrep[:8])), 2)).encode("hex")
		binrep = binrep[8:]
	if binrep:
		binrep = "".join(reversed(binrep))
		output += chr(int(binrep, 2)).encode("hex")
	return output.upper()

# Function to show „action pending“ message
def doPendingDialog():
	dlg = gtk.MessageDialog(wnd)
	dlg.set_markup("Waiting for response from the dongle")
	progress = gtk.ProgressBar()
	dlg.vbox.add(progress)
	def do_pulse(*w):
		if progress.get_window() is None:
			return False
		progress.pulse()
		return True
	glib.timeout_add(100, do_pulse)
	dlg.set_modal(True)
	dlg.show_all()
	return dlg

# Function to show error dialog
def doError(message):
	dlg = gtk.MessageDialog()
	dlg.set_markup(message)
	dlg.set_title("Error")
	dlg.add_button("Ok", gtk.RESPONSE_OK)
	dlg.run()
	sys.exit(1)

# Function to ask for sth
def doAsk(question):
	dlg = gtk.MessageDialog()
	dlg.set_markup(question)
	dlg.set_title("Question")
	entry = gtk.Entry()
	dlg.vbox.add(entry)
	entry.show()
	dlg.add_button("Ok", gtk.RESPONSE_OK)
	dlg.add_button("Cancel", gtk.RESPONSE_CANCEL)
	if dlg.run() == gtk.RESPONSE_CANCEL:
		dlg.destroy()
		return False
	retVal = entry.get_text()
	dlg.destroy()
	return retVal

def sendNWait(send):
	usbTTY.write(send + "\r\n")
	usbTTY.flush()
	globals()["pending"] = doPendingDialog()

def ttyReceivedCallback(source, condition):
	if condition != glib.IO_IN:
		doError("Connection to dongle lost")
	textInput = usbTTY.readline().strip()
	print textInput

	if "cmglMode" in globals():
		if textInput[0:2] == "AT": return True
		if textInput == "": return True
		if textInput[:5] == "+CMGL": 
			msgNo = textInput[7:textInput.find(",")]
			appendText("Message no. " + msgNo + " follows:")
			return True
		if textInput == "OK":
			del globals()["cmglMode"]
			if "pending" in globals():
				globals()["pending"].destroy()
				del globals()["pending"]
			appendText("No further messages")
			appendText("")
		else:
			try:
				appendText(" * " + pduDecodeSMS(textInput))
			except:
				appendText("Did not understand " + textInput)
		return True
	
	if len(textInput) == 0 or textInput[0] != "+":
		# Not of interest for us
		return True

	if "pending" in globals():
		globals()["pending"].destroy()
		del globals()["pending"]

	if textInput[0:5] == "+CPIN":
		result = textInput[7:]
		if result != 'READY':
			pin = doAsk("The card asks for " + result)
			if pin == False:
				sys.exit(1)
			sendNWait('AT+CPIN="' + pin + '"')
			usbTTY.flush()
			def send_ask_pin(*w):
				usbTTY.write("AT+CPIN?\r\n")
				return False
			glib.timeout_add(300, send_ask_pin)
		else:
			appendText("Pin ok; dongle accessible")
			usbTTY.write("AT+CNUM\r\n")
			usbTTY.flush()
			usbTTY.write("AT+CMGF=0\r\n")
			usbTTY.flush()
			usbTTY.write("AT+CNMI=1,1,0,0,0\r\n")
		return True
	
	if textInput[0:5] == "+CUSD":
		textInput = textInput[textInput.find('"') + 1:textInput.rfind('"')]
		appendText(pduDecode(textInput))
		appendText("")
	
	if textInput[:5] == "+CNUM":
		textInput = textInput[10:]
		textInput = textInput[textInput.find('"') + 1:textInput.rfind('"')]
		appendText("Own phone number is " + textInput)
		appendText("")
	
	if textInput[:5] == "+CMTI":
		appendText("A new text message has arrived!")

	return True

# Open TTY
try:
	usbTTY = serial.Serial("/dev/ttyUSB3")
	glib.io_add_watch(usbTTY.fileno(), glib.IO_IN | glib.IO_ERR, ttyReceivedCallback)
except:
	doError("Failed to upen USB-tty. Make sure to have the dongle attached, <i>/dev/ttyUSB3</i> accessible " +
		"and the <i>usbserial</i> module loaded")


# Create main window
wnd = gtk.Window()
wnd.set_title("USB UMTS dongle recharge tool")
wnd.set_size_request(640, 240)
hbox = gtk.HBox()
vbox = gtk.VBox()
textView = gtk.TextView()
textView.set_editable(False)
textView.set_cursor_visible(False)
textView.set_size_request(440, 240)
textView.set_wrap_mode(gtk.WRAP_WORD)
buffer = textView.get_buffer()
def appendText(text):
	buffer.insert(buffer.get_end_iter(), text.strip() + "\n")
	textView.scroll_to_iter(buffer.get_end_iter(), 0.25)
checkButton = gtk.Button("Check balance")
def checkAction(widget):
	appendText("Sending request for balance check")
	sendNWait('AT+CUSD=1,"' + pduEncode("*101#") + '",15')
checkButton.connect("clicked", checkAction)
rechargeButton = gtk.Button("Recharge")
def rechargeAction(widget):
	code = doAsk("Enter recharge code:")
	if code == False: return
	appendText("Sending recharge-request")
	sendNWait('AT+CUSD=1,"' + pduEncode("*103*" + code + "#") + '",15')
rechargeButton.connect("clicked", rechargeAction)
readTextButton = gtk.Button("Read new text messages")
def readTextAction(widget):
	appendText("Receiving text messages")
	sendNWait("AT+CMGL")
	globals()["cmglMode"] = True
readTextButton.connect("clicked", readTextAction)
readAllTextButton = gtk.Button("Read all text messages")
def readAllTextAction(widget):
	appendText("Receiving text messages")
	sendNWait("AT+CMGL=4")
	globals()["cmglMode"] = True
readAllTextButton.connect("clicked", readAllTextAction)
deleteTextButton = gtk.Button("Delete a text message")
def deleteTextAction(widget):
	delw = doAsk("Enter message no. to be deleted")
	if delw == False: return True
	usbTTY.write("AT+CMGD=" + delw + "\r\n")
deleteTextButton.connect("clicked", deleteTextAction)
sendCusdButton = gtk.Button("Send CUSD")
def sendCusdAction(widget):
	cusd = doAsk("CUSD to send: (e.g. *101#)")
	if cusd == False:
		return
	sendNWait('AT+CUSD=1,"' + pduEncode(cusd) + '",15')
sendCusdButton.connect("clicked", sendCusdAction)
wnd.connect("hide", lambda *f: gtk.main_quit())
hbox.add(vbox)
hbox.pack_end(textView, padding=5)
vbox.pack_start(checkButton, False, padding=3)
vbox.pack_start(rechargeButton, False, padding=3)
vbox.pack_start(readTextButton, False, padding=3)
vbox.pack_start(readAllTextButton, False, padding=3)
vbox.pack_start(deleteTextButton, False, padding=3)
vbox.pack_start(sendCusdButton, False, padding=3)
wnd.add(hbox)
wnd.show_all()
usbTTY.write("ATZ\r\n")
sendNWait("AT+CPIN?")

# Run
gtk.main()