icydb_schema/

proposal.rs

//! Canonical database-scoped proposal envelope.

use std::collections::{BTreeMap, BTreeSet};

use serde::{Deserialize, Serialize};
use sha2::{Digest, Sha256};

use crate::{
    ConstraintFragmentKind, ConstraintSourceKey, EntityFragment, EntitySourceKey,
    EnumVariantFragment, FieldFragment, FieldSourceKey, FieldType, IndexSourceKey,
    MAX_SCHEMA_ASSIGNMENTS, MAX_SCHEMA_CAPABILITIES, MAX_SCHEMA_PROPOSAL_FRAGMENTS,
    MAX_SCHEMA_REMOVALS, NamedTypeFragment, RelationFragment, RelationPathStepFragment,
    RelationSourceFragment, RelationSourceKey, ScalarLiteral, ScalarType, SchemaContractError,
    SchemaFragment, SchemaMigrationPlan, SchemaMigrationRename, SchemaMigrationTransform,
    SchemaProposalDigest, SchemaSubmissionKey, SourceCheckExpr, SourceCheckInstruction,
    SourceRuleOperation, TargetDatabaseIdentity, TargetStoreIdentity, TargetedRuleFragment,
    TypeSourceKey, check_len, encode_schema_fragment, encode_schema_proposal,
};

/// Sole maintained proposal contract version.
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
#[repr(transparent)]
pub struct ProposalContractVersion(u16);

impl ProposalContractVersion {
    /// Current pre-1.0 hard-cut proposal contract version.
    pub const CURRENT: Self = Self(1);

    /// Construct a version token for decoding and incompatibility tests.
    #[must_use]
    pub const fn from_raw(value: u16) -> Self {
        Self(value)
    }

    /// Return the raw version value.
    #[must_use]
    pub const fn get(self) -> u16 {
        self.0
    }
}

/// Feature required by one proposal.
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
#[repr(transparent)]
pub struct SchemaCapability(u16);

impl SchemaCapability {
    /// Exact composite record and enum contracts.
    pub const EXACT_COMPOSITE_TYPES: Self = Self(1);
    /// Accepted row-local constraints.
    pub const ACCEPTED_CHECKS: Self = Self(2);
    /// Secondary indexes.
    pub const SECONDARY_INDEXES: Self = Self(3);
    /// Restrictive relations.
    pub const RESTRICTIVE_RELATIONS: Self = Self(4);
    /// Accepted database defaults.
    pub const INSERT_DEFAULTS: Self = Self(5);
    /// Generated values.
    pub const GENERATED_VALUES: Self = Self(6);
    /// Managed created/updated timestamps.
    pub const MANAGED_TIMESTAMPS: Self = Self(7);
    /// Explicit versioned source migration declarations.
    pub const VERSIONED_MIGRATIONS: Self = Self(8);

    /// Construct a raw token for incompatibility testing and transport.
    #[must_use]
    pub const fn from_raw(value: u16) -> Self {
        Self(value)
    }

    /// Return the raw capability number.
    #[must_use]
    pub const fn get(self) -> u16 {
        self.0
    }

    const fn is_supported(self) -> bool {
        matches!(self.0, 1..=8)
    }
}

/// Expected accepted-schema head used for optimistic application.
#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
#[cfg_attr(not(target_family = "wasm"), derive(ic_wire::Wire))]
pub enum ExpectedAcceptedHead {
    /// The target database has no accepted schema.
    Empty,
    /// The target must match this exact accepted head.
    Exact {
        /// Nonzero accepted-schema revision.
        revision: u64,
        /// Opaque accepted-schema fingerprint.
        fingerprint: crate::ExpectedSchemaFingerprint,
    },
}

impl ExpectedAcceptedHead {
    const fn validate(&self) -> Result<(), SchemaContractError> {
        match self {
            Self::Exact { revision: 0, .. } => Err(SchemaContractError::InvalidReferenceList),
            Self::Empty | Self::Exact { .. } => Ok(()),
        }
    }
}

/// Explicit entity-to-store routing in the target database.
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct EntityStoreAssignment {
    entity: EntitySourceKey,
    store: TargetStoreIdentity,
}

impl EntityStoreAssignment {
    /// Construct one opaque routing assignment.
    #[must_use]
    pub const fn new(entity: EntitySourceKey, store: TargetStoreIdentity) -> Self {
        Self { entity, store }
    }

    /// Borrow the routed entity source key.
    #[must_use]
    pub const fn entity(&self) -> &EntitySourceKey {
        &self.entity
    }

    /// Return the opaque target-store identity.
    #[must_use]
    pub const fn store(&self) -> TargetStoreIdentity {
        self.store
    }
}

/// Explicit hard-cut removal operation.
#[derive(Clone, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
pub enum SchemaRemoval {
    /// Remove an entity.
    Entity(EntitySourceKey),
    /// Remove one field from an entity.
    Field {
        /// Owning entity.
        entity: EntitySourceKey,
        /// Field identity.
        field: FieldSourceKey,
    },
    /// Remove a named type.
    Type(TypeSourceKey),
    /// Remove one accepted constraint.
    Constraint {
        /// Owning entity.
        entity: EntitySourceKey,
        /// Constraint identity.
        constraint: ConstraintSourceKey,
    },
    /// Remove one index.
    Index {
        /// Owning entity.
        entity: EntitySourceKey,
        /// Index identity.
        index: IndexSourceKey,
    },
    /// Remove one relation.
    Relation {
        /// Owning entity.
        entity: EntitySourceKey,
        /// Relation identity.
        relation: RelationSourceKey,
    },
}

/// Canonical current-form database-scoped proposal.
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct SchemaProposal {
    version: ProposalContractVersion,
    capabilities: Vec<SchemaCapability>,
    target_database: TargetDatabaseIdentity,
    submission_key: SchemaSubmissionKey,
    expected_head: ExpectedAcceptedHead,
    fragments: Vec<SchemaFragment>,
    assignments: Vec<EntityStoreAssignment>,
    removals: Vec<SchemaRemoval>,
    migration: Option<SchemaMigrationPlan>,
}

