Allow passing data when subscribing for state updates

Source/Caches/HACacheKeyStates.swift

@@ -2,12 +2,12 @@ import Foundation
 
 /// Key for the cache
 internal struct HACacheKeyStates: HACacheKey {
-    static func create(connection: HAConnection) -> HACache<HACachedStates> {
+    static func create(connection: HAConnection, data: [String: Any]) -> HACache<HACachedStates> {
         .init(
             connection: connection,
             subscribe:
             .init(
-                subscription: .subscribeEntities(),
+                subscription: .subscribeEntities(data: data),
                 transform: { info in
                     .replace(processUpdates(
                         info: info,

Source/Caches/HACachedStates.swift

@@ -1,6 +1,10 @@
 public extension HACachesContainer {
     /// Cache of entity states, see `HACachedStates` for values.
-    var states: HACache<HACachedStates> { self[HACacheKeyStates.self] }
+    /// - Parameter data: The data passed to connection request
+    /// - Returns: The cache object with states
+    func states(_ data: [String: Any] = [:]) -> HACache<HACachedStates> {
+        self[HACacheKeyStates.self, data]
+    }
 }
 
 /// Cached version of all entity states

Source/Caches/HACachedUser.swift

@@ -5,7 +5,7 @@ public extension HACachesContainer {
 
 /// Key for the cache
 private struct HACacheKeyCurrentUser: HACacheKey {
-    static func create(connection: HAConnection) -> HACache<HAResponseCurrentUser> {
+    static func create(connection: HAConnection, data: [String: Any]) -> HACache<HAResponseCurrentUser> {
         .init(
             connection: connection,
             populate: .init(request: .currentUser(), transform: \.incoming),

Source/Caches/HACachesContainer.swift

@@ -7,8 +7,12 @@ public protocol HACacheKey {
     /// This is called exactly once per connection per cache key.
     ///
     /// - Parameter connection: The connection to create on
+    /// - Parameter data: The data passed to connection request
     /// - Returns: The cache you want to associate with the key
-    static func create(connection: HAConnection) -> HACache<Value>
+    static func create(
+        connection: HAConnection,
+        data: [String: Any]
+    ) -> HACache<Value>
 }
 
 /// Container for caches
@@ -56,15 +60,15 @@ public class HACachesContainer {
     /// - SeeAlso: `HACachesContainer` class description for how to use keys to retrieve caches.
     /// - Subscript: The key to look up
     /// - Returns: Either the existing cache for the key, or a new one created on-the-fly if none was available
-    public subscript<KeyType: HACacheKey>(_ key: KeyType.Type) -> HACache<KeyType.Value> {
+    public subscript<KeyType: HACacheKey>(_ key: KeyType.Type, data: [String: Any] = [:]) -> HACache<KeyType.Value> {
         // ObjectIdentifier is globally unique per class _or_ meta type, and we're using meta type here
         let key = ObjectIdentifier(KeyType.self)
 
         if let value = values[key] as? HACache<KeyType.Value> {
             return value
         }
 
-        let value = KeyType.create(connection: connection)
+        let value = KeyType.create(connection: connection, data: data)
         values[key] = value
         return value
     }

Source/Convenience/States.swift

@@ -11,9 +11,10 @@ public extension HATypedSubscription {
     }
 
     /// Listen for compressed state changes of all entities
+    /// - Parameter data: The data passed to connection request
     /// - Returns: A typed subscriptions that can be sent via `HAConnection`
-    static func subscribeEntities() -> HATypedSubscription<HACompressedStatesUpdates> {
-        .init(request: .init(type: .subscribeEntities, data: [:]))
+    static func subscribeEntities(data: [String: Any]) -> HATypedSubscription<HACompressedStatesUpdates> {
+        .init(request: .init(type: .subscribeEntities, data: data))
     }
 }
 

Tests/HACachedStates.test.swift

@@ -19,7 +19,7 @@ internal class HACachedStatesTests: XCTestCase {
     }
 
     func testPopulateAndSubscribeInfo() throws {
-        let cache = container.states
+        let cache = container.states()
         XCTAssertNil(cache.populateInfo)
         XCTAssertNil(cache.subscribeInfo)
         let subscribeOnlyInfo = try XCTUnwrap(cache.subscribeOnlyInfo)

Tests/HACachesContainer.test.swift

@@ -41,7 +41,7 @@ internal class HACachesContainerTests: XCTestCase {
 }
 
 private struct Key1: HACacheKey {
-    static func create(connection: HAConnection) -> HACache<Value1> {
+    static func create(connection: HAConnection, data: [String: Any]) -> HACache<Value1> {
         .init(
             connection: connection,
             populate: HACachePopulateInfo<Value1>(
@@ -61,7 +61,7 @@ private struct Value1: HADataDecodable {
 }
 
 private struct Key2: HACacheKey {
-    static func create(connection: HAConnection) -> HACache<Value1> {
+    static func create(connection: HAConnection, data: [String: Any]) -> HACache<Value1> {
         .init(
             connection: connection,
             populate: HACachePopulateInfo<Value1>(

Tests/States.test.swift

@@ -141,7 +141,7 @@ internal class StatesTests: XCTestCase {
     }
 
     func testSubscribeEntitiesRequest() throws {
-        let request = HATypedSubscription<HACompressedStatesUpdates>.subscribeEntities()
+        let request = HATypedSubscription<HACompressedStatesUpdates>.subscribeEntities(data: [:])
         XCTAssertEqual(request.request.type, .subscribeEntities)
     }
 }