# ============LICENSE_START=======================================================
# Copyright (c) 2017-2021 AT&T Intellectual Property. All rights reserved.
# Copyright (c) 2019 Pantheon.tech. All rights reserved.
# Copyright (c) 2020 Deutsche Telekom. All rights reserved.
# Copyright (c) 2021 Samsung Electronics. All rights reserved.
# Copyright (c) 2021 Fujitsu Ltd.
# ================================================================================
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============LICENSE_END=========================================================
#
##  Author Kiran Mandal (km386e)

"""
trapd_vnf_table verifies the successful creation of DB Tables.
"""
import logging
import os
import yaml
import os.path as path
import time
import subprocess

import get_logger
import db_monitoring as dbmon
import htbtworker as pm
import misshtbtd as db
import cbs_polling as cbs

prog_name = os.path.basename(__file__)
hb_properties_file = path.abspath(path.join(__file__, "../../config/hbproperties.yaml"))
_logger = logging.getLogger(__name__)


def hb_properties():
    # Read the hbproperties.yaml for postgress and CBS related data
    s = open(hb_properties_file, "r")
    a = yaml.full_load(s)
    ip_address = a["pg_ipAddress"]
    port_num = a["pg_portNum"]
    user_name = a["pg_userName"]
    password = a["pg_passwd"]
    dbName = a["pg_dbName"]
    db_name = dbName.lower()
    cbs_polling_required = a["CBS_polling_allowed"]
    cbs_polling_interval = a["CBS_polling_interval"]
    s.close()
    return ip_address, port_num, user_name, password, db_name, cbs_polling_required, cbs_polling_interval


def verify_DB_creation_1(user_name, password, ip_address, port_num, db_name):
    connection_db = pm.postgres_db_open(user_name, password, ip_address, port_num, db_name)
    try:
        _db_status = pm.db_table_creation_check(connection_db, "vnf_table_1")
    except Exception as e:
        return None

    return _db_status


def verify_DB_creation_2(user_name, password, ip_address, port_num, db_name):

    connection_db = pm.postgres_db_open(user_name, password, ip_address, port_num, db_name)
    try:
        _db_status = pm.db_table_creation_check(connection_db, "vnf_table_2")
    except Exception as e:
        return None

    return _db_status


def verify_DB_creation_hb_common(user_name, password, ip_address, port_num, db_name):

    connection_db = pm.postgres_db_open(user_name, password, ip_address, port_num, db_name)
    try:
        _db_status = pm.db_table_creation_check(connection_db, "hb_common")
    except Exception as e:
        return None

    return _db_status


def verify_cbspolling():
    os.environ["pytest"] = "test"
    os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static"
    cbs.poll_cbs(10)
    os.unsetenv("pytest")
    os.unsetenv("SERVICE_NAME")


def verify_fetch_json_file():
    os.environ["pytest"] = "test"
    os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static"
    os.environ["CONSUL_HOST"] = "localhost"
    os.environ["HOSTNAME"] = "mvp-dcaegen2-heartbeat-static"
    try:
        db.fetch_json_file()
        result = True
    except Exception as e:
        result = False
    print(result)
    os.unsetenv("pytest")
    os.unsetenv("SERVICE_NAME")
    os.unsetenv("CONSUL_HOST")
    os.unsetenv("HOSTNAME")
    return result


def verify_misshtbtdmain():
    os.environ["pytest"] = "test"
    os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static"
    os.environ["CONSUL_HOST"] = "localhost"
    os.environ["HOSTNAME"] = "mvp-dcaegen2-heartbeat-static"

    try:
        db.main()
        result = True
    except Exception as e:
        result = False
    print(result)
    os.unsetenv("pytest")
    os.unsetenv("SERVICE_NAME")
    os.unsetenv("CONSUL_HOST")
    os.unsetenv("HOSTNAME")
    return result


def verify_dbmonitoring():
    os.environ["pytest"] = "test"
    os.environ["SERVICE_NAME"] = "mvp-dcaegen2-heartbeat-static"
    os.environ["CONSUL_HOST"] = "localhost"
    os.environ["HOSTNAME"] = "mvp-dcaegen2-heartbeat-static"
    try:
        jsfile = db.fetch_json_file()
        ip_address, port_num, user_name, password, db_name, cbs_polling_required, cbs_polling_interval = hb_properties()
        hbc_pid, hbc_state, hbc_srcName, hbc_time = db.read_hb_common(
            user_name, password, ip_address, port_num, db_name
        )
        dbmon.db_monitoring(hbc_pid, jsfile, user_name, password, ip_address, port_num, db_name)
        result = True
    except Exception as e:
        print("Message process error - %s" % e)
        result = False
    print(result)
    os.unsetenv("pytest")
    os.unsetenv("SERVICE_NAME")
    os.unsetenv("CONSUL_HOST")
    os.unsetenv("HOSTNAME")
    return result


def verify_dbmon_startup():
    try:
        p = subprocess.Popen(["./miss_htbt_service/db_monitoring.py"], stdout=subprocess.PIPE, shell=True)
        time.sleep(1)
    except Exception as e:
        return None
    return True


def verify_sendControlLoop_VNF_ONSET():
    try:
        pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/"
        _CL_return = dbmon.sendControlLoopEvent(
            "ONSET",
            pol_url,
            "1.0",
            "vFireWall",
            "pscope",
            "VNF",
            "srcname1",
            1541234567,
            "SampleCLName",
            "1.0",
            "genVnfName",
        )
    except Exception as e:
        return None
    return _CL_return


def verify_sendControlLoop_VM_ONSET():
    try:
        pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/"
        _CL_return = dbmon.sendControlLoopEvent(
            "ONSET",
            pol_url,
            "1.0",
            "vFireWall",
            "pscope",
            "VM",
            "srcname1",
            1541234567,
            "SampleCLName",
            "1.0",
            "genVnfName",
        )
    except Exception as e:
        return None
    return _CL_return


def verify_sendControlLoop_VNF_ABATED():
    try:
        pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/"
        _CL_return = dbmon.sendControlLoopEvent(
            "ABATED",
            pol_url,
            "1.0",
            "vFireWall",
            "pscope",
            "VNF",
            "srcname1",
            1541234567,
            "SampleCLName",
            "1.0",
            "genVnfName",
        )
    except Exception as e:
        return None
    return _CL_return


def verify_sendControlLoop_VM_ABATED():
    try:
        pol_url = "http://10.12.5.252:3904/events/unauthenticated.DCAE_CL_OUTPUT/"
        _CL_return = dbmon.sendControlLoopEvent(
            "ABATED",
            pol_url,
            "1.0",
            "vFireWall",
            "pscope",
            "VM",
            "srcname1",
            1541234567,
            "SampleCLName",
            "1.0",
            "genVnfName",
        )
    except Exception as e:
        return None
    return _CL_return