Bridge Argument.varDecl

Author: atrick

SwiftCompilerSources/Sources/SIL/Argument.swift

@@ -30,7 +30,11 @@ public class Argument : Value, Hashable {
   }
 
   public var isReborrow: Bool { bridged.isReborrow() }
-  
+
+  public var varDecl: VarDecl? {
+    VarDecl(bridged: bridged.getVarDecl())
+  }
+
   public static func ==(lhs: Argument, rhs: Argument) -> Bool {
     lhs === rhs
   }

include/swift/SIL/SILBridging.h

@@ -899,6 +899,7 @@ struct BridgedArgument {
   BRIDGED_INLINE bool isSelf() const;
   BRIDGED_INLINE bool isReborrow() const;
   BRIDGED_INLINE bool hasResultDependsOn() const;
+  SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedNullableVarDecl getVarDecl() const;
 };
 
 struct OptionalBridgedArgument {

include/swift/SIL/SILBridgingImpl.h

@@ -471,6 +471,12 @@ bool BridgedArgument::hasResultDependsOn() const {
 
 bool BridgedArgument::isReborrow() const { return getArgument()->isReborrow(); }
 
+BridgedNullableVarDecl BridgedArgument::getVarDecl() const {
+  // TODO: why does AST bridging force a non-const pointer?
+  return {llvm::dyn_cast_or_null<swift::VarDecl>(
+      const_cast<swift::ValueDecl*>(getArgument()->getDecl()))};
+}
+
 //===----------------------------------------------------------------------===//
 //                            BridgedSubstitutionMap
 //===----------------------------------------------------------------------===//