//===--- SwiftToClangInteropContext.cpp - Interop context -------*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2022 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

#include "SwiftToClangInteropContext.h"
#include "swift/AST/Decl.h"
#include "swift/IRGen/IRABIDetailsProvider.h"

using namespace swift;

SwiftToClangInteropContext::SwiftToClangInteropContext(
    ModuleDecl &mod, const IRGenOptions &irGenOpts)
    : mod(mod), irGenOpts(irGenOpts) {}

SwiftToClangInteropContext::~SwiftToClangInteropContext() = default;

IRABIDetailsProvider &SwiftToClangInteropContext::getIrABIDetails() {
  if (!irABIDetails)
    irABIDetails = std::make_unique<IRABIDetailsProvider>(mod, irGenOpts);
  return *irABIDetails;
}

void SwiftToClangInteropContext::runIfStubForDeclNotEmitted(
    StringRef stubName, llvm::function_ref<void(void)> function) {
  auto result = emittedStubs.insert(stubName);
  if (result.second)
    function();
}

void SwiftToClangInteropContext::recordExtensions(
    const NominalTypeDecl *typeDecl, const ExtensionDecl *ext) {
  auto it = extensions.insert(
      std::make_pair(typeDecl, std::vector<const ExtensionDecl *>()));
  it.first->second.push_back(ext);
}

llvm::ArrayRef<const ExtensionDecl *>
SwiftToClangInteropContext::getExtensionsForNominalType(
    const NominalTypeDecl *typeDecl) const {
  auto exts = extensions.find(typeDecl);
  if (exts != extensions.end())
    return exts->getSecond();
  return {};
}