mirror of
https://github.com/home-assistant/iOS.git
synced 2026-02-17 22:02:08 -06:00
94fe0c568f
<!-- Thank you for submitting a Pull Request and helping to improve Home Assistant. Please complete the following sections to help the processing and review of your changes. Please do not delete anything from this template. --> ## Summary <!-- Provide a brief summary of the changes you have made and most importantly what they aim to achieve --> CC: @Penait1 ## Screenshots <!-- If this is a user-facing change not in the frontend, please include screenshots in light and dark mode. --> ## Link to pull request in Documentation repository <!-- Pull requests that add, change or remove functionality must have a corresponding pull request in the Companion App Documentation repository (https://github.com/home-assistant/companion.home-assistant). Please add the number of this pull request after the "#" --> Documentation: home-assistant/companion.home-assistant# ## Any other notes <!-- If there is any other information of note, like if this Pull Request is part of a bigger change, please include it here. -->
81 lines
2.7 KiB
Swift
81 lines
2.7 KiB
Swift
import AppIntents
|
|
import Foundation
|
|
import PromiseKit
|
|
import SFSafeSymbols
|
|
import Shared
|
|
|
|
@available(iOS 16.4, macOS 13.0, watchOS 9.0, *)
|
|
struct IntentSceneEntity: AppEntity {
|
|
static let typeDisplayRepresentation = TypeDisplayRepresentation(name: "Scene")
|
|
|
|
static let defaultQuery = IntentSceneAppEntityQuery()
|
|
|
|
var id: String
|
|
var entityId: String
|
|
var serverId: String
|
|
var serverName: String
|
|
var displayString: String
|
|
var iconName: String
|
|
var displayRepresentation: DisplayRepresentation {
|
|
DisplayRepresentation(title: "\(displayString)")
|
|
}
|
|
|
|
init(
|
|
id: String,
|
|
entityId: String,
|
|
serverId: String,
|
|
serverName: String,
|
|
displayString: String,
|
|
iconName: String
|
|
) {
|
|
self.id = id
|
|
self.entityId = entityId
|
|
self.serverId = serverId
|
|
self.serverName = serverName
|
|
self.displayString = displayString
|
|
self.iconName = iconName
|
|
}
|
|
}
|
|
|
|
@available(iOS 16.4, macOS 13.0, watchOS 9.0, *)
|
|
struct IntentSceneAppEntityQuery: EntityQuery, EntityStringQuery {
|
|
func entities(for identifiers: [String]) async throws -> [IntentSceneEntity] {
|
|
getSceneEntities().flatMap(\.1).filter { identifiers.contains($0.id) }
|
|
}
|
|
|
|
func entities(matching string: String) async throws -> IntentItemCollection<IntentSceneEntity> {
|
|
.init(sections: getSceneEntities(matching: string).map { (key: Server, value: [IntentSceneEntity]) in
|
|
.init(
|
|
.init(stringLiteral: key.info.name),
|
|
items: value.filter({ $0.displayString.lowercased().contains(string.lowercased()) })
|
|
)
|
|
})
|
|
}
|
|
|
|
func suggestedEntities() async throws -> IntentItemCollection<IntentSceneEntity> {
|
|
.init(sections: getSceneEntities().map { (key: Server, value: [IntentSceneEntity]) in
|
|
.init(.init(stringLiteral: key.info.name), items: value)
|
|
})
|
|
}
|
|
|
|
private func getSceneEntities(matching string: String? = nil) -> [(Server, [IntentSceneEntity])] {
|
|
var sceneEntities: [(Server, [IntentSceneEntity])] = []
|
|
let entities = ControlEntityProvider(domains: [.scene]).getEntities(matching: string)
|
|
|
|
for (server, values) in entities {
|
|
sceneEntities.append((server, values.map({ entity in
|
|
IntentSceneEntity(
|
|
id: entity.id,
|
|
entityId: entity.entityId,
|
|
serverId: entity.serverId,
|
|
serverName: server.info.name,
|
|
displayString: entity.name,
|
|
iconName: entity.icon ?? SFSymbol.moonStarsCircleFill.rawValue
|
|
)
|
|
})))
|
|
}
|
|
|
|
return sceneEntities
|
|
}
|
|
}
|