#!/usr/bin/python
# -*- coding: utf-8 -*-

# @License EPL-1.0 <http://spdx.org/licenses/EPL-1.0>
##############################################################################
# Copyright (c) 2017 Raghuram Vadapalli, Jaspreet Singh and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Eclipse Public License v1.0
# which accompanies this distribution, and is available at
# http://www.eclipse.org/legal/epl-v10.html
##############################################################################

"""
This script is used to parse dashboard config files, construct
JSON BODY and push it to ELK DB.

Usage: python push_dashboard.py host:port

"""

# stdlib
import json
import os
import sys
import glob

# 3rd party lib
from elasticsearch import Elasticsearch, RequestsHttpConnection, exceptions
import yaml

# User defined libs
import generate_visState as vis_gen
import generate_uiStateJSON as uiStateJSON_gen
import generate_dashVis as dash_gen
import generate_searchSourceJSON as searchSourceJSON_gen
import data_generate as data_gen


def p(x):
    print(json.dumps(x, indent=6, sort_keys=True))


# ELK DB host and port to be passed as ':' separated argument
if len(sys.argv) > 1:
    if ":" in sys.argv[1]:
        ELK_DB_HOST = sys.argv[1].split(":")[0]
        ELK_DB_PORT = sys.argv[1].split(":")[1]
else:
    print("Usage: python push_to_elk.py host:port")
    print("Unable to publish data to ELK. Exiting.")
    sys.exit()

try:
    es = Elasticsearch(
        hosts=[{"host": ELK_DB_HOST, "port": int(ELK_DB_PORT)}],
        scheme="https",
        connection_class=RequestsHttpConnection,
    )
except Exception as e:
    print("Unexpected Error Occurred. Exiting")
    print(e)
# print(es.info())


# sys.exit()
# Function to convert JSON object to string.
# Python puts 'true' as 'True' etc. which need handling.


def JSONToString(jobj):
    retval = str(jobj)
    retval = retval.replace("'", '"')
    retval = retval.replace(": ", ":")
    retval = retval.replace(", ", ",")
    retval = retval.replace("True", "true")
    retval = retval.replace("False", "false")
    retval = retval.replace("None", "null")
    return retval


# Clear .kibana index before pushing visualizations
try:
    index = ".kibana"
    res = es.indices.delete(index=index)
except Exception as e:
    print(e)
    # raise e
    print("Unable to push data to ElasticSearch")


# Create and push index-pattern to be used by visualizations

TEST_DATA_INDEX = "opendaylight-test"

INDEX_PATTERN_BODY = {
    "type": "index-pattern",
    "index-pattern": {"timeFieldName": "@timestamp", "title": TEST_DATA_INDEX},
}


KIBANA_CONFIG = {
    "config": {
        "defaultIndex": "pattern-for-{}".format(TEST_DATA_INDEX),
        "timepicker:timeDefaults": '{\n  "from": "now-5y",\n \
                                "to": "now",\n  "mode": "quick"\n}',
        "xPackMonitoring:showBanner": False,
    },
    "type": "config",
}

res = es.index(index=".kibana", doc_type="doc", id="config:6.2.4", body=KIBANA_CONFIG)


try:
    index = ".kibana"
    ES_ID = "index-pattern:pattern-for-{}".format(TEST_DATA_INDEX)
    res = es.index(index=index, doc_type="doc", id=ES_ID, body=INDEX_PATTERN_BODY)
    p(json.dumps(INDEX_PATTERN_BODY, indent=4))
    print(json.dumps(res, indent=4))
except Exception as e:
    print(e)
    # raise e
    print("Unable to push data to ElasticSearch")

try:
    viz_config_path = glob.glob("**/dashboard/viz_config.yaml")[0]
except IndexError:
    print("Visualization template file not found!")
    sys.exit()

try:
    dash_config_path = glob.glob("**/dashboard/dash_config.yaml")[0]
except IndexError:
    print("Dashboard configuration file not found!")
    sys.exit()

with open(dash_config_path, "r") as f:
    dash_config = yaml.safe_load(f)

with open(viz_config_path, "r") as f:
    viz_config = yaml.safe_load(f)


# Create and push visualizations
for dashboard_id, dashboard_content in dash_config.items():

    for _, i in dash_config[dashboard_id]["viz"].items():
        intermediate_format, visState = vis_gen.generate(
            i, viz_config[i["viz-template"]]
        )

        searchSourceJSON = searchSourceJSON_gen.generate(
            i, viz_config[i["viz-template"]], intermediate_format["index_pattern"]
        )

        uiStateJSON = uiStateJSON_gen.generate(i, viz_config[i["viz-template"]])

        # p(intermediate_format)
        # p(visState)

        # Template for visualization template
        VIZ_BODY = {
            "type": "visualization",
            "visualization": {
                "title": None,
                "visState": None,
                "uiStateJSON": "{}",
                "description": None,
                "version": 1,
                "kibanaSavedObjectMeta": {"searchSourceJSON": None},
            },
        }

        VIZ_BODY["visualization"]["title"] = intermediate_format["title"]
        VIZ_BODY["visualization"]["visState"] = JSONToString(visState)
        VIZ_BODY["visualization"]["uiStateJSON"] = JSONToString(uiStateJSON)
        VIZ_BODY["visualization"]["description"] = intermediate_format["desc"]
        VIZ_BODY["visualization"]["kibanaSavedObjectMeta"][
            "searchSourceJSON"
        ] = JSONToString(searchSourceJSON)

        p(VIZ_BODY)
        # Pushing visualization to Kibana
        index = ".kibana"
        ES_ID = "visualization:{}".format(i["id"])
        res = es.index(index=index, doc_type="doc", id=ES_ID, body=VIZ_BODY)
        print(json.dumps(res, indent=4))

    # Create and push dashboards

    # Template for dashboard body in Kibana
    DASH_BODY = {
        "type": "dashboard",
        "dashboard": {
            "title": None,
            "description": None,
            "panelsJSON": None,
            "optionsJSON": '{"darkTheme":false,\
                            "hidePanelTitles":false,"useMargins":true}',
            "version": 1,
            "kibanaSavedObjectMeta": {
                "searchSourceJSON": '{"query":{"language":"lucene", \
                                     "query":""}, \
                                     "filter":[],"highlightAll" \
                                      :true,"version":true}'
            },
        },
    }

    DASH_BODY["dashboard"]["title"] = dashboard_content["title"]
    DASH_BODY["dashboard"]["description"] = dashboard_content["desc"]
    DASH_BODY["dashboard"]["panelsJSON"] = JSONToString(
        dash_gen.generate(dashboard_content["viz"])
    )

    p(DASH_BODY)
    # Pushing dashboard to kibana
    index = ".kibana"
    ES_ID = "dashboard:{}".format(dashboard_content["id"])
    res = es.index(index=index, doc_type="doc", id=ES_ID, body=DASH_BODY)
    print(json.dumps(res, indent=4))