import os
import subprocess

from error_codes       import *
from errors            import is_error, print_errors
from errors_tsg        import ask_onboarding_error_codes
from install.install   import check_installation
from install.check_oms import get_oms_version
from .check_endpts     import check_internet_connect, check_agent_service_endpt, \
                              check_log_analytics_endpts
from .check_e2e        import check_e2e



def check_connection(interactive, err_codes=True, prev_success=NO_ERROR):
    print("CHECKING CONNECTION...")

    success = prev_success

    if (interactive and err_codes):
        if (ask_onboarding_error_codes() == USER_EXIT):
            return USER_EXIT

    # check if installed correctly
    print("Checking if installed correctly...")
    if (get_oms_version() == None):
        print_errors(ERR_OMS_INSTALL)
        print("Running the installation part of the troubleshooter in order to find the issue...")
        print("================================================================================")
        return check_installation(interactive, err_codes=False, prev_success=ERR_FOUND)

    # check general internet connectivity
    print("Checking if machine is connected to the internet...")
    checked_internet_connect = check_internet_connect()
    if (is_error(checked_internet_connect)):
        return print_errors(checked_internet_connect)
    else:
        success = print_errors(checked_internet_connect)

    # check if agent service endpoint connected
    print("Checking if agent service endpoint is connected...")
    checked_as_endpt = check_agent_service_endpt()
    if (is_error(checked_as_endpt)):
        return print_errors(checked_as_endpt)
    else:
        success = print_errors(checked_as_endpt)

    # check if log analytics endpoints connected
    print("Checking if log analytics endpoints are connected...")
    checked_la_endpts = check_log_analytics_endpts()
    if (is_error(checked_la_endpts)):
        return print_errors(checked_la_endpts)
    else:
        success = print_errors(checked_la_endpts)

    # check if queries are successful
    if (interactive):
        print("Checking if queries are successful...")
        checked_e2e = check_e2e()
        if (is_error(checked_e2e)):
            return print_errors(checked_e2e)
        else:
            success = print_errors(checked_e2e)
        
    return success