cw-cyber/contracts/hub-skills/src/validating.rs

use crate::error::ContractError;
use cid::{Cid, Version};
use cosmwasm_std::Response;
use cosmwasm_std::Uint64;
use std::str::FromStr;

/*
 * Basic rule /[a-z0-9-]/
 */
pub fn validate_by_basic_rule(val: String, field_name: String) -> Result<Response, ContractError> {
    for byte in val.as_bytes().iter() {
        // - && 0-9 && a-z
        if (*byte != 45) && (*byte < 48 || *byte > 57) && (*byte < 97 || *byte > 122) {
            return Err(ContractError::IncorrectInputData {
                val: format!(
                    "Incorrect value for field field {}. Allowed expression /[a-z0-9-]/",
                    field_name
                )
                .to_string(),
            });
        }
    }

    Ok(Response::default())
}

/*
 * Basic rule /[A-Z0-9]/
 */
pub fn validate_by_basic_uppercase_rule(
    val: String,
    field_name: String,
) -> Result<Response, ContractError> {
    for byte in val.as_bytes().iter() {
        // 0-9 && A-Z
        if (*byte < 48 || *byte > 57) && (*byte < 65 || *byte > 90) {
            return Err(ContractError::IncorrectInputData {
                val: format!(
                    "Incorrect value for field field {}. Allowed expression /[a-z0-9-]/",
                    field_name
                )
                .to_string(),
            });
        }
    }

    Ok(Response::default())
}

pub fn validate_datatype(val: String) -> Result<Response, ContractError> {
    for byte in val.as_bytes().iter() {
        // - && 0-9 && a-z
        if (*byte != 45) && (*byte < 48 || *byte > 57) && (*byte < 97 || *byte > 122) {
            return Err(ContractError::IncorrectInputData {
                val: "Incorrect data-type. a-z0-9- allowed".to_string(),
            });
        }
    }

    Ok(Response::default())
}

pub fn validate_url(val: String, field_name: String) -> Result<Response, ContractError> {
    for byte in val.as_bytes().iter() {
        // = & } { : / . - _ 0-9 a-z A-Z
        if (*byte != 61)
            && (*byte != 38)
            && (*byte != 125)
            && (*byte != 123)
            && (*byte != 58)
            && (*byte != 95)
            && (*byte != 45)
            && (*byte != 47)
            && (*byte != 46)
            && (*byte < 48 || *byte > 57)
            && (*byte < 97 || *byte > 122)
            && (*byte < 65 || *byte > 90)
        {
            return Err(ContractError::IncorrectInputData {
                val: format!("Incorrect value for field {}. Allowed only url", field_name)
                    .to_string(),
            });
        }
    }

    Ok(Response::default())
}

pub fn validate_by_int_range(
    from: Uint64,
    to: Uint64,
    field: Uint64,
    field_name: String,
) -> Result<Response, ContractError> {
    if field < from || field > to {
        return Err(ContractError::IncorrectInputData {
            val: format!("Incorrect value for field {}. Out from range", field_name).to_string(),
        });
    }

    Ok(Response::default())
}

pub fn validate_ipfs_cid(particle: String, field_name: String) -> Result<Response, ContractError> {
    let particle_value: Cid;
    let try_particle = Cid::from_str(&particle.clone());
    if try_particle.is_ok() {
        particle_value = try_particle.unwrap();

        if particle_value.version() != Version::V0 {
            return Err(ContractError::IncorrectInputData {
                val: format!(
                    "Incorrect value for field {}. Allowed only Ipfs hash",
                    field_name
                )
                .to_string(),
            });
        }
    } else {
        return Err(ContractError::IncorrectInputData {
            val: format!(
                "Incorrect value for field {}. Allowed only Ipfs hash",
                field_name
            )
            .to_string(),
        });
    }

    Ok(Response::default())
}