#!/usr/bin/python

"""
    Django-Twitter (Tango) utility functions. Huzzah.
"""

import urllib, urllib2

from urllib2 import HTTPError

try:
	import simplejson
except:
	print "Tango requires the simplejson library to work. http://www.undefined.org/python/"

try:
	import oauth
except:
	print "Tango requires oauth for authentication purposes. http://oauth.googlecode.com/svn/code/python/oauth/oauth.py"

class setup:
	def __init__(self, authtype = "OAuth", username = None, password = None, oauth_keys = None):
		self.authtype = authtype
		self.authenticated = False
		self.username = username
		self.password = password
		self.oauth_keys = oauth_keys
		if self.username is not None and self.password is not None:
			if self.authtype == "OAuth":
				pass
			elif self.authtype == "Basic":
				self.auth_manager = urllib2.HTTPPasswordMgrWithDefaultRealm()
				self.auth_manager.add_password(None, "http://twitter.com", self.username, self.password)
				self.handler = urllib2.HTTPBasicAuthHandler(self.auth_manager)
				self.opener = urllib2.build_opener(self.handler)
				self.authenticated = True
		else:
			pass
	
	def shortenURL(self, url_to_shorten):
		# Perhaps we should have fallbacks here in case the is.gd API limit gets hit? Maybe allow them to set the host?
		shortURL = urllib2.urlopen("http://is.gd/api.php?" + urllib.urlencode({"longurl": url_to_shorten})).read()
		return shortURL
	
	def constructApiURL(self, base_url, params):
		queryURL = base_url
		questionMarkUsed = False
		for param in params:
			if params[param] is not None:
				queryURL += (("&" if questionMarkUsed is True else "?") + param + "=" + params[param])
				questionMarkUsed = True
		return queryURL
	
	def getPublicTimeline(self):
		publicTimeline = simplejson.load(urllib2.urlopen("http://twitter.com/statuses/public_timeline.json"))
		formattedTimeline = []
		for tweet in publicTimeline:
			formattedTimeline.append(tweet['text'])
		return formattedTimeline
	
	def getUserTimeline(self, **kwargs): 
		# 99% API compliant, I think - need to figure out Gzip compression and auto-getting based on authentication
		# By doing this with kwargs and constructing a url outside, we can stay somewhat agnostic of API changes - it's all
		# based on what the user decides to pass. We just handle the heavy lifting! :D
		userTimelineURL = self.constructApiURL("http://twitter.com/statuses/user_timeline/" + self.username + ".json", kwargs)
		userTimeline = simplejson.load(urllib2.urlopen(userTimelineURL))
		formattedTimeline = []
		for tweet in userTimeline:
			formattedTimeline.append(tweet['text'])
		return formattedTimeline
	
	def getUserMentions(self, **kwargs):
		if self.authenticated is True:
			if self.authtype == "Basic":
				pass
			else:
				pass
		else:
			print "getUserMentions() requires you to be authenticated."
			pass
	
	def updateStatus(self, status, in_reply_to_status_id = None):
		if self.authenticated is True:
			if self.authtype == "Basic":
				try:
					self.opener.open("http://twitter.com/statuses/update.json?", urllib.urlencode({"status": status}, {"in_reply_to_status_id": in_reply_to_status_id}))
				except HTTPError, e:
					print e.code
					print e.headers
			else:
				print "Sorry, OAuth support is still forthcoming. Feel free to help out on this front!"
				pass
		else:
			print "updateStatus() requires you to be authenticated."
			pass
	
	def destroyStatus(self, id):
		if self.authenticated is True:
			self.http.request("http://twitter.com/status/destroy/" + id + ".json", "POST")
		else:
			print "destroyStatus() requires you to be authenticated."
			pass
	
	def getSearchTimeline(self, search_query, optional_page):
		params = urllib.urlencode({'q': search_query, 'rpp': optional_page}) # Doesn't hurt to do pages this way. *shrug*
		searchTimeline = simplejson.load(urllib2.urlopen("http://search.twitter.com/search.json", params))
		formattedTimeline = []
		for tweet in searchTimeline['results']:
			formattedTimeline.append(tweet['text'])
		return formattedTimeline
	
	def getCurrentTrends(self):
		# Returns an array of dictionary items containing the current trends
		trendingTopicsURL = "http://search.twitter.com/trends.json"
		trendingTopics = simplejson.load(urllib.urlopen(trendingTopicsURL))
		trendingTopicsArray = []
		for topic in trendingTopics['trends']:
			trendingTopicsArray.append({"name" : topic['name'], "url" : topic['url']})
		return trendingTopicsArray