ruma_common/push/

condition.rs

use std::{
    borrow::Cow, collections::BTreeMap, future::Future, ops::RangeBounds, pin::Pin, str::FromStr,
    sync::Arc,
};

use js_int::{Int, UInt};
use regex::bytes::Regex;
#[cfg(feature = "unstable-msc3931")]
use ruma_macros::StringEnum;
use serde::{Deserialize, Serialize};
use serde_json::Value as JsonValue;
use wildmatch::WildMatch;

use crate::{
    EventId, OwnedRoomId, OwnedUserId, UserId,
    power_levels::{NotificationPowerLevels, NotificationPowerLevelsKey},
    room_version_rules::RoomPowerLevelsRules,
    serde::JsonObject,
};
#[cfg(feature = "unstable-msc3931")]
use crate::{PrivOwnedStr, RoomVersionId};

mod condition_serde;
mod flattened_json;
mod room_member_count_is;

pub use self::{
    flattened_json::{FlattenedJson, FlattenedJsonValue, ScalarJsonValue},
    room_member_count_is::{ComparisonOperator, RoomMemberCountIs},
};

/// A condition that must apply for an associated push rule's action to be taken.
#[derive(Clone, Debug, Serialize)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
#[serde(tag = "kind", rename_all = "snake_case")]
pub enum PushCondition {
    /// A glob pattern match on a field of the event.
    EventMatch(EventMatchConditionData),

    /// Matches unencrypted messages where `content.body` contains the owner's display name in that
    /// room.
    #[deprecated]
    ContainsDisplayName,

    /// Matches the current number of members in the room.
    RoomMemberCount(RoomMemberCountConditionData),

    /// Takes into account the current power levels in the room, ensuring the sender of the event
    /// has high enough power to trigger the notification.
    SenderNotificationPermission(SenderNotificationPermissionConditionData),

    /// Apply the rule only to rooms that support a given feature.
    #[cfg(feature = "unstable-msc3931")]
    #[serde(rename = "org.matrix.msc3931.room_version_supports")]
    RoomVersionSupports(RoomVersionSupportsConditionData),

    /// Exact value match on a property of the event.
    EventPropertyIs(EventPropertyIsConditionData),

    /// Exact value match on a value in an array property of the event.
    EventPropertyContains(EventPropertyContainsConditionData),

    /// Matches a thread event based on the user's thread subscription status, as defined by
    /// [MSC4306].
    ///
    /// [MSC4306]: https://github.com/matrix-org/matrix-spec-proposals/pull/4306
    #[cfg(feature = "unstable-msc4306")]
    #[serde(rename = "io.element.msc4306.thread_subscription")]
    ThreadSubscription(ThreadSubscriptionConditionData),

    #[doc(hidden)]
    #[serde(untagged)]
    _Custom(_CustomPushCondition),
}

impl PushCondition {
    /// The value of the `kind` field for this condition.
    pub fn kind(&self) -> &str {
        match self {
            Self::EventMatch(_) => "event_match",
            #[allow(deprecated)]
            Self::ContainsDisplayName => "contains_display_name",
            Self::RoomMemberCount(_) => "room_member_count",
            Self::SenderNotificationPermission(_) => "sender_notification_permission",
            #[cfg(feature = "unstable-msc3931")]
            Self::RoomVersionSupports(_) => "org.matrix.msc3931.room_version_supports",
            Self::EventPropertyIs(_) => "event_property_is",
            Self::EventPropertyContains(_) => "event_property_contains",
            #[cfg(feature = "unstable-msc4306")]
            Self::ThreadSubscription(_) => "io.element.msc4306.thread_subscription",
            Self::_Custom(condition) => &condition.kind,
        }
    }

    /// The data of this `PushCondition`.
    ///
    /// The returned JSON object won't contain the `kind` field, use [`.kind()`][Self::kind] to
    /// access it.
    ///
    /// Prefer to use the public variants of `PushCondition` where possible; this method is meant to
    /// be used for custom conditions only.
    pub fn data(&self) -> Cow<'_, JsonObject> {
        fn serialize<T: Serialize>(obj: T) -> JsonObject {
            match serde_json::to_value(obj).expect("push condition serialization to succeed") {
                JsonValue::Object(obj) => obj,
                _ => panic!("all push condition variants must serialize to objects"),
            }
        }

        match self {
            Self::EventMatch(c) => Cow::Owned(serialize(c)),
            #[allow(deprecated)]
            Self::ContainsDisplayName => Cow::Owned(JsonObject::new()),
            Self::RoomMemberCount(c) => Cow::Owned(serialize(c)),
            Self::SenderNotificationPermission(c) => Cow::Owned(serialize(c)),
            #[cfg(feature = "unstable-msc3931")]
            Self::RoomVersionSupports(c) => Cow::Owned(serialize(c)),
            Self::EventPropertyIs(c) => Cow::Owned(serialize(c)),
            Self::EventPropertyContains(c) => Cow::Owned(serialize(c)),
            #[cfg(feature = "unstable-msc4306")]
            Self::ThreadSubscription(c) => Cow::Owned(serialize(c)),
            Self::_Custom(c) => Cow::Borrowed(&c.data),
        }
    }

    /// Check if this condition applies to the event.
    ///
    /// # Arguments
    ///
    /// * `event` - The flattened JSON representation of a room message event.
    /// * `context` - The context of the room at the time of the event. If the power levels context
    ///   is missing from it, conditions that depend on it will never apply.
    ///
    /// Returns `false` if this condition is unsupported.
    pub async fn applies(&self, event: &FlattenedJson, context: &PushConditionRoomCtx) -> bool {
        if event.get_str("sender").is_some_and(|sender| sender == context.user_id) {
            return false;
        }

        match self {
            Self::EventMatch(condition) => condition.applies(event, context),
            #[allow(deprecated)]
            Self::ContainsDisplayName => {
                let Some(value) = event.get_str("content.body") else { return false };
                value.matches_pattern(&context.user_display_name, true)
            }
            Self::RoomMemberCount(condition) => condition.applies(context),
            Self::SenderNotificationPermission(condition) => condition.applies(event, context),
            #[cfg(feature = "unstable-msc3931")]
            Self::RoomVersionSupports(condition) => condition.applies(context),
            Self::EventPropertyIs(condition) => condition.applies(event),
            Self::EventPropertyContains(condition) => condition.applies(event),
            #[cfg(feature = "unstable-msc4306")]
            Self::ThreadSubscription(condition) => condition.applies(event, context).await,
            Self::_Custom(_) => false,
        }
    }
}

