import json
import requests
import os
import os.path
import sys
import time
import psycopg2

class KubeconfigGetter():

    def _get_db_connection(self):
        conn = None
        try:
            conn = psycopg2.connect(
                host='xxxxx',
                port='5432',
                database="xxxxx",
                user="xxxxx",
                password='xxxxx')
        except Exception as e:
            print(e)
        return conn

    def get_cluster_name(self, fuze_spm_site_id):
        cluster_name = ''
        try:
            conn = self._get_db_connection()

            # create a cursor
            cur = conn.cursor()

            # execute a statement
            #print('select cluster_name, fuze_spm_site_id, fuze_spm_site_name, oam_vip_address from caas_cluster join caas_location on caas_cluster.location_id=caas_location.id where fuze_spm_site_id = ' + fuze_spm_site_id)
            sql = 'select cluster_name from caas_cluster join caas_location on caas_cluster.location_id=caas_location.id where fuze_spm_site_id=' + '\'' + fuze_spm_site_id + '\''
            cur.execute(sql)

            # display the PostgreSQL database server version
            cluster_name = cur.fetchone()[0]
            #print(cluster_name)

            # close the communication with the PostgreSQL
            cur.close()
        except (Exception, psycopg2.DatabaseError) as error:
            print(error)
            if conn is not None:
                conn.close()
                print('Database connection closed.')
        return str(cluster_name)

    def get_kubeconfig(self, site_name):
        #url = "http://192.168.31.20/orchestration/remote-regions/" + site_name + "/kubeconfig/vendor/samsung"
        url = "https://middleware.vcpfe.vzwops.com/orchestration/remote-regions/" + site_name + "/connection-details?team=samsung"
        url1 = "https://middleware.vcpfe.vzwops.com/orchestration/remote-regions/" + site_name + "/connection-details"
        resp = requests.get(url, verify=False)
        #print(resp.json())
        transaction_id = resp.json()['transaction_id']
        #print(transaction_id)
        url2 = url1 + "/" + transaction_id
        #print(url2)
        time.sleep(10)
        resp1 = requests.get(url2, verify=False)
        #print(resp1.json())
        kubeconfig = resp1.json()['remote_region_setup'][0]["kube_config"]
        #print(kubeconfig)
        if not os.path.exists("samsung-kubeconfigs"):
            os.makedirs("samsung-kubeconfigs")
        fp = open("samsung-kubeconfigs/" + site_name + ".conf", "w")
        fp.write(kubeconfig)
        fp.close()

if __name__ == '__main__':

    if len(sys.argv) < 2:
        print("python get-ss-kubeconfig.py <site-list-file>")
        print("  Make sure to populate database creds in this file.")
        print("  Use it from a server from which database is accessible (e.g.: app1 or app2)")
        exit(0)

    site_list_file = sys.argv[1]
    fp = open(site_list_file, "r")
    lines = fp.read()
    kubeconfigGetter = KubeconfigGetter()
    for fuze_site_id in lines.split("\n"):
        fuze_site_id = fuze_site_id.lstrip().rstrip()
        if fuze_site_id != '':
            #print(fuze_site_id)
            cluster_name = kubeconfigGetter.get_cluster_name(fuze_site_id)
            kubeconfigGetter.get_kubeconfig(cluster_name) 
    #fuze_site_id = sys.argv[1]
    #print(lines)
    #print(fuze_site_id + " " + cluster_name)
    print("Done")