1use common::fmt::QuoteStr;
2use surrealdb_types::{SqlFormat, ToSql, write_sql};
3
4use crate::language::Language;
5
6#[derive(Clone, Debug, PartialEq, Eq)]
7#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
8pub enum Filter {
9 Ascii,
10 EdgeNgram(u16, u16),
11 Lowercase,
12 Ngram(u16, u16),
13 Snowball(Language),
14 Uppercase,
15 Mapper(String),
16}
17
18impl ToSql for Filter {
19 fn fmt_sql(&self, f: &mut String, fmt: SqlFormat) {
20 match self {
21 Self::Ascii => f.push_str("ASCII"),
22 Self::EdgeNgram(min, max) => write_sql!(f, fmt, "EDGENGRAM({min},{max})"),
23 Self::Lowercase => f.push_str("LOWERCASE"),
24 Self::Ngram(min, max) => write_sql!(f, fmt, "NGRAM({min},{max})"),
25 Self::Snowball(lang) => write_sql!(f, fmt, "SNOWBALL({lang})"),
26 Self::Uppercase => f.push_str("UPPERCASE"),
27 Self::Mapper(path) => write_sql!(f, fmt, "MAPPER({})", QuoteStr(path)),
28 }
29 }
30}