more Sendable conformances & Sendable Context

Author: weissi

Sources/TSCBasic/ByteString.swift

@@ -23,7 +23,7 @@ import Foundation
 /// strings or and by eliminating wasted space in growable arrays). For
 /// construction of byte arrays, clients should use the `WritableByteStream` class
 /// and then convert to a `ByteString` when complete.
-public struct ByteString: ExpressibleByArrayLiteral, Hashable {
+public struct ByteString: ExpressibleByArrayLiteral, Hashable, Sendable {
     /// The buffer contents.
     @usableFromInline
     internal var _bytes: [UInt8]

Sources/TSCBasic/CodableResult.swift

@@ -49,6 +49,8 @@ public struct CodableResult<Success, Failure>: Codable where Success: Codable, F
     }
 }
 
+extension CodableResult: Sendable where Success: Sendable, Failure: Sendable {}
+
 extension CodableResult where Failure == StringError {
     public init(body: () throws -> Success) {
         do {

Sources/TSCBasic/DeltaAlgorithm.swift

@@ -27,7 +27,7 @@
 /// It is not an error to provide a predicate that does not satisfy these
 /// requirements, and the algorithm will generally produce reasonable results.
 /// However, it may run substantially more tests than with a good predicate.
-public struct DeltaAlgorithm<Change: Hashable> {
+public struct DeltaAlgorithm<Change: Hashable>: Sendable {
 
     public init() {}
 

Sources/TSCBasic/DiagnosticsEngine.swift

@@ -11,19 +11,19 @@
 import Dispatch
 
 /// The payload of a diagnostic.
-public protocol DiagnosticData: CustomStringConvertible {
+public protocol DiagnosticData: CustomStringConvertible, Sendable {
 }
 
 extension DiagnosticData {
     public var localizedDescription: String { self.description }
 }
 
-public protocol DiagnosticLocation: CustomStringConvertible {
+public protocol DiagnosticLocation: CustomStringConvertible, Sendable {
 }
 
-public struct Diagnostic: CustomStringConvertible {
+public struct Diagnostic: CustomStringConvertible, Sendable {
     /// The behavior associated with this diagnostic.
-    public enum Behavior {
+    public enum Behavior: Sendable {
         /// An error which will halt the operation.
         case error
 
@@ -38,7 +38,7 @@ public struct Diagnostic: CustomStringConvertible {
         case ignored
     }
 
-    public struct Message {
+    public struct Message: Sendable {
         /// The diagnostic's behavior.
         public let behavior: Behavior
 

Sources/TSCBasic/FileSystem.swift

@@ -13,8 +13,8 @@ import Foundation
 import Dispatch
 import SystemPackage
 
-public struct FileSystemError: Error, Equatable {
-    public enum Kind: Equatable {
+public struct FileSystemError: Error, Equatable, Sendable {
+    public enum Kind: Equatable, Sendable {
         /// Access to the path is denied.
         ///
         /// This is used when an operation cannot be completed because a component of

Sources/TSCBasic/HashAlgorithms.swift

@@ -12,7 +12,7 @@
 import CryptoKit
 #endif
 
-public protocol HashAlgorithm {
+public protocol HashAlgorithm: Sendable {
 
     /// Hashes the input bytes, returning the digest.
     ///
@@ -31,7 +31,7 @@ extension HashAlgorithm {
 /// SHA-256 implementation from Secure Hash Algorithm 2 (SHA-2) set of
 /// cryptographic hash functions (FIPS PUB 180-2).
 ///  Uses CryptoKit where available
-public struct SHA256: HashAlgorithm {
+public struct SHA256: HashAlgorithm, Sendable {
     private let underlying: HashAlgorithm
 
     public init() {
@@ -197,7 +197,7 @@ struct InternalSHA256: HashAlgorithm {
 #if canImport(CryptoKit)
 @available(*, deprecated, message: "use SHA256 which abstract over platform differences")
 @available(macOS 10.15, iOS 13, tvOS 13, watchOS 6, *)
-public struct CryptoKitSHA256: HashAlgorithm {
+public struct CryptoKitSHA256: HashAlgorithm, Sendable {
     let underlying = _CryptoKitSHA256()
     public init() {}
     public func hash(_ bytes: ByteString) -> ByteString {

Sources/TSCUtility/Context.swift

@@ -10,8 +10,46 @@
 
 import Foundation
 
-/// Typealias for an any typed dictionary for arbitrary usage to store context.
-public typealias Context = [ObjectIdentifier: Any]
+public struct Context: /* until we can remove the support for 'Any' values */ @unchecked Sendable {
+    private var backing: [ObjectIdentifier: Any] = [:]
+
+    @available(*, deprecated, message: "Values should be Sendable")
+    @_disfavoredOverload
+    public init(dictionaryLiteral keyValuePairs: (ObjectIdentifier, Any)...) {
+        self.backing = Dictionary(uniqueKeysWithValues: keyValuePairs)
+    }
+
+    public init(dictionaryLiteral keyValuePairs: (ObjectIdentifier, Sendable)...) {
+        self.backing = Dictionary(uniqueKeysWithValues: keyValuePairs)
+    }
+
+    @available(*, deprecated, message: "Values should be Sendable")
+    @_disfavoredOverload
+    public subscript(key: ObjectIdentifier) -> Any? {
+        get {
+            return self.backing[key]
+        }
+        set {
+            self.backing[key] = newValue
+        }
+    }
+
+    public subscript<Value>(key: ObjectIdentifier) -> Value? where Value: Sendable {
+        get {
+            return self.backing[key] as? Value
+        }
+        set {
+            self.backing[key] = newValue
+        }
+    }
+
+}
+
+@available(*, deprecated, renamed: "init()")
+extension Context: ExpressibleByDictionaryLiteral {
+    public typealias Key = ObjectIdentifier
+    public typealias Value = Any
+}
 
 extension Context {
     /// Get the value for the given type.
@@ -22,16 +60,36 @@ extension Context {
         return value
     }
 
+    public func get<T: Sendable>(_ type: T.Type = T.self) -> T {
+        guard let value = self.getOptional(type) else {
+            fatalError("no type \(T.self) in context")
+        }
+        return value
+    }
+
     /// Get the value for the given type, if present.
+    @available(*, deprecated, message: "Values should be Sendable")
+    @_disfavoredOverload
     public func getOptional<T>(_ type: T.Type = T.self) -> T? {
         guard let value = self[ObjectIdentifier(T.self)] else {
             return nil
         }
         return value as? T
     }
 
+    /// Get the value for the given type, if present.
+    public func getOptional<T: Sendable>(_ type: T.Type = T.self) -> T? {
+        return self[ObjectIdentifier(T.self)]
+    }
+
     /// Set a context value for a type.
+    @available(*, deprecated, message: "Values should be Sendable")
+    @_disfavoredOverload
     public mutating func set<T>(_ value: T) {
         self[ObjectIdentifier(T.self)] = value
     }
+
+    public mutating func set<Value: Sendable>(_ value: Value) {
+        self[ObjectIdentifier(Value.self)] = value
+    }
 }

Sources/TSCUtility/PolymorphicCodable.swift

@@ -42,6 +42,8 @@ public struct PolymorphicCodable<T: PolymorphicCodableProtocol>: Codable {
     }
 }
 
+extension PolymorphicCodable: Sendable where T: Sendable {}
+
 @propertyWrapper
 public struct PolymorphicCodableArray<T: PolymorphicCodableProtocol>: Codable {
     public let value: [PolymorphicCodable<T>]
@@ -54,3 +56,5 @@ public struct PolymorphicCodableArray<T: PolymorphicCodableProtocol>: Codable {
         return value.map{ $0.value }
     }
 }
+
+extension PolymorphicCodableArray: Sendable where T: Sendable {}

Sources/TSCUtility/SerializedDiagnostics.swift

@@ -11,7 +11,7 @@ import Foundation
 import TSCBasic
 
 /// Represents diagnostics serialized in a .dia file by the Swift compiler or Clang.
-public struct SerializedDiagnostics {
+public struct SerializedDiagnostics: Sendable {
   public enum Error: Swift.Error {
     case badMagic
     case unexpectedTopLevelRecord
@@ -59,9 +59,9 @@ extension SerializedDiagnostics.Error: CustomNSError {
 }
 
 extension SerializedDiagnostics {
-  public struct Diagnostic {
+  public struct Diagnostic: Sendable {
 
-    public enum Level: UInt64 {
+    public enum Level: UInt64, Sendable {
       case ignored, note, warning, error, fatal, remark
     }
     /// The diagnostic message text.
@@ -149,7 +149,7 @@ extension SerializedDiagnostics {
     }
   }
 
-  public struct SourceLocation: Equatable {
+  public struct SourceLocation: Equatable, Sendable {
     /// The filename associated with the diagnostic.
     public var filename: String
     public var line: UInt64
@@ -168,7 +168,7 @@ extension SerializedDiagnostics {
     }
   }
 
-  public struct FixIt {
+  public struct FixIt: Sendable {
     /// Start location.
     public var start: SourceLocation
     /// End location.

Sources/TSCUtility/Tracing.swift

@@ -10,7 +10,7 @@
 
 import Foundation
 
-public enum TracingEventType: String, Codable {
+public enum TracingEventType: String, Codable, Sendable {
     case asyncBegin
     case asyncEnd
 }
@@ -51,7 +51,7 @@ public protocol TracingEventProtocol {
     )
 }
 
-public protocol TracingCollectionProtocol {
+public protocol TracingCollectionProtocol: Sendable {
     var events: [TracingEventProtocol] { get set }
     init(_ events: [TracingEventProtocol])
 }
@@ -62,7 +62,7 @@ extension TracingCollectionProtocol {
     }
 }
 
-public struct TracingEvent: TracingEventProtocol, Codable {
+public struct TracingEvent: TracingEventProtocol, Codable, Sendable {
     public let cat: String
     public let name: String
     public let id: String
@@ -135,8 +135,24 @@ public struct TracingEvent: TracingEventProtocol, Codable {
     #endif
 }
 
-public class TracingCollection: TracingCollectionProtocol {
-    public var events: [TracingEventProtocol] = []
+public final class TracingCollection: TracingCollectionProtocol, @unchecked Sendable {
+    private let lock = NSLock()
+    private var _events: [TracingEventProtocol] = []
+
+    public var events: [TracingEventProtocol] {
+        get {
+            return self.lock.withLock {
+                self._events
+            }
+        }
+
+        set {
+            self.lock.withLock {
+                self._events = newValue
+            }
+        }
+    }
+
     public required init(_ events: [TracingEventProtocol] = []) {
         self.events = events
     }
@@ -154,10 +170,7 @@ extension Context {
 
     public var tracing: TracingCollectionProtocol? {
         get {
-            guard let collection = self[ObjectIdentifier(TracingCollectionProtocol.self)] as? TracingCollectionProtocol else {
-                return nil
-            }
-            return collection
+            return self[ObjectIdentifier(TracingCollectionProtocol.self)]
         }
         set {
             self[ObjectIdentifier(TracingCollectionProtocol.self)] = newValue

Sources/TSCUtility/Triple.swift

@@ -20,7 +20,7 @@ import TSCBasic
 /// @see Destination.target
 /// @see https://github.com/apple/swift-llvm/blob/stable/include/llvm/ADT/Triple.h
 ///
-public struct Triple: Encodable, Equatable {
+public struct Triple: Encodable, Equatable, Sendable {
     public let tripleString: String
 
     public let arch: Arch
@@ -36,7 +36,7 @@ public struct Triple: Encodable, Equatable {
         case unknownOS(os: String)
     }
 
-    public enum Arch: String, Encodable {
+    public enum Arch: String, Encodable, Sendable {
         case x86_64
         case x86_64h
         case i686
@@ -59,12 +59,12 @@ public struct Triple: Encodable, Equatable {
         case mips64el
     }
 
-    public enum Vendor: String, Encodable {
+    public enum Vendor: String, Encodable, Sendable {
         case unknown
         case apple
     }
 
-    public enum OS: String, Encodable, CaseIterable {
+    public enum OS: String, Encodable, CaseIterable, Sendable {
         case darwin
         case macOS = "macosx"
         case linux
@@ -73,7 +73,7 @@ public struct Triple: Encodable, Equatable {
         case openbsd
     }
 
-    public enum ABI: Encodable, Equatable, RawRepresentable {
+    public enum ABI: Encodable, Equatable, RawRepresentable, Sendable {
         case unknown
         case android
         case other(name: String)