Authentication.h

 
#ifndef PULSAR_AUTHENTICATION_H_
#define PULSAR_AUTHENTICATION_H_
 
#include <vector>
#include <string>
#include <map>
#include <memory>
#include <pulsar/Result.h>
#include <functional>
 
#pragma GCC visibility push(default)
 
namespace pulsar {
 
class ClientConfiguration;
class Authentication;
 
class AuthenticationDataProvider {
   public:
    virtual ~AuthenticationDataProvider();
    virtual bool hasDataForTls();
    virtual std::string getTlsCertificates();
    virtual std::string getTlsPrivateKey();
    virtual bool hasDataForHttp();
    virtual std::string getHttpAuthType();
    virtual std::string getHttpHeaders();
    virtual bool hasDataFromCommand();
    virtual std::string getCommandData();
 
   protected:
    AuthenticationDataProvider();
};
 
typedef std::shared_ptr<AuthenticationDataProvider> AuthenticationDataPtr;
typedef std::shared_ptr<Authentication> AuthenticationPtr;
typedef std::map<std::string, std::string> ParamMap;
 
class Authentication {
   public:
    virtual ~Authentication();
    virtual const std::string getAuthMethodName() const = 0;
    virtual Result getAuthData(AuthenticationDataPtr& authDataContent) const {
        authDataContent = authData_;
        return ResultOk;
    }
    static ParamMap parseDefaultFormatAuthParams(const std::string& authParamsString);
 
   protected:
    Authentication();
    AuthenticationDataPtr authData_;
    friend class ClientConfiguration;
};
 
class AuthFactory {
   public:
    static AuthenticationPtr Disabled();
 
    static AuthenticationPtr create(const std::string& pluginNameOrDynamicLibPath);
    static AuthenticationPtr create(const std::string& pluginNameOrDynamicLibPath,
                                    const std::string& authParamsString);
    static AuthenticationPtr create(const std::string& pluginNameOrDynamicLibPath, ParamMap& params);
 
   protected:
    static bool isShutdownHookRegistered_;
    static std::vector<void*> loadedLibrariesHandles_;
    static void release_handles();
};
 
class AuthTls : public Authentication {
   public:
    AuthTls(AuthenticationDataPtr&);
    ~AuthTls();
    static AuthenticationPtr create(ParamMap& params);
    static AuthenticationPtr create(const std::string& authParamsString);
    static AuthenticationPtr create(const std::string& certificatePath, const std::string& privateKeyPath);
    const std::string getAuthMethodName() const;
    Result getAuthData(AuthenticationDataPtr& authDataTls) const;
 
   private:
    AuthenticationDataPtr authDataTls_;
};
 
typedef std::function<std::string()> TokenSupplier;
 
class AuthToken : public Authentication {
   public:
    AuthToken(AuthenticationDataPtr&);
    ~AuthToken();
 
    static AuthenticationPtr create(ParamMap& params);
 
    static AuthenticationPtr create(const std::string& authParamsString);
 
    static AuthenticationPtr createWithToken(const std::string& token);
 
    static AuthenticationPtr create(const TokenSupplier& tokenSupplier);
 
    const std::string getAuthMethodName() const;
    Result getAuthData(AuthenticationDataPtr& authDataToken) const;
 
   private:
    AuthenticationDataPtr authDataToken_;
};
 
class AuthAthenz : public Authentication {
   public:
    AuthAthenz(AuthenticationDataPtr&);
    ~AuthAthenz();
    static AuthenticationPtr create(ParamMap& params);
    static AuthenticationPtr create(const std::string& authParamsString);
    const std::string getAuthMethodName() const;
    Result getAuthData(AuthenticationDataPtr& authDataAthenz) const;
 
   private:
    AuthenticationDataPtr authDataAthenz_;
};
 
}  // namespace pulsar
 
#pragma GCC visibility pop
 
#endif /* PULSAR_AUTHENTICATION_H_ */