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

use jrsonnet_evaluator::{
	error::{ErrorKind::RuntimeError, Result},
	function::builtin,
	typed::{Either, Either2},
	IBytes, IStr,
};

#[builtin]
pub fn builtin_encode_utf8(str: IStr) -> IBytes {
	str.cast_bytes()
}

#[builtin]
pub fn builtin_decode_utf8(arr: IBytes) -> Result<IStr> {
	Ok(arr
		.cast_str()
		.ok_or_else(|| RuntimeError("bad utf8".into()))?)
}

#[builtin]
pub fn builtin_base64(input: Either![IStr, IBytes]) -> String {
	use Either2::*;
	match input {
		A(l) => base64::encode(l.as_bytes()),
		B(a) => base64::encode(a.as_slice()),
	}
}

#[builtin]
pub fn builtin_base64_decode_bytes(str: IStr) -> Result<IBytes> {
	Ok(base64::decode(str.as_bytes())
		.map_err(|_| RuntimeError("bad base64".into()))?
		.as_slice()
		.into())
}

#[builtin]
pub fn builtin_base64_decode(str: IStr) -> Result<String> {
	let bytes = base64::decode(str.as_bytes()).map_err(|_| RuntimeError("bad base64".into()))?;
	Ok(String::from_utf8(bytes).map_err(|_| RuntimeError("bad utf8".into()))?)
}