from ennstatus.status.functions import (single_server, all_servers,

                                        all_servers_by_type)

from .model import Server

from .auth import httpauth

api_page = Blueprint('api', __name__)

gi4 = pygeoip.GeoIP('/usr/share/GeoIP/GeoIP.dat', pygeoip.MEMORY_CACHE)

gi6 = pygeoip.GeoIP('/usr/share/GeoIP/GeoIPv6.dat', pygeoip.MEMORY_CACHE)

@csrf.exempt

@api_page.route('/update', methods=('POST',))

@httpauth.login_required

def update():

    current_app.logger.info('Handling update')

    try:

        servers = current_app.config['ENNSTATUS_SERVERS']

    except KeyError as e:

        current_app.logger.error(str(e))

        return abort(500)

    username = httpauth.username()

    try:

        if request.remote_addr not in servers[username]['IPS']:

            current_app.logger.warn(

                'Unallowed IP {} tried to update data!'.format(

                    request.remote_addr

                )

            )

            return 'IP not allowed!\n', 403, {'Content-Type': 'text/plain'}

    except KeyError as e:

        current_app.logger.error(str(e))

        return abort(500)

    try:

        data = request.get_json()

    except BadRequest:

        current_app.logger.info('No JSON data supplied!')

        return 'No JSON data supplied!\n', 400, {'Content-Type': 'text/plain'}

    try:

        if username != data['name'].lower():

            current_app.logger.warn(

                'Unallowed user {} tried to update {}!'.format(

                    username, data['name']

                )

            )

            return ('You are not allowed to update this server\n',

                    403, {'Content-Type': 'text/plain'})

    except KeyError:

        return abort(409)

    if 'ip' in data:

        ip = data['ip']

    elif 'ip6' in data:

        ip = data['ip6']

    else:

        ip = request.remote_addr

        try:

            temp_ip = ipaddress.ip_address(ip)

        except ipaddress.AddressValueError:

            return 'IP not allowed!\n', 403, {'Content-Type': 'text/plain'}

        else:

            if temp_ip.version == 4:

                data['ip'] = ip

            elif temp_ip.verison == 6:

                data['ip6'] = ip

    try:

        country = gi4.country_name_by_addr(ip)

    except pygeoip.GeoIPError:

        country = gi6.country_name_by_addr(ip)

    data['country'] = country

    data['last_updated'] = strict_rfc3339.timestamp_to_rfc3339_utcoffset(

        datetime.utcnow().timestamp()

    )

    try:

        server = Server.from_dict(data)

    except Exception as e:

        current_app.logger.warning(' '.join([str(e), str(data)]))

        return str(e), 409, {'Content-Type': 'text/plain'}

    try:

        server.update_weights()

    except NotImplementedError:

        pass

    try:

        server.update_flags()

    except NotImplementedError:

        pass

    try:

        server.save()

    except Exception as e:

        current_app.logger.error(str(e))

        return str(e), 500, {'Content-Type': 'text/plain'}

    current_app.logger.info('Return result')

    return (

        server.json(), 201,

        {

            'Location': '/api/export/json/single?server_name={}'.format(

                server.name

            )

        }

    )

@api_page.route('/export', defaults={'server_type': 'all'})

@api_page.route(('/export/<any("all", "exit", "bridge", "relay", "single")'

                 ':server_type>'))

def export(server_type):

    current_app.logger.info('Handling export')

    if server_type == 'single':

        server_name = request.args.get('server_name', None)

        if server_name is not None:

            server = single_server(server_name)

            if server:

                return server.json(), 200, {'Content-Type': 'application/json'}

            else:

                current_app.logger.warning('Server not found!')

                return ('Server not found!\n',

                        404, {'Content-Type': 'text/plain'})

        else:

            current_app.logger.warning('No server_name specified!')

            return ('No server_name specified!\n',

                    400, {'Content-Type': 'text/plain'})

    else:

        if server_type == 'all':

            current_app.logger.info('Getting all servers!')

            servers = [

                json.loads(

                    server.json()

                ) for server in all_servers()

            ]

        else:

            current_app.logger.info('Getting all {}!'.format(server_type))

            servers = [

                json.loads(

                    server.json()

                ) for server in all_servers_by_type(server_type.lower())

            ]

        return json.dumps(servers), 200, {'Content-Type': 'application/json'}

@api_page.route('/fingerprints', defaults={'server_type': 'all'})

@api_page.route('/fingerprints/<any("all", "exit", "relay"):server_type>')

def fingerprint(server_type):

    if server_type == 'all':

        servers = [server.fingerprint for server in all_servers()

                   if server.type != 'bridge']

    else:

        servers = [server.fingerprint

                   for server in all_servers_by_type(server_type.lower())]

    return '\n'.join(servers), 200, {'Content-Type': 'text/plain'}

from setuptools import setup, find_packages

def _get_requirements():

    with open('requirements.in', encoding='utf-8') as f:

        lines = f.readlines()

    lines = [line[:-1] for line in lines if not line.startswith('#')]

    return lines

setup(name='Ennstatus',

      description=('Ennstatus provides the user with vital information about '

                   'the status of the organizations Tor servers.'),

      author='Frënn vun der Ënn',

      author_email='info@enn.lu',

      url='https://bitbucket.org/fvde/ennstatus',

      license='GPLv3+',

      packages=find_packages(),

      install_requires=_get_requirements(),

      include_package_data=True,

      entry_points={

          'console_scripts': [

              'ennstatuscli = ennstatus.cli:cli',

          ]

      },

      classifiers=['Development Status :: 5 - Production/Stable',

                   'Environment :: Web Environment',

                   'Operating System :: POSIX',

                   'Programming Language :: Python',

                   'Programming Language :: Python :: 3',

                   'Programming Language :: Python :: 3.3',

                   'Topic :: Internet',

                   'Topic :: Internet :: WWW/HTTP',

                   'Topic :: Internet :: WWW/HTTP :: Dynamic Content',

                   'Topic :: Internet :: WWW/HTTP :: WSGI',

                   'Topic :: Internet :: WWW/HTTP :: WSGI :: Application',

                   ('License :: OSI Approved :: '

                    'GNU General Public License v3 or later (GPLv3+)'),

                   ]

      )