impl From<EventMatchConditionData> for PushCondition {
    fn from(value: EventMatchConditionData) -> Self {
        Self::EventMatch(value)
    }
}

impl From<RoomMemberCountConditionData> for PushCondition {
    fn from(value: RoomMemberCountConditionData) -> Self {
        Self::RoomMemberCount(value)
    }
}

impl From<SenderNotificationPermissionConditionData> for PushCondition {
    fn from(value: SenderNotificationPermissionConditionData) -> Self {
        Self::SenderNotificationPermission(value)
    }
}

#[cfg(feature = "unstable-msc3931")]
impl From<RoomVersionSupportsConditionData> for PushCondition {
    fn from(value: RoomVersionSupportsConditionData) -> Self {
        Self::RoomVersionSupports(value)
    }
}

impl From<EventPropertyIsConditionData> for PushCondition {
    fn from(value: EventPropertyIsConditionData) -> Self {
        Self::EventPropertyIs(value)
    }
}

impl From<EventPropertyContainsConditionData> for PushCondition {
    fn from(value: EventPropertyContainsConditionData) -> Self {
        Self::EventPropertyContains(value)
    }
}

#[cfg(feature = "unstable-msc4306")]
impl From<ThreadSubscriptionConditionData> for PushCondition {
    fn from(value: ThreadSubscriptionConditionData) -> Self {
        Self::ThreadSubscription(value)
    }
}

/// Data for the `event_match` [`PushCondition`].
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct EventMatchConditionData {
    /// The [dot-separated path] of the property of the event to match.
    ///
    /// [dot-separated path]: https://spec.matrix.org/v1.18/appendices/#dot-separated-property-paths
    pub key: String,

    /// The glob-style pattern to match against.
    ///
    /// Patterns with no special glob characters should be treated as having asterisks
    /// prepended and appended when testing the condition.
    pub pattern: String,
}

impl EventMatchConditionData {
    /// Construct a new `EventMatchConditionData` with the given path and pattern.
    pub fn new(key: String, pattern: String) -> Self {
        Self { key, pattern }
    }

    /// Check if this condition applies to the event.
    ///
    /// # Arguments
    ///
    /// * `event` - The flattened JSON representation of a room message event.
    /// * `context` - The context of the room at the time of the event.
    fn applies(&self, event: &FlattenedJson, context: &PushConditionRoomCtx) -> bool {
        check_event_match(event, &self.key, &self.pattern, context)
    }
}

/// Data for the `room_member_count` [`PushCondition`].
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct RoomMemberCountConditionData {
    /// The condition on the current number of members in the room.
    pub is: RoomMemberCountIs,
}

impl RoomMemberCountConditionData {
    /// Construct a new `RoomMemberCountConditionData` with the given condition.
    pub fn new(is: RoomMemberCountIs) -> Self {
        Self { is }
    }

    /// Check if this condition applies to the context.
    ///
    /// # Arguments
    ///
    /// * `context` - The context of the room at the time of the event.
    fn applies(&self, context: &PushConditionRoomCtx) -> bool {
        self.is.contains(&context.member_count)
    }
}

/// Data for the `sender_notification_permission` [`PushCondition`].
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct SenderNotificationPermissionConditionData {
    /// The field in the power level event the user needs a minimum power level for.
    ///
    /// Fields must be specified under the `notifications` property in the power level event's
    /// `content`.
    pub key: NotificationPowerLevelsKey,
}

impl SenderNotificationPermissionConditionData {
    /// Construct a new `EventMatchConditionData` with the given field.
    pub fn new(key: NotificationPowerLevelsKey) -> Self {
        Self { key }
    }

    /// Check if this condition applies to the event.
    ///
    /// # Arguments
    ///
    /// * `event` - The flattened JSON representation of a room message event.
    /// * `context` - The context of the room at the time of the event.
    ///
    /// Returns `false` if the power levels are missing from the context, or if the `sender` of the
    /// event is missing or invalid.
    fn applies(&self, event: &FlattenedJson, context: &PushConditionRoomCtx) -> bool {
        let Some(power_levels) = &context.power_levels else { return false };
        let Some(sender_id) = event.get_str("sender") else { return false };
        let Ok(sender_id) = <&UserId>::try_from(sender_id) else { return false };

        power_levels.has_sender_notification_permission(sender_id, &self.key)
    }
}

/// Data for the `room_version_supports` [`PushCondition`].
#[cfg(feature = "unstable-msc3931")]
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct RoomVersionSupportsConditionData {
    /// The feature the room must support for the push rule to apply.
    pub feature: RoomVersionFeature,
}

#[cfg(feature = "unstable-msc3931")]
impl RoomVersionSupportsConditionData {
    /// Construct a new `RoomVersionSupportsConditionData` with the given feature.
    pub fn new(feature: RoomVersionFeature) -> Self {
        Self { feature }
    }

    /// Check if this condition applies to the event.
    ///
    /// # Arguments
    ///
    /// * `context` - The context of the room at the time of the event.
    ///
    /// Returns `false` if the feature is not supported.
    fn applies(&self, context: &PushConditionRoomCtx) -> bool {
        match &self.feature {
            RoomVersionFeature::ExtensibleEvents => {
                context.supported_features.contains(&RoomVersionFeature::ExtensibleEvents)
            }
            RoomVersionFeature::_Custom(_) => false,
        }
    }
}

/// Features supported by room versions.
#[cfg(feature = "unstable-msc3931")]
#[doc = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/src/doc/string_enum.md"))]
#[derive(Clone, StringEnum)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub enum RoomVersionFeature {
    /// m.extensible_events
    ///
    /// The room supports [extensible events].
    ///
    /// [extensible events]: https://github.com/matrix-org/matrix-spec-proposals/pull/1767
    #[cfg(feature = "unstable-msc3932")]
    #[ruma_enum(rename = "org.matrix.msc3932.extensible_events")]
    ExtensibleEvents,

    #[doc(hidden)]
    _Custom(PrivOwnedStr),
}

