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
use crate::errors::*; use crate::engine::ctx::State; use crate::json::LuaJsonValue; use crate::hlua::{self, AnyLuaValue}; use serde_json; use std::sync::Arc; use std::net::IpAddr; pub fn asn_lookup(lua: &mut hlua::Lua, state: Arc<State>) { lua.set("asn_lookup", hlua::function1(move |ip: String| -> Result<AnyLuaValue> { let ip = ip.parse::<IpAddr>() .context("Failed to parse IP") .map_err(|err| state.set_error(err.into()))?; let lookup = state.asn().lookup(ip) .map_err(|err| state.set_error(err))?; let lookup = serde_json::to_value(lookup) .map_err(|e| state.set_error(e.into()))?; Ok(LuaJsonValue::from(lookup).into()) })) } pub fn geoip_lookup(lua: &mut hlua::Lua, state: Arc<State>) { lua.set("geoip_lookup", hlua::function1(move |ip: String| -> Result<AnyLuaValue> { let ip = ip.parse::<IpAddr>() .context("Failed to parse IP") .map_err(|err| state.set_error(err.into()))?; let lookup = state.geoip().lookup(ip) .map_err(|err| state.set_error(err))?; let lookup = serde_json::to_value(lookup) .map_err(|e| state.set_error(e.into()))?; Ok(LuaJsonValue::from(lookup).into()) })) } #[cfg(test)] mod tests { use crate::engine::ctx::Script; #[test] fn verify_asn_lookup() { let script = Script::load_unchecked(r#" function run() x = asn_lookup('1.1.1.1') print(x) end "#).expect("Failed to load script"); script.test().expect("Script failed"); } #[test] fn verify_geoip_lookup() { let script = Script::load_unchecked(r#" function run() x = geoip_lookup('1.1.1.1') print(x) end "#).expect("Failed to load script"); script.test().expect("Script failed"); } }