impl SchemaProposal {
    /// Compose the public transport form from already selected fragments.
    ///
    /// This validates contract-local closure only. IcyDB still treats the
    /// result as untrusted and resolves target ownership, accepted references,
    /// capabilities, and catalog-native mutation semantics during application.
    ///
    /// # Errors
    ///
    /// Returns a typed contract error for bounds, duplicate definitions,
    /// ambiguous routing, removal conflicts, or malformed nested data.
    #[expect(
        clippy::too_many_arguments,
        clippy::too_many_lines,
        reason = "composition receives and validates one complete atomic public envelope"
    )]
    pub fn try_compose(
        mut capabilities: Vec<SchemaCapability>,
        target_database: TargetDatabaseIdentity,
        submission_key: SchemaSubmissionKey,
        expected_head: ExpectedAcceptedHead,
        mut fragments: Vec<SchemaFragment>,
        mut assignments: Vec<EntityStoreAssignment>,
        mut removals: Vec<SchemaRemoval>,
        migration: Option<SchemaMigrationPlan>,
    ) -> Result<Self, SchemaContractError> {
        check_len(
            "proposal capabilities",
            capabilities.len(),
            MAX_SCHEMA_CAPABILITIES,
        )?;
        check_len(
            "proposal fragments",
            fragments.len(),
            MAX_SCHEMA_PROPOSAL_FRAGMENTS,
        )?;
        check_len(
            "proposal assignments",
            assignments.len(),
            MAX_SCHEMA_ASSIGNMENTS,
        )?;
        check_len("proposal removals", removals.len(), MAX_SCHEMA_REMOVALS)?;
        expected_head.validate()?;
        crate::compact_sort_unstable_by(&mut capabilities, Ord::cmp);
        ensure_no_adjacent_duplicates(&capabilities)?;
        if capabilities
            .iter()
            .any(|capability| !capability.is_supported())
        {
            return Err(SchemaContractError::UnsupportedCapability);
        }
        let declares_migration_capability = capabilities
            .binary_search(&SchemaCapability::VERSIONED_MIGRATIONS)
            .is_ok();
        if declares_migration_capability != migration.is_some() {
            return Err(SchemaContractError::InvalidMigrationPlan);
        }
        if let Some(plan) = &migration {
            plan.validate()?;
        }
        for fragment in &fragments {
            fragment.validate()?;
        }
        let mut keyed_fragments = fragments
            .into_iter()
            .map(|fragment| encode_schema_fragment(&fragment).map(|bytes| (bytes, fragment)))
            .collect::<Result<Vec<_>, _>>()?;
        // Canonically equal fragment bytes describe equal fragments, while
        // duplicate assignment/removal keys reject below; no stable tie is observable.
        crate::compact_sort_unstable_by(&mut keyed_fragments, |left, right| left.0.cmp(&right.0));
        fragments = keyed_fragments
            .into_iter()
            .map(|(_, fragment)| fragment)
            .collect();
        crate::compact_sort_unstable_by(&mut assignments, |left, right| {
            left.entity.cmp(&right.entity)
        });
        ensure_no_adjacent_duplicates_by(&assignments, |assignment| &assignment.entity)?;
        crate::compact_sort_unstable_by(&mut removals, Ord::cmp);
        ensure_no_adjacent_duplicates(&removals)?;

        let mut entity_definitions = BTreeMap::new();
        let mut type_definitions = BTreeMap::new();
        let mut field_definitions = BTreeSet::new();
        let mut constraint_definitions = BTreeSet::new();
        let mut index_definitions = BTreeSet::new();
        let mut relation_definitions = BTreeSet::new();
        let mut entity_names = BTreeSet::new();
        let mut type_names = BTreeSet::new();
        for fragment in &fragments {
            for entity in fragment.entities() {
                if entity_definitions
                    .insert(entity.source_key().clone(), entity)
                    .is_some()
                {
                    return Err(SchemaContractError::DuplicateSourceKey);
                }
                if !entity_names.insert(entity.name()) {
                    return Err(SchemaContractError::DuplicateName);
                }
                for field in entity.fields() {
                    field_definitions
                        .insert((entity.source_key().clone(), field.source_key().clone()));
                }
                for constraint in entity.constraints() {
                    constraint_definitions
                        .insert((entity.source_key().clone(), constraint.source_key().clone()));
                }
                for index in entity.indexes() {
                    index_definitions
                        .insert((entity.source_key().clone(), index.source_key().clone()));
                }
                for relation in entity.relations() {
                    relation_definitions
                        .insert((entity.source_key().clone(), relation.source_key().clone()));
                }
            }
            for r#type in fragment.types() {
                if type_definitions
                    .insert(r#type.source_key().clone(), r#type)
                    .is_some()
                {
                    return Err(SchemaContractError::DuplicateSourceKey);
                }
                if !type_names.insert(r#type.name()) {
                    return Err(SchemaContractError::DuplicateName);
                }
            }
        }
        for assignment in &assignments {
            if !entity_definitions.contains_key(assignment.entity()) {
                return Err(SchemaContractError::InvalidReferenceList);
            }
        }
        if assignments.len() != entity_definitions.len() {
            return Err(SchemaContractError::MissingEntityStoreAssignment);
        }
        for removal in &removals {
            let collides = match removal {
                SchemaRemoval::Entity(entity) => entity_definitions.contains_key(entity),
                SchemaRemoval::Field { entity, field } => {
                    field_definitions.contains(&(entity.clone(), field.clone()))
                }
                SchemaRemoval::Type(r#type) => type_definitions.contains_key(r#type),
                SchemaRemoval::Constraint { entity, constraint } => {
                    constraint_definitions.contains(&(entity.clone(), constraint.clone()))
                }
                SchemaRemoval::Index { entity, index } => {
                    index_definitions.contains(&(entity.clone(), index.clone()))
                }
                SchemaRemoval::Relation { entity, relation } => {
                    relation_definitions.contains(&(entity.clone(), relation.clone()))
                }
            };
            if collides {
                return Err(SchemaContractError::DefinitionRemovalConflict);
            }
        }
        validate_proposal_closure(
            &expected_head,
            &entity_definitions,
            &type_definitions,
            &removals,
        )?;
        if let Some(plan) = &migration {
            validate_migration_plan(plan, &entity_definitions, &type_definitions)?;
        }

        Ok(Self {
            version: ProposalContractVersion::CURRENT,
            capabilities,
            target_database,
            submission_key,
            expected_head,
            fragments,
            assignments,
            removals,
            migration,
        })
    }

    /// Return the contract version.
    #[must_use]
    pub const fn version(&self) -> ProposalContractVersion {
        self.version
    }

    /// Borrow required capabilities in canonical order.
    #[must_use]
    pub fn capabilities(&self) -> &[SchemaCapability] {
        &self.capabilities
    }

    /// Return the target database identity.
    #[must_use]
    pub const fn target_database(&self) -> TargetDatabaseIdentity {
        self.target_database
    }

    /// Borrow the submission key.
    #[must_use]
    pub const fn submission_key(&self) -> &SchemaSubmissionKey {
        &self.submission_key
    }

    /// Borrow the optimistic accepted-head condition.
    #[must_use]
    pub const fn expected_head(&self) -> &ExpectedAcceptedHead {
        &self.expected_head
    }

    /// Borrow reusable fragments.
    #[must_use]
    pub fn fragments(&self) -> &[SchemaFragment] {
        &self.fragments
    }

    /// Borrow canonical entity-to-store assignments.
    #[must_use]
    pub fn assignments(&self) -> &[EntityStoreAssignment] {
        &self.assignments
    }

    /// Borrow explicit removals.
    #[must_use]
    pub fn removals(&self) -> &[SchemaRemoval] {
        &self.removals
    }

    /// Borrow the optional coordinated source migration plan.
    #[must_use]
    pub const fn migration(&self) -> Option<&SchemaMigrationPlan> {
        self.migration.as_ref()
    }

    /// Compute the canonical proposal digest.
    ///
    /// # Errors
    ///
    /// Returns a typed encoding error if the proposal no longer satisfies the
    /// current bounded contract.
    pub fn digest(&self) -> Result<SchemaProposalDigest, SchemaContractError> {
        let bytes = encode_schema_proposal(self)?;
        let digest: [u8; 32] = Sha256::digest(bytes).into();
        Ok(SchemaProposalDigest::from_bytes(digest))
    }

    pub(crate) fn validate_current(&self) -> Result<(), SchemaContractError> {
        if self.version != ProposalContractVersion::CURRENT {
            return Err(SchemaContractError::UnsupportedVersion {
                found: self.version.get(),
                supported: ProposalContractVersion::CURRENT.get(),
            });
        }
        let rebuilt = Self::try_compose(
            self.capabilities.clone(),
            self.target_database,
            self.submission_key.clone(),
            self.expected_head.clone(),
            self.fragments.clone(),
            self.assignments.clone(),
            self.removals.clone(),
            self.migration.clone(),
        )?;
        if rebuilt != *self {
            return Err(SchemaContractError::NonCanonical);
        }
        Ok(())
    }
}