#[cfg(feature = "unstable-msc3931")]
impl RoomVersionFeature {
    /// Get the default features for the given room version.
    pub fn list_for_room_version(version: &RoomVersionId) -> Vec<Self> {
        match version {
            RoomVersionId::V1
            | RoomVersionId::V2
            | RoomVersionId::V3
            | RoomVersionId::V4
            | RoomVersionId::V5
            | RoomVersionId::V6
            | RoomVersionId::V7
            | RoomVersionId::V8
            | RoomVersionId::V9
            | RoomVersionId::V10
            | RoomVersionId::V11
            | RoomVersionId::V12
            | RoomVersionId::_Custom(_) => vec![],
            #[cfg(feature = "unstable-msc2870")]
            RoomVersionId::MSC2870 => vec![],
        }
    }
}

/// Data for the `event_property_is` [`PushCondition`].
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct EventPropertyIsConditionData {
    /// The [dot-separated path] of the property of the event to match.
    ///
    /// [dot-separated path]: https://spec.matrix.org/v1.18/appendices/#dot-separated-property-paths
    pub key: String,

    /// The value to match against.
    pub value: ScalarJsonValue,
}

impl EventPropertyIsConditionData {
    /// Construct a new `EventPropertyIsConditionData` with the given path and value.
    pub fn new(key: String, value: ScalarJsonValue) -> Self {
        Self { key, value }
    }

    /// Check if this condition applies to the event.
    ///
    /// # Arguments
    ///
    /// * `event` - The flattened JSON representation of a room message event.
    fn applies(&self, event: &FlattenedJson) -> bool {
        event.get(&self.key).is_some_and(|v| *v == self.value)
    }
}

/// Data for the `event_property_contains` [`PushCondition`].
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct EventPropertyContainsConditionData {
    /// The [dot-separated path] of the property of the event to match.
    ///
    /// [dot-separated path]: https://spec.matrix.org/v1.18/appendices/#dot-separated-property-paths
    pub key: String,

    /// The value to match against.
    pub value: ScalarJsonValue,
}

impl EventPropertyContainsConditionData {
    /// Construct a new `EventPropertyContainsConditionData` with the given path and value.
    pub fn new(key: String, value: ScalarJsonValue) -> Self {
        Self { key, value }
    }

    /// Check if this condition applies to the event.
    ///
    /// # Arguments
    ///
    /// * `event` - The flattened JSON representation of a room message event.
    fn applies(&self, event: &FlattenedJson) -> bool {
        event
            .get(&self.key)
            .and_then(FlattenedJsonValue::as_array)
            .is_some_and(|a| a.contains(&self.value))
    }
}

/// Data for the `thread_subscription` [`PushCondition`].
#[derive(Clone, Debug, Serialize, Deserialize)]
#[cfg(feature = "unstable-msc4306")]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct ThreadSubscriptionConditionData {
    /// Whether the user must be subscribed (`true`) or unsubscribed (`false`) to the thread
    /// for the condition to match.
    pub subscribed: bool,
}

#[cfg(feature = "unstable-msc4306")]
impl ThreadSubscriptionConditionData {
    /// Construct a new `ThreadSubscriptionConditionData` with the given subscribed condition.
    pub fn new(subscribed: bool) -> Self {
        Self { subscribed }
    }

    /// Check if this condition applies to the event.
    ///
    /// # Arguments
    ///
    /// * `event` - The flattened JSON representation of a room message event.
    /// * `context` - The context of the room at the time of the event.
    ///
    /// Returns `false` if the `has_thread_subscription_fn` in the context is missing or if the
    /// thread root of the event is missing or invalid.
    async fn applies(&self, event: &FlattenedJson, context: &PushConditionRoomCtx) -> bool {
        let Some(has_thread_subscription_fn) = &context.has_thread_subscription_fn else {
            // If we don't have a function to check thread subscriptions, we can't
            // determine if the condition applies.
            return false;
        };

        // The event must have a relation of type `m.thread`.
        if event.get_str("content.m\\.relates_to.rel_type") != Some("m.thread") {
            return false;
        }

        // Retrieve the thread root event ID.
        let Some(Ok(thread_root)) =
            event.get_str("content.m\\.relates_to.event_id").map(<&EventId>::try_from)
        else {
            return false;
        };

        let is_subscribed = has_thread_subscription_fn(thread_root).await;

        self.subscribed == is_subscribed
    }
}

/// An unknown push condition.
#[doc(hidden)]
#[derive(Clone, Debug, Serialize)]
#[allow(clippy::exhaustive_structs)]
pub struct _CustomPushCondition {
    /// The kind of the condition.
    kind: String,

    /// The additional fields that the condition contains.
    #[serde(flatten)]
    data: JsonObject,
}

/// Check whether the given key of the given event matches the given pattern, with the given
/// context.
pub(super) fn check_event_match(
    event: &FlattenedJson,
    key: &str,
    pattern: &str,
    context: &PushConditionRoomCtx,
) -> bool {
    let value = match key {
        "room_id" => context.room_id.as_str(),
        _ => match event.get_str(key) {
            Some(v) => v,
            None => return false,
        },
    };

    value.matches_pattern(pattern, key == "content.body")
}

/// The context of the room associated to an event to be able to test all push conditions.
#[derive(Clone)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct PushConditionRoomCtx {
    /// The ID of the room.
    pub room_id: OwnedRoomId,

    /// The number of members in the room.
    pub member_count: UInt,

    /// The user's matrix ID.
    pub user_id: OwnedUserId,

    /// The display name of the current user in the room.
    pub user_display_name: String,

    /// The room power levels context for the room.
    ///
    /// If this is missing, push rules that require this will never match.
    pub power_levels: Option<PushConditionPowerLevelsCtx>,

    /// The list of features this room's version or the room itself supports.
    #[cfg(feature = "unstable-msc3931")]
    pub supported_features: Vec<RoomVersionFeature>,

    /// A closure that returns a future indicating if the given thread (represented by its thread
    /// root event id) is subscribed to by the current user, where subscriptions are defined as per
    /// [MSC4306].
    ///
    /// [MSC4306]: https://github.com/matrix-org/matrix-spec-proposals/pull/4306
    #[cfg(feature = "unstable-msc4306")]
    has_thread_subscription_fn: Option<Arc<HasThreadSubscriptionFn>>,

    /// When the `unstable-msc4306` feature is enabled with the field above, it changes the auto
    /// trait implementations of the struct to `!RefUnwindSafe` and `!UnwindSafe`. So we use
    /// `PhantomData` to keep the same bounds on the field when the feature is disabled, to always
    /// have the same auto trait implementations.
    #[cfg(not(feature = "unstable-msc4306"))]
    has_thread_subscription_fn: std::marker::PhantomData<Arc<HasThreadSubscriptionFn>>,
}

