smartbotic/lib/credentials/credential_client.hpp

#pragma once

#include <memory>
#include <string>
#include <vector>
#include <grpcpp/grpcpp.h>
#include "proto/credentials.grpc.pb.h"
#include "credential_types.hpp"
#include "common/error.hpp"

namespace smartbotic::credentials {

// Configuration for credential client
struct CredentialClientConfig {
    std::string address = "localhost:9013";  // Credential service address
    int timeout_ms = 5000;
};

// gRPC client for credential service
class CredentialClient {
public:
    explicit CredentialClient(const CredentialClientConfig& config);
    ~CredentialClient() = default;

    // Get HTTP authentication header for a credential
    common::Result<HttpAuth> getHttpAuth(const std::string& credential_id,
                                         const std::string& workflow_id = "");

    // Get IMAP credentials
    common::Result<ImapData> getImapCredentials(const std::string& credential_id,
                                                 const std::string& workflow_id = "");

    // The mTLS identity a workflow presents. The private key comes back over
    // this call and is held in memory for the length of the run, never written
    // anywhere.
    common::Result<ClientCertificateData> getClientCertificate(const std::string& credential_id,
                                                               const std::string& workflow_id = "");

    // Also satisfied by an imap credential for the same account.
    common::Result<SmtpData> getSmtpCredentials(const std::string& credential_id,
                                                const std::string& workflow_id = "");

    // Get MySQL credentials
    common::Result<MysqlData> getMysqlCredentials(const std::string& credential_id,
                                                   const std::string& workflow_id = "");

    // Get PostgreSQL credentials
    common::Result<PostgresqlData> getPostgresqlCredentials(const std::string& credential_id,
                                                             const std::string& workflow_id = "");

    // List available credentials
    common::Result<std::vector<CredentialInfo>> listCredentials(const std::string& workflow_id = "");

    // Check if connected
    bool isConnected() const;

private:
    CredentialClientConfig config_;
    std::shared_ptr<grpc::Channel> channel_;
    std::unique_ptr<proto::CredentialService::Stub> stub_;
};

} // namespace smartbotic::credentials