#!/usr/bin/env python3
###
# ============LICENSE_START=======================================================
# ORAN SMO PACKAGE - PYTHONSDK TESTS
# ================================================================================
# Copyright (C) 2022 AT&T Intellectual Property. All rights
#                             reserved.
# ================================================================================
# 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============================================
# ===================================================================
#
###
"""Closed Loop Apex usecase tests module."""
# This usecase has limitations due to Clamp issue.
# 1. make sure using the policy-clamp-be version 6.2.0-snapshot-latest at this the moment
import time
import logging.config
import subprocess
import os
from subprocess import check_output
import pytest
from waiting import wait
from onapsdk.configuration import settings
from oransdk.utils.jinja import jinja_env
from oransdk.policy.clamp import ClampToscaTemplate
from smo.cl_usecase import ClCommissioningUtils

# Set working dir as python script location
abspath = os.path.abspath(__file__)
dname = os.path.dirname(abspath)
os.chdir(dname)

logging.config.dictConfig(settings.LOG_CONFIG)
logger = logging.getLogger("test Control Loops for Clamp K8S usecase")
clcommissioning_utils = ClCommissioningUtils()
clamp = ClampToscaTemplate(settings.CLAMP_BASICAUTH)

chartmuseum_port = "8080"
usecase_name = "script_usecase"
chartmuseum_ip = "http://test-chartmuseum.test:8080"
global app_name

@pytest.fixture(autouse=True)
def setup_simulators(request):
    """Prepare the test environment before the executing the tests."""
    logger.info("Test class setup for Closed Loop tests of %s", request.node.name)

    deploy_chartmuseum()

    # Add the remote repo to Clamp k8s pod
    logger.info("Add the remote repo to Clamp k8s pod")
    k8s_pod = subprocess.run("kubectl get pods -n onap | grep k8s | awk '{print $1}'", shell=True, check=True, stdout=subprocess.PIPE).stdout.decode('utf-8').strip()

    repo_url = subprocess.run("kubectl get services -n test | grep test-chartmuseum | awk '{print $3}'", shell=True, check=True, stdout=subprocess.PIPE).stdout.decode('utf-8').strip()+":8080"
    logger.info("k8s: %s, repo_url:%s", k8s_pod, repo_url)
    cmd = f"kubectl exec -it -n onap {k8s_pod} -- sh -c \"helm repo add chartmuseum http://{repo_url}\""
    check_output(cmd, shell=True).decode('utf-8')
    cmd = f"kubectl exec -it -n onap {k8s_pod} -- sh -c \"helm repo update\""
    check_output(cmd, shell=True).decode('utf-8')
    cmd = f"kubectl exec -it -n onap {k8s_pod} -- sh -c \"helm search repo -l oru-app\""
    result = check_output(cmd, shell=True).decode('utf-8')
    if result == '':
        logger.info("Failed to update the K8s pod repo")
    logger.info("Test Session setup completed successfully for %s", request.node.name)

    ### Cleanup code
    yield
    # Finish and delete the cl instance
    clcommissioning_utils.clean_instance(usecase_name)
    wait(lambda: is_rapp_down(app_name), sleep_seconds=5, timeout_seconds=60, waiting_for="Rapp is down")
    # Remove the remote repo to Clamp k8s pod
    cmd = f"kubectl exec -it -n onap {k8s_pod} -- sh -c \"helm repo remove chartmuseum\""
    check_output(cmd, shell=True).decode('utf-8')
    cmd = "kubectl delete namespace test"
    check_output(cmd, shell=True).decode('utf-8')
    cmd = "helm repo remove test"
    check_output(cmd, shell=True).decode('utf-8')
    time.sleep(10)
    logger.info("Test Session cleanup done for %s", request.node.name)


def deploy_chartmuseum():
    """Start chartmuseum pod and populate with the nedded helm chart."""
    logger.info("Start to deploy chartmuseum")
    cmd = "helm repo add test https://chartmuseum.github.io/charts"
    check_output(cmd, shell=True).decode('utf-8')
    cmd = "kubectl create namespace test"
    check_output(cmd, shell=True).decode('utf-8')

    cmd = "helm install test test/chartmuseum --version 3.1.0 --namespace test --set env.open.DISABLE_API=false"
    check_output(cmd, shell=True).decode('utf-8')
    wait(lambda: is_chartmuseum_up(), sleep_seconds=10, timeout_seconds=60, waiting_for="chartmuseum to be ready")

    time.sleep(10)
    chartmuseum_url = subprocess.run("kubectl get services -n test | grep test-chartmuseum | awk '{print $3}'", shell=True, check=True, stdout=subprocess.PIPE).stdout.decode('utf-8').strip()+":8080"
    cmd = f"curl -X POST --data-binary @{dname}/resources/cl-test-helm-chart/oru-app-1.0.0.tgz http://{chartmuseum_url}/api/charts"
    check_output(cmd, shell=True).decode('utf-8')
    cmd = f"curl -X POST --data-binary @{dname}/resources/cl-test-helm-chart/odu-app-1.0.0.tgz http://{chartmuseum_url}/api/charts"
    check_output(cmd, shell=True).decode('utf-8')
    cmd = f"curl -X POST --data-binary @{dname}/resources/cl-test-helm-chart/odu-app-ics-version-1.0.0.tgz http://{chartmuseum_url}/api/charts"
    check_output(cmd, shell=True).decode('utf-8')