#[cfg(not(target_family = "wasm"))]
type HasThreadSubscriptionFuture<'a> = Pin<Box<dyn Future<Output = bool> + Send + 'a>>;

#[cfg(target_family = "wasm")]
type HasThreadSubscriptionFuture<'a> = Pin<Box<dyn Future<Output = bool> + 'a>>;

#[cfg(not(target_family = "wasm"))]
type HasThreadSubscriptionFn =
    dyn for<'a> Fn(&'a EventId) -> HasThreadSubscriptionFuture<'a> + Send + Sync;

#[cfg(target_family = "wasm")]
type HasThreadSubscriptionFn = dyn for<'a> Fn(&'a EventId) -> HasThreadSubscriptionFuture<'a>;

impl std::fmt::Debug for PushConditionRoomCtx {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let mut debug_struct = f.debug_struct("PushConditionRoomCtx");

        debug_struct
            .field("room_id", &self.room_id)
            .field("member_count", &self.member_count)
            .field("user_id", &self.user_id)
            .field("user_display_name", &self.user_display_name)
            .field("power_levels", &self.power_levels);

        #[cfg(feature = "unstable-msc3931")]
        debug_struct.field("supported_features", &self.supported_features);

        debug_struct.finish_non_exhaustive()
    }
}

impl PushConditionRoomCtx {
    /// Create a new `PushConditionRoomCtx`.
    pub fn new(
        room_id: OwnedRoomId,
        member_count: UInt,
        user_id: OwnedUserId,
        user_display_name: String,
    ) -> Self {
        Self {
            room_id,
            member_count,
            user_id,
            user_display_name,
            power_levels: None,
            #[cfg(feature = "unstable-msc3931")]
            supported_features: Vec::new(),
            has_thread_subscription_fn: Default::default(),
        }
    }

    /// Set a function to check if the user is subscribed to a thread, so as to define the push
    /// rules defined in [MSC4306].
    ///
    /// [MSC4306]: https://github.com/matrix-org/matrix-spec-proposals/pull/4306
    #[cfg(feature = "unstable-msc4306")]
    pub fn with_has_thread_subscription_fn(
        self,
        #[cfg(not(target_family = "wasm"))]
        has_thread_subscription_fn: impl for<'a> Fn(
            &'a EventId,
        ) -> HasThreadSubscriptionFuture<'a>
        + Send
        + Sync
        + 'static,
        #[cfg(target_family = "wasm")]
        has_thread_subscription_fn: impl for<'a> Fn(
            &'a EventId,
        ) -> HasThreadSubscriptionFuture<'a>
        + 'static,
    ) -> Self {
        Self { has_thread_subscription_fn: Some(Arc::new(has_thread_subscription_fn)), ..self }
    }

    /// Add the given power levels context to this `PushConditionRoomCtx`.
    pub fn with_power_levels(self, power_levels: PushConditionPowerLevelsCtx) -> Self {
        Self { power_levels: Some(power_levels), ..self }
    }
}

/// The room power levels context to be able to test the corresponding push conditions.
///
/// Should be constructed using `From<RoomPowerLevels>`.
#[derive(Clone, Debug)]
#[cfg_attr(not(ruma_unstable_exhaustive_types), non_exhaustive)]
pub struct PushConditionPowerLevelsCtx {
    /// The power levels of the users of the room.
    pub users: BTreeMap<OwnedUserId, Int>,

    /// The default power level of the users of the room.
    pub users_default: Int,

    /// The notification power levels of the room.
    pub notifications: NotificationPowerLevels,

    /// The tweaks for determining the power level of a user.
    pub rules: RoomPowerLevelsRules,
}

impl PushConditionPowerLevelsCtx {
    /// Create a new `PushConditionPowerLevelsCtx`.
    pub fn new(
        users: BTreeMap<OwnedUserId, Int>,
        users_default: Int,
        notifications: NotificationPowerLevels,
        rules: RoomPowerLevelsRules,
    ) -> Self {
        Self { users, users_default, notifications, rules }
    }

    /// Whether the given user has the permission to notify for the given key.
    pub fn has_sender_notification_permission(
        &self,
        user_id: &UserId,
        key: &NotificationPowerLevelsKey,
    ) -> bool {
        let Some(notification_power_level) = self.notifications.get(key) else {
            // We don't know the required power level for the key.
            return false;
        };

        if self
            .rules
            .privileged_creators
            .as_ref()
            .is_some_and(|creators| creators.contains(user_id))
        {
            return true;
        }

        let user_power_level = self.users.get(user_id).unwrap_or(&self.users_default);

        user_power_level >= notification_power_level
    }
}

/// Additional functions for character matching.
trait CharExt {
    /// Whether or not this char can be part of a word.
    fn is_word_char(&self) -> bool;
}

impl CharExt for char {
    fn is_word_char(&self) -> bool {
        self.is_ascii_alphanumeric() || *self == '_'
    }
}

/// Additional functions for string matching.
trait StrExt {
    /// Get the length of the char at `index`. The byte index must correspond to
    /// the start of a char boundary.
    fn char_len(&self, index: usize) -> usize;

    /// Get the char at `index`. The byte index must correspond to the start of
    /// a char boundary.
    fn char_at(&self, index: usize) -> char;

    /// Get the index of the char that is before the char at `index`. The byte index
    /// must correspond to a char boundary.
    ///
    /// Returns `None` if there's no previous char. Otherwise, returns the char.
    fn find_prev_char(&self, index: usize) -> Option<char>;

    /// Matches this string against `pattern`.
    ///
    /// The pattern can be a glob with wildcards `*` and `?`.
    ///
    /// The match is case insensitive.
    ///
    /// If `match_words` is `true`, checks that the pattern is separated from other words.
    fn matches_pattern(&self, pattern: &str, match_words: bool) -> bool;

    /// Matches this string against `pattern`, with word boundaries.
    ///
    /// The pattern can be a glob with wildcards `*` and `?`.
    ///
    /// A word boundary is defined as the start or end of the value, or any character not in the
    /// sets `[A-Z]`, `[a-z]`, `[0-9]` or `_`.
    ///
    /// The match is case sensitive.
    fn matches_word(&self, pattern: &str) -> bool;

