use crate::sql::comment::mightbespace;
use crate::sql::error::Error::ParserError;
use crate::sql::error::IResult;
use nom::bytes::complete::take_while;
use nom::bytes::complete::take_while_m_n;
use nom::character::complete::char;
use nom::character::is_alphanumeric;
use nom::multi::many1;
use nom::Err::Error;
use std::ops::RangeBounds;

pub fn colons(i: &str) -> IResult<&str, ()> {
	let (i, _) = mightbespace(i)?;
	let (i, _) = many1(char(';'))(i)?;
	let (i, _) = mightbespace(i)?;
	Ok((i, ()))
}

pub fn commas(i: &str) -> IResult<&str, ()> {
	let (i, _) = mightbespace(i)?;
	let (i, _) = char(',')(i)?;
	let (i, _) = mightbespace(i)?;
	Ok((i, ()))
}

#[inline]
pub fn is_digit(chr: char) -> bool {
	(0x30..=0x39).contains(&(chr as u8))
}

#[inline]
pub fn val_char(chr: char) -> bool {
	is_alphanumeric(chr as u8) || chr == '_'
}

#[inline]
pub fn escape(s: &str, f: &dyn Fn(char) -> bool, c: &str) -> String {
	for x in s.chars() {
		if !f(x) {
			return format!("{}{}{}", c, s, c);
		}
	}
	s.to_owned()
}

pub fn take_u32(i: &str) -> IResult<&str, u32> {
	let (i, v) = take_while(is_digit)(i)?;
	match v.parse::<u32>() {
		Ok(v) => Ok((i, v)),
		_ => Err(Error(ParserError(i))),
	}
}

pub fn take_u64(i: &str) -> IResult<&str, u64> {
	let (i, v) = take_while(is_digit)(i)?;
	match v.parse::<u64>() {
		Ok(v) => Ok((i, v)),
		_ => Err(Error(ParserError(i))),
	}
}

pub fn take_usize(i: &str) -> IResult<&str, usize> {
	let (i, v) = take_while(is_digit)(i)?;
	match v.parse::<usize>() {
		Ok(v) => Ok((i, v)),
		_ => Err(Error(ParserError(i))),
	}
}

pub fn take_digits(i: &str, n: usize) -> IResult<&str, u32> {
	let (i, v) = take_while_m_n(n, n, is_digit)(i)?;
	match v.parse::<u32>() {
		Ok(v) => Ok((i, v)),
		_ => Err(Error(ParserError(i))),
	}
}

pub fn take_digits_range(i: &str, n: usize, range: impl RangeBounds<u32>) -> IResult<&str, u32> {
	let (i, v) = take_while_m_n(n, n, is_digit)(i)?;
	match v.parse::<u32>() {
		Ok(v) if range.contains(&v) => Ok((i, v)),
		_ => Err(Error(ParserError(i))),
	}
}