//===--- SourceLoc.swift - SourceLoc bridging utilities ------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//

public struct SourceLoc {
  /// Points into a source file.
  let locationInFile: UnsafePointer<UInt8>

  public init?(locationInFile: UnsafePointer<UInt8>?) {
    guard let locationInFile = locationInFile else {
      return nil
    }
    self.locationInFile = locationInFile
  }

  public init?(bridged: swift.SourceLoc) {
    guard bridged.isValid() else {
      return nil
    }
    self.locationInFile = bridged.getOpaquePointerValue().assumingMemoryBound(to: UInt8.self)
  }

  public var bridged: swift.SourceLoc {
    .init(llvm.SMLoc.getFromPointer(locationInFile))
  }
}

extension SourceLoc {
  public func advanced(by n: Int) -> SourceLoc {
    SourceLoc(locationInFile: locationInFile.advanced(by: n))!
  }
}

extension Optional where Wrapped == SourceLoc {
  public var bridged: swift.SourceLoc {
    self?.bridged ?? .init()
  }
}

public struct CharSourceRange {
  private let start: SourceLoc
  private let byteLength: Int

  public init(start: SourceLoc, byteLength: Int) {
    self.start = start
    self.byteLength = byteLength
  }

  public init?(bridged: BridgedCharSourceRange) {
    guard let start = SourceLoc(bridged: bridged.start) else {
      return nil
    }
    self.init(start: start, byteLength: bridged.byteLength)
  }

  public var bridged: BridgedCharSourceRange {
    .init(start: start.bridged, byteLength: byteLength)
  }
}

extension Optional where Wrapped == CharSourceRange {
  public var bridged: BridgedCharSourceRange {
    self?.bridged ?? .init(start: .init(), byteLength: 0)
  }
}