    /// Translate the wildcards in `self` to a regex syntax.
    ///
    /// `self` must only contain wildcards.
    fn wildcards_to_regex(&self) -> String;
}

impl StrExt for str {
    fn char_len(&self, index: usize) -> usize {
        let mut len = 1;
        while !self.is_char_boundary(index + len) {
            len += 1;
        }
        len
    }

    fn char_at(&self, index: usize) -> char {
        let end = index + self.char_len(index);
        let char_str = &self[index..end];
        char::from_str(char_str)
            .unwrap_or_else(|_| panic!("Could not convert str '{char_str}' to char"))
    }

    fn find_prev_char(&self, index: usize) -> Option<char> {
        if index == 0 {
            return None;
        }

        let mut pos = index - 1;
        while !self.is_char_boundary(pos) {
            pos -= 1;
        }
        Some(self.char_at(pos))
    }

    fn matches_pattern(&self, pattern: &str, match_words: bool) -> bool {
        let value = &self.to_lowercase();
        let pattern = &pattern.to_lowercase();

        if match_words {
            value.matches_word(pattern)
        } else {
            WildMatch::new(pattern).matches(value)
        }
    }

    fn matches_word(&self, pattern: &str) -> bool {
        if self == pattern {
            return true;
        }
        if pattern.is_empty() {
            return false;
        }

        let has_wildcards = pattern.contains(['?', '*']);

        if has_wildcards {
            let mut chunks: Vec<String> = vec![];
            let mut prev_wildcard = false;
            let mut chunk_start = 0;

            for (i, c) in pattern.char_indices() {
                if matches!(c, '?' | '*') && !prev_wildcard {
                    if i != 0 {
                        chunks.push(regex::escape(&pattern[chunk_start..i]));
                        chunk_start = i;
                    }

                    prev_wildcard = true;
                } else if prev_wildcard {
                    let chunk = &pattern[chunk_start..i];
                    chunks.push(chunk.wildcards_to_regex());

                    chunk_start = i;
                    prev_wildcard = false;
                }
            }

            let len = pattern.len();
            if !prev_wildcard {
                chunks.push(regex::escape(&pattern[chunk_start..len]));
            } else if prev_wildcard {
                let chunk = &pattern[chunk_start..len];
                chunks.push(chunk.wildcards_to_regex());
            }

            // The word characters in ASCII compatible mode (with the `-u` flag) match the
            // definition in the spec: any character not in the set `[A-Za-z0-9_]`.
            let regex = format!(r"(?-u:^|\W|\b){}(?-u:\b|\W|$)", chunks.concat());
            let re = Regex::new(&regex).expect("regex construction should succeed");
            re.is_match(self.as_bytes())
        } else {
            match self.find(pattern) {
                Some(start) => {
                    let end = start + pattern.len();

                    // Look if the match has word boundaries.
                    let word_boundary_start = !self.char_at(start).is_word_char()
                        || !self.find_prev_char(start).is_some_and(|c| c.is_word_char());

                    if word_boundary_start {
                        let word_boundary_end = end == self.len()
                            || !self.find_prev_char(end).unwrap().is_word_char()
                            || !self.char_at(end).is_word_char();

                        if word_boundary_end {
                            return true;
                        }
                    }

                    // Find next word.
                    let non_word_str = &self[start..];
                    let Some(non_word) = non_word_str.find(|c: char| !c.is_word_char()) else {
                        return false;
                    };

                    let word_str = &non_word_str[non_word..];
                    let Some(word) = word_str.find(|c: char| c.is_word_char()) else {
                        return false;
                    };

                    word_str[word..].matches_word(pattern)
                }
                None => false,
            }
        }
    }

    fn wildcards_to_regex(&self) -> String {
        // Simplify pattern to avoid performance issues:
        // - The glob `?**?**?` is equivalent to the glob `???*`
        // - The glob `???*` is equivalent to the regex `.{3,}`
        let question_marks = self.matches('?').count();

        if self.contains('*') {
            format!(".{{{question_marks},}}")
        } else {
            format!(".{{{question_marks}}}")
        }
    }
}

#[cfg(test)]
mod tests {
    use std::collections::BTreeMap;

    use assert_matches2::assert_matches;
    use js_int::{Int, int, uint};
    use macro_rules_attribute::apply;
    use serde_json::{Value as JsonValue, from_value as from_json_value, json};
    use smol_macros::test;

    use super::{
        EventMatchConditionData, EventPropertyContainsConditionData, EventPropertyIsConditionData,
        FlattenedJson, PushCondition, PushConditionPowerLevelsCtx, PushConditionRoomCtx,
        RoomMemberCountConditionData, RoomMemberCountIs, SenderNotificationPermissionConditionData,
        StrExt,
    };
    use crate::{
        OwnedUserId, assert_to_canonical_json_eq, owned_room_id, owned_user_id,
        power_levels::{NotificationPowerLevels, NotificationPowerLevelsKey},
        room_version_rules::{AuthorizationRules, RoomPowerLevelsRules},
    };

    #[test]
    fn serialize_event_match_condition() {
        assert_to_canonical_json_eq!(
            PushCondition::EventMatch(EventMatchConditionData::new(
                "content.msgtype".into(),
                "m.notice".into()
            )),
            json!({
                "key": "content.msgtype",
                "kind": "event_match",
                "pattern": "m.notice"
            }),
        );
    }

    #[test]
    #[allow(deprecated)]
    fn serialize_contains_display_name_condition() {
        assert_to_canonical_json_eq!(
            PushCondition::ContainsDisplayName,
            json!({ "kind": "contains_display_name" }),
        );
    }

    #[test]
    fn serialize_room_member_count_condition() {
        assert_to_canonical_json_eq!(
            PushCondition::RoomMemberCount(RoomMemberCountConditionData::new(
                RoomMemberCountIs::from(uint!(2))
            )),
            json!({
                "is": "2",
                "kind": "room_member_count"
            }),
        );
    }

    #[test]
    fn serialize_sender_notification_permission_condition() {
        assert_to_canonical_json_eq!(
            PushCondition::SenderNotificationPermission(
                SenderNotificationPermissionConditionData::new("room".into())
            ),
            json!({
                "key": "room",
                "kind": "sender_notification_permission"
            }),
        );
    }

