import time
from datetime import datetime
import os
import json
import sys
import os.path
from setup import Setup
from database import Database
import requests
import urllib3
import socket
import ray
import pdb

#ray.init(num_cpus = 25) # Specify this system CPUs.

from ray.exceptions import (
    RaySystemError,
    RayError,
    RayTaskError,
    ObjectStoreFullError,
)

apis = ['/api/v1/instance',
        '/api/v1/nodeinfo',
        '/nodeinfo/2.0',
        '/nodeinfo/2.0.json',
        '/nodeinfo/2.1.json',
        '/main/nodeinfo/2.0',
        '/api/statusnet/config',
        '/api/nodeinfo/2.0.json',
        '/api/nodeinfo',
        '/wp-json/nodeinfo/2.0',
        '/api/v1/instance/nodeinfo/2.0',
        '/.well-known/x-nodeinfo2'
        ]

def is_json(myjson):

    try:
        json_object = json.loads(myjson)
    except ValueError as e:
        return False
    return True

@ray.remote
def getsoft(server):

    if server.find(".") == -1:
        return
    if server.find("@") != -1:
        return
    if server.find("/") != -1:
        return
    if server.find(":") != -1:
        return

    soft = ''

    is_nodeinfo = False

    url = 'https://' + server

    try:

        response = requests.get(url + '/.well-known/nodeinfo', headers = setup.user_agent, timeout=3)

        if response.status_code == 200:

            try:

                response_json = response.json()

                if len(response_json['links']) == 1:

                    nodeinfo = response_json['links'][0]['href'].replace(f'https://{server}','')

                elif len(response_json['links']) == 2:

                    nodeinfo = response_json['links'][1]['href'].replace(f'https://{server}','')

                try:

                    nodeinfo_data = requests.get(url + nodeinfo, headers = setup.user_agent, timeout=3)

                    if nodeinfo_data.status_code == 200:

                        nodeinfo_json = nodeinfo_data.json()

                        is_nodeinfo = True

                    else:

                        print(f"{nodeinfo} not responding: error code {nodeinfo_data.status_code}")

                except:

                    pass

            except:

                print(f'{server} is not responding: error code {response.status_code}')
                print('*********************************************************************')

                pass
        else:

            for api in apis:

                try:

                    response = requests.get(url + api, headers = setup.user_agent, timeout=3)

                    if response.status_code == 200:

                        if is_json(response.text):

                            nodeinfo_json = response.json()

                            if 'software' in nodeinfo_json:

                                nodeinfo = api

                                is_nodeinfo = True

                                break

                            elif 'title' in nodeinfo_json:

                                if nodeinfo_json['title'] == 'Zap':

                                    nodeinfo = api

                                    is_nodeinfo = True

                                    soft = 'zap'

                                    break

                            elif 'version' in nodeinfo_json:

                                nodeinfo = api

                                is_nodeinfo = True

                                break

                except:

                    pass

    except requests.exceptions.SSLError as errssl:

        pass

    except requests.exceptions.HTTPError as errh:

        pass

    except requests.exceptions.ConnectionError as errc:

        pass

    except requests.exceptions.ReadTimeout as to_err:

        pass

    except requests.exceptions.TooManyRedirects as tmr_err:

        pass

    except urllib3.exceptions.LocationParseError as lp_err:

        pass

    except requests.exceptions.InvalidURL as iu_err:

        pass

    except requests.exceptions.ChunkedEncodingError as chunk_err:

        print(f'ChunkedEncodingError! {server}')
        pass

    except ray.exceptions.RaySystemError as ray_sys_error:

        print(ray_sys_error) 
        pass

    else:

        if is_nodeinfo:

            if nodeinfo != '/api/v1/instance?':

                if nodeinfo != '/.well-known/x-nodeinfo2?':

                    try:

                        soft = nodeinfo_json['software']['name']
                        soft = soft.lower()
                        soft_version = nodeinfo_json['software']['version']
                        users = nodeinfo_json.get('usage').get('users').get('total') or '0'
                        if users > 1000000:
                            return

                        alive = True

                        db.write_api(server, soft, users, alive, nodeinfo, soft_version)

                        print(f"Server {server} ({soft} {soft_version}) is alive!")
                        print('*********************************************************************')

                        return

                    except:

                        pass

                else:

                    try:

                        soft = nodeinfo_json['server']['software']
                        soft = soft.lower()
                        soft_version = nodeinfo_json['server']['version']
                        users = nodeinfo_json['usage']['users']['total']
                        if users > 1000000:
                            return

                        alive = True

                        if soft == 'socialhome':

                            db.write_api(server, soft, users, alive, nodeinfo, soft_version)

                            print('*********************************************************************')
                            print(f"Server {serve}r ({soft} {soft_version}) is alive!")
                            print('*********************************************************************')

                        return

                    except:

                        pass

            if soft == '' and nodeinfo == "/api/v1/instance?":

                soft = 'mastodon'

                try:

                    users = nodeinfo_json['stats']['user_count']

                    if users > 1000000:

                        return

                except:

                    users = 0

                try:

                    soft_version = nodeinfo_json['version']

                except:

                    soft_version = 'unknown'

                alive = True

                db.write_api(server, soft, users, alive, nodeinfo, soft_version)

                print('*********************************************************************')
                print(f"Server {server} ({soft}) is alive!")

            elif soft == 'zap' and nodeinfo == "/api/v1/instance?":

                soft = 'zap'
                users = nodeinfo_json['stats']['user_count']
                soft_version = nodeinfo_json['version']
                alive = True

                print(server, soft, users, alive, api)

                print('*********************************************************************')
                print(f"Server {server} ({soft}) is alive!")

        else:

            print(f'Server {server} is dead')
            print('*********************************************************************')

###############################################################################
# main

if __name__ == '__main__':

    ## name: fetchservers.py

    setup = Setup()

    db = Database()

    res = requests.get('https://' + 'mastodon.social' + setup.peers_api, headers = setup.user_agent, timeout=3)

    hostname_peers = res.json()

    start = datetime.now()

    program = 'fetchservers'

    finish = start

    db.save_time(program, start, finish)

    now = start

    ray_start = time.time()

    try:

        results = ray.get([getsoft.remote(server) for server in hostname_peers])

        print(f"duration = {time.time() - ray_start}.\nprocessed servers: {len(results)}")

    except:

        pass

    finish = datetime.now()

    db.save_time(program, start, finish)