pub(crate) fn validate_migration_plan(
    plan: &SchemaMigrationPlan,
    entities: &BTreeMap<EntitySourceKey, &EntityFragment>,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> Result<(), SchemaContractError> {
    for transition in plan.transitions() {
        let entity = entities
            .get(transition.entity())
            .copied()
            .ok_or(SchemaContractError::InvalidMigrationReference)?;
        if transition.from().get().checked_add(1) != Some(entity.version().get()) {
            return Err(SchemaContractError::MigrationVersionGap);
        }
        let reachable = if transition.renames().iter().any(|rename| {
            matches!(
                rename,
                SchemaMigrationRename::NamedType { .. }
                    | SchemaMigrationRename::EnumVariant { .. }
                    | SchemaMigrationRename::RecordField { .. }
            )
        }) {
            entity_named_types(entity, types)
        } else {
            BTreeSet::new()
        };
        for rename in transition.renames() {
            validate_migration_rename_target(rename, transition, entity, types, &reachable)?;
        }
        for transform in transition.transforms() {
            validate_migration_transform_target(transform, entity, types)?;
        }
    }
    validate_shared_type_transition_closure(plan, entities, types)?;
    Ok(())
}

fn validate_shared_type_transition_closure(
    plan: &SchemaMigrationPlan,
    entities: &BTreeMap<EntitySourceKey, &EntityFragment>,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> Result<(), SchemaContractError> {
    // Deduplicate shared requirements by the same exact key used to canonicalize
    // declarations. Keep no cross-entity reachability cache: one large shared
    // graph must not allocate one retained copy per entity.
    let mut required = BTreeMap::<_, BTreeSet<_>>::new();
    for transition in plan.transitions() {
        for rename in transition.renames() {
            if let Some(target) = shared_rename_target_type(transition, rename) {
                required
                    .entry(target)
                    .or_default()
                    .insert(rename.sort_key());
            }
        }
    }
    if required.is_empty() {
        return Ok(());
    }
    for (source, entity) in entities {
        let transition = plan
            .transitions()
            .binary_search_by(|transition| transition.entity().cmp(source))
            .ok()
            .map(|index| &plan.transitions()[index]);
        // Traverse each consumer once, independent of the number of declarations.
        for target in entity_named_types(entity, types) {
            let Some(renames) = required.get(target) else {
                continue;
            };
            let transition = transition.ok_or(SchemaContractError::InvalidMigrationReference)?;
            for rename in renames {
                if transition
                    .renames()
                    .binary_search_by(|candidate| candidate.sort_key().cmp(rename))
                    .is_err()
                {
                    return Err(SchemaContractError::InvalidMigrationReference);
                }
            }
        }
    }
    Ok(())
}

fn shared_rename_target_type(
    transition: &crate::EntityMigration,
    rename: &SchemaMigrationRename,
) -> Option<TypeSourceKey> {
    match rename {
        SchemaMigrationRename::NamedType { to, .. } => Some(to.clone()),
        SchemaMigrationRename::EnumVariant { named_type, .. }
        | SchemaMigrationRename::RecordField { named_type, .. } => {
            Some(migration_target_type(transition, named_type))
        }
        SchemaMigrationRename::Field { .. }
        | SchemaMigrationRename::Relation { .. }
        | SchemaMigrationRename::Constraint { .. }
        | SchemaMigrationRename::Rule { .. } => None,
    }
}

fn validate_migration_rename_target(
    rename: &SchemaMigrationRename,
    transition: &crate::EntityMigration,
    entity: &EntityFragment,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
    reachable: &BTreeSet<&TypeSourceKey>,
) -> Result<(), SchemaContractError> {
    let valid = match rename {
        SchemaMigrationRename::Field { to, .. } => {
            entity.fields().iter().any(|field| field.source_key() == to)
        }
        SchemaMigrationRename::NamedType { to, .. } => reachable.contains(to),
        SchemaMigrationRename::EnumVariant { named_type, to, .. } => {
            let target_type = migration_target_type(transition, named_type);
            reachable.contains(&target_type)
                && matches!(
                    types.get(&target_type),
                    Some(NamedTypeFragment::Enum(target))
                        if target.variants().iter().any(|variant| variant.source_key() == to)
                )
        }
        SchemaMigrationRename::RecordField { named_type, to, .. } => {
            let target_type = migration_target_type(transition, named_type);
            reachable.contains(&target_type)
                && matches!(
                    types.get(&target_type),
                    Some(NamedTypeFragment::Record(target))
                        if target.fields().iter().any(|field| field.source_key() == to)
                )
        }
        SchemaMigrationRename::Relation { to, .. } => entity
            .relations()
            .iter()
            .any(|relation| relation.source_key() == to),
        SchemaMigrationRename::Constraint { to, .. } => entity
            .constraints()
            .iter()
            .any(|constraint| constraint.source_key() == to),
        SchemaMigrationRename::Rule { named_type, to, .. } => {
            let target_type = migration_target_type(transition, named_type);
            entity.constraints().iter().any(|constraint| {
                matches!(
                    constraint.kind(),
                    ConstraintFragmentKind::TargetedRule(rule)
                        if rule.target_type() == &target_type && rule.rule() == to
                )
            })
        }
    };
    if valid {
        Ok(())
    } else {
        Err(SchemaContractError::InvalidMigrationReference)
    }
}

fn entity_named_types<'schema>(
    entity: &'schema EntityFragment,
    types: &BTreeMap<TypeSourceKey, &'schema NamedTypeFragment>,
) -> BTreeSet<&'schema TypeSourceKey> {
    let mut pending: Vec<_> = entity
        .fields()
        .iter()
        .map(FieldFragment::field_type)
        .collect();
    let mut reachable = BTreeSet::new();
    while let Some(field_type) = pending.pop() {
        match field_type {
            FieldType::Scalar(_) => {}
            FieldType::List(item) => pending.push(item),
            FieldType::Named(source) => {
                if reachable.insert(source)
                    && let Some(definition) = types.get(source)
                {
                    push_named_type_field_types(definition, &mut pending);
                }
            }
        }
    }
    // Targeted rules contribute their exact target, not its dependency closure.
    // This preserves migration ownership independently of field-local rule checks.
    for constraint in entity.constraints() {
        if let ConstraintFragmentKind::TargetedRule(rule) = constraint.kind() {
            reachable.insert(rule.target_type());
        }
    }
    reachable
}