    #[test]
    fn deserialize_event_match_condition() {
        let json_data = json!({
            "key": "content.msgtype",
            "kind": "event_match",
            "pattern": "m.notice"
        });
        assert_matches!(
            from_json_value::<PushCondition>(json_data).unwrap(),
            PushCondition::EventMatch(condition)
        );
        assert_eq!(condition.key, "content.msgtype");
        assert_eq!(condition.pattern, "m.notice");
    }

    #[test]
    #[allow(deprecated)]
    fn deserialize_contains_display_name_condition() {
        assert_matches!(
            from_json_value::<PushCondition>(json!({ "kind": "contains_display_name" })).unwrap(),
            PushCondition::ContainsDisplayName
        );
    }

    #[test]
    fn deserialize_room_member_count_condition() {
        let json_data = json!({
            "is": "2",
            "kind": "room_member_count"
        });
        assert_matches!(
            from_json_value::<PushCondition>(json_data).unwrap(),
            PushCondition::RoomMemberCount(condition)
        );
        assert_eq!(condition.is, RoomMemberCountIs::from(uint!(2)));
    }

    #[test]
    fn deserialize_sender_notification_permission_condition() {
        let json_data = json!({
            "key": "room",
            "kind": "sender_notification_permission"
        });
        assert_matches!(
            from_json_value::<PushCondition>(json_data).unwrap(),
            PushCondition::SenderNotificationPermission(condition)
        );
        assert_eq!(condition.key, NotificationPowerLevelsKey::Room);
    }

    #[test]
    fn words_match() {
        assert!("foo bar".matches_word("foo"));
        assert!(!"Foo bar".matches_word("foo"));
        assert!(!"foobar".matches_word("foo"));
        assert!("foobar foo".matches_word("foo"));
        assert!(!"foobar foobar".matches_word("foo"));
        assert!(!"foobar bar".matches_word("bar bar"));
        assert!("foobar bar bar".matches_word("bar bar"));
        assert!(!"foobar bar barfoo".matches_word("bar bar"));
        assert!("conduit ⚡️".matches_word("conduit ⚡️"));
        assert!("conduit ⚡️".matches_word("conduit"));
        assert!("conduit ⚡️".matches_word("⚡️"));
        assert!("conduit⚡️".matches_word("conduit"));
        assert!("conduit⚡️".matches_word("⚡️"));
        assert!("⚡️conduit".matches_word("conduit"));
        assert!("⚡️conduit".matches_word("⚡️"));
        assert!("Ruma Dev👩‍💻".matches_word("Dev"));
        assert!("Ruma Dev👩‍💻".matches_word("👩‍💻"));
        assert!("Ruma Dev👩‍💻".matches_word("Dev👩‍💻"));

        // Regex syntax is escaped
        assert!(!"matrix".matches_word(r"\w*"));
        assert!(r"\w".matches_word(r"\w*"));
        assert!(!"matrix".matches_word("[a-z]*"));
        assert!("[a-z] and [0-9]".matches_word("[a-z]*"));
        assert!(!"m".matches_word("[[:alpha:]]?"));
        assert!("[[:alpha:]]!".matches_word("[[:alpha:]]?"));

        // From the spec: <https://spec.matrix.org/v1.18/client-server-api/#conditions-1>
        assert!("An example event.".matches_word("ex*ple"));
        assert!("exple".matches_word("ex*ple"));
        assert!("An exciting triple-whammy".matches_word("ex*ple"));
    }

    #[test]
    fn patterns_match() {
        // Word matching without glob
        assert!("foo bar".matches_pattern("foo", true));
        assert!("Foo bar".matches_pattern("foo", true));
        assert!(!"foobar".matches_pattern("foo", true));
        assert!("".matches_pattern("", true));
        assert!(!"foo".matches_pattern("", true));
        assert!("foo bar".matches_pattern("foo bar", true));
        assert!(" foo bar ".matches_pattern("foo bar", true));
        assert!("baz foo bar baz".matches_pattern("foo bar", true));
        assert!("foo baré".matches_pattern("foo bar", true));
        assert!(!"bar foo".matches_pattern("foo bar", true));
        assert!("foo bar".matches_pattern("foo ", true));
        assert!("foo ".matches_pattern("foo ", true));
        assert!("foo  ".matches_pattern("foo ", true));
        assert!(" foo  ".matches_pattern("foo ", true));

        // Word matching with glob
        assert!("foo bar".matches_pattern("foo*", true));
        assert!("foo bar".matches_pattern("foo b?r", true));
        assert!(" foo bar ".matches_pattern("foo b?r", true));
        assert!("baz foo bar baz".matches_pattern("foo b?r", true));
        assert!("foo baré".matches_pattern("foo b?r", true));
        assert!(!"bar foo".matches_pattern("foo b?r", true));
        assert!("foo bar".matches_pattern("f*o ", true));
        assert!("foo ".matches_pattern("f*o ", true));
        assert!("foo  ".matches_pattern("f*o ", true));
        assert!(" foo  ".matches_pattern("f*o ", true));

        // Glob matching
        assert!(!"foo bar".matches_pattern("foo", false));
        assert!("foo".matches_pattern("foo", false));
        assert!("foo".matches_pattern("foo*", false));
        assert!("foobar".matches_pattern("foo*", false));
        assert!("foo bar".matches_pattern("foo*", false));
        assert!(!"foo".matches_pattern("foo?", false));
        assert!("fooo".matches_pattern("foo?", false));
        assert!("FOO".matches_pattern("foo", false));
        assert!("".matches_pattern("", false));
        assert!("".matches_pattern("*", false));
        assert!(!"foo".matches_pattern("", false));

        // From the spec: <https://spec.matrix.org/v1.18/client-server-api/#conditions-1>
        assert!("Lunch plans".matches_pattern("lunc?*", false));
        assert!("LUNCH".matches_pattern("lunc?*", false));
        assert!(!" lunch".matches_pattern("lunc?*", false));
        assert!(!"lunc".matches_pattern("lunc?*", false));
    }

    fn sender() -> OwnedUserId {
        owned_user_id!("@worthy_whale:server.name")
    }

    fn push_context() -> PushConditionRoomCtx {
        let mut users = BTreeMap::new();
        users.insert(sender(), int!(25));

        let power_levels = PushConditionPowerLevelsCtx {
            users,
            users_default: int!(50),
            notifications: NotificationPowerLevels { room: int!(50) },
            rules: RoomPowerLevelsRules::new(&AuthorizationRules::V1, None),
        };

        let mut ctx = PushConditionRoomCtx::new(
            owned_room_id!("!room:server.name"),
            uint!(3),
            owned_user_id!("@gorilla:server.name"),
            "Groovy Gorilla".into(),
        );
        ctx.power_levels = Some(power_levels);
        ctx
    }

