//===--- ASTBridging.cpp - AST bridging functions -------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 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 "swift/AST/ASTBridging.h"

#include "swift/AST/DiagnosticEngine.h"
#include "swift/Basic/BridgingUtils.h"

using namespace swift;

namespace {
/// BridgedDiagnosticEngine -> DiagnosticEngine *.
DiagnosticEngine *getDiagnosticEngine(const BridgedDiagnosticEngine &bridged) {
  return static_cast<DiagnosticEngine *>(bridged.object);
}

/// BridgedDiagnosticArgument -> DiagnosticArgument
DiagnosticArgument
getDiagnosticArgument(const BridgedDiagnosticArgument &bridged) {
  switch (bridged.kind) {
  case BridgedDiagnosticArgumentKind_StringRef:
    return {getStringRef(bridged.value.stringRefValue)};
  case BridgedDiagnosticArgumentKind_Int:
    return {(int)bridged.value.intValue};
  }
  llvm_unreachable("unhandled enum value");
}

} // namespace

void DiagnosticEngine_diagnose(
    BridgedDiagnosticEngine bridgedEngine, SourceLoc loc,
    BridgedDiagID bridgedDiagID,
    BridgedArrayRef /*BridgedDiagnosticArgument*/ bridgedArguments,
    BridgedCharSourceRange bridgedHighlight,
    BridgedArrayRef /*BridgedDiagnosticFixIt*/ bridgedFixIts) {
  auto *D = getDiagnosticEngine(bridgedEngine);

  auto diagID = static_cast<DiagID>(bridgedDiagID);
  SmallVector<DiagnosticArgument, 2> arguments;
  for (auto bridgedArg :
       getArrayRef<BridgedDiagnosticArgument>(bridgedArguments)) {
    arguments.push_back(getDiagnosticArgument(bridgedArg));
  }
  auto inflight = D->diagnose(loc, diagID, arguments);

  // Add highlight.
  auto highlight = getCharSourceRange(bridgedHighlight);
  if (highlight.isValid()) {
    inflight.highlightChars(highlight.getStart(), highlight.getEnd());
  }

  // Add fix-its.
  for (auto bridgedFixIt : getArrayRef<BridgedDiagnosticFixIt>(bridgedFixIts)) {
    auto range = CharSourceRange(bridgedFixIt.start,
                                 bridgedFixIt.byteLength);
    auto text = getStringRef(bridgedFixIt.text);
    inflight.fixItReplaceChars(range.getStart(), range.getEnd(), text);
  }
}

bool DiagnosticEngine_hadAnyError(BridgedDiagnosticEngine bridgedEngine) {
  auto *D = getDiagnosticEngine(bridgedEngine);
  return D->hadAnyError();
}