import simplejson as json
import urllib2
import urllib
from datetime import datetime
import unicodedata 

class NotLoggedIn(Exception):
    pass

class LoginError(Exception):
    pass

class Comment(dict):
    """ Abstraction for comment data provided by JSON 
        Comments can be identifed by Kind = 1 """


    def __init__(self, data):

        dict.__init__(self)
        self['kind'] = 1
        self['id'] = data['id']
        self['post'] = data['link_id'][3:]
        self['body'] = data['body']
        self['ups'] = data['likes']
        self['downs'] = data['downs']
        self['subreddit_id'] = data['subreddit_id']
        self['subreddit'] = data['subreddit']
        self['author'] = data['author']
        self['permalink'] = u'http://reddit.com/comments/%s/c/%s' % (self['post'], self['id'])

    def __getattr__(self, name):
        return dict.__getitem__(self, name)

    def __unicode__(self):
        return u'/r/%s - %s' % (self['subreddit'], ['self.author'])

    def __str__(self):
        return self.__unicode__()


class Message(dict):
    """ Abstract for a Reddit Message """

    def __init__(self, msg=None):
        if msg:
            for k in msg.keys():
                self[k] = msg[k]

    def __getattr__(self, name):
        return dict.__getitem__(self, name)

    def __unicode__(self):
        return u"%s: %s" % (self.author, self.subject)

    def __str__(self):
        return self.__unicode__()

class Inbox():
    """
    Reddit Inbox class, accesses a user's inbox and provides a iterable
    list of messages

    >>> inbox = Inbox(username='testuser', password='testpassword')
    >>> len(inbox)
    5

    """

    REDDIT = "http://www.reddit.com"
    REDDIT_API_LOGIN = "%s/api/login" % REDDIT
    REDDIT_API_INBOX = "%s/message/inbox/.json?mark=false" % REDDIT
    REDDIT_API_COMPOSE = "%s/api/compose/" % REDDIT

    def __init__(self, username=None, password=None):
        if username and password:
            self.login(username, password)

    def login(self, username, password):
        data = { 'user': username,
                 'passwd': password, 
                 'api_type': 'json' }
        url = "%s/%s" % (self.REDDIT_API_LOGIN, username)

        jsondoc = json.load(self._url_request(url, data))

        if jsondoc and 'json' in jsondoc:
            if 'data' in jsondoc['json']:
                self.login_cookie = jsondoc['json']['data']['cookie']
                self.modhash = jsondoc['json']['data']['modhash']
                if self.login_cookie:
                    return True
            elif 'errors' in jsondoc['json']:
                raise LoginError(jsondoc['json']['errors'])

        return False            

    @property
    def _inbox_data(self):

        if not self.login_cookie:
            raise NotLoggedIn

        if not hasattr(self, '__inbox_cache'):
            inbox = json.load(self._opener.open(self.REDDIT_API_INBOX))['data']
            
            self.__inbox_cache = []
            for msg in inbox['children']:
                self.__inbox_cache.append(Message(msg['data']))
             
        return self.__inbox_cache

    def _url_request(self, url, data):
        if not hasattr(self, '_opener'):
            self._opener = urllib2.build_opener(urllib2.HTTPCookieProcessor())
            urllib2.install_opener(self._opener)

        req = urllib2.Request(url, urllib.urlencode(data))
        return self._opener.open(req)

    def __len__(self):
        return len(self._inbox_data)

    def __getitem__(self, name):
        return self._inbox_data[name]

    def __iter__(self):
        return self._inbox_data.__iter__()

    def send(self, to, subject, text):
        if not hasattr(self, 'login_cookie') or not self.login_cookie:
            raise NotLoggedIn

        data = { 'to': to,
                 'subject': subject.encode('utf-8'),
                 'text': text.encode('utf-8'), 
                 'uh': self.modhash,
                 'thing_id': '' }
        url = "%s" % (self.REDDIT_API_COMPOSE)

        jsondoc = json.load(self._url_request(url, data))