    fn first_flattened_event() -> FlattenedJson {
        FlattenedJson::from_value(json!({
            "sender": "@worthy_whale:server.name",
            "content": {
                "msgtype": "m.text",
                "body": "@room Give a warm welcome to Groovy Gorilla",
            },
        }))
    }

    fn second_flattened_event() -> FlattenedJson {
        FlattenedJson::from_value(json!({
            "sender": "@party_bot:server.name",
            "content": {
                "msgtype": "m.notice",
                "body": "Everybody come to party!",
            },
        }))
    }

    #[apply(test!)]
    async fn event_match_applies() {
        let context = push_context();
        let first_event = first_flattened_event();
        let second_event = second_flattened_event();

        let correct_room = PushCondition::EventMatch(EventMatchConditionData::new(
            "room_id".into(),
            "!room:server.name".into(),
        ));
        let incorrect_room = PushCondition::EventMatch(EventMatchConditionData::new(
            "room_id".into(),
            "!incorrect:server.name".into(),
        ));

        assert!(correct_room.applies(&first_event, &context).await);
        assert!(!incorrect_room.applies(&first_event, &context).await);

        let keyword = PushCondition::EventMatch(EventMatchConditionData::new(
            "content.body".into(),
            "come".into(),
        ));

        assert!(!keyword.applies(&first_event, &context).await);
        assert!(keyword.applies(&second_event, &context).await);

        let msgtype = PushCondition::EventMatch(EventMatchConditionData::new(
            "content.msgtype".into(),
            "m.notice".into(),
        ));

        assert!(!msgtype.applies(&first_event, &context).await);
        assert!(msgtype.applies(&second_event, &context).await);
    }

    #[apply(test!)]
    async fn room_member_count_is_applies() {
        let context = push_context();
        let event = first_flattened_event();

        let member_count_eq = PushCondition::RoomMemberCount(RoomMemberCountConditionData::new(
            RoomMemberCountIs::from(uint!(3)),
        ));
        let member_count_gt = PushCondition::RoomMemberCount(RoomMemberCountConditionData::new(
            RoomMemberCountIs::from(uint!(2)..),
        ));
        let member_count_lt = PushCondition::RoomMemberCount(RoomMemberCountConditionData::new(
            RoomMemberCountIs::from(..uint!(3)),
        ));

        assert!(member_count_eq.applies(&event, &context).await);
        assert!(member_count_gt.applies(&event, &context).await);
        assert!(!member_count_lt.applies(&event, &context).await);
    }

    #[apply(test!)]
    #[allow(deprecated)]
    async fn contains_display_name_applies() {
        let context = push_context();
        let first_event = first_flattened_event();
        let second_event = second_flattened_event();

        let contains_display_name = PushCondition::ContainsDisplayName;

        assert!(contains_display_name.applies(&first_event, &context).await);
        assert!(!contains_display_name.applies(&second_event, &context).await);
    }

    #[apply(test!)]
    async fn sender_notification_permission_applies() {
        let context = push_context();
        let first_event = first_flattened_event();
        let second_event = second_flattened_event();

        let sender_notification_permission = PushCondition::SenderNotificationPermission(
            SenderNotificationPermissionConditionData::new("room".into()),
        );

        assert!(!sender_notification_permission.applies(&first_event, &context).await);
        assert!(sender_notification_permission.applies(&second_event, &context).await);
    }

    #[cfg(feature = "unstable-msc3932")]
    #[apply(test!)]
    async fn room_version_supports_applies() {
        use assign::assign;

        use super::{RoomVersionFeature, RoomVersionSupportsConditionData};

        let context_not_matching = push_context();
        let context_matching = assign!(
            PushConditionRoomCtx::new(
                owned_room_id!("!room:server.name"),
                uint!(3),
                owned_user_id!("@gorilla:server.name"),
                "Groovy Gorilla".into(),
            ), {
                power_levels: context_not_matching.power_levels.clone(),
                supported_features: vec![RoomVersionFeature::ExtensibleEvents],
            }
        );

        let simple_event = FlattenedJson::from_value(json!({
            "sender": "@worthy_whale:server.name",
            "content": {
                "msgtype": "org.matrix.msc3932.extensible_events",
                "body": "@room Give a warm welcome to Groovy Gorilla",
            },
        }));

        let room_version_condition = PushCondition::RoomVersionSupports(
            RoomVersionSupportsConditionData::new(RoomVersionFeature::ExtensibleEvents),
        );

        assert!(room_version_condition.applies(&simple_event, &context_matching).await);
        assert!(!room_version_condition.applies(&simple_event, &context_not_matching).await);
    }

