Add set_name(), set_count(), set_wear() and set_metadata() to Lua Ite…

…mStack
minetest · Sep 28, 2013 · a9c0961 · a9c0961
1 parent 7313928
commit a9c0961
Show file tree

Hide file tree

Showing 3 changed files with 79 additions and 0 deletions.
diff --git a/doc/lua_api.txt b/doc/lua_api.txt
@@ -1658,9 +1658,13 @@ ItemStack: A stack of items.
 methods:
 - is_empty(): return true if stack is empty
 - get_name(): returns item name (e.g. "default:stone")
+- set_name(itemname)
 - get_count(): returns number of items on the stack
+- set_count(count)
 - get_wear(): returns tool wear (0-65535), 0 for non-tools
+- set_wear(wear)
 - get_metadata(): returns metadata (a string attached to an item stack)
+- set_metadata(metadata)
 - clear(): removes all items from the stack, making it empty
 - replace(item): replace the contents of this stack (item can also
     be an itemstring or table)

diff --git a/src/script/lua_api/l_item.cpp b/src/script/lua_api/l_item.cpp
@@ -57,6 +57,20 @@ int LuaItemStack::l_get_name(lua_State *L)
 	return 1;
 }
 
+// set_name(self, name)
+int LuaItemStack::l_set_name(lua_State *L)
+{
+	NO_MAP_LOCK_REQUIRED;
+	LuaItemStack *o = checkobject(L, 1);
+	ItemStack &item = o->m_stack;
+	item.name = luaL_checkstring(L, 2);
+
+	if (item.name == "" || item.empty())
+		item.clear();
+
+	return 1;
+}
+
 // get_count(self) -> number
 int LuaItemStack::l_get_count(lua_State *L)
 {
@@ -67,6 +81,20 @@ int LuaItemStack::l_get_count(lua_State *L)
 	return 1;
 }
 
+// set_count(self, number)
+int LuaItemStack::l_set_count(lua_State *L)
+{
+	NO_MAP_LOCK_REQUIRED;
+	LuaItemStack *o = checkobject(L, 1);
+	ItemStack &item = o->m_stack;
+	item.count = luaL_checkinteger(L, 2);
+
+	if (item.name == "" || item.empty())
+		item.clear();
+
+	return 1;
+}
+
 // get_wear(self) -> number
 int LuaItemStack::l_get_wear(lua_State *L)
 {
@@ -77,6 +105,20 @@ int LuaItemStack::l_get_wear(lua_State *L)
 	return 1;
 }
 
+// set_wear(self, number)
+int LuaItemStack::l_set_wear(lua_State *L)
+{
+	NO_MAP_LOCK_REQUIRED;
+	LuaItemStack *o = checkobject(L, 1);
+	ItemStack &item = o->m_stack;
+	item.wear = luaL_checkinteger(L, 2);
+
+	if (item.wear > 65535)
+		item.clear();
+
+	return 1;
+}
+
 // get_metadata(self) -> string
 int LuaItemStack::l_get_metadata(lua_State *L)
 {
@@ -87,6 +129,23 @@ int LuaItemStack::l_get_metadata(lua_State *L)
 	return 1;
 }
 
+// set_metadata(self, string)
+int LuaItemStack::l_set_metadata(lua_State *L)
+{
+	NO_MAP_LOCK_REQUIRED;
+	LuaItemStack *o = checkobject(L, 1);
+	ItemStack &item = o->m_stack;
+
+	size_t len = 0;
+	const char *ptr = luaL_checklstring(L, 2, &len);
+	if (ptr)
+		item.metadata.assign(ptr, len);
+	else
+		item.metadata = "";
+
+	return 1;
+}
+
 // clear(self) -> true
 int LuaItemStack::l_clear(lua_State *L)
 {
@@ -363,9 +422,13 @@ const char LuaItemStack::className[] = "ItemStack";
 const luaL_reg LuaItemStack::methods[] = {
 	luamethod(LuaItemStack, is_empty),
 	luamethod(LuaItemStack, get_name),
+	luamethod(LuaItemStack, set_name),
 	luamethod(LuaItemStack, get_count),
+	luamethod(LuaItemStack, set_count),
 	luamethod(LuaItemStack, get_wear),
+	luamethod(LuaItemStack, set_wear),
 	luamethod(LuaItemStack, get_metadata),
+	luamethod(LuaItemStack, set_metadata),
 	luamethod(LuaItemStack, clear),
 	luamethod(LuaItemStack, replace),
 	luamethod(LuaItemStack, to_string),

diff --git a/src/script/lua_api/l_item.h b/src/script/lua_api/l_item.h
@@ -41,15 +41,27 @@ class LuaItemStack : public ModApiBase {
 	// get_name(self) -> string
 	static int l_get_name(lua_State *L);
 
+	// set_name(self, name)
+	static int l_set_name(lua_State *L);
+
 	// get_count(self) -> number
 	static int l_get_count(lua_State *L);
 
+	// set_count(self, number)
+	static int l_set_count(lua_State *L);
+
 	// get_wear(self) -> number
 	static int l_get_wear(lua_State *L);
 
+	// set_wear(self, number)
+	static int l_set_wear(lua_State *L);
+
 	// get_metadata(self) -> string
 	static int l_get_metadata(lua_State *L);
 
+	// set_metadata(self, string)
+	static int l_set_metadata(lua_State *L);
+
 	// clear(self) -> true
 	static int l_clear(lua_State *L);