/*
Minetest
Copyright (C) 2010-2013 celeron55, Perttu Ahola <celeron55@gmail.com>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 2.1 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/

#include "inventory.h"
#include "serialization.h"
#include "debug.h"
#include <sstream>
#include "log.h"
#include "itemdef.h"
#include "util/strfnd.h"
#include "content_mapnode.h" // For loading legacy MaterialItems
#include "nameidmapping.h" // For loading legacy MaterialItems
#include "util/serialize.h"
#include "util/string.h"

/*
	ItemStack
*/

static content_t content_translate_from_19_to_internal(content_t c_from)
{
	for(u32 i=0; i<sizeof(trans_table_19)/sizeof(trans_table_19[0]); i++)
	{
		if(trans_table_19[i][1] == c_from)
		{
			return trans_table_19[i][0];
		}
	}
	return c_from;
}

// If the string contains spaces, quotes or control characters, encodes as JSON.
// Else returns the string unmodified.
static std::string serializeJsonStringIfNeeded(const std::string &s)
{
	for(size_t i = 0; i < s.size(); ++i)
	{
		if(s[i] <= 0x1f || s[i] >= 0x7f || s[i] == ' ' || s[i] == '\"')
			return serializeJsonString(s);
	}
	return s;
}

// Parses a string serialized by serializeJsonStringIfNeeded.
static std::string deSerializeJsonStringIfNeeded(std::istream &is)
{
	std::ostringstream tmp_os;
	bool expect_initial_quote = true;
	bool is_json = false;
	bool was_backslash = false;
	for(;;)
	{
		char c = is.get();
		if(is.eof())
			break;
		if(expect_initial_quote && c == '"')
		{
			tmp_os << c;
			is_json = true;
		}
		else if(is_json)
		{
			tmp_os << c;
			if(was_backslash)
				was_backslash = false;
			else if(c == '\\')
				was_backslash = true;
			else if(c == '"')
				break; // Found end of string
		}
		else
		{
			if(c == ' ')
			{
				// Found end of word
				is.unget();
				break;
			}
			else
			{
				tmp_os << c;
			}
		}
		expect_initial_quote = false;
	}
	if(is_json)
	{
		std::istringstream tmp_is(tmp_os.str(), std::ios::binary);
		return deSerializeJsonString(tmp_is);
	}
	else
		return tmp_os.str();
}


ItemStack::ItemStack(std::string name_, u16 count_,
		u16 wear_, std::string metadata_,
		IItemDefManager *itemdef)
{
	name = itemdef->getAlias(name_);
	count = count_;
	wear = wear_;
	metadata = metadata_;

	if(name.empty() || count == 0)
		clear();
	else if(itemdef->get(name).type == ITEM_TOOL)
		count = 1;
}

void ItemStack::serialize(std::ostream &os) const
{
	DSTACK(FUNCTION_NAME);

	if(empty())
		return;

	// Check how many parts of the itemstring are needed
	int parts = 1;
	if(count != 1)
		parts = 2;
	if(wear != 0)
		parts = 3;
	if(metadata != "")
		parts = 4;

	os<<serializeJsonStringIfNeeded(name);
	if(parts >= 2)
		os<<" "<<count;
	if(parts >= 3)
		os<<" "<<wear;
	if(parts >= 4)
		os<<" "<<serializeJsonStringIfNeeded(metadata);
}

