"""
:mod: FileCatalogProxyHandler

.. module: FileCatalogProxyHandler
  :synopsis: This is a service which represents a DISET proxy to the File
  Catalog

"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

# imports
import os
import six

# from DIRAC
from DIRAC import gLogger, S_OK, S_ERROR
from DIRAC.Core.DISET.RequestHandler import RequestHandler
from DIRAC.Resources.Catalog.FileCatalogFactory import FileCatalogFactory
from DIRAC.Core.Utilities.Subprocess import pythonCall
from DIRAC.FrameworkSystem.Client.ProxyManagerClient import gProxyManager

__RCSID__ = "c59810311 (2021-10-01 10:44:25 +0200) Chris Burr <christopher.burr@cern.ch>"


def initializeFileCatalogProxyHandler(serviceInfo):
    """service initalisation"""
    return S_OK()


class FileCatalogProxyHandler(RequestHandler):
    """
    .. class:: FileCatalogProxyHandler
    """

    types_callProxyMethod = [list(six.string_types), list(six.string_types), [tuple, list], dict]

    def export_callProxyMethod(self, fcName, methodName, args, kargs):
        """A generic method to call methods of the Storage Element."""
        res = pythonCall(120, self.__proxyWrapper, fcName, methodName, args, kargs)
        if res["OK"]:
            return res["Value"]
        else:
            return res

    def __proxyWrapper(self, fcName, methodName, args, kwargs):
        """The wrapper will obtain the client proxy and set it up in the environment.
            The required functionality is then executed and returned to the client.

        :param self: self reference
        :param str name: fcn name
        :param tuple args: fcn args
        :param dict kwargs: fcn keyword args
        """
        result = self.__prepareSecurityDetails()
        if not result["OK"]:
            return result
        proxyLocation = result["Value"]
        try:
            result = FileCatalogFactory().createCatalog(fcName)
            if result["OK"]:
                fileCatalog = result["Value"]
                method = getattr(fileCatalog, methodName)
            else:
                return result
        except AttributeError as error:
            errStr = "%s proxy: no method named %s" % (fcName, methodName)
            gLogger.exception(errStr, methodName, error)
            return S_ERROR(errStr)
        try:
            result = method(*args, **kwargs)
            if os.path.exists(proxyLocation):
                os.remove(proxyLocation)
            return result
        except Exception as error:
            if os.path.exists(proxyLocation):
                os.remove(proxyLocation)
            errStr = "%s proxy: Exception while performing %s" % (fcName, methodName)
            gLogger.exception(errStr, error)
            return S_ERROR(errStr)

    def __prepareSecurityDetails(self, vomsFlag=True):
        """Obtains the connection details for the client"""
        try:
            credDict = self.getRemoteCredentials()
            clientDN = credDict["DN"]
            clientUsername = credDict["username"]
            clientGroup = credDict["group"]
            gLogger.debug("Getting proxy for %s@%s (%s)" % (clientUsername, clientGroup, clientDN))
            if vomsFlag:
                result = gProxyManager.downloadVOMSProxyToFile(clientDN, clientGroup)
            else:
                result = gProxyManager.downloadProxyToFile(clientDN, clientGroup)
            if not result["OK"]:
                return result
            gLogger.debug("Updating environment.")
            os.environ["X509_USER_PROXY"] = result["Value"]
            return result
        except Exception as error:
            exStr = "__getConnectionDetails: Failed to get client connection details."
            gLogger.exception(exStr, "", error)
            return S_ERROR(exStr)