########################################################################
# 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__ = "cd15833a2 (2021-04-08 13:55:16 +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)