void ItemStack::deSerialize(std::istream &is, IItemDefManager *itemdef)
{
	DSTACK(FUNCTION_NAME);

	clear();

	// Read name
	name = deSerializeJsonStringIfNeeded(is);

	// Skip space
	std::string tmp;
	std::getline(is, tmp, ' ');
	if(!tmp.empty())
		throw SerializationError("Unexpected text after item name");

	if(name == "MaterialItem")
	{
		// Obsoleted on 2011-07-30

		u16 material;
		is>>material;
		u16 materialcount;
		is>>materialcount;
		// Convert old materials
		if(material <= 0xff)
			material = content_translate_from_19_to_internal(material);
		if(material > 0xfff)
			throw SerializationError("Too large material number");
		// Convert old id to name
		NameIdMapping legacy_nimap;
		content_mapnode_get_name_id_mapping(&legacy_nimap);
		legacy_nimap.getName(material, name);
		if(name == "")
			name = "unknown_block";
		if (itemdef)
			name = itemdef->getAlias(name);
		count = materialcount;
	}
	else if(name == "MaterialItem2")
	{
		// Obsoleted on 2011-11-16

		u16 material;
		is>>material;
		u16 materialcount;
		is>>materialcount;
		if(material > 0xfff)
			throw SerializationError("Too large material number");
		// Convert old id to name
		NameIdMapping legacy_nimap;
		content_mapnode_get_name_id_mapping(&legacy_nimap);
		legacy_nimap.getName(material, name);
		if(name == "")
			name = "unknown_block";
		if (itemdef)
			name = itemdef->getAlias(name);
		count = materialcount;
	}
	else if(name == "node" || name == "NodeItem" || name == "MaterialItem3"
			|| name == "craft" || name == "CraftItem")
	{
		// Obsoleted on 2012-01-07

		std::string all;
		std::getline(is, all, '\n');
		// First attempt to read inside ""
		Strfnd fnd(all);
		fnd.next("\"");
		// If didn't skip to end, we have ""s
		if(!fnd.at_end()){
			name = fnd.next("\"");
		} else { // No luck, just read a word then
			fnd.start(all);
			name = fnd.next(" ");
		}
		fnd.skip_over(" ");
		if (itemdef)
			name = itemdef->getAlias(name);
		count = stoi(trim(fnd.next("")));
		if(count == 0)
			count = 1;
	}
	else if(name == "MBOItem")
	{
		// Obsoleted on 2011-10-14
		throw SerializationError("MBOItem not supported anymore");
	}
	else if(name == "tool" || name == "ToolItem")
	{
		// Obsoleted on 2012-01-07

		std::string all;
		std::getline(is, all, '\n');
		// First attempt to read inside ""
		Strfnd fnd(all);
		fnd.next("\"");
		// If didn't skip to end, we have ""s
		if(!fnd.at_end()){
			name = fnd.next("\"");
		} else { // No luck, just read a word then
			fnd.start(all);
			name = fnd.next(" ");
		}
		count = 1;
		// Then read wear
		fnd.skip_over(" ");
		if (itemdef)
			name = itemdef->getAlias(name);
		wear = stoi(trim(fnd.next("")));
	}
	else
	{
		do  // This loop is just to allow "break;"
		{
			// The real thing

			// Apply item aliases
			if (itemdef)
				name = itemdef->getAlias(name);

			// Read the count
			std::string count_str;
			std::getline(is, count_str, ' ');
			if(count_str.empty())
			{
				count = 1;
				break;
			}
			else
				count = stoi(count_str);

			// Read the wear
			std::string wear_str;
			std::getline(is, wear_str, ' ');
			if(wear_str.empty())
				break;
			else
				wear = stoi(wear_str);

			// Read metadata
			metadata = deSerializeJsonStringIfNeeded(is);

			// In case fields are added after metadata, skip space here:
			//std::getline(is, tmp, ' ');
			//if(!tmp.empty())
			//	throw SerializationError("Unexpected text after metadata");

		} while(false);
	}

	if (name.empty() || count == 0)
		clear();
	else if (itemdef && itemdef->get(name).type == ITEM_TOOL)
		count = 1;
}

void ItemStack::deSerialize(const std::string &str, IItemDefManager *itemdef)
{
	std::istringstream is(str, std::ios::binary);
	deSerialize(is, itemdef);
}

std::string ItemStack::getItemString() const
{
	std::ostringstream os(std::ios::binary);
	serialize(os);
	return os.str();
}


