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

# Copyright(C) 2009-2014  Florent Fourcot, Romain Bignon
#
# 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 decimal import Decimal
from datetime import date, timedelta
import re
import hashlib

from weboob.capabilities.bank import Account
from weboob.capabilities.base import NotAvailable
from weboob.tools.browser2.page import HTMLPage, LoggedPage, method, ListElement, ItemElement
from weboob.tools.browser2.filters import Attr, CleanText, CleanDecimal, Filter, Field, MultiFilter, Env, Date, Lower
from weboob.tools.capabilities.bank.transactions import FrenchTransaction


__all__ = ['AccountsList']


class Transaction(FrenchTransaction):
    PATTERNS = [(re.compile(u'^retrait dab (?P<dd>\d{2})/(?P<mm>\d{2})/(?P<yy>\d{4}) (?P<text>.*)'), FrenchTransaction.TYPE_WITHDRAWAL),
                (re.compile(u'^carte (?P<dd>\d{2})/(?P<mm>\d{2})/(?P<yy>\d{4}) (?P<text>.*)'), FrenchTransaction.TYPE_CARD),
                (re.compile(u'^virement (sepa )?(emis vers|recu|emis)? (?P<text>.*)'), FrenchTransaction.TYPE_TRANSFER),
                (re.compile(u'^cheque (?P<text>.*)'), FrenchTransaction.TYPE_CHECK),
                (re.compile(u'^prelevement (?P<text>.*)'), FrenchTransaction.TYPE_ORDER),
                (re.compile(u'^prélèvement sepa en faveur de (?P<text>.*)'), FrenchTransaction.TYPE_ORDER),
                ]


class AddPref(MultiFilter):
    prefixes = {u'Courant': u'CC-', u'Livret A': 'LA-', u'Orange': 'LEO-',
            u'Durable': u'LDD-', u"Titres": 'TITRE-', u'PEA': u'PEA-'}

    def filter(self, values):
        el, label = values
        for key, pref in self.prefixes.items():
            if key in label:
                return pref + el
        return el


class AddType(Filter):
    types = {u'Courant': Account.TYPE_CHECKING, u'Livret A': Account.TYPE_SAVINGS,
             u'Orange': Account.TYPE_SAVINGS, u'Durable': Account.TYPE_SAVINGS,
             u'Titres': Account.TYPE_MARKET, u'PEA': Account.TYPE_MARKET}

    def filter(self, label):
        for key, acc_type in self.types.items():
            if key in label:
                return acc_type
        return Account.TYPE_UNKNOWN


class Hashmd5(MultiFilter):
    def filter(self, values):
        concat = ''
        for value in values:
            concat += u'%s' % value
        return hashlib.md5(concat.encode('utf-8')).hexdigest()

class AccountsList(LoggedPage, HTMLPage):

    i = 0

    @method
    class get_list(ListElement):
        item_xpath = '//a[@class="mainclic"]'

        class item(ItemElement):
            klass = Account

            obj_currency = u'EUR'
            obj__id = CleanText('span[@class="account-number"]')
            obj_label = CleanText('span[@class="title"]')
            obj_id = AddPref(Field('_id'), Field('label'))
            obj_type = AddType(Field('label'))
            obj_balance = CleanDecimal('span[@class="solde"]/label')
            obj_coming = NotAvailable
            obj__jid = Attr('//input[@name="javax.faces.ViewState"]', 'value')


    @method
    class get_transactions(ListElement):
        item_xpath = '//table'
        i = 0


        class item(ItemElement):
            klass = Transaction

            monthvalue = {u'janv.': '01', u'févr.': '02', u'mars': '03', u'avr.': '04',
                          u'mai': '05', u'juin': '06', u'juil.': '07', u'août': '08',
                          u'sept.': '09', u'oct.': '10', u'nov.': '11', u'déc.': '12',
                         }
            catvalue = {u'virt': u"Virement", u'autre': u"Autre",
                        u'plvt': u'Prélèvement', u'cb_ret': u"Carte retrait",
                        u'cb_ach': u'Carte achat', u'chq': u'Chèque',
                        u'frais': u'Frais bancaire', u'sepaplvt': u'Prélèvement'}

            # we use lower for compatibility with the old website
            obj_raw = Transaction.Raw(Lower('.//td[@class="lbl"]'))
            obj_amount = CleanDecimal('.//td[starts-with(@class, "amount")]')
            obj__textdate = Env('_textdate')
            obj_date = Date(Field('_textdate'), dayfirst=True)
            obj_rdate = Field('date')
            obj_id = Hashmd5(Field('_textdate'), Field('raw'), Field('amount'))


            def condition(self):
                if self.el.find('.//td[@class="date"]') is None:
                    return False
                if AccountsList.i < self.env['index']:
                    AccountsList.i += 1
                    return False
                return True

            def parse(self, table):
                textdate = table.find('.//td[@class="date"]').text_content()
                # Do not parse transactions already parsed
                if textdate == 'hier':
                    textdate = (date.today() - timedelta(days=1)).strftime('%d/%m/%Y')
                elif textdate == "aujourd'hui":
                    textdate = date.today().strftime('%d/%m/%Y')
                else:
                    frenchmonth = textdate.split(' ')[1]
                    month = self.monthvalue[frenchmonth]
                    textdate = textdate.replace(' ', '')
                    textdate = textdate.replace(frenchmonth, '/%s/' %month)
                self.env['_textdate'] = textdate
                category = table.find('.//td[@class="picto"]/span')
                category = unicode(category.attrib['class'].split('-')[0].lower())
                try:
                    category = self.catvalue[category]
                except:
                    pass
                self.env['category'] = category

    def get_history_jid(self):
        span = self.doc.xpath('//span[@id="index:panelASV"]')
        if len(span) > 1:
            # Assurance Vie, we do not support this kind of account.
            return None

        span = Attr('//span[starts-with(@id, "index:j_id")]', 'id')(self.doc)
        jid = span.split(':')[1]
        return jid

    def islast(self):
        havemore = self.doc.getroot().cssselect('.show-more-transactions')
        if len(havemore) == 0:
            return True

        nomore = self.doc.getroot().cssselect('.no-more-transactions')
        return (len(nomore) > 0)