Improve alias bindings

Author: CodaFi

Sources/LLVM/Alias.swift

@@ -11,4 +11,22 @@ public struct Alias: IRGlobal {
   public func asLLVM() -> LLVMValueRef {
     return llvm
   }
+
+  /// Access the target value of this alias.
+  public var aliasee: IRValue {
+    get { return LLVMAliasGetAliasee(llvm) }
+    set { LLVMAliasSetAliasee(llvm, newValue.asLLVM()) }
+  }
+
+  /// Retrieves the previous alias in the module, if there is one.
+  public func previous() -> Alias? {
+    guard let previous = LLVMGetPreviousGlobalAlias(llvm) else { return nil }
+    return Alias(llvm: previous)
+  }
+
+  /// Retrieves the next alias in the module, if there is one.
+  public func next() -> Alias? {
+    guard let next = LLVMGetNextGlobalAlias(llvm) else { return nil }
+    return Alias(llvm: next)
+  }
 }

Sources/LLVM/Module.swift

@@ -230,6 +230,29 @@ public final class Module: CustomStringConvertible {
     }
   }
 
+  /// Retrieves the sequence of aliases that make up this module.
+  public var aliases: AnySequence<Alias> {
+    var current = firstAlias
+    return AnySequence<Alias> {
+      return AnyIterator<Alias> {
+        defer { current = current?.next() }
+        return current
+      }
+    }
+  }
+
+  /// Retrieves the first alias in this module, if there are any aliases.
+  public var firstAlias: Alias? {
+    guard let fn = LLVMGetFirstGlobalAlias(llvm) else { return nil }
+    return Alias(llvm: fn)
+  }
+
+  /// Retrieves the last alias in this module, if there are any aliases.
+  public var lastAlias: Alias? {
+    guard let fn = LLVMGetLastGlobalAlias(llvm) else { return nil }
+    return Alias(llvm: fn)
+  }
+
   /// The current debug metadata version number.
   public static var debugMetadataVersion: UInt32 {
     return LLVMDebugMetadataVersion();
@@ -307,6 +330,18 @@ extension Module {
     return Function(llvm: fn)
   }
 
+  /// Searches for and retrieves an alias with the given name in this module
+  /// if that name references an existing alias.
+  ///
+  /// - parameter name: The name of the alias to search for.
+  ///
+  /// - returns: A representation of an alias with the given
+  ///   name or nil if no such named alias exists.
+  public func alias(named name: String) -> Alias? {
+    guard let alias = LLVMGetNamedGlobalAlias(llvm, name, name.count) else { return nil }
+    return Alias(llvm: alias)
+  }
+
   /// Searches for and retrieves a comdat section with the given name in this
   /// module.  If none is found, one with that name is created and returned.
   ///