    #[apply(test!)]
    async fn event_property_is_applies() {
        use crate::push::condition::ScalarJsonValue;

        let context = push_context();
        let event = FlattenedJson::from_value(json!({
            "sender": "@worthy_whale:server.name",
            "content": {
                "msgtype": "m.text",
                "body": "Boom!",
                "org.fake.boolean": false,
                "org.fake.number": 13,
                "org.fake.null": null,
            },
        }));

        let string_match = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            "content.body".to_owned(),
            "Boom!".into(),
        ));
        assert!(string_match.applies(&event, &context).await);

        let string_no_match = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            "content.body".to_owned(),
            "Boom".into(),
        ));
        assert!(!string_no_match.applies(&event, &context).await);

        let wrong_type = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            "content.body".to_owned(),
            false.into(),
        ));
        assert!(!wrong_type.applies(&event, &context).await);

        let bool_match = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            r"content.org\.fake\.boolean".to_owned(),
            false.into(),
        ));
        assert!(bool_match.applies(&event, &context).await);

        let bool_no_match = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            r"content.org\.fake\.boolean".to_owned(),
            true.into(),
        ));
        assert!(!bool_no_match.applies(&event, &context).await);

        let int_match = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            r"content.org\.fake\.number".to_owned(),
            int!(13).into(),
        ));
        assert!(int_match.applies(&event, &context).await);

        let int_no_match = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            r"content.org\.fake\.number".to_owned(),
            int!(130).into(),
        ));
        assert!(!int_no_match.applies(&event, &context).await);

        let null_match = PushCondition::EventPropertyIs(EventPropertyIsConditionData::new(
            r"content.org\.fake\.null".to_owned(),
            ScalarJsonValue::Null,
        ));
        assert!(null_match.applies(&event, &context).await);
    }

    #[apply(test!)]
    async fn event_property_contains_applies() {
        use crate::push::condition::ScalarJsonValue;

        let context = push_context();
        let event = FlattenedJson::from_value(json!({
            "sender": "@worthy_whale:server.name",
            "content": {
                "org.fake.array": ["Boom!", false, 13, null],
            },
        }));

        let wrong_key = PushCondition::EventPropertyContains(
            EventPropertyContainsConditionData::new("send".to_owned(), false.into()),
        );
        assert!(!wrong_key.applies(&event, &context).await);

        let string_match =
            PushCondition::EventPropertyContains(EventPropertyContainsConditionData::new(
                r"content.org\.fake\.array".to_owned(),
                "Boom!".into(),
            ));
        assert!(string_match.applies(&event, &context).await);

        let string_no_match =
            PushCondition::EventPropertyContains(EventPropertyContainsConditionData::new(
                r"content.org\.fake\.array".to_owned(),
                "Boom".into(),
            ));
        assert!(!string_no_match.applies(&event, &context).await);

        let bool_match =
            PushCondition::EventPropertyContains(EventPropertyContainsConditionData::new(
                r"content.org\.fake\.array".to_owned(),
                false.into(),
            ));
        assert!(bool_match.applies(&event, &context).await);

        let bool_no_match =
            PushCondition::EventPropertyContains(EventPropertyContainsConditionData::new(
                r"content.org\.fake\.array".to_owned(),
                true.into(),
            ));
        assert!(!bool_no_match.applies(&event, &context).await);

        let int_match =
            PushCondition::EventPropertyContains(EventPropertyContainsConditionData::new(
                r"content.org\.fake\.array".to_owned(),
                int!(13).into(),
            ));
        assert!(int_match.applies(&event, &context).await);

        let int_no_match =
            PushCondition::EventPropertyContains(EventPropertyContainsConditionData::new(
                r"content.org\.fake\.array".to_owned(),
                int!(130).into(),
            ));
        assert!(!int_no_match.applies(&event, &context).await);

        let null_match =
            PushCondition::EventPropertyContains(EventPropertyContainsConditionData::new(
                r"content.org\.fake\.array".to_owned(),
                ScalarJsonValue::Null,
            ));
        assert!(null_match.applies(&event, &context).await);
    }

    #[apply(test!)]
    async fn room_creators_always_have_notification_permission() {
        let mut context = push_context();
        context.power_levels = Some(PushConditionPowerLevelsCtx {
            users: BTreeMap::new(),
            users_default: Int::MIN,
            notifications: NotificationPowerLevels { room: Int::MAX },
            rules: RoomPowerLevelsRules::new(&AuthorizationRules::V12, Some(sender())),
        });

        let first_event = first_flattened_event();

        let sender_notification_permission = PushCondition::SenderNotificationPermission(
            SenderNotificationPermissionConditionData::new(NotificationPowerLevelsKey::Room),
        );

        assert!(sender_notification_permission.applies(&first_event, &context).await);
    }

    #[cfg(feature = "unstable-msc4306")]
    #[apply(test!)]
    async fn thread_subscriptions_match() {
        use super::ThreadSubscriptionConditionData;
        use crate::{EventId, event_id};

        let context = push_context().with_has_thread_subscription_fn(|event_id: &EventId| {
            Box::pin(async move {
                // Simulate thread subscriptions for testing.
                event_id == event_id!("$subscribed_thread")
            })
        });

        let subscribed_thread_event = FlattenedJson::from_value(json!({
            "event_id": "$thread_response",
            "sender": "@worthy_whale:server.name",
            "content": {
                "msgtype": "m.text",
                "body": "response in thread $subscribed_thread",
                "m.relates_to": {
                    "rel_type": "m.thread",
                    "event_id": "$subscribed_thread",
                    "is_falling_back": true,
                    "m.in_reply_to": {
                        "event_id": "$prev_event",
                    },
                },
            },
        }));

        let unsubscribed_thread_event = FlattenedJson::from_value(json!({
            "event_id": "$thread_response2",
            "sender": "@worthy_whale:server.name",
            "content": {
                "msgtype": "m.text",
                "body": "response in thread $unsubscribed_thread",
                "m.relates_to": {
                    "rel_type": "m.thread",
                    "event_id": "$unsubscribed_thread",
                    "is_falling_back": true,
                    "m.in_reply_to": {
                        "event_id": "$prev_event2",
                    },
                },
            },
        }));

        let non_thread_related_event = FlattenedJson::from_value(json!({
            "event_id": "$thread_response2",
            "sender": "@worthy_whale:server.name",
            "content": {
                "m.relates_to": {
                    "rel_type": "m.reaction",
                    "event_id": "$subscribed_thread",
                    "key": "👍",
                },
            },
        }));

        let subscribed_thread_condition =
            PushCondition::ThreadSubscription(ThreadSubscriptionConditionData::new(true));
        assert!(subscribed_thread_condition.applies(&subscribed_thread_event, &context).await);
        assert!(!subscribed_thread_condition.applies(&unsubscribed_thread_event, &context).await);
        assert!(!subscribed_thread_condition.applies(&non_thread_related_event, &context).await);

        let unsubscribed_thread_condition =
            PushCondition::ThreadSubscription(ThreadSubscriptionConditionData::new(false));
        assert!(unsubscribed_thread_condition.applies(&unsubscribed_thread_event, &context).await);
        assert!(!unsubscribed_thread_condition.applies(&subscribed_thread_event, &context).await);
        assert!(!unsubscribed_thread_condition.applies(&non_thread_related_event, &context).await);
    }

    #[test]
    fn custom_condition_roundtrip() {
        let json = json!({
            "kind": "local_dev_custom",
            "foo": "bar"
        });

        let condition = from_json_value::<PushCondition>(json.clone()).unwrap();
        assert_eq!(condition.kind(), "local_dev_custom");
        let data = condition.data();
        assert_eq!(data.len(), 1);
        assert_matches!(data.get("foo"), Some(JsonValue::String(foo)));
        assert_eq!(foo, "bar");

        assert_to_canonical_json_eq!(condition, json);
    }
}