import re

import json

from datetime import datetime

from ast import literal_eval

import pygeoip

FINGERPRINT_REGEX = re.compile(r'^[A-Z0-9]{40}$', re.I)

IP_REGEX = (r'^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}'

            r'(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$')

IP_REGEX = re.compile(IP_REGEX)

PRIVATE_IP_REGEX = (r'(^127\.)|(^192\.168\.)|(^10\.)|(^172\.1[6-9]\.)'

                    r'|(^172\.2[0-9]\.)|(^172\.3[0-1]\.)|'

                    r'255\.255\.255\.255')

PRIVATE_IP_REGEX = re.compile(PRIVATE_IP_REGEX)

DATE_FORMAT = '%d-%m-%Y %H:%M:%S'

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

def check_json_format(server):

    for key in ('server_type', 'server_name', 'tor_status'):

        if key not in server:

            raise ValueError('%s key not present!\n' % key)

    if server['server_type'] not in ('Exit', 'Relay', 'Bridge'):

        error_message = ('server_type has not the right content!'

                         ' is: %s must be one of: Exit, Relay or Bridge\n') \

            % server['server_type']

        raise ValueError(error_message)

    if not server['tor_status'] in ('Online', 'Offline'):

        error_message = ('tor_status has not the right content!'

                         ' is: %s must be one of: Online or Offline\n') \

            % server['tor_status']

    if server['server_type'] in ('Exit', 'Relay'):

        if 'fingerprint' not in server:

            raise ValueError('fingerprint key not present!\n')

        else:

            if FINGERPRINT_REGEX.match(server['fingerprint']) is None:

                raise ValueError('fingerprint has not the right format!\n')

    else:

        if 'obfs' not in server:

            raise ValueError('obfs key not present!\n')

        else:

            if server['obfs'] not in ('True', 'False'):

                error_message = ('obfs has not the right content!'

                                 ' is: %s must be one of: True or False\n') \

                    % server['obfs']

                raise ValueError(error_message)

    if 'ip' in server:

        if IP_REGEX.match(server['ip']) is None:

            raise ValueError('ip is not the right format!\n')

        elif PRIVATE_IP_REGEX.match(server['ip']) is not None:

            raise ValueError('ip is not accepted!\n')

    return True

def update_server(server, ip):

    server['last_updated'] = datetime.utcnow().strftime(DATE_FORMAT)

    server['server_status'] = 'Online'

    server['country'] = gi4.country_name_by_addr(ip)

    if server['server_type'] == 'Bridge':

        server['obfs'] = literal_eval(server['obfs'])

    else:

        if 'obfs' in server:

            del server['obfs']

        server['ip'] = ip

    try:

        filename = ''.join(['data/', server['server_name'].lower(), '.json'])

        with open(filename, 'w', encoding='utf-8') as fb:

            json.dump(server, fb)

        return server

    except Exception as e:

        return e

{% macro colorize_status(status) %}

  {% if status == "Online" %}

    {% set color = "text-success" %}

  {% elif status == "Unknown" %}

    {% set color = "text-warning" %}

  {% else %}

    {% set color = "text-danger" %}

  {% endif %}

  <p class={{ color }}>{{ status }}</p>

{% endmacro %}

{% macro colorize_obfs(obfs) %}

  {% if obfs %}

    {% set color = "text-success" %}

  {% else %}

    {% set color = "text-danger" %}

  {% endif %}

  <p class={{ color }}>{{ obfs }}</p>

{% endmacro %}

{% macro create_country(country) %}

  {% set country_class = "flag-" + country|lower|replace(' ', '-') %}

  <i class={{ country_class }}></i> {{ country|title }}

{% endmacro %}

{% macro create_fingerprint(fingerprint) %}

  <a href="http://globe.enn.lu/#/relay/{{ fingerprint }}">{{ fingerprint|upper}}</a>

{% endmacro %}

{% macro create_name(name) %}

  <a href="http://{{ name|lower }}.enn.lu">{{ name }}</a>

{% endmacro %}

{% macro create_server_table(server_type, servers) %}

  {% if server_type in ('Exit', 'Relay') %}

    {% set headers = ['#', 'Name', 'IP', 'Server Status', 'Tor Status', 'Country', 'Fingerprint', 'Last Updated (UTC)'] %}

  {% else %}

  {% endif %}

  <h2>{{ server_type }}</h2>

  <table class="table table-bordered table-striped">

    <thead>

      <tr>

        {% for name in headers %}

          <th>{{ name }}</th>

        {% endfor %}

      </tr>

    </thead>

    <tbody>

      {% for server in servers %}

        <tr>

          <td>{{ loop.index }}</td>

          {% if server_type in ('Exit', 'Relay') %}

            <td>{{ create_name(server['server_name']) }}</td>

          {% else %}

            <td>{{ server['server_name'] }}</td>

          {% endif %}

          {% if server_type in ('Exit', 'Relay') %}

            <td>{{ server['ip'] }}</td>

          {% endif %}

          {% for status in [server['server_status'], server['tor_status']] %}

            <td>{{ colorize_status(status) }}</td>

          {% endfor %}

          <td>{{ create_country(server['country']) }}</td>

            <td>{{ colorize_obfs(server['obfs']) }}</td>

          {% endif %}

          <td>{{ server['last_updated'] }}</td>

        </tr>

      {% endfor %}

    </tbody>

  </table>

{% endmacro %}