// Copyright 2011 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/ssl/ssl_client_auth_cache.h"

#include "base/check.h"
#include "net/cert/x509_certificate.h"
#include "net/ssl/ssl_private_key.h"

namespace net {

SSLClientAuthCache::SSLClientAuthCache() = default;

SSLClientAuthCache::~SSLClientAuthCache() = default;

bool SSLClientAuthCache::Lookup(const HostPortPair& server,
                                scoped_refptr<X509Certificate>* certificate,
                                scoped_refptr<SSLPrivateKey>* private_key) {
  DCHECK(certificate);

  auto iter = cache_.find(server);
  if (iter == cache_.end())
    return false;

  *certificate = iter->second.first;
  *private_key = iter->second.second;
  return true;
}

void SSLClientAuthCache::Add(const HostPortPair& server,
                             scoped_refptr<X509Certificate> certificate,
                             scoped_refptr<SSLPrivateKey> private_key) {
  cache_[server] = std::pair(std::move(certificate), std::move(private_key));

  // TODO(wtc): enforce a maximum number of entries.
}

bool SSLClientAuthCache::Remove(const HostPortPair& server) {
  return cache_.erase(server);
}

void SSLClientAuthCache::Clear() {
  cache_.clear();
}

base::flat_set<HostPortPair> SSLClientAuthCache::GetCachedServers() const {
  // TODO(mattm): If views become permitted by Chromium style maybe we could
  // avoid the intermediate vector by using:
  // auto keys = std::views::keys(m);
  // base::flat_set<HostPortPair>(base::sorted_unique, keys.begin(),
  //                              keys.end());

  // Use the flat_set underlying container type (currently a std::vector), so we
  // can move the keys into the set instead of copying them.
  base::flat_set<HostPortPair>::container_type keys;
  keys.reserve(cache_.size());
  for (const auto& [key, _] : cache_) {
    keys.push_back(key);
  }
  // `cache_` is a std::map, so the keys are already sorted.
  return base::flat_set<HostPortPair>(base::sorted_unique, std::move(keys));
}

}  // namespace net