lyra-scripting/src/lua/ecs/query/mod.rs

@@ -40,4 +40,100 @@ impl LuaQuery {
             QueryInner::Function(function) => function.call((world, lua_en)),
         }
     }
+}
+
+#[derive(Debug, Clone)]
+pub enum LuaQueryResult {
+    None,
+    AlwaysNone,
+    FilterPass,
+    FilterDeny,
+    Some(mlua::Value),
+}
+
+impl mlua::IntoLua for LuaQueryResult {
+    fn into_lua(self, lua: &mlua::Lua) -> mlua::Result<mlua::Value> {
+        let t = lua.create_table()?;
+        t.set("enum_ty", "query_result")?;
+
+        match self {
+            LuaQueryResult::None => {
+                t.set("result", "none")?;
+            },
+            LuaQueryResult::AlwaysNone => {
+                t.set("result", "always_none")?;
+            },
+            LuaQueryResult::FilterPass => {
+                t.set("result", "filter_pass")?;
+            },
+            LuaQueryResult::FilterDeny => {
+                t.set("result", "filter_deny")?;
+            },
+            LuaQueryResult::Some(value) => {
+                t.set("result", "some")?;
+                t.set("val", value)?;
+            },
+        }
+
+        t.into_lua(lua)
+    }
+}
+
+#[inline(always)]
+fn from_lua_error_query_result(ty: &'static str, msg: &str) -> mlua::Error {
+    mlua::Error::FromLuaConversionError {
+        from: ty,
+        to: "QueryResult".into(),
+        message: Some(msg.into()),
+    }
+}
+
+#[inline(always)]
+fn malformed_table_error_query_result(ty: &'static str, missing_field: &str) -> mlua::Error {
+    mlua::Error::FromLuaConversionError {
+        from: ty,
+        to: "QueryResult".into(),
+        message: Some(format!("malformed table, cannot convert, failed to get field '{}'", missing_field)),
+    }
+}
+
+impl mlua::FromLua for LuaQueryResult {
+    fn from_lua(value: mlua::Value, _: &mlua::Lua) -> mlua::Result<Self> {
+        let ty = value.type_name();
+        let table = value.as_table()
+            .ok_or(from_lua_error_query_result(ty, "expected Table"))?;
+
+        let var_name: String = table.get("enum_ty")
+            .map_err(|_| malformed_table_error_query_result(ty, "enum_ty"))?;
+        if var_name != "query_result" {
+            return Err(mlua::Error::FromLuaConversionError {
+                from: ty,
+                to: "QueryResult".into(),
+                message: Some(format!("mismatched enum_ty: '{}'", var_name)),
+            });
+        }
+
+        let result: String = table.get("result")
+            .map_err(|_| malformed_table_error_query_result(ty, "result"))?;
+        let result_str = result.as_str();
+
+        match result_str {
+            "none" => Ok(Self::None),
+            "always_none" => Ok(Self::AlwaysNone),
+            "filter_pass" => Ok(Self::FilterPass),
+            "filter_deny" => Ok(Self::FilterDeny),
+            "value" => {
+                let val: mlua::Value = table.get("val")
+                    .map_err(|_| malformed_table_error_query_result(ty, "val"))?;
+                Ok(Self::Some(val))
+            },
+            _ => {
+                Err(mlua::Error::FromLuaConversionError {
+                    from: ty,
+                    to: "QueryResult".into(),
+                    message: Some(format!("unknown result type: '{}'", result_str)),
+                })
+            }
+        }
+    }
 }