Add bytes module

Author: khvzak

src/bytes.rs

@@ -2,8 +2,8 @@ use std::ops::{Deref, DerefMut};
 
 use bytes::Bytes;
 use mlua::{
-    BorrowedBytes, Error, FromLua, Lua, MetaMethod, Result, String as LuaString, UserData, UserDataMethods,
-    UserDataRegistry, Value,
+    BorrowedBytes, Error, FromLua, Lua, MetaMethod, Result, String as LuaString, Table, UserData,
+    UserDataMethods, UserDataRegistry, Value,
 };
 
 /// A Lua userdata wrapper around [`Bytes`].
@@ -28,6 +28,10 @@ impl DerefMut for LuaBytes {
 
 impl UserData for LuaBytes {
     fn register(registry: &mut UserDataRegistry<Self>) {
+        registry.add_function("new", |_, data: LuaString| {
+            Ok(Self(Bytes::copy_from_slice(&data.as_bytes())))
+        });
+
         registry.add_method("len", |_, this, ()| Ok(this.len()));
 
         registry.add_method("is_empty", |_, this, ()| Ok(this.is_empty()));
@@ -95,3 +99,18 @@ impl Deref for AsBytesRefImpl<'_> {
         }
     }
 }
+
+/// A loader for the `bytes` module.
+fn loader(lua: &Lua) -> Result<Table> {
+    let t = lua.create_table()?;
+    t.set("Bytes", lua.create_proxy::<LuaBytes>()?)?;
+    Ok(t)
+}
+
+/// Registers the `bytes` module in the given Lua state.
+pub fn register(lua: &Lua, name: Option<&str>) -> Result<Table> {
+    let name = name.unwrap_or("@bytes");
+    let value = loader(lua)?;
+    lua.register_module(name, &value)?;
+    Ok(value)
+}

tests/lib.rs

@@ -9,6 +9,7 @@ async fn run_file(modname: &str) -> Result<()> {
 
     // Preload all modules
     mlua_stdlib::assertions::register(&lua, None)?;
+    mlua_stdlib::bytes::register(&lua, None)?;
     mlua_stdlib::env::register(&lua, None)?;
     let testing = mlua_stdlib::testing::register(&lua, None)?;
     mlua_stdlib::time::register(&lua, None)?;
@@ -95,6 +96,7 @@ macro_rules! include_tests {
 
 include_tests! {
     assertions,
+    bytes,
     env,
     time,
     #[cfg(feature = "json")] json,

tests/lua/bytes_tests.lua

@@ -0,0 +1,29 @@
+local bytes = require("@bytes")
+
+testing:test("Bytes.split_off", function(t)
+    local b = bytes.Bytes.new("hello world")
+    local other = b:split_off(6)
+    t.assert_eq(tostring(b), "hello ")
+    t.assert_eq(tostring(other), "world")
+end)
+
+testing:test("Bytes.split_to", function(t)
+    local b = bytes.Bytes.new("hello world")
+    local other = b:split_to(5)
+    t.assert_eq(tostring(b), " world")
+    t.assert_eq(tostring(other), "hello")
+end)
+
+testing:test("Bytes.clear", function(t)
+    local b = bytes.Bytes.new("hello")
+    b:clear()
+    t.assert_eq(b:len(), 0)
+    t.assert_eq(b:is_empty(), true)
+end)
+
+testing:test("Bytes.truncate", function(t)
+    local b = bytes.Bytes.new("hello")
+    b:truncate(2)
+    t.assert_eq(tostring(b), "he")
+    t.assert_eq(b:len(), 2)
+end)