#!/usr/bin/env python2
# -*- coding: UTF-8 -*-
# File: msg.py
# Date: Sat Nov 22 23:23:52 2014 +0800
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>

from datetime import datetime
from bs4 import BeautifulSoup
from .utils import ensure_bin_str, ensure_unicode

TYPE_MSG = 1
TYPE_IMG = 3
TYPE_SPEAK = 34
TYPE_VIDEO = 43
TYPE_EMOJI = 47
TYPE_LOCATION = 48
TYPE_LINK = 49  # link share or file from web
TYPE_VOIP = 50
TYPE_SYSTEM = 10000

class WeChatMsg(object):
    """ fields in concern"""
    FIELDS = ["msgSvrId","type","isSend","createTime","talker","content","imgPath"]
    FILTER_TYPES = [TYPE_SYSTEM]

    @staticmethod
    def filter_types(tp):
        if tp in WeChatMsg.FILTER_TYPES or tp > 10000 or tp < 0:
            return True
        return False

    def __init__(self, row):
        """ row: a tuple corresponding to FIELDS"""
        assert len(row) == len(WeChatMsg.FIELDS)
        for f, v in zip(WeChatMsg.FIELDS, row):
            setattr(self, f, v)
        self.createTime = datetime.fromtimestamp(self.createTime / 1000)
        self.talker_name = None
        if self.content:
            self.content = ensure_unicode(self.content)

    def msg_str(self):
        # TODO: fix more types
        if self.type == TYPE_LOCATION:
            soup = BeautifulSoup(self.content)
            loc = soup.find('location')
            label = loc['label']
            ret = label
            try:
                poiname = loc['poiname']
                if poiname:
                    ret = poiname
            except:
                pass
            return ret + " ({},{})".format(loc['x'], loc['y'])
        elif self.type == TYPE_VOIP:
            return "REQUEST VIDEO CHAT"
        elif self.type == TYPE_LINK:
            soup = BeautifulSoup(self.content)
            url = soup.find('url').text
            if not url:
                title = soup.find('title').text
                assert title, "No title or url found in TYPE_LINK"
                return u"FILE:{}".format(title)
            return u"URL:{}".format(url)
        elif self.type == TYPE_VIDEO:
            return "VIDEO"
        else:
            return self.content

    def __repr__(self):
        ret = u"{}|{}:{}:{}".format(
            self.type,
            (self.talker if not self.talker_name else self.talker_name) \
                if not self.isSend else 'me',
            self.createTime,
            ensure_unicode(self.msg_str())).encode('utf-8')
        if self.imgPath:
            ret = u"{}|img:{}".format(ensure_unicode(ret.strip()), self.imgPath)
            return ret.encode('utf-8')
        else:
            return ret

    def __lt__(self, r):
        return self.createTime < r.createTime