1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#![allow(clippy::needless_return)]
#![deny(clippy::implicit_return)]
use yttrium_key_base as key_base;
#[cfg(feature = "loader")]
#[no_mangle]
pub fn key_create() -> *mut dyn key_base::Key {
	return Box::into_raw(Box::new(std_parameter {
		info: create_key_info(),
		function: key_function,
	}));
}

pub fn safe_create() -> Box<dyn key_base::Key> {
	return Box::new(std_parameter {
		info: create_key_info(),
		function: key_function,
	});
}


/*
Parameters:
Optional, the string to split on. If empty returns the entire parameter string
Required, the index of the split string to return
*/
fn create_key_info() -> key_base::KeyInfo {
	return key_base::KeyInfo {
		name: String::from("parameter"),
		parameters_required: vec![0, 2],
	};
}

#[allow(non_camel_case_types)]
struct std_parameter {
	pub info: key_base::KeyInfo,
	pub function: fn(parameter: &[String], environment: &mut key_base::environment::Environment) -> Result<String, String>,
}

impl key_base::Key for std_parameter {
	fn get_key_info(&self) -> &key_base::KeyInfo {
		return &self.info;
	}

	fn get_key_function(&self) -> fn(parameter: &[String], environment: &mut key_base::environment::Environment) -> Result<String, String> {
		return self.function;
	}
}

fn key_function(parameter: &[String], environment: &mut key_base::environment::Environment) -> Result<String, String> {
	if let key_base::environment::events::EventType::Message(event) = &mut environment.event_info {
		if parameter.is_empty() {
			return Ok(event.parameter.clone());
		} else {
			match parameter[1].parse::<usize>() {
				Ok(index) => {
					if !event.split_parameters.contains_key(&parameter[0]) {
						//This is hacky and I don't like this
						let split = event.parameter.split(&parameter[0]).map(String::from).collect();
						event.split_parameters.insert(parameter[0].clone(), split);
					}
					let split = event.split_parameters.get(&parameter[0]).unwrap();
					if split.len() >= index + 1 {
						return Ok(split[index].clone());
					} else {
						return Err(format!("`parameter` split by `{}` didn't have `{}` elements", parameter[0], index + 1));
					}
				}
				Err(error) => {
					return Err(format!("Invalid number passed to `parameter`: `{}`", error.to_string()));
				}
			}
		}
	} else {
		return Err(String::from("`parameter` called on invalid event type"));
	}
}