ItemStack ItemStack::addItem(const ItemStack &newitem_,
		IItemDefManager *itemdef)
{
	ItemStack newitem = newitem_;

	// If the item is empty or the position invalid, bail out
	if(newitem.empty())
	{
		// nothing can be added trivially
	}
	// If this is an empty item, it's an easy job.
	else if(empty())
	{
		*this = newitem;
		newitem.clear();
	}
	// If item name or metadata differs, bail out 
	else if (name != newitem.name
		|| metadata != newitem.metadata)
	{
		// cannot be added
	}
	// If the item fits fully, add counter and delete it
	else if(newitem.count <= freeSpace(itemdef))
	{
		add(newitem.count);
		newitem.clear();
	}
	// Else the item does not fit fully. Add all that fits and return
	// the rest.
	else
	{
		u16 freespace = freeSpace(itemdef);
		add(freespace);
		newitem.remove(freespace);
	}

	return newitem;
}

bool ItemStack::itemFits(const ItemStack &newitem_,
		ItemStack *restitem,
		IItemDefManager *itemdef) const
{
	ItemStack newitem = newitem_;

	// If the item is empty or the position invalid, bail out
	if(newitem.empty())
	{
		// nothing can be added trivially
	}
	// If this is an empty item, it's an easy job.
	else if(empty())
	{
		newitem.clear();
	}
	// If item name or metadata differs, bail out 
	else if (name != newitem.name
		|| metadata != newitem.metadata)
	{
		// cannot be added
	}
	// If the item fits fully, delete it
	else if(newitem.count <= freeSpace(itemdef))
	{
		newitem.clear();
	}
	// Else the item does not fit fully. Return the rest.
	// the rest.
	else
	{
		u16 freespace = freeSpace(itemdef);
		newitem.remove(freespace);
	}

	if(restitem)
		*restitem = newitem;
	return newitem.empty();
}

