surrealpatch/lib/src/fnc/string.rs

use crate::dbs::Runtime;
use crate::err::Error;
use crate::sql::value::Value;
use slug::slugify;

pub fn concat(_: &Runtime, args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.into_iter().map(|x| x.as_strand().value).collect::<Vec<_>>().concat().into())
}

pub fn ends_with(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let val = args.remove(0).as_strand().value;
	let chr = args.remove(0).as_strand().value;
	Ok(val.ends_with(&chr).into())
}

pub fn join(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let chr = args.remove(0).as_strand().value;
	let val = args.into_iter().map(|x| x.as_strand().value);
	let val = val.collect::<Vec<_>>().join(&chr);
	Ok(val.into())
}

pub fn length(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let val = args.remove(0).as_strand().value;
	let num = val.chars().count() as i64;
	Ok(num.into())
}

pub fn lowercase(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_strand().value.to_lowercase().into())
}

pub fn repeat(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let val = args.remove(0).as_strand().value;
	let num = args.remove(0).as_int() as usize;
	Ok(val.repeat(num).into())
}

pub fn replace(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let val = args.remove(0).as_strand().value;
	let old = args.remove(0).as_strand().value;
	let new = args.remove(0).as_strand().value;
	Ok(val.replace(&old, &new).into())
}

pub fn reverse(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_strand().value.chars().rev().collect::<String>().into())
}

pub fn slice(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let val = args.remove(0).as_strand().value;
	let beg = args.remove(0).as_int() as usize;
	let lim = args.remove(0).as_int() as usize;
	let val = val.chars().skip(beg).take(lim).collect::<String>();
	Ok(val.into())
}

pub fn slug(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(slugify(&args.remove(0).as_strand().value).into())
}

pub fn split(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let val = args.remove(0).as_strand().value;
	let chr = args.remove(0).as_strand().value;
	let val = val.split(&chr).collect::<Vec<&str>>();
	Ok(val.into())
}

pub fn starts_with(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	let val = args.remove(0).as_strand().value;
	let chr = args.remove(0).as_strand().value;
	Ok(val.starts_with(&chr).into())
}

pub fn trim(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_strand().value.trim().into())
}

pub fn uppercase(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_strand().value.to_uppercase().into())
}

pub fn words(_: &Runtime, mut args: Vec<Value>) -> Result<Value, Error> {
	Ok(args.remove(0).as_strand().value.split(' ').collect::<Vec<&str>>().into())
}