LuaTableBuilder#toNbtCompound

Author: diamonddevv

autodocs/autodoc.lua

@@ -2,7 +2,7 @@
 
 --#region Information
 -- GENERATED AUTODOC
--- Generated: 2026-01-20T10:55:53.180503
+-- Generated: 2026-01-20T18:25:29.060359500
 -- Luafy Version: 2.0.0
 -- Format: Lua LS library file
 --#endregion
@@ -102,6 +102,11 @@ function ScriptResult.await_error() end
 function ScriptResult.release() end
 
 
+--- NBT Compound Tag
+---@class NbtTable
+local NbtTable = {}
+
+
 --- An entity.
 ---@class Entity
 local Entity = {}
@@ -210,10 +215,16 @@ function ItemStack.get_item_type() end
 function ItemStack.get_item_id() end
 
 --- Gets a component from this stack as NBT.
----@param component_id string The id of the component to fetch.
+---@param component_id string The id of the component type to fetch.
 ---@return table
 function ItemStack.get_component(component_id) end
 
+--- Sets a component from this stack as NBT.
+---@param component_id string The id of the component type.
+---@param nbt table The data to write. Will be encoded into the item stack.
+---@return nil
+function ItemStack.set_component(component_id, nbt) end
+
 
 --#endregion
 
@@ -311,6 +322,11 @@ function luafy.provide_hello_world() end
 ---@return string
 function luafy.get_luaj_version() end
 
+--- temp; convert table to nbt
+---@param table table table
+---@return NbtTable
+function luafy.nbt(table) end
+
 math = {}
 
 --- Creates a 3-component vector object.

src/main/java/dev/diamond/luafy/lua/LuaTableBuilder.java

@@ -2,20 +2,19 @@
 
 import dev.diamond.luafy.script.LuaScript;
 import dev.diamond.luafy.script.object.AbstractScriptObject;
-import net.minecraft.nbt.IntTag;
+import net.minecraft.nbt.*;
 import org.luaj.vm2.LuaString;
 import org.luaj.vm2.LuaTable;
 import org.luaj.vm2.LuaValue;
 import org.luaj.vm2.Varargs;
 import org.luaj.vm2.lib.VarArgFunction;
 
+import java.util.Arrays;
 import java.util.Collection;
 import java.util.List;
+import java.util.Optional;
 import java.util.function.Consumer;
 import java.util.function.Function;
-import net.minecraft.nbt.CompoundTag;
-import net.minecraft.nbt.ListTag;
-import net.minecraft.nbt.Tag;
 
 public class LuaTableBuilder {
     private final LuaTable table;
@@ -144,12 +143,35 @@ private static LuaValue fromNbt(Tag element) {
         };
     }
 
+    public static CompoundTag toNbtCompound(LuaTable table) {
+        CompoundTag nbt = new CompoundTag();
+        for (LuaValue key : table.keys()) {
+            toNbt(table.get(key)).ifPresent(value -> nbt.put(MetamethodImpl.tostring(key), value));
+        }
+        return nbt;
+    }
 
-    private static Tag toNbt(LuaValue value) {
+    private static Optional<Tag> toNbt(LuaValue value) {
         return switch (value.type()) {
-            case LuaValue.TINT ->
+            case LuaValue.TINT ->       Optional.of(IntTag.valueOf(value.toint()));
+            case LuaValue.TNUMBER ->    Optional.of(FloatTag.valueOf(value.tofloat()));
+            case LuaValue.TBOOLEAN ->   Optional.of(ByteTag.valueOf(value.toboolean()));
+            case LuaValue.TSTRING ->    Optional.of(StringTag.valueOf(MetamethodImpl.tostring(value)));
+            case LuaValue.TTABLE ->     {
+                LuaTable table = value.checktable();
+
+                if (Arrays.stream(table.keys()).allMatch(k -> k.type() == LuaValue.TINT)) {
+                    ListTag list = new ListTag();
+                    for (LuaValue key : table.keys()) {
+                        toNbt(table.get(key)).ifPresent(list::add);
+                    }
+                    yield Optional.of(list);
+                } else {
+                    yield Optional.of(toNbtCompound(table));
+                }
+            }
 
-            default -> new CompoundTag();
+            default -> Optional.empty();
         };
     }
 }

src/main/java/dev/diamond/luafy/script/api/LuafyApi.java

@@ -46,6 +46,14 @@ public void addFunctions(ScriptApiBuilder apiBuilder) {
             builder.add("get_luaj_version", args -> {
                 return LuaString.valueOf(Luafy.LUAJ_VER);
             }, "Returns the version of LuaJ used by the mod.", args -> {}, Argtypes.STRING);
+
+
+
+            builder.add("nbt", args -> {
+                return LuaTableBuilder.provide(ScriptObjects.NBT_TABLE, LuaTableBuilder.toNbtCompound(args.arg1().checktable()), script);
+            }, "temp; convert table to nbt", args -> {
+                args.add("table", Argtypes.TABLE, "table");
+            }, ScriptObjects.NBT_TABLE);
         });
 
     }

src/main/java/dev/diamond/luafy/script/object/NbtTableScriptObject.java

@@ -30,7 +30,8 @@ public void toTable(CompoundTag obj, LuaTableBuilder builder, LuaScript script)
 
     @Override
     public CompoundTag toThing(LuaTable table, CommandSourceStack src, LuaScript script) {
-        return null;
+        LuaTable backing = table.get(PROP_TABLE).checktable();
+        return LuaTableBuilder.toNbtCompound(backing);
     }
 
     @Override

src/main/java/dev/diamond/luafy/script/object/game/ItemStackScriptObject.java

@@ -87,7 +87,7 @@ public void toTable(ItemStack obj, LuaTableBuilder builder, LuaScript script) {
             // get the component
             DataComponentType<Object> type = (DataComponentType<Object>) BuiltInRegistries.DATA_COMPONENT_TYPE.get(Identifier.parse(key)).orElseThrow().value();
 
-            CompoundTag nbt = LuaTableBuilder.toNbtCompound();
+            CompoundTag nbt = LuaTableBuilder.toNbtCompound(data);
 
             return LuaValue.NIL;
         });