# -*- coding: utf-8 -*-

# Copyright(C) 2008-2011  Romain Bignon, Christophe Benz
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.


import math
import re
import datetime
import random
import urllib

from weboob.tools.browser import BaseBrowser, BrowserIncorrectPassword, BrowserUnavailable
from weboob.tools.json import json

from weboob.capabilities.base import UserError
from weboob.capabilities.chat import ChatException, ChatMessage
from weboob.capabilities.messages import CantSendMessage


__all__ = ['AuMBrowser']


class AuMException(UserError):
    ERRORS = {"0.0.0":  "Bad signature",
              "0.0.1":  "Malformed request",
              "0.0.2":  "Not logged",
              "1.1.1":  "No member has this login",
              "1.1.2":  "Password don't match",
              "1.1.3":  "User has been banned",
              "1.12.1": "Invalid country",
              "1.12.1": "Invalid region",
              "4.0.1":  "Member not found",
              "4.1.1":  "Thread doesn't exist",
              "4.1.2":  "Cannot write to this member",
              "5.1.1":  "Member tergeted doesn't exist",
              "5.1.2":  "Sex member targeted is not the opposite of the member logged",
              "5.1.3":  "Not possible to send a charm",
              "5.1.4":  "Not possible to send a charm because the 5 charms has been already used",
              "5.1.5":  "Not possible because the guy has already send a charm to this girl",
              "5.1.6":  "No more money",
              "5.1.7":  "Not possible to add to basket",
              "5.2.1":  "Member doesn't exist",
              "5.3.1":  "Member doesn't exist",
             }

    def __init__(self, code):
        Exception.__init__(self, self.ERRORS.get(code, code))
        self.code = code

