import ipaddress

import json

import functools

import statistics

from pathlib import Path

from datetime import datetime

import jsonschema

import strict_rfc3339

import requests

from flask import current_app

from pkg_resources import resource_filename

from ..utils import check_ip

schema = json.load(

    open(

        resource_filename('ennstatus.api', 'schema/server.json'),

        encoding='utf-8'

    )

)

validate = functools.partial(

    jsonschema.validate,

    schema=schema,

    format_checker=jsonschema.FormatChecker()

)

def calculate_weight(data):

    obj = {}

    for subkey in ('1_week', '1_month', '3_months', '1_year', '5_years'):

        subdata = data[subkey]

        factor = subdata['factor']

        values = [x * factor for x in subdata['values'] if x is not None]

        if values:

            obj[subkey] = statistics.mean(values) * 100

        else:

            obj[subkey] = None

    return obj

class ServerEncoder(json.JSONEncoder):

    def default(self, obj):

        if isinstance(obj, (ipaddress.IPv4Address, ipaddress.IPv6Address)):

            return str(obj)

        if isinstance(obj, datetime):

            return strict_rfc3339.timestamp_to_rfc3339_utcoffset(

                obj.timestamp()

            )

        return json.JSONEncoder.default(self, obj)

class ServerDecoder(json.JSONDecoder):

    def decode(self, json_string):

        default_obj = super().decode(json_string)

        for key in ('ip', 'ip6'):

            if key in default_obj:

                current_app.logger.debug('{}: {}'.format(

                    key, default_obj[key]

                )

                )

                default_obj[key] = ipaddress.ip_address(default_obj[key])

        current_app.logger.debug('Loading last_updated')

        default_obj['last_updated'] = datetime.fromtimestamp(

            strict_rfc3339.rfc3339_to_timestamp(default_obj['last_updated'])

        )

        return default_obj

class Server:

    def __init__(self, *args, **kwargs):

        self.name = kwargs['name']

        self.type = kwargs['type']

        self.status = kwargs.get('status')

        self.fingerprint = kwargs['fingerprint']

        self.last_updated = kwargs['last_updated']

        self.country = kwargs['country']

        if self.type == 'bridge':

            self.obfs = kwargs.get('obfs')

            self.fteproxy = kwargs.get('fteproxy')

            self.flashproxy = kwargs.get('flashproxy')

            self.meek = kwargs.get('meek')

        else:

            self.ip = kwargs['ip']

            if 'ip6' in kwargs:

                self.ip6 = kwargs['ip6']

            default_weights = {

                '1_week': None,

                '1_month': None,

                '3_months': None,

                '1_year': None,

                '5_years': None

            }

            self.mean_consensus_weight = kwargs.get(

                'mean_consensus_weight',

                default_weights

            )

            self.mean_guard_probability = kwargs.get(

                'mean_guard_probability',

                default_weights

            )

            self.mean_exit_probability = kwargs.get(

                'mean_exit_probability',

                default_weights

            )

            self.mean_consensus_weight_fraction = kwargs.get(

                'mean_consensus_weight_fraction',

                default_weights

            )

            self.mean_middle_probability = kwargs.get(

                'mean_middle_probability',

                default_weights

            )

    @classmethod

    def from_file_by_name(cls, name):

        filepath = Path('data') / (name.lower() + '.json')

        current_app.logger.info('Loading {}'.format(str(filepath)))

        if filepath.exists() and filepath.is_file():

            try:

                with filepath.open(encoding='utf-8') as f:

                    data = json.load(f, cls=ServerDecoder)

            except (IOError, ValueError):

                current_app.logger.error('IOError or ValueError')

                return False

            else:

                return cls(**data)

        else:

            current_app.logger.error('File error!')

            return False

    @classmethod

    def from_json(cls, server):

        try:

            if cls.check_json_format(json.loads(server)):

                decoded = json.loads(server, cls=ServerDecoder)

                return cls(**decoded)

        except (jsonschema.ValidationError, ValueError) as e:

            raise e

    @classmethod

    def from_dict(cls, server):

        return cls.from_json(json.dumps(server))

    def json(self):

        return json.dumps(self.__dict__, cls=ServerEncoder)

    @staticmethod

    def check_json_format(server):

        try:

            validate(server)

        except jsonschema.ValidationError as e:

            raise e

        for key in ('ip', 'ip6'):

            if key in server:

                address = ipaddress.ip_address(server[key])

                if not check_ip(address):

                    raise ValueError('{} is not accepted!\n'.format(key))

        return True

    def save(self):

        filepath = Path('data') / (self.name.lower() + '.json')

        try:

            with filepath.open(mode='w', encoding='utf-8') as f:

                json.dump(self.__dict__, f, cls=ServerEncoder)

        except Exception as e:

            raise e

    def update_weights(self):

        if self.type not in ('exit', 'relay'):

            raise NotImplementedError

        url = 'https://onionoo.torproject.org/weights?lookup={}'.format(

            self.fingerprint

        )

        data = requests.get(url)

        try:

            data.raise_for_status()

        except requests.HTTPError as e:

            raise e

        else:

            data = data.json()['relays'][0]

        self.mean_consensus_weight = calculate_weight(data['consensus_weight'])

        self.mean_exit_probability = calculate_weight(data['exit_probability'])

        self.mean_guard_probability = calculate_weight(

            data['guard_probability']

        )

        self.mean_middle_probability = calculate_weight(

            data['middle_probability']

        )

        self.mean_consensus_weight_fraction = calculate_weight(

            data['consensus_weight_fraction']

        )

    def check_status(self):

        now = datetime.utcnow()

        delta = now - self.last_updated

        if delta.seconds >= 3600:

            self.status = False

        elif delta.seconds >= 600:

            self.status = None

