summaryrefslogtreecommitdiff
path: root/test.py
blob: 19564bbda2df816f7ec89d1353f8c50a57394847 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
import sys
import httplib
import urllib
import time
from threading import Thread, Lock

class Timer(Thread):
	def __init__(self,func,interval=0.50):
		Thread.__init__(self)
		self.func = func
		self.interval = interval

		self.running = False

	def run(self):
		self.running = True
		while self.running:
			self.func()
			time.sleep(self.interval)

	def stop(self):
		self.running = False


class OmegleChat(object):
	headers = {
		"Host" : "omegle.com",
		"Content-type": "application/x-www-form-urlencoded; charset=utf-8",
		"Accept": "application/json"
	}

	def __init__(self,poll_interval=0.5):
		self.conn = httplib.HTTPConnection('www.omegle.com')
#		self.conn.set_debuglevel(100)

		self.conn_lock = Lock()

		self.timer = Timer(self.events, poll_interval)

	def start(self):
		self.conn_lock.acquire()
		self.conn.request("POST", "/start", {}, OmegleChat.headers)
		self.conn_lock.release()
		r=self.conn.getresponse()
		body=r.read()
		id=body.split("\"")
		if id.__len__() == 3:
			self.id = id[1]
			print "Connected, id=%s" % self.id
		else:
			raise Exception("Bad response: %s" % body)
		#Poll events
		self.timer.start()

	def disconnect(self):
		self.timer.stop()

		self.conn_lock.acquire()
		self.conn.request("POST",
			"/disconnect",
			urllib.urlencode({'id' : self.id}),
			OmegleChat.headers)
		r = self.conn.getresponse()
		self.conn_lock.release()
		body = r.read()
		if body != "win":
			raise Exception("/disconnect; Bad response: %s" % body)

	def events(self):
		conn = httplib.HTTPConnection('www.omegle.com')
		conn.request("POST", 
			"/events", 
			urllib.urlencode({'id' : self.id}),
			OmegleChat.headers)
		r=conn.getresponse()
		body=r.read()
		self.on_event(conn,body)
		conn.close()

	def send(self,msg):
		self.conn_lock.acquire()
		self.conn.request("POST",
			"/send",
			urllib.urlencode({'id':self.id,'msg':msg}),
			OmegleChat.headers)
		r=self.conn.getresponse()
		self.conn_lock.release()
		body = r.read()
		if body != "win":
			raise Exception("/send; Bad response %s" % body)

	def typing(self,msg):
		self.conn_lock.acquire()
		self.conn.request("POST",
			"/typing",
			urllib.urlencode({'id':self.id}),
			OmegleChat.headers)
		r=self.conn.getresponse()
		self.conn_lock.release()
		body = r.read()
		if body != "win":
			raise Exception("/typing; Bad response %s" % body)


	def on_event(self,conn,body):
		print "Event: %s" % body

chat = OmegleChat()
chat.start()

while 1==1:
	try:
		cmd=sys.stdin.readline().strip()
	except KeyboardInterrupt:
		chat.disconnect()
		sys.exit(0)
	if cmd=="quit":
		chat.disconnect()
		break
	else:
		chat.send(cmd)