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

# Copyright(C) 2010  Romain Bignon
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

from __future__ import with_statement

from weboob.capabilities.geolocip import ICapGeolocIp, IpLocation
from weboob.capabilities.base import NotAvailable
from weboob.tools.backend import BaseBackend
from weboob.tools.browser import BaseBrowser
from weboob.tools.value import ValuesDict, Value


__all__ = ['GeolocIpBackend']


class GeolocIpBackend(BaseBackend, ICapGeolocIp):
    NAME = 'geolocip'
    MAINTAINER = 'Romain Bignon'
    EMAIL = 'romain@weboob.org'
    VERSION = '0.4'
    LICENSE = 'GPLv3'
    DESCRIPTION = u"IP Adresses geolocalisation with the API of the site www.geolocalise-ip.com "
    CONFIG = ValuesDict(Value('email',   label='Username'),
                        Value('password',label='Password', masked=True))
    BROWSER = BaseBrowser

    def create_default_browser(self):
        return self.create_browser(self.config['email'], self.config['password'])

    def get_location(self, ipaddr):
        with self.browser:
            args = {'email':     self.config['email'],
                    'pass':      self.config['password'],
                    'ip':        str(ipaddr)
                   }
            content = self.browser.readurl(self.browser.buildurl('http://www.geolocalise-ip.com/api.php', **args))
            tab = {'ville' : NotAvailable ,\
                    'region' : NotAvailable ,\
                    'cp' : NotAvailable ,\
                    'pays' : NotAvailable ,\
                    'lt' : NotAvailable ,\
                    'lg' : NotAvailable ,\
                    'host' : NotAvailable ,\
                    'tld' : NotAvailable ,\
                    'fai' : NotAvailable}

            for line in content.split('&'):
                if not '=' in line:
                    continue
                key, value = line.split('=', 1)
                tab[key] = value

            if 'erreur' in tab and tab['erreur'][0] == '1':
                raise Exception(tab['erreur'][1:].replace('<p>', '').replace('</p>', '').replace('<br />', '\n')\
                                .strip().decode('iso-8859-1'))

            iploc = IpLocation(ipaddr)

            if tab['ville'] != NotAvailable:
                iploc.city = tab['ville'].decode('iso-8859-15')
            iploc.region = tab['region']
            iploc.zipcode = tab['cp']
            iploc.country = tab['pays']
            iploc.lt = float(tab['lt'])
            iploc.lg = float(tab['lg'])
            iploc.host = tab['host']
            iploc.tld = tab['tld']
            if 'fai' in tab:
                iploc.isp = tab['fai']
            else:
                iploc.isp = NotAvailable
            return iploc