{

    "title": "Relay/Exit schema",

    "$schema": "http://json-schema.org/draft-04/schema#",

    "type": "object",

    "properties": {

        "name": {

            "type": "string"

        },

        "status": {

            "$ref": "#/definitions/status_data"

        },

        "country": {

            "type": "string",

            "enum": [

                "Afghanistan", "Aland Islands", "Albania", "Algeria", 

                "American Samoa", "Andorra", "Angola", "Anguilla",

                "Anonymous Proxy", "Antarctica", "Antigua and Barbuda",

                "Argentina", "Armenia", "Aruba", "Asia/Pacific Region",

                "Australia", "Austria", "Azerbaijan", "Bahamas", "Bahrain",

                "Bangladesh", "Barbados", "Belarus", "Belgium", "Belize",

                "Benin", "Bermuda", "Bhutan", "Bolivia",

                "Bonaire, Sint Eustatius and Saba", "Bosnia and Herzegovina",

                "Botswana", "Bouvet Island", "Brazil",

                "British Indian Ocean Territory", "Brunei Darussalam",

                "Bulgaria", "Burkina Faso", "Burundi", "Cambodia",

                "Cameroon", "Canada", "Cape Verde", "Cayman Islands",

                "Central African Republic", "Chad", "Chile", "China",

                "Christmas Island", "Cocos (Keeling) Islands", "Colombia",

                "Comoros", "Congo", "Congo, The Democratic Republic of the",

                "Cook Islands", "Costa Rica", "Cote D'Ivoire", "Croatia",

                "Cuba", "Cyprus", "Czech Republic", "Denmark", "Djibouti",

                "Dominica", "Dominican Republic", "Ecuador", "Egypt",

                "El Salvador", "Equatorial Guinea", "Eritrea", "Estonia",

                "Ethiopia", "Europe", "Falkland Islands (Malvinas)",

                "Faroe Islands", "Fiji", "Finland", "France",

                "France, Metropolitan", "French Guiana", "French Polynesia",

                "French Southern Territories", "Gabon", "Gambia", "Georgia",

                "Germany", "Ghana", "Gibraltar", "Greece", "Greenland",

                "Grenada", "Guadeloupe", "Guam", "Guatemala", "Guernsey",

                "Guinea", "Guinea-Bissau", "Guyana", "Haiti",

                "Heard Island and McDonald Islands",

                "Holy See (Vatican City State)", "Honduras", "Hong Kong",

                "Hungary", "Iceland", "India", "Indonesia",

                "Iran, Islamic Republic of", "Iraq", "Ireland", "Isle of Man",

                "Israel", "Italy", "Jamaica", "Japan", "Jersey", "Jordan",

                "Kazakhstan", "Kenya", "Kiribati",

                "Korea, Democratic People's Republic of",

                "Korea, Republic of", "Kuwait", "Kyrgyzstan",

                "Lao People's Democratic Republic", "Latvia", "Lebanon",

                "Lesotho", "Liberia", "Libya", "Liechtenstein", "Lithuania",

                "Luxembourg", "Macau", "Macedonia", "Madagascar", "Malawi",

                "Malaysia", "Maldives", "Mali", "Malta", "Marshall Islands",

                "Martinique", "Mauritania", "Mauritius", "Mayotte", "Mexico",

                "Micronesia, Federated States of", "Moldova, Republic of",

                "Monaco", "Mongolia", "Montenegro", "Montserrat", "Morocco",

                "Mozambique", "Myanmar", "Namibia", "Nauru", "Nepal",

                "Netherlands", "Netherlands Antilles", "New Caledonia",

                "New Zealand", "Nicaragua", "Niger", "Nigeria", "Niue",

                "Norfolk Island", "Northern Mariana Islands", "Norway",

                "Oman", "Other", "Pakistan", "Palau", "Palestinian Territory",

                "Panama", "Papua New Guinea", "Paraguay", "Peru",

                "Philippines", "Pitcairn Islands", "Poland", "Portugal",

                "Puerto Rico", "Qatar", "Reunion", "Romania",

                "Russian Federation", "Rwanda", "Saint Barthelemy",

                "Saint Helena", "Saint Kitts and Nevis", "Saint Lucia",

                "Saint Martin", "Saint Pierre and Miquelon",

                "Saint Vincent and the Grenadines", "Samoa", "San Marino",

                "Sao Tome and Principe", "Satellite Provider", "Saudi Arabia",

                "Senegal", "Serbia", "Seychelles", "Sierra Leone", "Singapore",

                "Slovakia", "Slovenia", "Solomon Islands", "Somalia",

                "South Africa", "South Georgia and the South Sandwich Islands",

                "South Sudan", "Spain", "Sri Lanka", "Sudan", "Suriname",

                "Svalbard and Jan Mayen", "Swaziland", "Sweden", "Switzerland",

                "Syrian Arab Republic", "Taiwan", "Tajikistan",

                "Tanzania, United Republic of", "Thailand", "Timor-Leste",

                "Togo", "Tokelau", "Tonga", "Trinidad and Tobago", "Tunisia",

                "Turkey", "Turkmenistan", "Turks and Caicos Islands", "Tuvalu",

                "Uganda", "Ukraine", "United Arab Emirates", "United Kingdom",

                "United States", "United States Minor Outlying Islands",

                "Uruguay", "Uzbekistan", "Vanuatu", "Venezuela", "Vietnam",

                "Virgin Islands, British", "Virgin Islands, U.S.",

                "Wallis and Futuna", "Western Sahara", "Yemen", "Zambia",

                "Zimbabwe"

            ]

        },

        "fingerprint": {

            "type": "string",

            "pattern": "^[a-zA-Z0-9]{40}$"

        },

        "last_updated": {

            "type": "string",

            "format": "date-time"

        },

            "type": [

                "string",

                "null"

            ],

            "default": null

        },

        "type": {

            "type": "string"

        }

    },

    "required": [

        "name",

        "status",

        "last_updated",

        "country",

        "fingerprint",

        "type"

    ],

    "definitions": {

        "weights": {

            "type": "object",

            "properties": {

                "1_week": {

                    "$ref": "#/definitions/weights_data"

                },

                "1_month": {

                    "$ref": "#/definitions/weights_data"

                },

                "3_months": {

                    "$ref": "#/definitions/weights_data"

                },

                "1_year": {

                    "$ref": "#/definitions/weights_data"

                },

                "5_years": {

                    "$ref": "#/definitions/weights_data"

                }

            },

            "required": [

                "1_week",

                "1_month",

                "3_months",

                "1_year",

                "5_years"

            ],

            "additionalProperties": false

        },

        "weights_data": {

            "type": [

                "number",

                "null"

            ],

            "default": null

        },

        "status_data": {

            "type": [

                "boolean",

                "null"

            ],

            "default": null

        }

    },

    "anyOf": [

        {

            "properties": {

                "type": {

                    "enum": [

                        "exit",

                        "relay"

                    ]

                },

                "ip": {

                    "type": "string",

                    "format": "ipv4"

                },

                "ip6": {

                    "type": "string",

                    "format": "ipv6"

                },

                "mean_consensus_weight": {

                    "$ref": "#/definitions/weights"

                },

                "mean_guard_probability": {

                    "$ref": "#/definitions/weights"

                },

                "mean_exit_probability": {

                    "$ref": "#/definitions/weights"

                },

                "mean_middle_probability": {

                    "$ref": "#/definitions/weights"

                },

                "mean_consensus_weight_fraction": {

                    "$ref": "#/definitions/weights"

                }

            },

            "required": [

                "ip"

            ],

            "not": {

                "required": [

                    "obfs",

                    "fteproxy",

                    "flashproxy",

                    "meek"

                ]

            }

        },

        {

            "properties": {

                "type": {

                    "enum": [

                        "bridge"

                    ]

                },

                "obfs": {

                    "type": [

                        "boolean",

                        "null"

                    ]

                },

                "fteproxy": {

                    "type": [

                        "boolean",

                        "null"

                    ]

                },

                "flashproxy": {

                    "type": [

                        "boolean",

                        "null"

                    ]

                },

                "meek": {

                    "type": [

                        "boolean",

                        "null"

                    ]

                }

            },

            "required": [

                "obfs",

                "fteproxy",

                "flashproxy",

                "meek"

            ],

            "not": {

                "required": [

                    "ip",

                    "ip6",

                    "mean_consensus_weight",

                    "mean_guard_probability",

                    "mean_exit_probability"

                ]

            }

        }

    ]

} 

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+)'),

                   ]

      )