surrealpatch/lib/src/fnc/is.rs

use crate::dbs::Runtime;
use crate::err::Error;
use crate::sql::value::Value;
use once_cell::sync::Lazy;
use regex::Regex;
use std::char;

#[rustfmt::skip] static UUID_RE: Lazy<Regex> = Lazy::new(|| Regex::new(r"^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$").unwrap());
#[rustfmt::skip] static USER_RE: Lazy<Regex> = Lazy::new(|| Regex::new(r"^(?i)[a-z0-9.!#$%&'*+/=?^_`{|}~-]+\z").unwrap());
#[rustfmt::skip] static HOST_RE: Lazy<Regex> = Lazy::new(|| Regex::new(r"(?i)^[a-z0-9](?:[a-z0-9-]{0,61}[a-z0-9])?(?:\.[a-z0-9](?:[a-z0-9-]{0,61}[a-z0-9])?)*$").unwrap());
#[rustfmt::skip] static DOMAIN_RE: Lazy<Regex> = Lazy::new(|| Regex::new(r"^([a-zA-Z0-9_]{1}[a-zA-Z0-9_-]{0,62}){1}(\.[a-zA-Z0-9_]{1}[a-zA-Z0-9_-]{0,62})*[\._]?$",).unwrap());
#[rustfmt::skip] static SEMVER_RE: Lazy<Regex> = Lazy::new(|| Regex::new("^v?(?:0|[1-9]\\d*)\\.(?:0|[1-9]\\d*)\\.(?:0|[1-9]\\d*)(-(0|[1-9]\\d*|\\d*[a-zA-Z-][0-9a-zA-Z-]*)(\\.(0|[1-9]\\d*|\\d*[a-zA-Z-][0-9a-zA-Z-]*))*)?(\\+[0-9a-zA-Z-]+(\\.[0-9a-zA-Z-]+)*)?$").unwrap());
#[rustfmt::skip] static LATITUDE_RE: Lazy<Regex> = Lazy::new(|| Regex::new("^[-+]?([1-8]?\\d(\\.\\d+)?|90(\\.0+)?)$").unwrap());
#[rustfmt::skip] static LONGITUDE_RE: Lazy<Regex> = Lazy::new(|| Regex::new("^[-+]?([1-8]?\\d(\\.\\d+)?|90(\\.0+)?)$").unwrap());

pub fn alphanum(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_string().chars().all(char::is_alphanumeric).into())
}

pub fn alpha(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_string().chars().all(char::is_alphabetic).into())
}

pub fn ascii(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_string().chars().all(|x| char::is_ascii(&x)).into())
}

pub fn domain(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(DOMAIN_RE.is_match(args.remove(0).as_string().as_str()).into())
}

pub fn email(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	// Convert to a String
	let val = args.remove(0).as_string();
	// Convert to a &str
	let val = val.as_str();
	// Check if value is empty
	if val.is_empty() {
		return Ok(Value::False);
	}
	// Ensure the value contains @
	if !val.contains('@') {
		return Ok(Value::False);
	}
	// Reverse split the value by @
	let parts: Vec<&str> = val.rsplitn(2, '@').collect();
	// Check the first part matches
	if !USER_RE.is_match(parts[1]) {
		return Ok(Value::False);
	}
	// Check the second part matches
	if !HOST_RE.is_match(parts[0]) {
		return Ok(Value::False);
	}
	// The email is valid
	Ok(Value::True)
}

pub fn hexadecimal(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_string().chars().all(|x| char::is_ascii_hexdigit(&x)).into())
}

pub fn latitude(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(LATITUDE_RE.is_match(args.remove(0).as_string().as_str()).into())
}

pub fn longitude(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(LONGITUDE_RE.is_match(args.remove(0).as_string().as_str()).into())
}

pub fn numeric(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_string().chars().all(char::is_numeric).into())
}

pub fn semver(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(SEMVER_RE.is_match(args.remove(0).as_string().as_str()).into())
}

pub fn uuid(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(UUID_RE.is_match(args.remove(0).as_string().as_str()).into())
}