########################################################################
# File :   IdProviderFactory.py
# Author : A.T.
########################################################################

"""  The Identity Provider Factory instantiates IdProvider objects
     according to their configuration
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from DIRAC import S_OK, S_ERROR, gLogger
from DIRAC.Core.Utilities.ObjectLoader import ObjectLoader
from DIRAC.ConfigurationSystem.Client.Helpers.Resources import getInfoAboutProviders

__RCSID__ = "45ddde113 (2021-10-01 11:19:55 +0200) Chris Burr <christopher.burr@cern.ch>"


class IdProviderFactory(object):

    #############################################################################
    def __init__(self):
        """Standard constructor"""
        self.log = gLogger.getSubLogger("IdProviderFactory")

    #############################################################################
    def getIdProvider(self, idProvider):
        """This method returns a IdProvider instance corresponding to the supplied name.

        :param str idProvider: the name of the Identity Provider

        :return: S_OK(IdProvider)/S_ERROR()
        """
        result = getInfoAboutProviders(of="Id", providerName=idProvider, option="all", section="all")
        if not result["OK"]:
            return result
        pDict = result["Value"]
        pDict["ProviderName"] = idProvider
        pType = pDict["ProviderType"]

        self.log.verbose("Creating IdProvider", "of %s type with the name %s" % (pType, idProvider))
        subClassName = "%sIdProvider" % (pType)

        result = ObjectLoader().loadObject("Resources.IdProvider.%s" % subClassName)
        if not result["OK"]:
            self.log.error("Failed to load object", "%s: %s" % (subClassName, result["Message"]))
            return result

        pClass = result["Value"]
        try:
            provider = pClass()
            provider.setParameters(pDict)
        except Exception as x:
            msg = "IdProviderFactory could not instantiate %s object: %s" % (subClassName, str(x))
            self.log.exception()
            self.log.error(msg)
            return S_ERROR(msg)

        return S_OK(provider)