class AuMBrowser(BaseBrowser):
    DOMAIN = 'api.adopteunmec.com'
    APIKEY = 'fb0123456789abcd'

    consts = None
    search_query = None
    my_sex = 0
    my_id = 0
    my_name = u''
    my_coords = (0,0)

    def id2url(self, id):
        return 'http://www.adopteunmec.com/index.php/profile/%s' % id

    def url2id(func):
        def inner(self, id, *args, **kwargs):
            m = re.match('^http://.*adopteunmec.com.*/(\d+)$', str(id))
            if m:
                id = int(m.group(1))
            else:
                m = re.match('^http://.*adopteunmec.com/index.php/profile/(\d+).*', str(id))
                if m:
                    id = int(m.group(1))
            return func(self, id, *args, **kwargs)
        return inner

    def api_request(self, command, action, parameter='', data=None, nologin=False):
        if data is None:
            # Always do POST requests.
            data = ''
        elif isinstance(data, (list,tuple,dict)):
            data = urllib.urlencode(data)
        elif isinstance(data, unicode):
            data = data.encode('utf-8')

        url = self.buildurl(self.absurl('/api.php'), S=self.APIKEY,
                                                     C=command,
                                                     A=action,
                                                     P=parameter,
                                                     O='json')
        buf = self.openurl(url, data).read()

        try:
            r = json.loads(buf[buf.find('{'):])
        except ValueError:
            raise ValueError(buf)

        #pprint(r)

        if 'errors' in r and r['errors'] != '0' and len(r['errors']) > 0:
            code = r['errors'][0]
            if code in (u'0.0.2', u'1.1.1', u'1.1.2'):
                if not nologin:
                    self.login()
                    return self.api_request(command, action, parameter, data, nologin=True)
                else:
                    raise BrowserIncorrectPassword(AuMException.ERRORS[code])
            else:
                raise AuMException(code)
        return r

    def login(self):
        r = self.api_request('me', 'login', data={'login': self.username,
                                                  'pass':  self.password,
                                                 }, nologin=True)
        self.my_sex = r['result']['me']['sex']
        self.my_id = int(r['result']['me']['id'])
        self.my_name = r['result']['me']['pseudo']
        self.my_coords = (float(r['result']['me']['lat']), float(r['result']['me']['lng']))
        return r

    #def register(self, password, sex, birthday_d, birthday_m, birthday_y, zipcode, country, godfather=None):
    #    if not self.is_on_page(RegisterPage):
    #        self.location('http://www.adopteunmec.com/register2.php')
    #    self.page.register(password, sex, birthday_d, birthday_m, birthday_y, zipcode, country)
    #    if godfather:
    #        if not self.is_on_page(AccountPage):
    #            self.location('http://www.adopteunmec.com/account.php')
    #        self.page.set_godfather(godfather)

    #@pageaccess
    #def add_photo(self, name, f):
    #    if not self.is_on_page(EditPhotoPage):
    #        self.location('/edit.php?type=1')
    #    return self.page.add_photo(name, f)

    #@pageaccess
    #def set_nickname(self, nickname):
    #    if not self.is_on_page(EditAnnouncePage):
    #        self.location('/edit.php?type=2')
    #    return self.page.set_nickname(nickname)

    #@pageaccess
    #def set_announce(self, title=None, description=None, lookingfor=None):
    #    if not self.is_on_page(EditAnnouncePage):
    #        self.location('/edit.php?type=2')
    #    return self.page.set_announce(title, description, lookingfor)

    #@pageaccess
    #def set_description(self, **args):
    #    if not self.is_on_page(EditDescriptionPage):
    #        self.location('/edit.php?type=3')
    #    return self.page.set_description(**args)

    def check_login(func):
        def inner(self, *args, **kwargs):
            if self.my_id == 0:
                self.login()
            return func(self, *args, **kwargs)
        return inner

    def get_consts(self):
        if self.consts is not None:
            return self.consts

        self.consts = []
        for i in xrange(2):
            r = self.api_request('me', 'all_values', data={'sex': i})
            self.consts.append(r['result']['values'])

        return self.consts

    @check_login
    def score(self):
        r = self.api_request('member', 'view', data={'id': self.my_id})
        return int(r['result']['member']['popu']['popu'])

    @check_login
    def get_my_name(self):
        return self.my_name

    @check_login
    def get_my_id(self):
        return self.my_id

    @check_login
    def nb_new_mails(self):
        r = self.api_request('me', '[default]')
        return r['result']['news']['newMails']

    @check_login
    def nb_new_baskets(self):
        r = self.api_request('me', '[default]')
        return r['result']['news']['newBaskets']

    @check_login
    def nb_new_visites(self):
        r = self.api_request('me', '[default]')
        return r['result']['news']['newVisits']

    @check_login
    def nb_available_charms(self):
        r = self.login()
        return r['result']['flashs']

    @check_login
    def nb_godchilds(self):
        r = self.api_request('member', 'view', data={'id': self.my_id})
        return int(r['result']['member']['popu']['invits'])

    @check_login
    def get_baskets(self):
        r = self.api_request('me', 'basket')
        return r['result']['basket']

    @check_login
    def get_flashs(self):
        r = self.api_request('me', 'flashs')
        return r['result']['all']

    @check_login
    def get_visits(self):
        r = self.api_request('me', 'visits')
        return r['result']['news'] + r['result']['olds']

    @check_login
    def get_threads_list(self, count=30):
        r = self.api_request('message', '[default]', '%d,0' % count)
        return r['result']['threads']

    @check_login
    @url2id
    def get_thread_mails(self, id, count=30):
        r = self.api_request('message', 'thread', data={'memberId': id, 'count': count})
        return r['result']['thread']

    @check_login
    @url2id
    def post_mail(self, id, content):
        # It seems it is not needed anymore.
        #new_content = u''
        #for c in content:
        #    try:
        #        new_content += '&%s;' % codepoint2name[ord(c)]
        #    except KeyError:
        #        new_content += c

        content = content.replace('\n', '\r\n').encode('utf-8', 'replace')

        try:
            self.api_request('message', 'new', data={'memberId': id, 'message': content})
        except AuMException, e:
            raise CantSendMessage(unicode(e))

    @check_login
    @url2id
    def delete_thread(self, id):
        r = self.api_request('message', 'delete', data={'id_user': id})
        self.logger.debug('Thread deleted: %r' % r)

    @check_login
    @url2id
    def send_charm(self, id):
        try:
            self.api_request('member', 'addBasket', data={'id': id})
        except AuMException:
            return False
        else:
            return True

    @check_login
    @url2id
    def add_basket(self, id):
        try:
            self.api_request('member', 'addBasket', data={'id': id})
        except AuMException:
            return False
        else:
            return True

    @url2id
    def deblock(self, id):
        self.readurl('http://www.adopteunmec.com/fajax_postMessage.php?action=deblock&to=%s' % id)
        return True

    @url2id
    def report_fake(self, id):
        return self.readurl('http://www.adopteunmec.com/fake.php', 'id=%s' % id)

    @url2id
    def rate(self, id, what, rating):
        result = self.openurl('http://www.adopteunmec.com/fajax_vote.php', 'member=%s&what=%s&rating=%s' % (id, what, rating)).read()
        return float(result)

    def search_profiles(self, **kwargs):
        if self.search_query is None:
            r = self.api_request('searchs', '[default]')
            self.search_query = r['result']['search']['query']

        params = {}
        for key, value in json.loads(self.search_query).iteritems():
            if isinstance(value, dict):
                for k, v in value.iteritems():
                    params['%s%s' % (key, k.capitalize())] = v
            else:
                params[key] = value or ''
        r = self.api_request('searchs', 'advanced', '30,0', params)
        ids = [s['id'] for s in r['result']['search']]
        return set(ids)

    @url2id
    def get_profile(self, id, with_pics=True):
        r = self.api_request('member', 'view', data={'id': id})
        if not 'result' in r:
            print r
        profile = r['result']['member']


        # Calculate distance in km.
        profile['dist'] = 0.0
        if 'lat' in profile and 'lng' in profile:
            coords = (float(profile['lat']), float(profile['lng']))

            R = 6371
            lat1 = math.radians(self.my_coords[0])
            lat2 = math.radians(coords[0])
            lon1 = math.radians(self.my_coords[1])
            lon2 = math.radians(coords[1])
            dLat = lat2 - lat1
            dLong = lon2 - lon1
            a= pow(math.sin(dLat/2), 2) + math.cos(lat1) * math.cos(lat2) * pow(math.sin(dLong/2), 2)
            c= 2 * math.atan2(math.sqrt(a), math.sqrt(1-a))
            profile['dist'] = R * c

        if with_pics:
            r = self.api_request('member', 'pictures', data={'id': id})
            profile['pictures'] = []
            for pic in r['result']['pictures']:
                d = {'hidden': False}
                d.update(pic)
                profile['pictures'].append(d)

        return profile

    def _get_chat_infos(self):
        try:
            data = json.load(self.openurl('http://www.adopteunmec.com/1.1_cht_get.php?anticache=%f' % random.random()))
        except ValueError:
            raise BrowserUnavailable()

        if data['error']:
            raise ChatException(u'Error while getting chat infos. json:\n%s' % data)
        return data

    def iter_contacts(self):
        def iter_dedupe(contacts):
            yielded_ids = set()
            for contact in contacts:
                if contact['id'] not in yielded_ids:
                    yield contact
                yielded_ids.add(contact['id'])

        data = self._get_chat_infos()
        return iter_dedupe(data['contacts'])

    def iter_chat_messages(self, _id=None):
        data = self._get_chat_infos()
        if data['messages'] is not None:
            for message in data['messages']:
                yield ChatMessage(id_from=message['id_from'], id_to=message['id_to'], message=message['message'], date=message['date'])

    def send_chat_message(self, _id, message):
        url = 'http://www.adopteunmec.com/1.1_cht_send.php?anticache=%f' % random.random()
        data = dict(id=_id, message=message)
        headers = {
                'Content-type': 'application/x-www-form-urlencoded',
                'Accept': 'text/plain',
                'Referer': 'http://www.adopteunmec.com/chat.php',
                'Origin': 'http://www.adopteunmec.com',
                }
        request = self.request_class(url, urllib.urlencode(data), headers)
        response = self.openurl(request).read()
        try:
            datetime.datetime.strptime(response,  '%Y-%m-%d %H:%M:%S')
            return True
        except ValueError:
            return False