def is_chartmuseum_up() -> bool:
    """Check if the chartmuseum is up."""
    cmd = "kubectl get pods --field-selector status.phase=Running -n test"
    result = check_output(cmd, shell=True).decode('utf-8')
    logger.info("Checking if chartmuseum is UP: %s", result)
    if result == '':
        logger.info("chartmuseum is Down")
        return False
    logger.info("chartmuseum is Up")
    return True


def is_rapp_up(appname) -> bool:
    """Check if the rapp is up."""
    cmd = "kubectl get pods -n nonrtric | grep " + appname + " | wc -l"
    result = check_output(cmd, shell=True).decode('utf-8')
    logger.info("Checking if %s is up :%s", appname, result)
    if int(result) == 1:
        logger.info("%s is Up", appname.upper())
        return True
    logger.info("%s is Down", appname.upper())
    return False

def is_rapp_down(appname) -> bool:
    """Check if the rapp is down."""
    cmd = "kubectl get pods -n nonrtric | grep " + appname + " | wc -l"
    result = check_output(cmd, shell=True).decode('utf-8')
    logger.info("Checking if %s is down :%s", appname, result)
    if int(result) == 0:
        logger.info("%s is Down", appname.upper())
        return True
    logger.info("%s is Up", appname.upper())
    return False

def test_cl_oru_app_deploy():
    chart_version = "1.0.0"
    chart_name = "oru-app"
    release_name = "oru-app"
    global app_name
    app_name = chart_name
    """The Closed Loop O-RU Fronthaul Recovery usecase Apex version."""
    logger.info("Upload tosca to commissioning")
    commissioning_payload = jinja_env().get_template("commission_k8s.json.j2").render(chartmuseumIp=chartmuseum_ip, chartmuseumPort=chartmuseum_port, chartVersion=chart_version, chartName=chart_name, releaseName=release_name)
    instance_payload = jinja_env().get_template("create_instance_k8s.json.j2").render(chartmuseumIp=chartmuseum_ip, chartmuseumPort=chartmuseum_port, chartVersion=chart_version, chartName=chart_name, releaseName=release_name, instanceName=usecase_name)
    assert clcommissioning_utils.create_instance(usecase_name, commissioning_payload, instance_payload) is True

    logger.info("Check if oru-app is up")
    wait(lambda: is_rapp_up(chart_name), sleep_seconds=5, timeout_seconds=300, waiting_for="Oru app to be up")

def test_cl_odu_app_smo_deploy():
    chart_version = "1.0.0"
    chart_name = "odu-app"
    release_name = "odu-app"
    global app_name
    app_name = chart_name
    """The O-DU Slice Assurance SMO Version use case."""
    logger.info("Upload tosca to commissioning")
    commissioning_payload = jinja_env().get_template("commission_k8s.json.j2").render(chartmuseumIp=chartmuseum_ip, chartmuseumPort=chartmuseum_port, chartVersion=chart_version, chartName=chart_name, releaseName=release_name)
    instance_payload = jinja_env().get_template("create_instance_k8s.json.j2").render(chartmuseumIp=chartmuseum_ip, chartmuseumPort=chartmuseum_port, chartVersion=chart_version, chartName=chart_name, releaseName=release_name, instanceName=usecase_name)
    assert clcommissioning_utils.create_instance(usecase_name, commissioning_payload, instance_payload) is True

    logger.info("Check if odu-app smo version is up")
    wait(lambda: is_rapp_up(chart_name), sleep_seconds=5, timeout_seconds=300, waiting_for="Odu app smo version to be up")

def test_cl_odu_app_ics_deploy():
    chart_version = "1.0.0"
    chart_name = "odu-app-ics-version"
    release_name = "odu-app-ics-version"
    global app_name
    app_name = chart_name
    """The O-DU Slice Assurance ICS Version use case."""
    logger.info("Upload tosca to commissioning")
    commissioning_payload = jinja_env().get_template("commission_k8s.json.j2").render(chartmuseumIp=chartmuseum_ip, chartmuseumPort=chartmuseum_port, chartVersion=chart_version, chartName=chart_name, releaseName=release_name)
    instance_payload = jinja_env().get_template("create_instance_k8s.json.j2").render(chartmuseumIp=chartmuseum_ip, chartmuseumPort=chartmuseum_port, chartVersion=chart_version, chartName=chart_name, releaseName=release_name, instanceName=usecase_name)
    assert clcommissioning_utils.create_instance(usecase_name, commissioning_payload, instance_payload) is True

    logger.info("Check if odu-app ics version is up")
    wait(lambda: is_rapp_up(chart_name), sleep_seconds=5, timeout_seconds=300, waiting_for="Odu app ics version to be up")