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

# Copyright(C) 2012 Florent Fourcot
#
# 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/>.


from weboob.tools.browser import BasePage
from weboob.capabilities.bill import Detail, Bill
from datetime import datetime, date, time


__all__ = ['HistoryPage', 'DetailsPage']


def convert_price(div):
    try:
        price = div.find('div[@class="horsForfait"]/p/span').text
        price = price.encode('utf-8', 'replace').replace('€', '').replace(',', '.')
        return float(price)
    except:
        return 0.


class DetailsPage(BasePage):

    def on_loaded(self):
        self.details = []
        self.datebills = []
        num = self.document.xpath('//div[@class="infosLigneDetail pointer"]')[0].text
        num = num.split("-")[2].strip()

        # National parsing
        divnat = self.document.xpath('//div[@class="national"]')[0]
        self.parse_div(divnat, "National : %s | International : %s", False)

        # International parsing
        divint = self.document.xpath('//div[@class="international hide"]')[0]
        self.parse_div(divint, u"Appels émis : %s | Appels reçus : %s", True)

        for trbill in self.document.xpath('//tr[@class="derniereFacture"]'):
            mydate = trbill.find('td/input').attrib['onclick'].split("'")[1]
            bill = Bill()
            bill.label = mydate
            bill.id = num + "." + mydate
            bill.date = date(int(mydate[0:4]), int(mydate[4:6]), int(mydate[6:8]))
            bill.format = 'html'
            self.datebills.append(bill)

    def parse_div(self, divglobal, string, inter=False):
        divs = divglobal.xpath('div[@class="detail"]')
        # Two informations in one div...
        div = divs.pop(0)
        voice = self.parse_voice(div, string, inter)
        self.details.append(voice)
        self.iter_divs(divs, inter)

    def iter_divs(self, divs, inter=False):
        for div in divs:
            detail = Detail()

            detail.label = div.find('div[@class="titreDetail"]/p').text_content()
            if inter:
                detail.label = detail.label + " (international)"
            detail.infos = div.find('div[@class="consoDetail"]/p').text_content().lstrip()
            detail.price = convert_price(div)

            self.details.append(detail)

    def parse_voice(self, div, string, inter=False):
        voice = Detail()
        voice.label = div.find('div[@class="titreDetail"]/p').text_content()
        if inter:
            voice.label = voice.label + " (international)"
        voice.price = convert_price(div)
        voice1 = div.xpath('div[@class="consoDetail"]/p/span')[0].text
        voice2 = div.xpath('div[@class="consoDetail"]/p/span')[1].text
        voice.infos = string % (voice1, voice2)

        return voice

    def get_details(self):
        return self.details

    def date_bills(self):
        return self.datebills


def _get_date(detail):
    return detail.datetime


class HistoryPage(BasePage):

    def on_loaded(self):
        self.calls = []
        for tr in self.document.xpath('//tr'):
            tds = tr.xpath('td')
            if tds[0].text == None or tds[0].text == "Date":
                pass
            else:
                detail = Detail()
                mydate = date(*reversed([int(x) for x in tds[0].text.split(' ')[0].split("/")]))
                mytime = time(*[int(x) for x in tds[0].text.split(' ')[1].split(":")])
                detail.datetime = datetime.combine(mydate, mytime)
                detail.label = tds[1].text.lstrip().rstrip() + " " + tds[2].text.lstrip().rstrip() + " " + tds[3].text.lstrip().rstrip()
                try:
                    detail.price = float(tds[4].text[0:4].replace(',', '.'))
                except:
                    detail.price = 0.

                self.calls.append(detail)

    def get_calls(self):
        return sorted(self.calls, key=_get_date, reverse=True)