#!/usr/bin/env python3
# -*- coding:utf-8 -*-
#############################################################################
# Copyright (c) 2020 Huawei Technologies Co.,Ltd.
#
# openGauss is licensed under Mulan PSL v2.
# You can use this software according to the terms
# and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#
#          http://license.coscl.org.cn/MulanPSL2
#
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS,
# WITHOUT WARRANTIES OF ANY KIND,
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
# See the Mulan PSL v2 for more details.
# ----------------------------------------------------------------------------
# Description  : Resetreplconninfo.py is a utility to reset local replconninfo.
#############################################################################

import getopt
import os
import sys
import subprocess

sys.path.append(sys.path[0] + "/../")
from gspylib.common.GaussLog import GaussLog
from gspylib.common.Common import DefaultValue
from gspylib.common.ErrorCode import ErrorCode

########################################################################
# Global variables define
########################################################################
g_opts = None


########################################################################
class CmdOptions():
    """
    """

    def __init__(self):
        """
        """
        self.action = ""
        self.clusterUser = ""


def usage():
    """
Resetreplconninfo.py is a utility to reset replconninfos on local node.

Usage:
  python3 Resetreplconninfo.py --help
  python3 Resetreplconninfo.py -U omm -t reset

General options:
  -U                                 Cluster user.
  -t                                 reset.
  --help                             Show help information for this utility,
                                     and exit the command line mode.
    """
    print(usage.__doc__)


def parseCommandLine():
    """
    function: parse command line
    """
    try:
        opts, args = getopt.getopt(sys.argv[1:], "U:t:h", ["help"])
    except Exception as e:
        usage()
        GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50000"] % str(e))

    if len(args) > 0:
        usage()
        GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50000"]
                               % str(args[0]))

    global g_opts
    g_opts = CmdOptions()

    for (key, value) in opts:
        if key == "-h" or key == "--help":
            usage()
            sys.exit(0)
        elif key == "-t":
            g_opts.action = value
        elif key == "-U":
            g_opts.clusterUser = value


def checkParameter():
    """
    function: check parameter
    """
    if g_opts.clusterUser == "":
        GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50001"] % 'U' + ".")
    if g_opts.action == "":
        GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50001"] % 't' + ".")
    if g_opts.action != "reset":
        GaussLog.exitWithError(ErrorCode.GAUSS_500["GAUSS_50004"] % 't')


class Resetreplconninfo():
    """
    class: Resetreplconninfo
    """

    def __init__(self):
        """
        function: configure all instance on local node
        """
        # get mpprc file
        envfile = os.getenv('MPPDB_ENV_SEPARATE_PATH')
        if envfile is not None and envfile != "":
            self.userProfile = \
                envfile.replace("\\", "\\\\").replace('"', '\\"\\"')
        else:
            self.userProfile = "~/.bashrc"

    def __getStatusByOM(self):
        """
        function :Get the environment parameter.
        output : String
        """
        cmd = "source %s;gs_om -t status --detail" % self.userProfile
        (status, output) = subprocess.getstatusoutput(cmd)
        if status != 0:
            raise Exception(ErrorCode.GAUSS_514["GAUSS_51400"]
                            % cmd + " Error: \n%s" % output)
        return output.split("\n")[-1]

    def resetRepl(self):
        """
        function: reset Repl
        input : NA
        output: NA
        """
        status_list = self.__getStatusByOM().split('|')
        repl_list = ['replconninfo' + str(i) for i in
                     range(1, len(status_list))]

        localhost = DefaultValue.GetHostIpOrName()
        remote_ip_dict = {}
        for info_all in status_list:
            info = info_all.split()
            if info[1] == localhost:
                local_dndir = info[4]
            else:
                remote_ip_dict[info[2]] = info[6]
        head_cmd = "source %s;" % self.userProfile
        for repl in repl_list:
            cmd = head_cmd + 'gs_guc check -N %s -D %s -c "%s"' % \
                  (localhost, local_dndir, repl)
            status, output = subprocess.getstatusoutput(cmd)
            if status != 0:
                raise Exception(ErrorCode.GAUSS_514["GAUSS_51400"] %
                                cmd + " Error:\n%s" % output)
            # get remote ip and check iscascade
            replinfo_all = output.split('\n')[-2].strip().split("'")
            replinfo_value = replinfo_all[1].split()
            for remoteip in remote_ip_dict:
                if remoteip in replinfo_all[1]:
                    if remote_ip_dict[remoteip] == "Cascade" and \
                            "iscascade=true" not in replinfo_value:
                        replinfo_value.append("iscascade=true")
                    elif remote_ip_dict[remoteip] != "Cascade" and \
                            "iscascade=true" in replinfo_value:
                        replinfo_value.remove("iscascade=true")
                    else:
                        break
                    replinfo_all = \
                        replinfo_all[0] + "'" + " ".join(replinfo_value) + "'"
                    cmd = head_cmd + 'gs_guc reload -N %s -D %s -c "%s"' % \
                          (localhost, local_dndir, replinfo_all)
                    status, output = subprocess.getstatusoutput(cmd)
                    if status != 0:
                        raise Exception(ErrorCode.GAUSS_514["GAUSS_51400"] %
                                        cmd + " Error:\n%s" % output)
                    break


if __name__ == '__main__':
    try:
        # parse and check input parameters
        parseCommandLine()
        checkParameter()

        # reset replconninfos
        reseter = Resetreplconninfo()
        reseter.resetRepl()

    except Exception as e:
        GaussLog.exitWithError(str(e))

    sys.exit(0)