PureXML/Sources/Schema/ParticleRestriction.swift

extension PureXML.Schema {
    /// XSD 1.0 "Particle Valid (Restriction)" for the common cases: a complex
    /// type derived by restriction must have a content model that accepts a
    /// subset of its base's. Covers element-vs-element name matching,
    /// element-vs-wildcard admission, wildcard narrowing, occurrence-range
    /// subsumption, and the group recursions (sequence/all order-preserving with
    /// emptiable skips, choice mapping, sequence-into-choice, sequence-into-all).
    /// The spec's effective-total-range arithmetic for MapAndSum is approximated
    /// by per-particle checks, a documented simplification.
    enum ParticleRestriction {
        /// The reason a restriction's content is not a valid subset of its
        /// base's, or nil when it is.
        static func violation(restricted: ContentType, base: ContentType) -> String? {
            switch (restricted, base) {
            case (.empty, .empty), (.simpleContent, _), (_, .simpleContent):
                nil
            case let (.empty, .elementOnly(baseParticle)), let (.empty, .mixed(baseParticle)):
                emptiable(baseParticle) ? nil : "the base content is required, so the restriction cannot be EMPTY"
            case (.mixed, .elementOnly):
                "a restriction cannot add mixed content to an element-only base"
            case (.elementOnly, .empty), (.mixed, .empty):
                "the base content is EMPTY, so the restriction cannot add content"
            case let (.elementOnly(restrictedParticle), .elementOnly(baseParticle)),
                 let (.elementOnly(restrictedParticle), .mixed(baseParticle)),
                 let (.mixed(restrictedParticle), .mixed(baseParticle)):
                valid(restrictedParticle, baseParticle)
                    ? nil
                    : "the restricted content model is not a subset of the base's"
            }
        }

        /// Whether `restricted` accepts a subset of `base` (the pairwise check).
        static func valid(_ restricted: Particle, _ base: Particle) -> Bool {
            guard rangeSubsumed(restricted, base) else { return false }
            switch (restricted.term, base.term) {
            case let (.element(restrictedName, _), .element(baseName, _)):
                return sameName(restrictedName, baseName)
            case let (.element(name, _), .wildcard(wildcard)):
                return wildcard.admits(name)
            case let (.wildcard(restrictedWildcard), .wildcard(baseWildcard)):
                return narrows(restrictedWildcard, baseWildcard)
            case let (.element, .group(baseGroup)):
                // RecurseAsIfGroup: the lone element, as a one-particle sequence,
                // against the base group.
                return groupValid(Group(compositor: .sequence, particles: [restricted.withUnitRange()]), baseGroup)
            case let (.group(restrictedGroup), .group(baseGroup)):
                return groupValid(restrictedGroup, baseGroup)
            case let (.group(restrictedGroup), .wildcard(wildcard)):
                // NSRecurseCheckCardinality, namespace part: every leaf element
                // the group can contain must be admitted by the wildcard.
                return leafNames(of: restrictedGroup).allSatisfy { wildcard.admits($0) }
            default:
                return false
            }
        }

        private static func groupValid(_ restricted: Group, _ base: Group) -> Bool {
            switch (restricted.compositor, base.compositor) {
            case (.sequence, .sequence), (.all, .all):
                recurse(restricted.particles, base.particles, skippedMustBeEmptiable: true)
            case (.choice, .choice):
                recurse(restricted.particles, base.particles, skippedMustBeEmptiable: false)
            case (.sequence, .choice):
                // MapAndSum (simplified): each restricted particle fits some branch.
                restricted.particles.allSatisfy { particle in
                    base.particles.contains { valid(particle, $0) }
                }
            case (.sequence, .all):
                // RecurseUnordered: map onto distinct base particles in any order.
                recurseUnordered(restricted.particles, base.particles)
            default:
                false
            }
        }

        /// Order-preserving mapping of the restricted particles onto the base's.
        /// A skipped base particle must be emptiable for sequence/all (Recurse);
        /// for choice (RecurseLax) skipping is free.
        private static func recurse(_ restricted: [Particle], _ base: [Particle], skippedMustBeEmptiable: Bool) -> Bool {
            var baseIndex = 0
            for particle in restricted {
                var matched = false
                while baseIndex < base.count {
                    let candidate = base[baseIndex]
                    baseIndex += 1
                    if valid(particle, candidate) {
                        matched = true
                        break
                    }
                    if skippedMustBeEmptiable, !emptiable(candidate) { return false }
                }
                if !matched { return false }
            }
            if skippedMustBeEmptiable {
                while baseIndex < base.count {
                    if !emptiable(base[baseIndex]) { return false }
                    baseIndex += 1
                }
            }
            return true
        }

        /// Any-order mapping onto distinct base particles; unmapped base
        /// particles must be emptiable.
        private static func recurseUnordered(_ restricted: [Particle], _ base: [Particle]) -> Bool {
            var used = [Bool](repeating: false, count: base.count)
            for particle in restricted {
                guard let index = base.indices.first(where: { !used[$0] && valid(particle, base[$0]) }) else {
                    return false
                }
                used[index] = true
            }
            return base.indices.allSatisfy { used[$0] || emptiable(base[$0]) }
        }

        /// Occurrence-range subsumption: the restriction may not occur less often
        /// than the base requires, nor more often than the base allows.
        private static func rangeSubsumed(_ restricted: Particle, _ base: Particle) -> Bool {
            guard restricted.minOccurs >= base.minOccurs else { return false }
            guard let baseMax = base.maxOccurs else { return true }
            guard let restrictedMax = restricted.maxOccurs else { return false }
            return restrictedMax <= baseMax
        }

        /// Whether a particle can match no children at all.
        static func emptiable(_ particle: Particle) -> Bool {
            if particle.minOccurs == 0 { return true }
            switch particle.term {
            case .element, .wildcard:
                return false
            case let .group(group):
                switch group.compositor {
                case .sequence, .all:
                    return group.particles.allSatisfy(emptiable)
                case .choice:
                    return group.particles.contains(where: emptiable)
                }
            }
        }

        /// Whether the restricted wildcard admits no namespace the base refuses.
        private static func narrows(_ restricted: Wildcard, _ base: Wildcard) -> Bool {
            switch (restricted.namespace, base.namespace) {
            case (_, .any):
                true
            case (.other, .other):
                restricted.targetNamespace == base.targetNamespace
            case let (.enumerated(narrow), .enumerated(wide)):
                narrow.isSubset(of: wide)
            case let (.enumerated(narrow), .other):
                !narrow.contains("") && !narrow.contains(base.targetNamespace ?? "")
            default:
                false
            }
        }

        private static func sameName(_ lhs: PureXML.Model.QualifiedName, _ rhs: PureXML.Model.QualifiedName) -> Bool {
            lhs.localName == rhs.localName && (lhs.namespaceURI ?? "") == (rhs.namespaceURI ?? "")
        }

        /// Every element name a group can contain (its alphabet).
        private static func leafNames(of group: Group) -> [PureXML.Model.QualifiedName] {
            group.particles.flatMap { particle -> [PureXML.Model.QualifiedName] in
                switch particle.term {
                case let .element(name, _): [name]
                case let .group(inner): leafNames(of: inner)
                case .wildcard: []
                }
            }
        }
    }
}

private extension PureXML.Schema.Particle {
    /// A copy occurring exactly once, for RecurseAsIfGroup's wrapping.
    func withUnitRange() -> PureXML.Schema.Particle {
        .init(minOccurs: minOccurs, maxOccurs: maxOccurs, term: term)
    }
}