include "nullesc.lua"

AdvDupe2.NetFile = ""
local AutoSave = false
local uploading = false
local uploading_timeout = 0
local uploading_timeout_time = 10

local function CheckFileNameCl(path)
	if file.Exists(path..".txt", "DATA") then
		for i = 1, AdvDupe2.FileRenameTryLimit do
			if not file.Exists(path.."_"..i..".txt", "DATA") then
				return path.."_"..i
			end
		end
	end
	return path
end

--[[
	Name: AdvDupe2_ReceiveFile
	Desc: Receive file data from the server when downloading to the client
	Params: 
	Returns:
]]
local function AdvDupe2_ReceiveFile(len, ply, len2)
	local status = net.ReadInt(8)
	
	if(status==1)then AdvDupe2.NetFile = "" end
	AdvDupe2.NetFile=AdvDupe2.NetFile..net.ReadString()

	if(status==2)then
		local path = ""
		if(AutoSave)then
			if(LocalPlayer():GetInfo("advdupe2_auto_save_overwrite")~="1")then
				path = CheckFileNameCl(AdvDupe2.AutoSavePath)
			end
		else
			path = CheckFileNameCl(AdvDupe2.SavePath)
		end
		file.Write(path..".txt", AdvDupe2.Null.invesc(AdvDupe2.NetFile))
		
		if(!file.Exists(path..".txt", "DATA"))then
			AdvDupe2.NetFile = ""
			AdvDupe2.Notify("File was not saved!",NOTIFY_ERROR,5)
			return
		end
		
		local errored = false
		if(LocalPlayer():GetInfo("advdupe2_debug_openfile")=="1")then
			if(not file.Exists(path..".txt", "DATA"))then AdvDupe2.Notify("File does not exist", NOTIFY_ERROR) return end
			
			local read = file.Read(path..".txt")
			if not read then AdvDupe2.Notify("File could not be read", NOTIFY_ERROR) return end
			AdvDupe2.Decode(read, function(success,dupe,info,moreinfo) 
									if(success)then
										AdvDupe2.Notify("DEBUG CHECK: File successfully opens. No EOF errors.")
									else
										AdvDupe2.Notify("DEBUG CHECK: File contains EOF errors.", NOTIFY_ERROR)
										errored = true
									end
										end)
		end
		
		local filename = string.Explode("/", path)
		filename = filename[#filename]
		if(AutoSave)then
			if(IsValid(AdvDupe2.FileBrowser.AutoSaveNode))then
				local add = true
				for i=1, #AdvDupe2.FileBrowser.AutoSaveNode.Files do
					if(filename==AdvDupe2.FileBrowser.AutoSaveNode.Files[i].Label:GetText())then
						add=false
						break
					end
				end
				if(add)then
					AdvDupe2.FileBrowser.AutoSaveNode:AddFile(filename)
					AdvDupe2.FileBrowser.Browser.pnlCanvas:Sort(AdvDupe2.FileBrowser.AutoSaveNode)
				end
			end
		else
			AdvDupe2.FileBrowser.Browser.pnlCanvas.ActionNode:AddFile(filename)
			AdvDupe2.FileBrowser.Browser.pnlCanvas:Sort(AdvDupe2.FileBrowser.Browser.pnlCanvas.ActionNode)
		end
		AdvDupe2.NetFile = ""
		if(!errored)then
			AdvDupe2.Notify("File successfully saved!",NOTIFY_GENERIC, 5)
		end
		return
	end

end
net.Receive("AdvDupe2_ReceiveFile", AdvDupe2_ReceiveFile)

function AdvDupe2.LoadGhosts(dupe, info, moreinfo, name, preview)

	AdvDupe2.RemoveGhosts()
	AdvDupe2.Ghosting = true
	
	if(preview)then
		AdvDupe2.Preview = true
		if(AdvDupe2.HeadEnt)then
			AdvDupe2.PHeadEnt = AdvDupe2.HeadEnt
			AdvDupe2.PHeadZPos = AdvDupe2.HeadZPos
			AdvDupe2.PHeadPos = AdvDupe2.HeadPos*1
			AdvDupe2.PHeadOffset = AdvDupe2.HeadOffset*1
			AdvDupe2.PHeadAngle = AdvDupe2.HeadAngle*1
			AdvDupe2.GhostToPreview = table.Copy(AdvDupe2.GhostToSpawn)
		end
	else
		AdvDupe2.PHeadEnt = nil
		AdvDupe2.PHeadZPos = nil
		AdvDupe2.PHeadPos = nil
		AdvDupe2.PHeadOffset = nil
		AdvDupe2.PHeadAngle = nil
		AdvDupe2.GhostToPreview = nil
		AdvDupe2.Preview=false
	end
	
	AdvDupe2.GhostToSpawn = {}
	local count = 0
	
	local time
	local desc
	local date
	local creator
	
	if(info.ad1)then
		time = moreinfo["Time"] or ""
		desc = info["Description"] or ""
		date = info["Date"] or ""
		creator = info["Creator"] or ""

		AdvDupe2.HeadEnt = tonumber(moreinfo.Head)
		local spx,spy,spz = moreinfo.StartPos:match("^(.-),(.-),(.+)$")
		AdvDupe2.HeadPos = Vector(tonumber(spx) or 0, tonumber(spy) or 0, tonumber(spz) or 0)
		local z = (tonumber(moreinfo.HoldPos:match("^.-,.-,(.+)$")) or 0)*-1
		AdvDupe2.HeadZPos = z
		AdvDupe2.HeadPos.Z = AdvDupe2.HeadPos.Z + z

		local Pos
		local Ang
		for k,v in pairs(dupe["Entities"])do
			
			if(v.SavedParentIdx)then 
				if(not v.BuildDupeInfo)then v.BuildDupeInfo = {} end
				v.BuildDupeInfo.DupeParentID = v.SavedParentIdx
				Pos = v.LocalPos*1
				Ang = v.LocalAngle*1
			else
				Pos = nil
				Ang = nil
			end
			for i,p in pairs(v.PhysicsObjects)do
				p.Pos = Pos or (p.LocalPos*1)
				p.Pos.Z = p.Pos.Z - z
				p.Angle = Ang or (p.LocalAngle*1)
				p.LocalPos = nil
				p.LocalAngle = nil
			end
			v.LocalPos = nil
			v.LocalAngle = nil
			AdvDupe2.GhostToSpawn[count] = {Model=v.Model, PhysicsObjects=v.PhysicsObjects}
			if(AdvDupe2.HeadEnt == k)then
				AdvDupe2.HeadEnt = count
			end
			count = count + 1
		end
		
		AdvDupe2.HeadOffset = AdvDupe2.GhostToSpawn[AdvDupe2.HeadEnt].PhysicsObjects[0].Pos
		AdvDupe2.HeadAngle = AdvDupe2.GhostToSpawn[AdvDupe2.HeadEnt].PhysicsObjects[0].Angle
	
	else
		time = info["time"]
		desc = dupe["Description"]
		date = info["date"]
		creator = info["name"]

		AdvDupe2.HeadEnt = dupe["HeadEnt"].Index
		AdvDupe2.HeadZPos = dupe["HeadEnt"].Z
		AdvDupe2.HeadPos = dupe["HeadEnt"].Pos
		AdvDupe2.HeadOffset = dupe["Entities"][AdvDupe2.HeadEnt].PhysicsObjects[0].Pos
		AdvDupe2.HeadAngle = dupe["Entities"][AdvDupe2.HeadEnt].PhysicsObjects[0].Angle
		
		for k,v in pairs(dupe["Entities"])do
			AdvDupe2.GhostToSpawn[count] = {Model=v.Model, PhysicsObjects=v.PhysicsObjects}
			if(AdvDupe2.HeadEnt == k)then
				AdvDupe2.HeadEnt = count
			end
			count = count + 1
		end
	end
	
	if(not preview)then
		AdvDupe2.Info.File:SetText("File: "..name)
		AdvDupe2.Info.Creator:SetText("Creator: "..creator)
		AdvDupe2.Info.Date:SetText("Date: "..date)
		AdvDupe2.Info.Time:SetText("Time: "..time)
		AdvDupe2.Info.Size:SetText("Size: "..string.NiceSize(tonumber(info.size) or 0))
		AdvDupe2.Info.Desc:SetText("Desc: "..(desc or ""))
		AdvDupe2.Info.Entities:SetText("Entities: "..table.Count(dupe["Entities"]))
		AdvDupe2.Info.Constraints:SetText("Constraints: "..table.Count(dupe["Constraints"]))
	end
	
	AdvDupe2.StartGhosting()

end

--[[
	Name: InitializeUpload
	Desc: When the client clicks upload, prepares to send data to the server
	Params: File Data, Path to save
	Returns:
]]			
function AdvDupe2.InitializeUpload(ReadPath, ReadArea)
	if(uploading and CurTime()<uploading_timeout)then AdvDupe2.Notify("Already opening file, please wait.", NOTIFY_ERROR) return end
	uploading_timeout = CurTime()+uploading_timeout_time
	if(ReadArea==0)then
		ReadPath = AdvDupe2.DataFolder.."/"..ReadPath..".txt"
	elseif(ReadArea==1)then
		ReadPath = AdvDupe2.DataFolder.."/-Public-/"..ReadPath..".txt"
	else
		ReadPath = "adv_duplicator/"..ReadPath..".txt"
	end
	
	if(not file.Exists(ReadPath, "DATA"))then AdvDupe2.Notify("File does not exist", NOTIFY_ERROR) return end
	
	local read = file.Read(ReadPath)
	if not read then AdvDupe2.Notify("File could not be read", NOTIFY_ERROR) return end
	local name = string.Explode("/", ReadPath)
	name = name[#name]
	name = string.sub(name, 1, #name-4)
	
	uploading = true
	
	AdvDupe2.Decode(read, function(success, dupe, info, moreinfo) 
		if(success)then
			AdvDupe2.PendingDupe = { read, dupe, info, moreinfo, name }
			RunConsoleCommand("AdvDupe2_InitReceiveFile")
		else
			uploading = false
			AdvDupe2.Notify("File could not be decoded. Upload Canceled.", NOTIFY_ERROR)
		end 
	end)
end

--[[
	Name: SendFileToServer
	Desc: Send chunks of the file data to the server
	Params: end of file
	Returns:
]]
local function SendFileToServer(eof)
	uploading_timeout = CurTime()+uploading_timeout_time

	if(AdvDupe2.LastPos+eof>AdvDupe2.Length)then
		eof = AdvDupe2.Length
	end

	local data = string.sub(AdvDupe2.File, AdvDupe2.LastPos, AdvDupe2.LastPos+eof)

	AdvDupe2.LastPos = AdvDupe2.LastPos+eof+1
	AdvDupe2.ProgressBar.Percent = math.min(math.floor((AdvDupe2.LastPos/AdvDupe2.Length)*100),100)

	net.Start("AdvDupe2_ReceiveFile")
		net.WriteBit(AdvDupe2.LastPos>=AdvDupe2.Length)
		net.WriteString(data)
	net.SendToServer()
	
end

usermessage.Hook("AdvDupe2_ReceiveNextStep",function(um)
	if AdvDupe2.PendingDupe then
		local read,dupe,info,moreinfo,name = unpack( AdvDupe2.PendingDupe )
		
		AdvDupe2.PendingDupe = nil
		AdvDupe2.LoadGhosts(dupe, info, moreinfo, name )
		AdvDupe2.File = AdvDupe2.Null.esc(read)
		AdvDupe2.LastPos = 0
		AdvDupe2.Length = string.len(AdvDupe2.File)
		AdvDupe2.InitProgressBar("Opening:")
	end
	
	if uploading then
		SendFileToServer(um:ReadShort())
	end
end)

usermessage.Hook("AdvDupe2_UploadRejected",function(um)
	if uploading then
		uploading = false
		AdvDupe2.PendingDupe = nil
		AdvDupe2.File = nil
		AdvDupe2.LastPos = nil
		AdvDupe2.Length = nil
	end
	if(um:ReadBool())then AdvDupe2.RemoveProgressBar() end
end)

concommand.Add("AdvDupe2_SaveType", function(ply, cmd, args)
	AutoSave = args[1]=="1"
end)