import ipaddress

import json

from datetime import datetime

from flask import (Blueprint, request, current_app, jsonify, render_template,

                   abort)

import strict_rfc3339

import pygeoip

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

                                        all_servers_by_type)

from .model import Server

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)

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

def update():

    current_app.logger.info('Handling update')

        current_app.logger.warn('Unallowed IP %s tried to update data!'

                                % request.remote_addr)

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

    data = request.get_json()

    if data is None:

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

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

    if 'ip' in data:

        ip = data['ip']

    elif 'ip6' in data:

        ip = data['ip6']

    else:

        ip = request.remote_addr

    try:

        ip = ipaddress.ip_address(ip)

    except ipaddress.AddressValueError:

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

    if ip.version == 4:

        data['country'] = gi4.country_name_by_addr(str(ip))

    elif ip.version == 6:

        data['country'] = gi6.country_name_by_addr(str(ip))

    else:

        data['country'] = None

    data['last_updated'] = strict_rfc3339.timestamp_to_rfc3339_utcoffset(

        datetime.utcnow().timestamp()

    )

    try:

        server = Server.from_json(json.dumps(data))

    except Exception as e:

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

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

    if server.type in ('exit', 'relay'):

        server.update_weights()

    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=%s'

             % server.name})

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

                                     'export_format': 'json'})

@api_page.route('/export/<any("json", "xml"):export_format>',

                defaults={'server_type': 'all'})

@api_page.route(('/export/<any("json", "xml"):export_format>'

                 '/<any("all", "exit", "bridge", "relay", "single")'

                 ':server_type>'))

def export(export_format, 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:

                if export_format == 'json':

                    current_app.logger.info('Returning server as json!')

                    return (server.json(), 200,

                            {'Content-Type': 'application/json'})

                else:

                    current_app.logger.info('Returning server as xml!')

                    return (

                        render_template(

                            'api/export/xml/single_server.xml',

                            server=server),

                        200, {'Content-Type': 'text/xml'})

            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 = [server.json() for server in all_servers()]

        else:

            current_app.logger.info('Getting all %s!' % server_type)

            servers = list(all_servers_by_type(server_type.lower()))

        if export_format == 'json':

            current_app.logger.info('Returning as json!')

            return str(servers)

        else:

            current_app.logger.info('Returning as xml!')

            return (render_template('api/export/xml/network.xml',

                                    servers=servers),

                    200, {'Content-Type': 'text/xml'})

Flask-Bootstrap==3.3.5.6

Flask-Mail==0.9.1

Flask-Moment==0.5.1

Flask-WTF==0.12

Flask==0.10.1

jsonschema==2.5.1

pygeoip==0.3.2

python-gnupg==0.3.7

requests==2.7.0

strict-rfc3339==0.5

#

# This file is autogenerated by pip-compile

# Make changes in requirements.in, then run this to update:

#

#    pip-compile requirements.in

#

blinker==1.4              # via flask-mail

dominate==2.1.12          # via flask-bootstrap

Flask-Bootstrap==3.3.5.6

Flask-Mail==0.9.1

Flask-Moment==0.5.1

Flask-WTF==0.12

flask==0.10.1

itsdangerous==0.24        # via flask

jinja2==2.8               # via flask

jsonschema==2.5.1

markupsafe==0.23          # via jinja2

pygeoip==0.3.2

python-gnupg==0.3.7

requests==2.7.0

strict-rfc3339==0.5

werkzeug==0.10.4          # via flask, flask-wtf

wtforms==2.0.2            # via flask-wtf