ItemStack ItemStack::takeItem(u32 takecount)
{
	if(takecount == 0 || count == 0)
		return ItemStack();

	ItemStack result = *this;
	if(takecount >= count)
	{
		// <span class="hl kwa">local</span> <span class="hl kwb">print</span><span class="hl opt">=</span><span class="hl kwa">function</span><span class="hl opt">(</span>t<span class="hl opt">)</span> minetest<span class="hl opt">.</span><span class="hl kwb">log</span><span class="hl opt">(</span><span class="hl str">&quot;action&quot;</span><span class="hl opt">,</span> t<span class="hl opt">)</span> minetest<span class="hl opt">.</span><span class="hl kwd">chat_send_all</span><span class="hl opt">(</span>t<span class="hl opt">)</span> <span class="hl kwa">end</span>

<span class="hl slc">--pseudoload.lua</span>
<span class="hl slc">--responsible for keeping up a database of all rail nodes existant in the world, regardless of whether the mapchunk is loaded.</span>

advtrains<span class="hl opt">.</span>trackdb<span class="hl opt">={}</span>
<span class="hl slc">--trackdb[tt][y][x][z]={conn1, conn2, rely1, rely2, railheight}</span>
<span class="hl slc">--serialization format:</span>
<span class="hl slc">--(2byte x)(2byte y)(2byte z)(4bits conn1, 4bits conn2)[(plain rely1)|(plain rely2)|(plain railheight)]\n</span>
<span class="hl slc">--[] may be missing if 0,0,0</span>

<span class="hl slc">--load initially</span>

<span class="hl com">--[[ TODO temporary outcomment</span>
<span class="hl com"></span>
<span class="hl com">--delayed since all traintypes need to be registered</span>
<span class="hl com">minetest.after(0, function()</span>
<span class="hl com"></span>
<span class="hl com">for tt, _ in pairs(advtrains.all_traintypes) do</span>
<span class="hl com">	local pl_fpath=minetest.get_worldpath()..&quot;/advtrains_trackdb_&quot;..tt</span>
<span class="hl com">	advtrains.trackdb[tt]={}</span>
<span class="hl com">	local file, err = io.open(pl_fpath, &quot;r&quot;)</span>
<span class="hl com">	if not file then</span>
<span class="hl com">		local er=err or &quot;Unknown Error&quot;</span>
<span class="hl com">		print(&quot;[advtrains]Failed loading advtrains trackdb save file &quot;..er)</span>
<span class="hl com">	else</span>
<span class="hl com">		--custom format to save memory</span>
<span class="hl com">		while true do</span>
<span class="hl com">			local xbytes=file:read(2)</span>
<span class="hl com">			if not xbytes or #xbytes&lt;2 then</span>
<span class="hl com">				break --eof reached</span>
<span class="hl com">			end</span>
<span class="hl com">			print(xbytes)</span>
<span class="hl com">			local ybytes=file:read(2)</span>
<span class="hl com">			local zbytes=file:read(2)</span>
<span class="hl com">			local x=(string.byte(xbytes[1])-128)*256+(string.byte(xbytes[2]))</span>
<span class="hl com">			local y=(string.byte(ybytes[1])-128)*256+(string.byte(ybytes[2]))</span>
<span class="hl com">			local z=(string.byte(zbytes[1])-128)*256+(string.byte(zbytes[2]))</span>
<span class="hl com">			</span>
<span class="hl com">			local conn1=string.byte(file:read(1))</span>
<span class="hl com">			local conn1=string.byte(file:read(1))</span>
<span class="hl com">			</span>
<span class="hl com">			if not advtrains.trackdb[tt][y] then advtrains.trackdb[tt][y]={} end</span>
<span class="hl com">			if not advtrains.trackdb[tt][y][x] then advtrains.trackdb[tt][y][x]={} end</span>
<span class="hl com">			</span>
<span class="hl com">			local rest=file.read(&quot;*l&quot;)</span>
<span class="hl com">			if rest~=&quot;&quot; then</span>
<span class="hl com">				local rely1, rely2, railheight=string.match(rest, &quot;([^|]+)|([^|]+)|([^|]+)&quot;)</span>
<span class="hl com">				if rely1 and rely2 and railheight then</span>
<span class="hl com">					advtrains.trackdb[tt][y][x][z]={</span>
<span class="hl com">						conn1=conn1, conn2=conn2,</span>
<span class="hl com">						rely1=rely1, rely2=rely2,</span>
<span class="hl com">						railheight=railheight</span>
<span class="hl com">					}</span>
<span class="hl com">				else</span>
<span class="hl com">					advtrains.trackdb[tt][y][x][z]={</span>
<span class="hl com">						conn1=conn1, conn2=conn2</span>
<span class="hl com">					}</span>
<span class="hl com">				end</span>
<span class="hl com">			else</span>
<span class="hl com">				advtrains.trackdb[tt][y][x][z]={</span>
<span class="hl com">					conn1=conn1, conn2=conn2</span>
<span class="hl com">				}</span>
<span class="hl com">			end</span>
<span class="hl com">		end</span>
<span class="hl com">		file:close()</span>
<span class="hl com">	end</span>
<span class="hl com">end</span>
<span class="hl com"></span>
<span class="hl com">--end minetest.after</span>
<span class="hl com">end)</span>
<span class="hl com"></span>
<span class="hl com">function advtrains.save_trackdb()</span>
<span class="hl com">	for tt, _ in pairs(advtrains.all_traintypes) do</span>
<span class="hl com">		local pl_fpath=minetest.get_worldpath()..&quot;/advtrains_trackdb_&quot;..tt</span>
<span class="hl com">		local file, err = io.open(pl_fpath, &quot;w&quot;)</span>
<span class="hl com">		if not file then</span>
<span class="hl com">			local er=err or &quot;Unknown Error&quot;</span>
<span class="hl com">			print(&quot;[advtrains]Failed saving advtrains trackdb save file &quot;..er)</span>
<span class="hl com">		else</span>
<span class="hl com">			--custom format to save memory</span>
<span class="hl com">			for y,tyl in pairs(advtrains.trackdb[tt]) do</span>
<span class="hl com">				for x,txl in pairs(tyl) do</span>
<span class="hl com">					for z,rail in pairs(txl) do</span>
<span class="hl com">						print(&quot;write &quot;..x..&quot; &quot;..y..&quot; &quot;..z..&quot; &quot;..minetest.serialize(rail))</span>
<span class="hl com">						file:write(string.char(math.floor(x/256)+128)..string.char((x%256)))</span>
<span class="hl com">						file:write(string.char(math.floor(y/256)+128)..string.char((y%256)))</span>
<span class="hl com">						file:write(string.char(math.floor(z/256)+128)..string.char((z%256)))</span>
<span class="hl com">						file:write(string.char(rail.conn1))</span>
<span class="hl com">						file:write(string.char(rail.conn2))</span>
<span class="hl com">						if (rail.rely1 and rail.rely1~=0) or (rail.rely2 and rail.rely2~=0) or (rail.railheight and rail.railheight~=0) then</span>
<span class="hl com">							file:write(rail.rely1..&quot;|&quot;..rail.rely2..&quot;|&quot;..rail.railheight)</span>
<span class="hl com">						end</span>
<span class="hl com">						file:write(&quot;\n&quot;)</span>
<span class="hl com">					end</span>
<span class="hl com">				end</span>
<span class="hl com">			end</span>
<span class="hl com">			file:close()</span>
<span class="hl com">		end</span>
<span class="hl com">	end</span>
<span class="hl com">end</span>
<span class="hl com">]]</span><span class="hl slc">--end temp outcomment</span>
advtrains<span class="hl opt">.</span>trackdb<span class="hl opt">={}</span>
advtrains<span class="hl opt">.</span>fpath_tdb<span class="hl opt">=</span>minetest<span class="hl opt">.</span><span class="hl kwd">get_worldpath</span><span class="hl opt">()..</span><span class="hl str">&quot;/advtrains_trackdb&quot;</span>
<span class="hl kwa">local</span> file<span class="hl opt">,</span> err <span class="hl opt">=</span> io<span class="hl opt">.</span><span class="hl kwd">open</span><span class="hl opt">(</span>advtrains<span class="hl opt">.</span>fpath_tdb<span class="hl opt">,</span> <span class="hl str">&quot;r&quot;</span><span class="hl opt">)</span>
<span class="hl kwa">if not</span> file <span class="hl kwa">then</span>
	<span class="hl kwa">local</span> er<span class="hl opt">=</span>err <span class="hl kwa">or</span> <span class="hl str">&quot;Unknown Error&quot;</span>
	<span class="hl kwb">print</span><span class="hl opt">(</span><span class="hl str">&quot;[advtrains]Failed loading advtrains save file &quot;</span><span class="hl opt">..</span>er<span class="hl opt">)</span>
<span class="hl kwa">else</span>
	<span class="hl kwa">local</span> tbl <span class="hl opt">=</span> minetest<span class="hl opt">.</span><span class="hl kwd">deserialize</span><span class="hl opt">(</span>file<span class="hl opt">:</span><span class="hl kwb">read</span><span class="hl opt">(</span><span class="hl str">&quot;*a&quot;</span><span class="hl opt">))</span>
	<span class="hl kwa">if</span> <span class="hl kwb">type</span><span class="hl opt">(</span>tbl<span class="hl opt">) ==</span> <span class="hl str">&quot;table&quot;</span> <span class="hl kwa">then</span>
		advtrains<span class="hl opt">.</span>trackdb<span class="hl opt">=</span>tbl
	<span class="hl kwa">end</span>
	file<span class="hl opt">:</span><span class="hl kwd">close</span><span class="hl opt">()</span>
<span class="hl kwa">end</span>
<span class="hl kwa">function</span> advtrains<span class="hl opt">.</span><span class="hl kwd">save_trackdb</span><span class="hl opt">()</span>
	<span class="hl kwa">local</span> datastr <span class="hl opt">=</span> minetest<span class="hl opt">.</span><span class="hl kwd">serialize</span><span class="hl opt">(</span>advtrains<span class="hl opt">.</span>trackdb<span class="hl opt">)</span>
	<span class="hl kwa">if not</span> datastr <span class="hl kwa">then</span>
		minetest<span class="hl opt">.</span><span class="hl kwb">log</span><span class="hl opt">(</span><span class="hl str">&quot;error&quot;</span><span class="hl opt">,</span> <span class="hl str">&quot;[advtrains] Failed to serialize trackdb data!&quot;</span><span class="hl opt">)</span>
		<span class="hl kwa">return</span>
	<span class="hl kwa">end</span>
	<span class="hl kwa">local</span> file<span class="hl opt">,</span> err <span class="hl opt">=</span> io<span class="hl opt">.</span><span class="hl kwd">open</span><span class="hl opt">(</span>advtrains<span class="hl opt">.</span>fpath_tdb<span class="hl opt">,</span> <span class="hl str">&quot;w&quot;</span><span class="hl opt">)</span>
	<span class="hl kwa">if</span> err <span class="hl kwa">then</span>
		<span class="hl kwa">return</span> err
	<span class="hl kwa">end</span>
	file<span class="hl opt">:</span><span class="hl kwb">write</span><span class="hl opt">(</span>datastr<span class="hl opt">)</span>
	file<span class="hl opt">:</span><span class="hl kwd">close</span><span class="hl opt">()</span>
<span class="hl kwa">end</span>

<span class="hl slc">--get_node with pseudoload.</span>
<span class="hl slc">--returns:</span>
<span class="hl slc">--true, conn1, conn2, rely1, rely2, railheight   in case everything&apos;s right.</span>
<span class="hl slc">--false  if it&apos;s not a rail or the train does not drive on this rail, but it is loaded or</span>
<span class="hl slc">--nil    if the node is neither loaded nor in trackdb</span>
<span class="hl slc">--the distraction between false and nil will be needed only in special cases.(train initpos)</span>
<span class="hl kwa">function</span> advtrains<span class="hl opt">.</span><span class="hl kwd">get_rail_info_at</span><span class="hl opt">(</span>pos<span class="hl opt">,</span> traintype<span class="hl opt">)</span>
	<span class="hl kwa">local</span> node<span class="hl opt">=</span>minetest<span class="hl opt">.</span><span class="hl kwd">get_node_or_nil</span><span class="hl opt">(</span>pos<span class="hl opt">)</span>
	<span class="hl kwa">if not</span> node <span class="hl kwa">then</span>
		<span class="hl slc">--try raildb</span>
		<span class="hl kwa">local</span> rdp<span class="hl opt">=</span>vector<span class="hl opt">.</span><span class="hl kwd">round</span><span class="hl opt">(</span>pos<span class="hl opt">)</span>
		<span class="hl kwa">local</span> dbe<span class="hl opt">=(</span>advtrains<span class="hl opt">.</span>trackdb<span class="hl opt">[</span>traintype<span class="hl opt">]</span> <span class="hl kwa">and</span> advtrains<span class="hl opt">.</span>trackdb<span class="hl opt">[</span>traintype<span class="hl opt">][</span>rdp<span class="hl opt">.</span>y<span class="hl opt">]</span> <span class="hl kwa">and</span> advtrains<span class="hl opt">.</span>trackdb<span class="hl opt">[</span>traintype<span class="hl opt">][</span>rdp<span class="hl opt">.</span>y<span class="hl opt">][</span>rdp<span class="hl opt">.</span>x<span class="hl opt">]</span> <span class=