AuthenticationOperator.cpp

// Local private includes
#include "AuthenticationOperator.hpp"

// Local public includes
#include "common/TraceException.hpp"
#include "common/DynaLog.hpp"

// Standard includes
#include <any>
#include <iostream>
#include <boost/uuid/uuid.hpp>
#include <boost/uuid/uuid_generators.hpp>
#include <boost/uuid/uuid_io.hpp>

namespace SDMS {

AuthenticationOperator::AuthenticationOperator(std::any &options) {
  try {
    m_authentication_manager = std::any_cast<IAuthenticationManager *>(options);
  } catch (std::bad_cast &error) {
    std::cerr << "Caught bad any cast in AuthenticationOperator constructor."
              << error.what() << std::endl;
  }
}

void AuthenticationOperator::execute(IMessage &message) {

  if (message.exists(MessageAttribute::KEY) == 0) {
    EXCEPT(1, "'KEY' attribute not defined.");
  }
  // 🔹 Generate correlation ID for this request
  boost::uuids::random_generator generator;
  boost::uuids::uuid uuid = generator();

  LogContext log_context;
  log_context.correlation_id = boost::uuids::to_string(uuid);
  m_authentication_manager->purge();

  std::string key = std::get<std::string>(message.get(MessageAttribute::KEY));

  std::string uid = "anon";
  if (m_authentication_manager->hasKey(key, log_context)) {
    m_authentication_manager->incrementKeyAccessCounter(key, log_context);
    
    try {
      uid = m_authentication_manager->getUID(key, log_context);
    } catch (const std::exception& e) {
      // Log the exception to help diagnose authentication issues
      std::cerr << "[AuthenticationOperator] Failed to get UID for key: " 
                << key.substr(0, 8) << "... Exception: " << e.what() << std::endl;
      // Keep uid as "anon" if we fail to get the actual UID
    }
  }

  message.set(MessageAttribute::ID, uid);
}

} // namespace SDMS