Init with request (#18)

* Allow initializing WebSocket with a URLRequest
* Use WebSocketKit utilities and add license info
* Update dependencies
* Update README

Author: atdrendel

.github/workflows/ci.yml

@@ -4,11 +4,11 @@ on: push
 
 jobs:
   test:
-    runs-on: macos-13
+    runs-on: macos-14
 
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - name: Select Xcode 15
-        run: sudo xcode-select -s /Applications/Xcode_15.0.app
+        run: sudo xcode-select -s /Applications/Xcode_15.4.app
       - name: Test
         run: swift test

LICENSE.md

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2024 Shareup Software Corporation
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Package.resolved

@@ -24,7 +24,7 @@ let package = Package(
         ),
         .package(
             url: "https://github.com/vapor/websocket-kit.git",
-            from: "2.14.0"
+            from: "2.15.0"
         ),
         .package(
             url: "https://github.com/apple/swift-nio.git",

Package.swift

@@ -2,11 +2,11 @@
 
 ## _(macOS, iOS, iPadOS, tvOS, and watchOS)_
 
-A concrete implementation of a WebSocket client implemented by wrapping Apple's [`NWConnection`](https://developer.apple.com/documentation/network/nwconnection).
+A concrete implementation of a WebSocket client implemented by wrapping Apple's [`URLSessionWebSocketTask`](https://developer.apple.com/documentation/foundation/urlsessionwebsockettask).
 
 The public interface of `WebSocket` is a simple struct whose public methods are exposed as closures. The reason for this design is to make it easy to inject fake WebSockets into your code for testing purposes.
 
-The actual implementation is `SystemWebSocket`, but this type is not publicly accessible. Instead, you can access it via `WebSocket.system(url:)`. `SystemWebSocket` tries its best to mirror the documented behavior of web browsers' [`WebSocket`](http://developer.mozilla.org/en-US/docs/Web/API/WebSocket). Please report any deviations as bugs.
+The actual implementation is `SystemWebSocket`, but this type is not publicly accessible. Instead, you can access it via `WebSocket.system(url:)` or `WebSocket.system(request:)`. `SystemWebSocket` tries its best to mirror the documented behavior of web browsers' [`WebSocket`](http://developer.mozilla.org/en-US/docs/Web/API/WebSocket). Please report any deviations as bugs.
 
 `WebSocket` exposes a simple API and makes heavy use of [Swift Concurrency](https://developer.apple.com/documentation/swift/swift_standard_library/concurrency).
 
@@ -49,3 +49,7 @@ try await socket.close()
 
 1. In your Terminal, navigate to the `websocket-apple` directory
 2. Run the tests using `swift test`
+
+## Notices
+
+This library includes code from [WebSocketKit](https://github.com/vapor/websocket-kit) and [SwiftNIO](https://github.com/apple/swift-nio), the use of which depends on their licenses.

README.md

@@ -2,14 +2,14 @@ import Foundation
 import Synchronized
 
 func webSocketTask(
-    for url: URL,
+    for request: URLRequest,
     options: WebSocketOptions,
     onOpen: @escaping @Sendable () async -> Void,
     onClose: @escaping @Sendable (WebSocketCloseCode, Data?) async -> Void
 ) -> URLSessionWebSocketTask {
     let session = session(for: options)
 
-    let task = session.webSocketTask(with: url)
+    let task = session.webSocketTask(with: request)
     task.maximumMessageSize = options.maximumMessageSize
 
     let delegate = session.delegate as! Delegate

Sources/WebSocket/SystemURLSession.swift

@@ -24,7 +24,8 @@ final actor SystemWebSocket: Publisher {
         return true
     } }
 
-    nonisolated let url: URL
+    nonisolated var url: URL { request.url! }
+    nonisolated let request: URLRequest
     nonisolated let options: WebSocketOptions
     nonisolated let onOpen: WebSocketOnOpen
     nonisolated let onClose: WebSocketOnClose
@@ -49,12 +50,12 @@ final actor SystemWebSocket: Publisher {
     )
 
     init(
-        url: URL,
+        request: URLRequest,
         options: WebSocketOptions = .init(),
         onOpen: @escaping WebSocketOnOpen = {},
         onClose: @escaping WebSocketOnClose = { _ in }
     ) async throws {
-        self.url = url
+        self.request = request
         self.options = options
         self.onOpen = onOpen
         self.onClose = onClose
@@ -180,7 +181,7 @@ private extension SystemWebSocket {
     func connect() throws {
         precondition(isUnopened)
         let task = webSocketTask(
-            for: url,
+            for: request,
             options: options,
             onOpen: { [weak self] in await self?.doOpen() },
             onClose: { [weak self] closeCode, reason async in

Sources/WebSocket/SystemWebSocket.swift

@@ -104,9 +104,24 @@ public extension WebSocket {
         options: WebSocketOptions = .init(),
         onOpen: @escaping WebSocketOnOpen = {},
         onClose: @escaping WebSocketOnClose = { _ in }
+    ) async throws -> Self {
+        try await system(
+            request: URLRequest(url: url),
+            options: options,
+            onOpen: onOpen,
+            onClose: onClose
+        )
+    }
+
+    /// System WebSocket implementation powered by `URLSessionWebSocketTask`.
+    static func system(
+        request: URLRequest,
+        options: WebSocketOptions = .init(),
+        onOpen: @escaping WebSocketOnOpen = {},
+        onClose: @escaping WebSocketOnClose = { _ in }
     ) async throws -> Self {
         let ws = try await SystemWebSocket(
-            url: url,
+            request: request,
             options: options,
             onOpen: onOpen,
             onClose: onClose