fn migration_target_type(
    transition: &crate::EntityMigration,
    accepted_before: &TypeSourceKey,
) -> TypeSourceKey {
    transition
        .renames()
        .iter()
        .find_map(|rename| match rename {
            SchemaMigrationRename::NamedType { from, to } if from == accepted_before => {
                Some(to.clone())
            }
            _ => None,
        })
        .unwrap_or_else(|| accepted_before.clone())
}

fn validate_migration_transform_target(
    transform: &SchemaMigrationTransform,
    entity: &EntityFragment,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> Result<(), SchemaContractError> {
    let target = entity
        .fields()
        .iter()
        .find(|field| field.source_key() == transform.target())
        .ok_or(SchemaContractError::InvalidMigrationReference)?;
    match transform {
        SchemaMigrationTransform::Fill { literal, .. }
        | SchemaMigrationTransform::Coalesce { literal, .. } => {
            validate_migration_literal_target(literal, target, types)
        }
        SchemaMigrationTransform::CheckedCast {
            target: target_scalar,
            ..
        } if target.field_type() == &FieldType::Scalar(*target_scalar) => Ok(()),
        SchemaMigrationTransform::Copy { .. } => Ok(()),
        SchemaMigrationTransform::CheckedCast { .. } => {
            Err(SchemaContractError::InvalidMigrationTransform)
        }
    }
}

fn validate_migration_literal_target(
    literal: &ScalarLiteral,
    target: &FieldFragment,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> Result<(), SchemaContractError> {
    let valid = match (target.field_type(), literal) {
        (FieldType::Scalar(scalar), literal) => scalar.accepts_literal(literal),
        (FieldType::Named(target_type), ScalarLiteral::EnumUnit { enum_type, variant })
            if target_type == enum_type =>
        {
            matches!(
                types.get(enum_type),
                Some(NamedTypeFragment::Enum(target_enum))
                    if target_enum
                        .variants()
                        .iter()
                        .any(|candidate| candidate.source_key() == variant)
            )
        }
        (FieldType::List(_) | FieldType::Named(_), _) => false,
    };
    if valid {
        Ok(())
    } else {
        Err(SchemaContractError::LiteralTypeMismatch)
    }
}

#[derive(Default)]
struct ProposalReferences {
    types: BTreeSet<TypeSourceKey>,
    relation_entities: BTreeSet<EntitySourceKey>,
    relation_fields: BTreeSet<(EntitySourceKey, FieldSourceKey)>,
}

pub(crate) fn validate_proposal_closure(
    expected_head: &ExpectedAcceptedHead,
    entities: &BTreeMap<EntitySourceKey, &EntityFragment>,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
    removals: &[SchemaRemoval],
) -> Result<(), SchemaContractError> {
    let mut references = ProposalReferences::default();
    for entity in entities.values() {
        collect_entity_references(entity, types, &mut references)?;
        validate_local_relation_targets(entity, entities, types)?;
    }
    for r#type in types.values() {
        collect_named_type_references(r#type, &mut references);
    }
    for removal in removals {
        let removes_reference = match removal {
            SchemaRemoval::Entity(entity) => references.relation_entities.contains(entity),
            SchemaRemoval::Field { entity, field } => references
                .relation_fields
                .contains(&(entity.clone(), field.clone())),
            SchemaRemoval::Type(r#type) => references.types.contains(r#type),
            SchemaRemoval::Constraint { .. }
            | SchemaRemoval::Index { .. }
            | SchemaRemoval::Relation { .. } => false,
        };
        if removes_reference {
            return Err(SchemaContractError::RemovedReference);
        }
    }
    if matches!(expected_head, ExpectedAcceptedHead::Empty)
        && (references
            .types
            .iter()
            .any(|reference| !types.contains_key(reference))
            || references
                .relation_entities
                .iter()
                .any(|reference| !entities.contains_key(reference)))
    {
        return Err(SchemaContractError::InvalidLocalReference);
    }
    Ok(())
}

fn collect_entity_references(
    entity: &EntityFragment,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
    references: &mut ProposalReferences,
) -> Result<(), SchemaContractError> {
    for field in entity.fields() {
        collect_field_references(field, types, references)?;
    }
    for relation in entity.relations() {
        references
            .relation_entities
            .insert(relation.target_entity().clone());
        references.relation_fields.extend(
            relation
                .target_fields()
                .iter()
                .cloned()
                .map(|field| (relation.target_entity().clone(), field)),
        );
        if let RelationSourceFragment::Nested { steps, .. } = relation.source() {
            references
                .types
                .extend(steps.iter().filter_map(|step| match step {
                    RelationPathStepFragment::EnterNamed { r#type } => Some(r#type.clone()),
                    RelationPathStepFragment::RecordMember { record, .. } => Some(record.clone()),
                    RelationPathStepFragment::EnumVariantPayload { r#enum, .. } => {
                        Some(r#enum.clone())
                    }
                    RelationPathStepFragment::OptionalSome
                    | RelationPathStepFragment::ListItems
                    | RelationPathStepFragment::SetItems
                    | RelationPathStepFragment::MapValues => None,
                }));
        }
    }
    for index in entity.indexes() {
        if let Some(predicate) = index.predicate() {
            collect_expression_enum_references(predicate, types, references)?;
        }
    }
    for constraint in entity.constraints() {
        match constraint.kind() {
            ConstraintFragmentKind::Check(expression) => {
                collect_expression_enum_references(expression, types, references)?;
            }
            ConstraintFragmentKind::TargetedRule(rule) => {
                references.types.insert(rule.target_type().clone());
                validate_targeted_rule(entity, rule, types)?;
            }
        }
    }
    Ok(())
}

fn validate_targeted_rule(
    entity: &EntityFragment,
    rule: &TargetedRuleFragment,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> Result<(), SchemaContractError> {
    let root = entity
        .fields()
        .iter()
        .find(|field| field.source_key() == rule.root())
        .ok_or(SchemaContractError::InvalidLocalReference)?;
    if types.contains_key(rule.target_type())
        && !field_type_reaches_target(root.field_type(), rule.target_type(), types)
    {
        return Err(SchemaContractError::InvalidRuleTarget);
    }
    let Some(target) = types.get(rule.target_type()) else {
        return Ok(());
    };
    let shape = resolve_rule_target_shape(target, types)?;
    if operation_matches_target(rule.operation(), shape) {
        Ok(())
    } else {
        Err(SchemaContractError::InvalidRuleTarget)
    }
}

fn field_type_reaches_target(
    root: &FieldType,
    target: &TypeSourceKey,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> bool {
    let mut pending = vec![root];
    let mut visited = BTreeSet::new();
    while let Some(field_type) = pending.pop() {
        match field_type {
            FieldType::Scalar(_) => {}
            FieldType::List(item) => pending.push(item),
            FieldType::Named(source) => {
                if source == target {
                    return true;
                }
                if !visited.insert(source) {
                    continue;
                }
                if let Some(definition) = types.get(source) {
                    push_named_type_field_types(definition, &mut pending);
                }
            }
        }
    }
    false
}

fn push_named_type_field_types<'types>(
    r#type: &'types NamedTypeFragment,
    pending: &mut Vec<&'types FieldType>,
) {
    match r#type {
        NamedTypeFragment::Record(record) => {
            pending.extend(
                record
                    .fields()
                    .iter()
                    .map(crate::RecordFieldFragment::field_type),
            );
        }
        NamedTypeFragment::Enum(r#enum) => {
            pending.extend(
                r#enum
                    .variants()
                    .iter()
                    .filter_map(|variant| variant.payload()),
            );
        }
        NamedTypeFragment::Newtype { inner, .. }
        | NamedTypeFragment::List { item: inner, .. }
        | NamedTypeFragment::Set { item: inner, .. } => pending.push(inner),
        NamedTypeFragment::Map { key, value, .. } => {
            pending.push(key);
            pending.push(value);
        }
        NamedTypeFragment::Tuple { members, .. } => {
            pending.extend(members.iter().map(crate::TupleElementFragment::field_type));
        }
    }
}

#[derive(Clone, Copy)]
enum RuleTargetShape {
    Collection,
    Scalar(ScalarType),
}

fn resolve_rule_target_shape(
    target: &NamedTypeFragment,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> Result<RuleTargetShape, SchemaContractError> {
    let mut current = target;
    let mut visited = BTreeSet::new();
    loop {
        if !visited.insert(current.source_key()) {
            return Err(SchemaContractError::InvalidRuleTarget);
        }
        match current {
            NamedTypeFragment::List { .. }
            | NamedTypeFragment::Set { .. }
            | NamedTypeFragment::Map { .. } => return Ok(RuleTargetShape::Collection),
            NamedTypeFragment::Newtype { inner, .. } => match inner {
                FieldType::Scalar(scalar) => return Ok(RuleTargetShape::Scalar(*scalar)),
                FieldType::List(_) => return Ok(RuleTargetShape::Collection),
                FieldType::Named(source) => {
                    current = types
                        .get(source)
                        .copied()
                        .ok_or(SchemaContractError::InvalidRuleTarget)?;
                }
            },
            NamedTypeFragment::Record(_)
            | NamedTypeFragment::Enum(_)
            | NamedTypeFragment::Tuple { .. } => {
                return Err(SchemaContractError::InvalidRuleTarget);
            }
        }
    }
}

fn operation_matches_target(operation: &SourceRuleOperation, shape: RuleTargetShape) -> bool {
    match (operation, shape) {
        (
            SourceRuleOperation::LengthRangeInclusive { .. },
            RuleTargetShape::Collection
            | RuleTargetShape::Scalar(ScalarType::Blob { .. } | ScalarType::Text { .. }),
        ) => true,
        (
            SourceRuleOperation::NumericMaximumInclusive { value }
            | SourceRuleOperation::NumericMinimumInclusive { value },
            RuleTargetShape::Scalar(scalar),
        ) => numeric_scalar(scalar) && numeric_rule_literal_matches(scalar, value),
        (
            SourceRuleOperation::NumericRangeInclusive { min, max },
            RuleTargetShape::Scalar(scalar),
        ) => {
            numeric_scalar(scalar)
                && numeric_rule_literal_matches(scalar, min)
                && numeric_rule_literal_matches(scalar, max)
        }
        (SourceRuleOperation::MultipleOf { divisor }, RuleTargetShape::Scalar(scalar)) => {
            exact_numeric_scalar(scalar) && numeric_rule_literal_matches(scalar, divisor)
        }
        _ => false,
    }
}

fn numeric_rule_literal_matches(scalar: ScalarType, literal: &crate::ScalarLiteral) -> bool {
    if let (ScalarType::Decimal { scale }, crate::ScalarLiteral::Decimal(value)) = (scalar, literal)
    {
        let value = value.normalize();
        return value.scale() <= scale && value.scale_to_integer(scale).is_some();
    }
    scalar.accepts_literal(literal)
}

const fn exact_numeric_scalar(scalar: ScalarType) -> bool {
    matches!(
        scalar,
        ScalarType::Decimal { .. }
            | ScalarType::Int8
            | ScalarType::Int16
            | ScalarType::Int32
            | ScalarType::Int64
            | ScalarType::Int128
            | ScalarType::Nat8
            | ScalarType::Nat16
            | ScalarType::Nat32
            | ScalarType::Nat64
            | ScalarType::Nat128
    )
}

const fn numeric_scalar(scalar: ScalarType) -> bool {
    matches!(
        scalar,
        ScalarType::Decimal { .. }
            | ScalarType::Float32
            | ScalarType::Float64
            | ScalarType::Int8
            | ScalarType::Int16
            | ScalarType::Int32
            | ScalarType::Int64
            | ScalarType::Int128
            | ScalarType::Nat8
            | ScalarType::Nat16
            | ScalarType::Nat32
            | ScalarType::Nat64
            | ScalarType::Nat128
            | ScalarType::U256
    )
}

fn collect_named_type_references(r#type: &NamedTypeFragment, references: &mut ProposalReferences) {
    match r#type {
        NamedTypeFragment::Record(record) => {
            for field in record.fields() {
                collect_field_type_reference(field.field_type(), references);
            }
        }
        NamedTypeFragment::Enum(r#enum) => {
            for variant in r#enum.variants() {
                if let Some(payload) = variant.payload() {
                    collect_field_type_reference(payload, references);
                }
            }
        }
        NamedTypeFragment::Newtype { inner, .. }
        | NamedTypeFragment::List { item: inner, .. }
        | NamedTypeFragment::Set { item: inner, .. } => {
            collect_field_type_reference(inner, references);
        }
        NamedTypeFragment::Map { key, value, .. } => {
            collect_field_type_reference(key, references);
            collect_field_type_reference(value, references);
        }
        NamedTypeFragment::Tuple { members, .. } => {
            for member in members {
                collect_field_type_reference(member.field_type(), references);
            }
        }
    }
}

#[cfg(test)]
mod u256_tests {
    use super::*;
    use crate::{ScalarLiteral, U256};

    #[test]
    fn u256_targeted_ranges_admit_while_multiple_of_remains_deferred() {
        let shape = RuleTargetShape::Scalar(ScalarType::U256);
        assert!(operation_matches_target(
            &SourceRuleOperation::NumericRangeInclusive {
                min: ScalarLiteral::U256(U256::ZERO),
                max: ScalarLiteral::U256(U256::MAX),
            },
            shape,
        ));
        assert!(!operation_matches_target(
            &SourceRuleOperation::MultipleOf {
                divisor: ScalarLiteral::U256(U256::ONE),
            },
            shape,
        ));
    }
}

fn collect_field_references(
    field: &FieldFragment,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
    references: &mut ProposalReferences,
) -> Result<(), SchemaContractError> {
    collect_field_type_reference(field.field_type(), references);
    if let crate::FieldInsertPolicy::Default(ScalarLiteral::EnumUnit { enum_type, variant }) =
        field.insert_policy()
    {
        let FieldType::Named(field_type) = field.field_type() else {
            return Err(SchemaContractError::LiteralTypeMismatch);
        };
        if field_type != enum_type {
            return Err(SchemaContractError::LiteralTypeMismatch);
        }
        collect_enum_literal_reference(enum_type, variant, types, references)?;
    }
    Ok(())
}

fn collect_field_type_reference(field_type: &FieldType, references: &mut ProposalReferences) {
    match field_type {
        FieldType::List(item) => collect_field_type_reference(item, references),
        FieldType::Named(reference) => {
            references.types.insert(reference.clone());
        }
        FieldType::Scalar(_) => {}
    }
}

fn collect_expression_enum_references(
    expression: &SourceCheckExpr,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
    references: &mut ProposalReferences,
) -> Result<(), SchemaContractError> {
    for instruction in expression.instructions() {
        if let SourceCheckInstruction::Literal(ScalarLiteral::EnumUnit { enum_type, variant }) =
            instruction
        {
            collect_enum_literal_reference(enum_type, variant, types, references)?;
        }
    }
    Ok(())
}

fn collect_enum_literal_reference(
    enum_type: &TypeSourceKey,
    variant: &TypeSourceKey,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
    references: &mut ProposalReferences,
) -> Result<(), SchemaContractError> {
    references.types.insert(enum_type.clone());
    let Some(local) = types.get(enum_type) else {
        return Ok(());
    };
    let NamedTypeFragment::Enum(local) = local else {
        return Err(SchemaContractError::InvalidEnumLiteral);
    };
    if local
        .variants()
        .iter()
        .all(|candidate| candidate.source_key() != variant)
    {
        return Err(SchemaContractError::InvalidEnumLiteral);
    }
    Ok(())
}

fn validate_local_relation_targets(
    source: &EntityFragment,
    entities: &BTreeMap<EntitySourceKey, &EntityFragment>,
    types: &BTreeMap<TypeSourceKey, &NamedTypeFragment>,
) -> Result<(), SchemaContractError> {
    for relation in source.relations() {
        let Some(target) = entities.get(relation.target_entity()) else {
            continue;
        };
        match relation.source() {
            RelationSourceFragment::Direct { fields } => {
                for (source_key, target_key) in fields.iter().zip(relation.target_fields()) {
                    let source_field = source
                        .fields()
                        .iter()
                        .find(|field| field.source_key() == source_key)
                        .ok_or(SchemaContractError::InvalidLocalReference)?;
                    let target_field = target
                        .fields()
                        .iter()
                        .find(|field| field.source_key() == target_key)
                        .ok_or(SchemaContractError::InvalidLocalReference)?;
                    let source_type = match source_field.field_type() {
                        FieldType::List(item) => item.as_ref(),
                        field_type => field_type,
                    };
                    if source_type != target_field.field_type() {
                        return Err(SchemaContractError::RelationTypeMismatch);
                    }
                }
            }
            RelationSourceFragment::Nested { .. } => {
                let terminal = validate_nested_relation_path(source, relation, types)?;
                let target_key = relation
                    .target_fields()
                    .first()
                    .ok_or(SchemaContractError::InvalidReferenceList)?;
                let target_field = target
                    .fields()
                    .iter()
                    .find(|field| field.source_key() == target_key)
                    .ok_or(SchemaContractError::InvalidLocalReference)?;
                if terminal != target_field.field_type() {
                    return Err(SchemaContractError::RelationTypeMismatch);
                }
            }
        }
    }
    Ok(())
}

fn validate_nested_relation_path<'schema>(
    source: &'schema EntityFragment,
    relation: &RelationFragment,
    types: &BTreeMap<TypeSourceKey, &'schema NamedTypeFragment>,
) -> Result<&'schema FieldType, SchemaContractError> {
    let RelationSourceFragment::Nested { root, steps } = relation.source() else {
        return Err(SchemaContractError::RelationTypeMismatch);
    };
    let root = source
        .fields()
        .iter()
        .find(|field| field.source_key() == root)
        .ok_or(SchemaContractError::InvalidLocalReference)?;
    let mut kind = root.field_type();
    let mut nullable = root.nullable();
    let mut entered = None;
    for step in steps {
        match step {
            RelationPathStepFragment::OptionalSome => {
                if !nullable || entered.is_some() {
                    return Err(SchemaContractError::RelationTypeMismatch);
                }
                nullable = false;
            }
            RelationPathStepFragment::EnterNamed { r#type } => {
                let FieldType::Named(named) = kind else {
                    return Err(SchemaContractError::RelationTypeMismatch);
                };
                if nullable || entered.is_some() || named != r#type {
                    return Err(SchemaContractError::RelationTypeMismatch);
                }
                match types.get(named).copied() {
                    Some(
                        NamedTypeFragment::Record(_)
                        | NamedTypeFragment::Enum(_)
                        | NamedTypeFragment::List { .. }
                        | NamedTypeFragment::Set { .. }
                        | NamedTypeFragment::Map { .. },
                    ) => {
                        entered = Some(named);
                    }
                    Some(NamedTypeFragment::Newtype { inner, .. }) => {
                        kind = inner;
                    }
                    Some(NamedTypeFragment::Tuple { .. }) | None => {
                        return Err(SchemaContractError::RelationTypeMismatch);
                    }
                }
            }
            RelationPathStepFragment::RecordMember { record, field } => {
                if nullable || entered != Some(record) {
                    return Err(SchemaContractError::RelationTypeMismatch);
                }
                let Some(NamedTypeFragment::Record(definition)) = types.get(record).copied() else {
                    return Err(SchemaContractError::RelationTypeMismatch);
                };
                let member = definition
                    .fields()
                    .iter()
                    .find(|member| member.source_key() == field)
                    .ok_or(SchemaContractError::InvalidLocalReference)?;
                kind = member.field_type();
                nullable = member.nullable();
                entered = None;
            }
            RelationPathStepFragment::EnumVariantPayload { r#enum, variant } => {
                if nullable || entered != Some(r#enum) {
                    return Err(SchemaContractError::RelationTypeMismatch);
                }
                let Some(NamedTypeFragment::Enum(definition)) = types.get(r#enum).copied() else {
                    return Err(SchemaContractError::RelationTypeMismatch);
                };
                let payload = definition
                    .variants()
                    .iter()
                    .find(|candidate| candidate.source_key() == variant)
                    .and_then(EnumVariantFragment::payload)
                    .ok_or(SchemaContractError::InvalidLocalReference)?;
                kind = payload;
                nullable = false;
                entered = None;
            }
            RelationPathStepFragment::ListItems
            | RelationPathStepFragment::SetItems
            | RelationPathStepFragment::MapValues => {
                if nullable {
                    return Err(SchemaContractError::RelationTypeMismatch);
                }
                kind = relation_collection_step_output(kind, entered.take(), step, types)?;
                nullable = false;
            }
        }
    }
    if nullable || entered.is_some() || !matches!(kind, FieldType::Scalar(_)) {
        return Err(SchemaContractError::RelationTypeMismatch);
    }
    Ok(kind)
}

fn relation_collection_step_output<'schema>(
    kind: &'schema FieldType,
    entered: Option<&TypeSourceKey>,
    step: &RelationPathStepFragment,
    types: &BTreeMap<TypeSourceKey, &'schema NamedTypeFragment>,
) -> Result<&'schema FieldType, SchemaContractError> {
    match (step, entered, kind) {
        (RelationPathStepFragment::ListItems, None, FieldType::List(item)) => Ok(item),
        (RelationPathStepFragment::ListItems, Some(named), FieldType::Named(current))
            if named == current =>
        {
            let Some(NamedTypeFragment::List { item, .. }) = types.get(current).copied() else {
                return Err(SchemaContractError::RelationTypeMismatch);
            };
            Ok(item)
        }
        (RelationPathStepFragment::SetItems, Some(named), FieldType::Named(current))
            if named == current =>
        {
            let Some(NamedTypeFragment::Set { item, .. }) = types.get(current).copied() else {
                return Err(SchemaContractError::RelationTypeMismatch);
            };
            Ok(item)
        }
        (RelationPathStepFragment::MapValues, Some(named), FieldType::Named(current))
            if named == current =>
        {
            let Some(NamedTypeFragment::Map { value, .. }) = types.get(current).copied() else {
                return Err(SchemaContractError::RelationTypeMismatch);
            };
            Ok(value)
        }
        _ => Err(SchemaContractError::RelationTypeMismatch),
    }
}

