App Schemas for Siri Integration (Calendar Domain)

import AppIntents
import CoreSpotlight
import SwiftData
import Foundation

// MARK: - Schematized Calendar Entity

@AppEntity
struct CalendarEntity: IndexedEntity {
    static let typeDisplayRepresentation = TypeDisplayRepresentation(name: "Calendar")

    var id: UUID

    @Property(title: "Title")
    var title: String

    var displayRepresentation: DisplayRepresentation {
        DisplayRepresentation(
            title: "\(title)",
            image: .init(systemName: "calendar")
        )
    }

    struct DefaultQuery: EntityQuery, EnumerableEntityQuery {
        @Dependency var calendarManager: CalendarManager

        func entities(for identifiers: [UUID]) async throws -> [CalendarEntity] {
            calendarManager.calendars
                .filter { identifiers.contains($0.id) }
                .map { CalendarEntity(id: $0.id, title: $0.title) }
        }

        func allEntities() async throws -> [CalendarEntity] {
            calendarManager.calendars
                .map { CalendarEntity(id: $0.id, title: $0.title) }
        }
    }

    init(id: UUID, title: String) {
        self.id = id
        self.title = title
    }
}

// MARK: - Donating Entities to Spotlight

@MainActor
class CalendarManager {
    var calendars: [CalendarModel] = []
    private let searchableIndex = CSSearchableIndex(name: "com.example.cometcal")

    func createCalendar(title: String) async throws -> CalendarModel {
        let model = CalendarModel(id: UUID(), title: title)
        calendars.append(model)
        let entity = CalendarEntity(id: model.id, title: model.title)
        try await searchableIndex.indexAppEntities([entity])
        return model
    }

    func updateCalendar(_ model: CalendarModel) async throws {
        if let idx = calendars.firstIndex(where: { $0.id == model.id }) {
            calendars[idx] = model
        }
        let entity = CalendarEntity(id: model.id, title: model.title)
        try await searchableIndex.indexAppEntities([entity])
    }

    func deleteCalendar(id: UUID) async throws {
        calendars.removeAll { $0.id == id }
        try await searchableIndex.deleteAppEntities(identifiedBy: [id], ofType: CalendarEntity.self)
    }
}

// MARK: - Lightweight Data Model

struct CalendarModel: Identifiable {
    var id: UUID
    var title: String
}