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
#![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_db_read {
		info: create_key_info(),
		function: key_function,
	}));
}

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


fn create_key_info() -> key_base::KeyInfo {
	return key_base::KeyInfo {
		name: String::from("db_read"),
		parameters_required: vec![2],
	};
}

#[allow(non_camel_case_types)]
struct std_db_read {
	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_db_read {
	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> {
	match environment.database_manager.get_database(&parameter[0]) {
		Some(result) => {
			match result.get_key(&parameter[1]) {
				Some(value) => {
					match value {
						key_base::databases::StringOrArray::String(string) => {
							return Ok(string);
						}
						key_base::databases::StringOrArray::Array(array) => {
							return Ok(array.join(""));
						}
					}
				}
				None => {
					return Ok(String::new());
				}
			}
		}
		None => {
			return Ok(String::new());
		}
	}
}