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")
    static let defaultQuery = CalendarEntityQuery()

    var id: UUID
    var title: String

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

struct CalendarEntityQuery: 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) }
    }
}

// MARK: - Schematized Event Entity

@AppEntity
struct EventEntity: IndexedEntity {
    static let typeDisplayRepresentation = TypeDisplayRepresentation(name: "Event")
    static let defaultQuery = EventEntityQuery()

    var id: UUID
    var title: String
    var startDate: Date
    var endDate: Date
    var calendar: CalendarEntity?
    var notes: String?
    var location: String?

    var displayRepresentation: DisplayRepresentation {
        DisplayRepresentation(
            title: "\(title)",
            subtitle: "\(startDate.formatted(date: .abbreviated, time: .shortened))",
            image: .init(systemName: "calendar.badge.clock")
        )
    }
}

struct EventEntityQuery: EntityQuery {
    @Dependency var calendarManager: CalendarManager

    func entities(for identifiers: [UUID]) async throws -> [EventEntity] {
        calendarManager.events
            .filter { identifiers.contains($0.id) }
            .map { event in
                EventEntity(
                    id: event.id,
                    title: event.title,
                    startDate: event.startDate,
                    endDate: event.endDate,
                    notes: event.notes,
                    location: event.location
                )
            }
    }
}

// MARK: - Donating events to the Spotlight semantic index

@MainActor
final class CalendarManager: ObservableObject {
    static let shared = CalendarManager()
    var events: [EventModel] = []
    var calendars: [CalendarModel] = []

    private let searchableIndex = CSSearchableIndex(name: "com.example.cometcal")

    func createEvent(_ model: EventModel) async throws {
        events.append(model)
        let entity = EventEntity(
            id: model.id,
            title: model.title,
            startDate: model.startDate,
            endDate: model.endDate,
            notes: model.notes,
            location: model.location
        )
        try await searchableIndex.indexAppEntities([entity])
    }

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

// MARK: - Minimal stub models

struct EventModel {
    var id: UUID = UUID()
    var title: String
    var startDate: Date
    var endDate: Date
    var notes: String?
    var location: String?
}

struct CalendarModel {
    var id: UUID = UUID()
    var title: String
}