from asyncio.log import logger
import yaml
import logging
import os
import pathlib
import sys
from kubernetes import client, config
from kubernetes.client.rest import ApiException
import requests
from string import Template
import string
import random
import redis
import time
import re

file = pathlib.Path('/tmp/redis.sock')

if file.exists():
    r = redis.Redis(unix_socket_path='/tmp/redis.sock', charset="utf-8", decode_responses=True)
else:
    r = redis.Redis("127.0.0.1", charset="utf-8", decode_responses=True)

if os.environ.get("DEV"):
    r.set("log_pod_regex", ".*")
    
# create logger
logging.basicConfig(level=os.environ.get("LOGLEVEL", "INFO"))
logging.info('Starting script for KubeInvaders programming mode')

configuration = client.Configuration()
token = os.environ["TOKEN"]
configuration.api_key = {"authorization": f"Bearer {token}"}
configuration.host = sys.argv[1]

configuration.insecure_skip_tls_verify = True
configuration.verify_ssl = False

client.Configuration.set_default(configuration)
client.Configuration.set_default(configuration)

api_instance = client.CoreV1Api()
batch_api = client.BatchV1Api()
namespace = "kubeinvaders"

#for key in r.scan_iter("log:*"):
#    r.delete(key)

while True:
    webtail_pods = []
    final_pod_list = []
    if r.exists("log_pod_regex") and r.exists('logs_enabled'):
        if r.get("logs_enabled") == 1:
            logging.info("Found regex log_pod_regex in Redis. Logs from all pods should be collected")
            log_pod_regex = r.get("log_pod_regex")
            try:
                api_response = api_instance.list_pod_for_all_namespaces()
            except ApiException as e:
                logging.info(e)
            logging.info(f"Going to search pod compliant with the regex on {len(api_response.items)} pods")
            for pod in api_response.items:
                if re.search(f"{log_pod_regex}", pod.metadata.name):
                    webtail_pods.append(pod)
                    logging.info(f"Taking log of {pod.metadata.name} because it is compliant with the regex {log_pod_regex}")

    try:
        api_response = api_instance.list_namespaced_pod(namespace="kubeinvaders")
    except ApiException as e:
        logging.info(e)

    final_pod_list = webtail_pods + api_response.items

    for pod in final_pod_list:
        if ((pod.metadata.name in webtail_pods) or (pod.metadata.labels.get('approle') != None and pod.metadata.labels['approle'] == 'chaosnode' and pod.status.phase != "Pending")):
            try:
                logging.info(f"Reading logs of {pod.metadata.name} on {pod.metadata.namespace}")
                api_response = api_instance.read_namespaced_pod_log(name=pod.metadata.name, namespace=pod.metadata.namespace)
                logrow = f"<div class='row'><div style='margin-top: 2%; background-color:#400075; color: #e9f0ef; text-align: center;' class='alert' role='alert'>Pod Name: {pod.metadata.name} (Log TTL: 30sec)</div><div style='margin-top: 1.5px; background-color:#000000; color: #e9f0ef; font-size: 12px; font-family: Courier New, Courier, monospace;' class='alert' role='alert'>{api_response}</div></div>"
                r.set(f"log:{pod.metadata.name}", logrow)
                r.expire(f"log:{pod.metadata.name}", 30)
                logging.info(f"Phase of {pod.metadata.name} is {pod.status.phase}")   
                if pod.status.phase == "Succeeded" and pod.metadata.labels['approle'] == 'chaosnode':
                    try:
                        api_response = api_instance.delete_namespaced_pod(pod.metadata.name, namespace = pod.metadata.namespace)
                        logging.info(f"Deleted pod {pod.metadata.name}")
                    except ApiException as e:
                        logging.info(e)
            except ApiException as e:
                logging.info(e)
    file = pathlib.Path('/var/www/html')
    if file.exists():
        log_html_file = pathlib.Path('/var/www/html/chaoslogs.html')
        if log_html_file.exists():
            with open("/var/www/html/chaoslogs.html", "w") as myfile:
                myfile.write('')
        for key in r.scan_iter("log:*"):
            with open("/var/www/html/chaoslogs.html", "a") as myfile:
                myfile.write(str(r.get(key)))
    time.sleep(1)