fn ensure_no_adjacent_duplicates<T>(values: &[T]) -> Result<(), SchemaContractError>
where
    T: Eq,
{
    if values.windows(2).any(|pair| pair[0] == pair[1]) {
        return Err(SchemaContractError::DuplicateSourceKey);
    }
    Ok(())
}

fn ensure_no_adjacent_duplicates_by<T, K>(
    values: &[T],
    key: impl Fn(&T) -> &K,
) -> Result<(), SchemaContractError>
where
    K: Eq,
{
    if values.windows(2).any(|pair| key(&pair[0]) == key(&pair[1])) {
        return Err(SchemaContractError::DuplicateSourceKey);
    }
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::decode_schema_proposal;
    use std::str::FromStr;

    fn empty_proposal_with(capabilities: Vec<SchemaCapability>) -> SchemaProposal {
        SchemaProposal::try_compose(
            capabilities,
            TargetDatabaseIdentity::from_bytes([1; 32]),
            SchemaSubmissionKey::try_new("proposal-version-test")
                .expect("submission key should admit"),
            ExpectedAcceptedHead::Empty,
            Vec::new(),
            Vec::new(),
            Vec::new(),
            None,
        )
        .expect("empty proposal should compose")
    }

    #[test]
    fn decoded_future_contract_version_fails_typed() {
        let proposal = empty_proposal_with(Vec::new());
        let mut bytes = crate::encode_schema_proposal(&proposal).expect("proposal should encode");
        bytes[5..7].copy_from_slice(&2_u16.to_be_bytes());

        assert_eq!(
            decode_schema_proposal(&bytes),
            Err(SchemaContractError::UnsupportedVersion {
                found: 2,
                supported: 1,
            }),
        );
    }

    #[test]
    fn decoded_unknown_capability_fails_typed() {
        let proposal = empty_proposal_with(vec![SchemaCapability::EXACT_COMPOSITE_TYPES]);
        let mut bytes = crate::encode_schema_proposal(&proposal).expect("proposal should encode");
        bytes[11..13].copy_from_slice(&u16::MAX.to_be_bytes());

        assert_eq!(
            decode_schema_proposal(&bytes),
            Err(SchemaContractError::UnsupportedCapability),
        );
    }

    #[test]
    fn targeted_numeric_operations_require_exact_target_literal_admission() {
        let decimal = |value| {
            crate::ScalarLiteral::Decimal(
                crate::Decimal::from_str(value).expect("decimal fixture should parse"),
            )
        };
        assert!(operation_matches_target(
            &SourceRuleOperation::MultipleOf {
                divisor: decimal("0.25"),
            },
            RuleTargetShape::Scalar(ScalarType::Decimal { scale: 2 }),
        ));
        assert!(!operation_matches_target(
            &SourceRuleOperation::MultipleOf {
                divisor: decimal("0.251"),
            },
            RuleTargetShape::Scalar(ScalarType::Decimal { scale: 2 }),
        ));
        assert!(!operation_matches_target(
            &SourceRuleOperation::MultipleOf {
                divisor: crate::ScalarLiteral::Float64(
                    crate::Float64::try_new(2.0).expect("finite float"),
                ),
            },
            RuleTargetShape::Scalar(ScalarType::Float64),
        ));
        assert!(operation_matches_target(
            &SourceRuleOperation::NumericMaximumInclusive {
                value: crate::ScalarLiteral::Float64(
                    crate::Float64::try_new(2.0).expect("finite float"),
                ),
            },
            RuleTargetShape::Scalar(ScalarType::Float64),
        ));
    }
}