import subprocess

import os

import time

import requests

def start_multiner_server():

    current_directory = os.getcwd()

    working_directory = "../resources/BERN2/scripts/"

    os.chdir(working_directory)

    run_path = "run_bern2.sh"

    stop_path = "stop_bern2.sh"

    print("Stopping any existing Multi-NER server instance.")

    stop_process = subprocess.Popen(["bash", stop_path], stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)

    stop_process.wait()

    print("Activating Mutli-NER Server... This can take approx. 1 minute")

    try:

        subprocess.Popen(["bash", run_path], stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)

        timeout = 120  # Adjust this value as needed

        # Define the server's URL that you want to check

        server_url = "http://localhost:8888"  # Update with the actual URL

        # Wait for the server to become available or reach the timeout

        start_time = time.time()

        while True:

            try:

                # Send a request to the server to check its availability

                response = requests.get(server_url)

                response.raise_for_status()  # Raises an exception for non-2xx status codes

                break  # Server is available, exit the loop

            except (requests.ConnectionError, requests.HTTPError) as e:

                if time.time() - start_time >= timeout:

                    print(f"Server did not become available within {timeout} seconds.")

                    break  # Timeout reached

                else:

                    # Wait for a short time before checking again

                    time.sleep(1)

        # Continue with other tasks

        print("Server is now available.")

    except subprocess.CalledProcessError as e:

        print(f"Error executing the script: {e}")

    os.chdir(current_directory)

def stop_multiner_server():

    current_directory = os.getcwd()

    working_directory = "../resources/BERN2/scripts/"

    os.chdir(working_directory)

    stop_path = "stop_bern2.sh"

    stop_process = subprocess.Popen(["bash", stop_path], stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)

    stop_process.wait()

    print("Multi-NER server instance terminated.")