#!/usr/bin/python
# -*- coding: utf-8 -*-

# Copyright 2008 Ryan Cox ( ryan.a.cox@gmail.com ) All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

'''A wrapper library for Tumblr's public web API: http://www.tumblr.com/api'''

__author__ = 'ryan.a.cox@gmail.com'
__version__ = '0.1'


from urllib2 import Request, urlopen, URLError, HTTPError
from urllib import urlencode, quote
import base64
import re

try:
    import simplejson
except ImportError:
    from django.utils import simplejson

GENERATOR = 'python-tumblr'
PAGESIZE = 50  


class TumblrError(Exception):
	''' General Tumblr error ''' 
	def __init__(self, msg):
		self.msg = msg 

	def __str__(self):
		return self.msg 

class TumblrAuthError(TumblrError):
	''' Wraps a 403 result '''
	pass

class TumblrRequestError(TumblrError):
	''' Wraps a 400 result '''
	pass

class TumblrIterator:
	def __init__(self, name, start, max, type):
		self.name = name
		self.start = start
		self.max = max 
		self.type = type
		self.results = None
		self.index = 0
		
	def __iter__(self):
		return self
	
	def next(self):
		if not self.results or (self.index == len(self.results['posts'])): 
			self.start += self.index
			self.index = 0
			url = "http://%s.tumblr.com/api/read/json?start=%s&num=%s" % (self.name,self.start, PAGESIZE)
			if self.type:
				url += "&type=" + self.type
			response = urlopen(url)
			page = response.read()
			m = re.match("^.*?({.*}).*$", page,re.DOTALL | re.MULTILINE | re.UNICODE)
			self.results = simplejson.loads(m.group(1))

		if (self.index >= self.max) or len(self.results['posts']) == 0:
			raise StopIteration

		self.index += 1
		return self.results['posts'][self.index-1]  

class Api:
	def __init__(self, name, email=None, password=None ):
		self.name = name
		self.is_authenticated = False
		self.email = email 
		self.password = password

	def auth_check(self):
		if self.is_authenticated:
			return
		url = 'http://www.tumblr.com/api/write'
		values = {	
				'action': 'authenticate',
				'generator' : GENERATOR, 
				'email': self.email, 
				'password' : self.password }

		data = urlencode(values)
		req = Request(url, data)
		try: 
			response = urlopen(req)
			page = response.read()
			self.url = page 
			self.is_authenticated = True
			return
		except HTTPError, e:
			if 403 == e.code:
				raise TumblrAuthError(str(e))
			if 400 == e.code:
				raise TumblrRequestError(str(e))
		except Exception, e:
			raise TumblrError(str(e))


	def write_regular(self, title=None, body=None, **args): 
		if title:	
			args['title'] = title
		if body: 
			args['body'] = body 
		args = self._fixnames(args)
		if not 'title' in args and not 'body' in args:
			raise TumblrError("Must supply either body or title argument")

		self.auth_check()
		args['type'] = 'regular'
		return self._write(args)

	def write_photo(self, source=None, **args): 
		if source:
			args['source'] = source 

		args = self._fixnames(args)
		if 'source' in args and 'data' in args:
			raise TumblrError("Must  NOT supply both source and data arguments")

		if not 'source' in args and not 'data' in args:
			raise TumblrError("Must supply source or data argument")
		
		self.auth_check()
		args['type'] = 'photo'
		return self._write(args)

	def write_quote(self, quote=None, **args): 
		if quote:
			args['quote'] = quote
		args = self._fixnames(args)
		if not 'quote' in args: 
			raise TumblrError("Must supply quote arguments")
		
		self.auth_check()
		args['type'] = 'quote'
		return self._write(args)

	def write_link(self, url=None, **args): 
		if url:
			args['url'] = url
		args = self._fixnames(args)
		if not 'url' in args:
			raise TumblrError("Must supply url argument")

		self.auth_check()
		args['type'] = 'link'
		return self._write(args)

	def write_conversation(self, conversation=None, **args): 
		if conversation:
			args['conversation'] = conversation
		args = self._fixnames(args)
		if not 'conversation' in args:
			raise TumblrError("Must supply conversation argument")

		self.auth_check()
		args['type'] = 'conversation'
		return self._write(args)

	def write_video(self, embed=None, **args): 
		if embed:
			args['embed'] = embed
		args = self._fixnames(args)
		if 'embed' in args and 'data' in args:
			raise TumblrError("Must  NOT supply both embed and data arguments")

		if not 'embed' in args and not 'data' in args:
			raise TumblrError("Must supply embed or data argument")
		
		self.auth_check()
		args['type'] = 'video'
		return self._write(args)

	def _fixnames(self, args):
		for key in args: 
			if '_' in key:
				value = args[key]
				del args[key]
				args[key.replace('_', '-')] = value
		return args 

	def _write(self, params, headers=None): 
		self.auth_check()
		url = 'http://www.tumblr.com/api/write'
		params['email'] = self.email
		params['password'] = self.password
		params['generator'] = GENERATOR
		data = urlencode(params)
		if headers:
			req = Request(url, data, headers)
		else:
			req = Request(url, data)
		newid = None
		try: 
			urlopen(req)
			raise TumblrError("Error writing post")

		except HTTPError, e:
			if 201 == e.code:
				newid = e.read() 
				return self.read(id=newid)
			raise TumblrError(e.read()) 

	def read(self, id=None, start=0,max=2**31-1,type=None): 
		if id:
			url = "http://%s.tumblr.com/api/read/json?id=%s" % (self.name,id)
			response = urlopen(url)
			page = response.read()
			m = re.match("^.*?({.*}).*$", page,re.DOTALL | re.MULTILINE | re.UNICODE)
			results = simplejson.loads(m.group(1))
			if len(results['posts']) == 0:
				return None 
				
			return results['posts'][0]  
		else:	
			return TumblrIterator(self.name,start,max,type)

if __name__ == "__main__":
	pass