App Schemas for Intelligent Siri Experiences

import AppIntents
import Foundation

// MARK: - Contact Entity

struct UnicornContact: AppEntity {
    static let typeDisplayRepresentation = TypeDisplayRepresentation(name: "Unicorn Contact")
    static let defaultQuery = UnicornContactQuery()

    let id: String
    let name: String

    var displayRepresentation: DisplayRepresentation {
        DisplayRepresentation(title: "\(name)")
    }
}

struct UnicornContactQuery: EntityStringQuery {
    func entities(for identifiers: [String]) async throws -> [UnicornContact] {
        // Return contacts matching the given IDs from your data store
        identifiers.map { UnicornContact(id: $0, name: $0) }
    }

    func entities(matching string: String) async throws -> [UnicornContact] {
        let all = [UnicornContact(id: "glow", name: "Glow"),
                   UnicornContact(id: "bubbles", name: "Bubbles"),
                   UnicornContact(id: "flare", name: "Flare")]
        return all.filter { $0.name.localizedCaseInsensitiveContains(string) }
    }
}

// MARK: - Message Entity

struct UnicornMessage: AppEntity {
    static let typeDisplayRepresentation = TypeDisplayRepresentation(name: "Message")
    static let defaultQuery = UnicornMessageQuery()

    let id: String
    let body: String
    let recipientID: String

    var displayRepresentation: DisplayRepresentation {
        DisplayRepresentation(title: "\(body)")
    }
}

struct UnicornMessageQuery: EntityStringQuery {
    func entities(for identifiers: [String]) async throws -> [UnicornMessage] {
        identifiers.map { UnicornMessage(id: $0, body: "", recipientID: "") }
    }
    func entities(matching string: String) async throws -> [UnicornMessage] { [] }
}

// MARK: - Send Message Intent (App Schema: messages domain)

struct SendUnicornMessageIntent: AppIntent {
    static let title: LocalizedStringResource = "Send Unicorn Message"
    static let description = IntentDescription("Send a message to a unicorn contact via UnicornChat.")

    // Schema-defined parameters Siri knows how to resolve
    @Parameter(title: "Recipient")
    var recipient: UnicornContact

    @Parameter(title: "Message")
    var messageBody: String

    func perform() async throws -> some ReturnsValue<UnicornMessage> {
        // Invoke your app's actual send logic here
        let sentMessage = UnicornMessage(
            id: UUID().uuidString,
            body: messageBody,
            recipientID: recipient.id
        )
        // e.g. UnicornChatStore.shared.send(sentMessage)
        return .result(value: sentMessage)
    }
}