File rollout server

--[[  Fileserver - Senders call for a file update on the machine.
-- This machine responds with the files from within the
-- File server folder.
--]]
-- Local Vars
local serverdir = "\\server\\"
local clientdir = "\\client\\"

-- setup rednet
local sModemSide = nil
for n,sSide in ipairs(rs.getSides()) do
  if peripheral.getType(sSide) == "modem" then
	sModemSide = sSide
	break
  end
end
if sModemSide == nil then
  print("No wireless modem attached")
  return
end
rednet.open(sModemSide)
-- end setup rednet

--	 *************
--	 * Functions *
--	 *************
-- Sending file function
local function sendFile(filename, computerID, filedir)
local file = filedir .. filename
if not fs.exists(file) or fs.isDir(file) then
  print("File can't be sent!")
  print("It is either non existant or is a directory!")
  return false
end

computerID = tonumber(computerID)

local sr = fs.open(file, "r")
print("Connecting")

while true do
  rednet.send(computerID, "bt.con")
  os.startTimer(1)
  local e,p1,p2,p3 = os.pullEvent()
  if e == "rednet_message" then
   local id,msg = p1,p2
   if msg == "bt.acc" then
	print("connected!")
	break
   end
  end
end

sleep(0.8)
rednet.send(computerID, "bt.fn:" .. filename)

sleep(0.5)

print("Sending...")
local data
--while true do
  data = textutils.serialize(sr.readAll())
  print (data)
--  if data == nil then
--   break
--  end
  rednet.send(computerID, "bt.file:" .. data)
  sleep(0.8)
--end
rednet.send(computerID, "bt.done")
print("File sent!")
end

-- Check if folder exists --
function FolderExists(strFolderName)
local fileHandle = fs.isDir(strFolderName)
if fileHandle == false then
  return false
else
  return true
end
end
-- function get Dir listing
function Dirlist(folder)
local list = fs.list(folder)
return list
end
-- Sending of Programs funct
local args = {}
function Sendprogs( list, directory, cmpId, ... )
  args = ( ... )
  if type(list) ~= "table" then
	return false
  else
	for i = 1,#list do
  err = sendFile(list[i], cmpId)
end
  end

end
-- one time grab of the directory on boot up
-- Main program
while true do
id,msg = "",""
id, msg = rednet.receive(0.8)
if msg == "update_client" then
  list = Dirlist(clientdir)
  print("Client update starting")
  for i = 1,#list do
   sendFile(list[i], id, clientdir)
  end
elseif msg == "update_server" then
  list = Dirlist(serverdir)
  print("Server update starting")
  for i = 1,#list do
   sendFile(list[i], id, serverdir)
  end
end
rednet.send(id, "bt.eol")
end
--[[

err = Sendprogs( list, filedir, cmpId )
--]]
-- EOF.

-- File Updater Program
-- Varibles
local filesrv = 1  -- Your update server ID.
local cmptype = "client"
-- setup rednet
local sModemSide = nil
for n,sSide in ipairs(rs.getSides()) do
  if peripheral.getType(sSide) == "modem" then
	sModemSide = sSide
	break
  end
end
if sModemSide == nil then
  print("No wireless modem attached")
  return
end
rednet.open(sModemSide)
-- end setup rednet

-- *************
-- * Functions *
-- *************
-- Receive File.
local function receiveFile()
shell.run("id")
print("Waiting for a bt connection...")
rednet.send(filesrv,"update_client")

while true do
  local id,msg
  local id,msg,p3 = rednet.receive()

  if msg == "bt.eol" then printError("Finished file update") shell.run("delete fileupdate") shell.run("rename fileupdater fileupdate") shell.run("startup") error() end
   if msg == "bt.con" then
	print("Connected to " .. id .. "!")
	local remoteID = id
	rednet.send(id, "bt.acc")
  
	local fileName
  
	print("Getting the file name...")
  
	local id,msg = "",""
	while true do
	 id,msg = rednet.receive()
	
	 if id == remoteID and string.sub(msg, 1, #"bt.fn:") == "bt.fn:" then
	  fileName = string.sub(msg, #"bt.fn:" + 1)
	  break
	 end
	end

	if fs.exists(fileName) then
	 print("File already exists, Deleting old file... " .. fileName)
	 fs.delete(fileName)
	end
  
	print("Receiving file: " .. fileName)
  
	local file = fs.open(fileName, "w")

	local id,msg

	while true do
	 id,msg = rednet.receive()
	 if id == remoteID then
	  if string.sub(msg, 1, #"bt.file:") == "bt.file:" then
	   local subbedData = textutils.unserialize(string.sub(msg, #"bt.file:" + 1))
	
	   if term.isColor() then
		term.setTextColor(colors.lightBlue)
	   end
	  
	   file.writeLine(subbedData)
	  elseif msg == "bt.done" then
	   file.flush()
	   file.close()
	
	   if term.isColor() then
		term.setTextColor(colors.white)
	   end
	   print("Received file!")
	
	   break
	  end
	 end
	end
   end
end
end
-- Main Program
print("updating files")
receiveFile()