Consolidates index checking in NSOrderedSet

Consolidates several index checks into object(at:) and adds a few index
checks where it makes sense.

Author: philium

Foundation/NSOrderedSet.swift

@@ -41,7 +41,7 @@ open class NSOrderedSet : NSObject, NSCopying, NSMutableCopying, NSSecureCoding,
         guard aCoder.allowsKeyedCoding else {
             preconditionFailure("Unkeyed coding is unsupported.")
         }
-        for idx in 0..<self.count {
+        for idx in _indices {
             aCoder.encode(_SwiftValue.store(self.object(at: idx)), forKey:"NS.object.\(idx)")
         }
     }
@@ -67,6 +67,7 @@ open class NSOrderedSet : NSObject, NSCopying, NSMutableCopying, NSSecureCoding,
     }
 
     open func object(at idx: Int) -> Any {
+        _validateSubscript(idx)
         return _SwiftValue.fetch(nonOptional: _orderedStorage[idx])
     }
 
@@ -120,11 +121,26 @@ open class NSOrderedSet : NSObject, NSCopying, NSMutableCopying, NSSecureCoding,
         if type(of: self) === NSOrderedSet.self || type(of: self) === NSMutableOrderedSet.self {
             return _orderedStorage.map { _SwiftValue.fetch(nonOptional: $0) }
         } else {
-            return (0..<count).map { idx in
+            return _indices.map { idx in
                 return self[idx]
             }
         }
     }
+    
+    /// The range of indices that are valid for subscripting the ordered set.
+    internal var _indices: Range<Int> {
+        return 0..<count
+    }
+    
+    /// Checks that an index is valid: 0 ≤ `index` ≤ `count`.
+    internal func _validateIndex(_ index: Int, file: StaticString = #file, line: UInt = #line) {
+        precondition(index <= count && index >= 0, "\(self): Index out of bounds", file: file, line: line)
+    }
+    
+    /// Checks that an index is valid for subscripting: 0 ≤ `index` < `count`.
+    internal func _validateSubscript(_ index: Int, file: StaticString = #file, line: UInt = #line) {
+        precondition(_indices.contains(index), "\(self): Index out of bounds", file: file, line: line)
+    }
 }
 
 extension NSOrderedSet : Sequence {
@@ -148,9 +164,6 @@ extension NSOrderedSet {
     open func objects(at indexes: IndexSet) -> [Any] {
         var entries = [Any]()
         for idx in indexes {
-            guard idx < count && idx >= 0 else {
-                fatalError("\(self): Index out of bounds")
-            }
             entries.append(object(at: idx))
         }
         return entries
@@ -177,7 +190,7 @@ extension NSOrderedSet {
             return false
         }
 
-        for idx in 0..<count {
+        for idx in _indices {
             if let value1 = object(at: idx) as? AnyHashable,
                let value2 = other.object(at: idx) as? AnyHashable {
                 if value1 != value2 {
@@ -338,9 +351,7 @@ extension NSOrderedSet {
 open class NSMutableOrderedSet : NSOrderedSet {
 
     open func insert(_ object: Any, at idx: Int) {
-        guard idx <= count && idx >= 0 else {
-            fatalError("\(self): Index out of bounds")
-        }
+        _validateIndex(idx)
 
         let value = _SwiftValue.store(object)
 
@@ -353,15 +364,12 @@ open class NSMutableOrderedSet : NSOrderedSet {
     }
 
     open func removeObject(at idx: Int) {
+        _validateSubscript(idx)
         _storage.remove(_orderedStorage[idx])
         _orderedStorage.remove(at: idx)
     }
 
     open func replaceObject(at idx: Int, with obj: Any) {
-        guard idx < count && idx >= 0 else {
-            fatalError("\(self): Index out of bounds")
-        }
-        
         let value = _SwiftValue.store(obj)
         let objectToReplace = _SwiftValue.store(object(at: idx))
         _orderedStorage[idx] = value
@@ -420,10 +428,6 @@ extension NSMutableOrderedSet {
     }
 
     open func exchangeObject(at idx1: Int, withObjectAt idx2: Int) {
-        guard idx1 < count && idx1 >= 0 && idx2 < count && idx2 >= 0 else {
-            fatalError("\(self): Index out of bounds")
-        }
-
         let object1 = self.object(at: idx1)
         let object2 = self.object(at: idx2)
         _orderedStorage[idx1] = _SwiftValue.store(object2)
@@ -451,6 +455,7 @@ extension NSMutableOrderedSet {
     }
 
     open func setObject(_ obj: Any, at idx: Int) {
+        _validateIndex(idx)
         let object = _SwiftValue.store(obj)
         _storage.insert(object)
         if idx == _orderedStorage.count {