Myros27

DefragBot V2

May 18th, 2025 (edited)
46
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Lua 35.98 KB | None | 0 0
  1. --[[
  2.     DefragBot v1.8 - Complete Merged Script
  3.     - Runs performDefragCycle in the main thread for easier debugging.
  4.     - Reverted announce to simpler cooldown (os.sleep(0.2) directly in announce).
  5.     - Kept robust pcall chat sending and file logging.
  6.     - CommandHandlers table fixed.
  7. ]]
  8.  
  9. --#region Configuration
  10. local CHAT_BOX_PERIPHERAL_NAME = "chatBox"
  11. local COMMAND_PREFIX = "@defrag"
  12. local CHAT_BOT_NAME = "DefragBot"
  13. local CHAT_BOT_BRACKETS = "[]"
  14. local CHAT_BOT_BRACKET_COLOR = "&3" -- Cyan
  15.  
  16. local DEBUG_LOGGING_ENABLED = true
  17. local LOG_FILE_NAME = "defrag.log"
  18. --#endregion
  19.  
  20. --#region Global State
  21. -- local lastAnnounceTime = 0 -- Removed for simpler announce
  22. local chatBox = nil
  23. local chatBoxFunctional = true -- Assume functional until a send error specifically
  24. --#endregion
  25.  
  26. --#region Logger Setup
  27. local function writeToLogFile(message) if not DEBUG_LOGGING_ENABLED then return end; local f,e=fs.open(LOG_FILE_NAME,"a"); if f then f.write(string.format("[%s] %s\n",os.date("%Y-%m-%d %H:%M:%S"),message));f.close() else print("LOG ERR: "..tostring(e))end end
  28. local function botLog(msg)local fm="["..CHAT_BOT_NAME.."-Debug] "..msg; print(fm); writeToLogFile(fm)end
  29. local function moveLog(msg)local fm="[MoveLib-Debug] "..msg; print(fm); writeToLogFile(fm)end
  30. --#endregion
  31.  
  32. --#region Movement Library Code (Integrated - Assumed same as v1.7)
  33. local AUTOMATA_PERIPHERAL_NAME="endAutomata";local POSITION_FILE="turtle_pos.json";local REFUEL_SLOT=16;local FUEL_ITEM_NAME_PART="coal";local automata=nil;local current_pos={x=nil,y=nil,z=nil};local current_dir=nil;local DIR_VECTORS={[0]={x=0,y=0,z=1},[1]={x=1,y=0,z=0},[2]={x=0,y=0,z=-1},[3]={x=-1,y=0,z=0}};local DIR_NAMES={[0]="North (+Z)",[1]="East (+X)",[2]="South (-Z)",[3]="West (-X)"};local function savePosition()if current_pos.x==nil or current_dir==nil then moveLog("No save, state unknown.");return end;local d={x=current_pos.x,y=current_pos.y,z=current_pos.z,dir=current_dir};local f,e=fs.open(POSITION_FILE,"w");if f then f.write(textutils.serialiseJSON(d));f.close();moveLog("Pos saved: X:"..d.x.." Y:"..d.y.." Z:"..d.z.." Dir:"..(DIR_NAMES[d.dir]or"Unk"))else moveLog("Err save pos: "..(e or"unk"))end end;local function loadPosition()if fs.exists(POSITION_FILE)then local f,e=fs.open(POSITION_FILE,"r");if f then local sD=f.readAll();f.close();local s,d=pcall(textutils.unserialiseJSON,sD);if s and d and d.x~=nil and d.y~=nil and d.z~=nil and d.dir~=nil then current_pos.x=d.x;current_pos.y=d.y;current_pos.z=d.z;current_dir=d.dir;moveLog("Pos loaded: X:"..current_pos.x.." Y:"..current_pos.y.." Z:"..current_pos.z.." Dir:"..(DIR_NAMES[current_dir]or"Unk"));return true else moveLog("Fail parse pos file: "..tostring(d))end else moveLog("Err open pos file: "..(e or"unk"))end else moveLog("Pos file not found.")end;return false end;local function dirNameToNumber(n)n=string.lower(n or"");if n=="n"or n=="north"then return 0 elseif n=="e"or n=="east"then return 1 elseif n=="s"or n=="south"then return 2 elseif n=="w"or n=="west"then return 3 end;moveLog("Warn: Invalid dir '"..(n or"nil").."'. Default N.");return 0 end;local function turnLeft()if current_dir==nil then moveLog("No turn: dir unk.");return false end;if turtle.turnLeft()then current_dir=(current_dir-1+4)%4;moveLog("Left. Dir:"..(DIR_NAMES[current_dir]or"Unk"));savePosition();return true else moveLog("Fail L turn.");return false end end;local function turnRight()if current_dir==nil then moveLog("No turn: dir unk.");return false end;if turtle.turnRight()then current_dir=(current_dir+1)%4;moveLog("Right. Dir:"..(DIR_NAMES[current_dir]or"Unk"));savePosition();return true else moveLog("Fail R turn.");return false end end;local function turnAround()if current_dir==nil then moveLog("No turn around: dir unk.");return false end;moveLog("Turn around...");if turnRight()and turnRight()then moveLog("Around. Dir:"..(DIR_NAMES[current_dir]or"Unk"));return true else moveLog("Fail turn around.");return false end end;local function forward()if current_pos.x==nil then moveLog("No move: pos unk.");return false end;if turtle.getFuelLevel()<1 and turtle.getFuelLimit()~=0 then moveLog("Fwd: <1 fuel.");return false end;if turtle.forward()then local v=DIR_VECTORS[current_dir];current_pos.x=current_pos.x+v.x;current_pos.y=current_pos.y+v.y;current_pos.z=current_pos.z+v.z;moveLog("Fwd. Pos: X:"..current_pos.x.." Y:"..current_pos.y.." Z:"..current_pos.z);savePosition();return true else moveLog("Fail fwd (block).");return false end end;local function back()if current_pos.x==nil then moveLog("No move: pos unk.");return false end;if turtle.getFuelLevel()<1 and turtle.getFuelLimit()~=0 then moveLog("Back: <1 fuel.");return false end;if turtle.back()then local v=DIR_VECTORS[current_dir];current_pos.x=current_pos.x-v.x;current_pos.y=current_pos.y-v.y;current_pos.z=current_pos.z-v.z;moveLog("Back. Pos: X:"..current_pos.x.." Y:"..current_pos.y.." Z:"..current_pos.z);savePosition();return true else moveLog("Fail back (block).");return false end end;local function up()if current_pos.y==nil then moveLog("No up: Y pos unk.");return false end;if turtle.getFuelLevel()<1 and turtle.getFuelLimit()~=0 then moveLog("Up: <1 fuel.");return false end;if turtle.up()then current_pos.y=current_pos.y+1;moveLog("Up. Pos: X:"..current_pos.x.." Y:"..current_pos.y.." Z:"..current_pos.z);savePosition();return true else moveLog("Fail up (block/no fuel).");return false end end;local function down()if current_pos.y==nil then moveLog("No down: Y pos unk.");return false end;if turtle.down()then current_pos.y=current_pos.y-1;moveLog("Down. Pos: X:"..current_pos.x.." Y:"..current_pos.y.." Z:"..current_pos.z);savePosition();return true else moveLog("Fail down (block).");return false end end;local function home()if not automata then automata=peripheral.find(AUTOMATA_PERIPHERAL_NAME);if not automata then moveLog("Err: EndAuto ('"..AUTOMATA_PERIPHERAL_NAME.."') miss.");return false end end;moveLog("Warp home...");local s,r=pcall(function()return automata.warpToPoint("home")end);if s and r then moveLog("Warp home OK.");current_pos.x=0;current_pos.y=0;current_pos.z=0;current_dir=0;moveLog("Pos reset home (000N).");savePosition();return true else moveLog("Fail warp home: "..tostring(r or"pcall err"));return false end end;local function refuel()moveLog("Refuel...");turtle.select(REFUEL_SLOT);local iBFS=turtle.getItemCount(REFUEL_SLOT);local iD=turtle.getItemDetail(REFUEL_SLOT);local iN="";if iD and iD.name then iN=string.lower(iD.name)else moveLog("Fuel slot "..REFUEL_SLOT.." empty/no name.")end;if iBFS>0 and string.find(iN,FUEL_ITEM_NAME_PART)then moveLog("Refuel from slot "..REFUEL_SLOT.." ("..(iD.name or"Unk Fuel").." x"..iBFS..")");if turtle.refuel(0)then local iAFS=turtle.getItemCount(REFUEL_SLOT);local cI=iBFS-iAFS;if cI>0 then moveLog("Refueled OK. Consumed "..cI.." "..(iD.name or"fuel")..".")else moveLog("Refueled, 0 consumed. Fuel: "..turtle.getFuelLevel())end else moveLog("turtle.refuel(0) fail. Fuel: "..turtle.getFuelLevel())end elseif iBFS>0 then moveLog("Item in fuel slot not fuel: "..(iD and iD.name or"Unk"))else moveLog("Fuel slot "..REFUEL_SLOT.." init empty.")end;local iTS=64-turtle.getItemCount(REFUEL_SLOT);local sTC=0;if iTS>0 then moveLog("Suck "..iTS.." to replenish fuel slot...");for _=1,iTS do if turtle.getItemCount(REFUEL_SLOT)>=64 then break end;local cSD=turtle.getItemDetail(REFUEL_SLOT);local cSN="";if cSD and cSD.name then cSN=string.lower(cSD.name)end;if turtle.getItemCount(REFUEL_SLOT)>0 and not string.find(cSN,FUEL_ITEM_NAME_PART)then moveLog("Slot "..REFUEL_SLOT.." has non-fuel ("..(cSD.name or"Unk").."). Stop suck.");break end;if turtle.suck()then sTC=sTC+1 else moveLog("Fail suck/chest empty after "..sTC.." items.");break end;sleep(0.1)end;if sTC>0 then moveLog("Sucked "..sTC.." items to replenish.")end else moveLog("Fuel slot full/non-fuel, no suck.")end;moveLog("Refuel finish. Fuel: "..turtle.getFuelLevel());return true end;local function setPos(x,y,z,dir)if type(x)~="number"or type(y)~="number"or type(z)~="number"then moveLog("Err: setPos xyz num.");return false end;current_pos.x=x;current_pos.y=y;current_pos.z=z;if type(dir)=="number"then current_dir=dir%4 else current_dir=dirNameToNumber(dir)end;moveLog("Pos set: X:"..x.." Y:"..y.." Z:"..z.." Dir:"..(DIR_NAMES[current_dir]or"Unk"));savePosition();return true end;local function turnToDir(td)if current_dir==nil then moveLog("No turnToDir: dir unk.");return false end;if current_dir==td then return true end;moveLog("Turn to dir: "..(DIR_NAMES[td]or"Unk"));local df=(td-current_dir+4)%4;if df==1 then return turnRight()elseif df==2 then return turnAround()elseif df==3 then return turnLeft()end;return false end;local function moveTo(tx,ty,tz,tdir)if current_pos.x==nil then moveLog("No moveTo: pos unk.");return false end;if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then if ty>current_pos.y then moveLog("moveTo: Low fuel ("..turtle.getFuelLevel()..") need up.")end end;moveLog(string.format("MoveTo: Tgt X%s Y%s Z%s Dir%s",tostring(tx),tostring(ty),tostring(tz),tostring(tdir)));local tdn;if type(tdir)=="number"then tdn=tdir%4 else tdn=dirNameToNumber(tdir)end;local att=0;local max_att=100;local max_stuck_ax=3;while(current_pos.x~=tx or current_pos.y~=ty or current_pos.z~=tz)and att<max_att do att=att+1;local moved=false;local cur_ax_stuck=0;if current_pos.y<ty then if up()then moved=true else cur_ax_stuck=cur_ax_stuck+1;moveLog("moveTo: Block UP("..cur_ax_stuck..")");if(current_pos.x~=tx or current_pos.z~=tz)and cur_ax_stuck<max_stuck_ax then elseif cur_ax_stuck<max_stuck_ax then if turnToDir(math.random(0,3))and forward()then moved=true end else moveLog("moveTo: Stuck Y(up). Abort Y.");ty=current_pos.y end end elseif current_pos.y>ty then if down()then moved=true else cur_ax_stuck=cur_ax_stuck+1;moveLog("moveTo: Block DOWN("..cur_ax_stuck..")");if(current_pos.x~=tx or current_pos.z~=tz)and cur_ax_stuck<max_stuck_ax then elseif cur_ax_stuck<max_stuck_ax then if turnToDir(math.random(0,3))and forward()then moved=true end else moveLog("moveTo: Stuck Y(down). Abort Y.");ty=current_pos.y end end end;if moved then goto continue_main_loop end;cur_ax_stuck=0;if current_pos.x<tx then if turnToDir(1)and forward()then moved=true else cur_ax_stuck=cur_ax_stuck+1;moveLog("moveTo: Block EAST("..cur_ax_stuck..")");if current_pos.z~=tz and cur_ax_stuck<max_stuck_ax then elseif cur_ax_stuck<max_stuck_ax then if turnToDir((math.random(0,1)*2))and forward()then moved=true end else moveLog("moveTo: Stuck X(east). Abort X.");tx=current_pos.x end end elseif current_pos.x>tx then if turnToDir(3)and forward()then moved=true else cur_ax_stuck=cur_ax_stuck+1;moveLog("moveTo: Block WEST("..cur_ax_stuck..")");if current_pos.z~=tz and cur_ax_stuck<max_stuck_ax then elseif cur_ax_stuck<max_stuck_ax then if turnToDir((math.random(0,1)*2))and forward()then moved=true end else moveLog("moveTo: Stuck X(west). Abort X.");tx=current_pos.x end end end;if moved then goto continue_main_loop end;cur_ax_stuck=0;if current_pos.z<tz then if turnToDir(0)and forward()then moved=true else cur_ax_stuck=cur_ax_stuck+1;moveLog("moveTo: Block NORTH("..cur_ax_stuck..")");if current_pos.x~=tx and cur_ax_stuck<max_stuck_ax then elseif cur_ax_stuck<max_stuck_ax then if turnToDir((math.random(0,1)*2)+1)and forward()then moved=true end else moveLog("moveTo: Stuck Z(north). Abort Z.");tz=current_pos.z end end elseif current_pos.z>tz then if turnToDir(2)and forward()then moved=true else cur_ax_stuck=cur_ax_stuck+1;moveLog("moveTo: Block SOUTH("..cur_ax_stuck..")");if current_pos.x~=tx and cur_ax_stuck<max_stuck_ax then elseif cur_ax_stuck<max_stuck_ax then if turnToDir((math.random(0,1)*2)+1)and forward()then moved=true end else moveLog("moveTo: Stuck Z(south). Abort Z.");tz=current_pos.z end end end;if moved then goto continue_main_loop end;if not moved then moveLog("moveTo: No move. Att: "..att);if cur_ax_stuck>=max_stuck_ax then break end end;::continue_main_loop::;sleep(0.05)end;if att>=max_att then moveLog("moveTo: Max att.")end;if not turnToDir(tdn)then moveLog("moveTo: Fail final turn.")end;local s=current_pos.x==tx and current_pos.y==ty and current_pos.z==tz and current_dir==tdn;if s then moveLog("moveTo: OK.")else moveLog(string.format("moveTo: Finish. Pos X%s Y%s Z%s Dir%s.",tostring(current_pos.x),tostring(current_pos.y),tostring(current_pos.z),(DIR_NAMES[current_dir]or"Unk")))end;return s end;local function initMoveLibrary()moveLog("Init MoveLib...");automata=peripheral.find(AUTOMATA_PERIPHERAL_NAME);if not automata then moveLog("CRIT WARN: EndAuto ('"..AUTOMATA_PERIPHERAL_NAME.."') miss. Home warp fail.")end;if not loadPosition()then moveLog("Pos unk/load fail. Warp home & set default.");if home()then moveLog("Init home OK.")else moveLog("CRIT: Fail init home. Pos unk.");current_pos.x=nil;current_pos.y=nil;current_pos.z=nil;current_dir=nil end end;moveLog("MoveLib Init. Pos: X:"..tostring(current_pos.x).." Y:"..tostring(current_pos.y).." Z:"..tostring(current_pos.z).." Dir:"..(current_dir and DIR_NAMES[current_dir]or"Unk"))end;local function getPosition()return current_pos end;local function getDirection()return current_dir end;local function getDirectionName()return current_dir and DIR_NAMES[current_dir]or"Unknown"end;local ml={l=turnLeft,turnLeft=turnLeft,r=turnRight,turnRight=turnRight,f=forward,forward=forward,b=back,back=back,a=turnAround,turnAround=turnAround,u=up,up=up,d=down,down=down,h=home,home=home,refuel=refuel,setPos=setPos,m=moveTo,moveTo=moveTo,getPosition=getPosition,getDirection=getDirection,getDirectionName=getDirectionName,init=initMoveLibrary};initMoveLibrary();
  34. --#endregion End of Integrated Movement Library
  35.  
  36. --#region DefragBot Specific Data & Logic
  37. local LOCATIONS={home={x=0,y=0,z=0,dir='N',name="Refuel Station"},io_port={x=1,y=0,z=0,dir='N',name="IO Port (Disk Input)"},hopper={x=1,y=1,z=0,dir='N',name="Hopper"},chest_temp_storage={x=1,y=2,z=0,dir='N',name="Chest (Temp Disk Storage)"}}
  38. local ME_DRIVE_BAY_RANGES={{y=0,x_start=-9,x_end=-2,dir='N',name="Lower Bay Row 1"},{y=2,x_start=-9,x_end=-2,dir='N',name="Lower Bay Row 2"},{y=4,x_start=-9,x_end=-2,dir='N',name="Lower Bay Row 3"},{y=9,x_start=-9,x_end=-2,dir='N',name="Upper Bay Row 1"},{y=11,x_start=-9,x_end=-2,dir='N',name="Upper Bay Row 2"},{y=13,x_start=-9,x_end=-2,dir='N',name="Upper Bay Row 3"},}
  39. local function getAllMeDriveBayCoordinates()local cs={};for _,br in ipairs(ME_DRIVE_BAY_RANGES)do for x=br.x_start,br.x_end do table.insert(cs,{x=x,y=br.y,z=0,dir=br.dir,name=br.name.." (X:"..x..")"})end end;return cs end
  40. local ALL_ME_DRIVE_LOCATIONS=getAllMeDriveBayCoordinates()
  41. local isDefragRunning=false;local fuelWarningSent=false
  42. --#endregion
  43.  
  44. --#region DefragBot Peripherals and Helpers
  45. local function sendFormattedChat(messageComponents) -- Simplified, always global
  46.     local plainText = ""
  47.     if type(messageComponents) == "table" then
  48.         for _,c_comp in ipairs(messageComponents) do
  49.             plainText = plainText .. (type(c_comp) == "table" and c_comp.text or tostring(c_comp) or "")
  50.         end
  51.     elseif messageComponents ~= nil then plainText = tostring(messageComponents)
  52.     else plainText = "nil_message_components_to_sendFormattedChat" end
  53.    
  54.     if not chatBox then -- Check if chatBox was found at startup
  55.         botLog("[NoChat] " .. plainText);
  56.         return false
  57.     end
  58.    
  59.     local jsonMessage_success, jsonMessage = pcall(textutils.serialiseJSON, messageComponents)
  60.     if not jsonMessage_success then
  61.         botLog("!!! textutils.serialiseJSON FAILED: " .. tostring(jsonMessage) .. " -- Original: " .. textutils.serialize(messageComponents,{compact=true,max_depth=2}))
  62.         print("ERROR: Could not serialize chat message!"); return false
  63.     end
  64.  
  65.     -- Use an anonymous function for the pcall to ensure 'self' is handled correctly by the chatBox method
  66.     local peripheral_call_success, peripheral_error = pcall(function()
  67.         -- Call using the documented signature for Advanced Peripherals' sendFormattedMessage
  68.         return chatBox.sendFormattedMessage(jsonMessage, CHAT_BOT_NAME, CHAT_BOT_BRACKETS, CHAT_BOT_BRACKET_COLOR)
  69.     end)
  70.    
  71.     if not peripheral_call_success then
  72.         botLog("!!! chatBox.sendFormattedMessage FAILED: " .. tostring(peripheral_error) .. " -- JSON was: " .. jsonMessage)
  73.         print("ERROR: Failed to send formatted message via chatBox: " .. tostring(peripheral_error))
  74.         -- Consider if chatBoxFunctional should be set to false here if this fails repeatedly
  75.         return false
  76.     end
  77.     return true
  78. end
  79.  
  80. local function announce(messageComponents)
  81.     -- Simpler announce: just calls sendFormattedChat and adds a fixed small delay.
  82.     local success = sendFormattedChat(messageComponents)
  83.     if not success then
  84.         botLog("Announce: sendFormattedChat failed.")
  85.     end
  86.     sleep(0.3) -- A slightly longer default delay than 0.2 to be safe with AP chatbox.
  87.     return success
  88. end
  89.  
  90. local COLORS={GOLD="gold",AQUA="aqua",GRAY="gray",RED="red",GREEN="green",YELLOW="yellow",WHITE="white"}
  91. --#endregion
  92.  
  93. --#region DefragBot Command Handlers
  94. local commandHandlers = {}
  95.  
  96. commandHandlers.help = function(u,_)
  97.     announce({{text="--- DefragBot Cmds ("..COMMAND_PREFIX..") ---",c=COLORS.GOLD,b=true}})
  98.     announce({{text=COMMAND_PREFIX.." help",c=COLORS.AQUA},{text=" - Help.",c=COLORS.GRAY}})
  99.     announce({{text=COMMAND_PREFIX.." cmd <ml.func()>",c=COLORS.AQUA},{text=" - Move func.",c=COLORS.GRAY}})
  100.     announce({{text=COMMAND_PREFIX.." mult <seq>",c=COLORS.AQUA},{text=" - Seq l,r,f,b,u,d,a.",c=COLORS.GRAY}})
  101.     announce({{text=COMMAND_PREFIX.." pos",c=COLORS.AQUA},{text=" - Pos/dir.",c=COLORS.GRAY}})
  102.     announce({{text=COMMAND_PREFIX.." fuel",c=COLORS.AQUA},{text=" - Fuel.",c=COLORS.GRAY}})
  103.     announce({{text=COMMAND_PREFIX.." unstuck",c=COLORS.AQUA},{text=" - Warp home, reset pos.",c=COLORS.GRAY}})
  104.     announce({{text=COMMAND_PREFIX.." startdefrag",c=COLORS.AQUA},{text=" - Start ME Drive defrag.",c=COLORS.GRAY}})
  105.     announce({{text=COMMAND_PREFIX.." stopdefrag",c=COLORS.AQUA},{text=" - Stop defrag.",c=COLORS.GRAY}})
  106. end
  107. commandHandlers.cmd = function(u,a)
  108.     if #a==0 then announce({{text="Usage: "..COMMAND_PREFIX.." cmd <ml.func()>",c=COLORS.YELLOW}});return end
  109.     local cs=table.concat(a," ");botLog("Usr "..u.." exec: "..cs)
  110.     local fn,e=load("return "..cs,"ucmd","t",{ml=ml})
  111.     if not fn then announce({{text="Err parsing: ",c=COLORS.RED},{text=tostring(e),c=COLORS.YELLOW}});botLog("Parse err: "..tostring(e));return end
  112.     local s,r=pcall(fn)
  113.     if s then announce({{text="Cmd exec. Res: ",c=COLORS.GREEN},{text=tostring(r),c=COLORS.WHITE}});botLog("Cmd res: "..tostring(r))
  114.     else announce({{text="Err exec: ",c=COLORS.RED},{text=tostring(r),c=COLORS.YELLOW}});botLog("Exec err: "..tostring(r))end
  115. end
  116. commandHandlers.mult = function(u,a)
  117.     if #a~=1 or type(a[1])~="string"then announce({{text="Usage: "..COMMAND_PREFIX.." mult <seq>",c=COLORS.YELLOW}});announce({{text="Seq: lrfbuda. Ex: lffr",c=COLORS.GRAY}});return end
  118.     local sq=string.lower(a[1]);botLog("Usr "..u.." mult: "..sq);announce({{text="Exec seq: ",c=COLORS.AQUA},{text=sq,c=COLORS.WHITE}});local sc=true
  119.     for i=1,#sq do
  120.         local mc=string.sub(sq,i,i);local mf=nil
  121.         if mc=="l"then mf=ml.l elseif mc=="r"then mf=ml.r elseif mc=="f"then mf=ml.f elseif mc=="b"then mf=ml.b
  122.         elseif mc=="u"then mf=ml.u elseif mc=="d"then mf=ml.d elseif mc=="a"then mf=ml.a end
  123.         if mf then
  124.             announce({{text="Exec: ",c=COLORS.GRAY},{text=mc,c=COLORS.YELLOW}})
  125.             local ms=mf()
  126.             if not ms then
  127.                 announce({{text="Move '",c=COLORS.RED},{text=mc,c=COLORS.YELLOW},{text="' fail. Stop.",c=COLORS.RED}})
  128.                 if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 and not fuelWarningSent then
  129.                     announce({{text="CRIT: Fuel low ("..turtle.getFuelLevel()..")! Unstuck?",c=COLORS.DARK_RED,b=true}});fuelWarningSent=true
  130.                 end
  131.                 sc=false;break
  132.             end
  133.         else announce({{text="Unk char '",c=COLORS.RED},{text=mc,c=COLORS.YELLOW},{text="' in seq. Stop.",c=COLORS.RED}});sc=false;break end
  134.     end
  135.     if sc then announce({{text="Seq finish.",c=COLORS.GREEN}})end
  136. end
  137. commandHandlers.pos = function(u,_)
  138.     local pD=ml.getPosition();local dNS=ml.getDirectionName()
  139.     if pD and pD.x~=nil then announce({{text="Pos: ",c=COLORS.AQUA},{text="X:"..tostring(pD.x).." Y:"..tostring(pD.y).." Z:"..tostring(pD.z),c=COLORS.WHITE},{text=" Facing: "..dNS,c=COLORS.WHITE}})
  140.     else announce({{text="Pos unknown.",c=COLORS.YELLOW}})end
  141. end
  142. commandHandlers.fuel = function(u,_)
  143.     local l,lm=turtle.getFuelLevel(),turtle.getFuelLimit()
  144.     announce({{text="Fuel: ",c=COLORS.AQUA},{text=tostring(l)..(lm>0 and(" / "..tostring(lm))or""),c=COLORS.WHITE}})
  145. end
  146. commandHandlers.unstuck = function(u,_)
  147.     announce({{text="Unstuck: warp home...",c=COLORS.YELLOW}});botLog(u.." unstuck.")
  148.     isDefragRunning=false
  149.     if ml.h()then announce({{text="Warp home & reset pos OK.",c=COLORS.GREEN}})
  150.     else announce({{text="Fail warp home. Manual help?",c=COLORS.RED}})end
  151. end
  152. --#endregion
  153.  
  154. --#region DefragBot Core Logic
  155. local function handleMoveFailure(actionDescription, criticalMoveFailed)
  156.     botLog("!!! INSIDE handleMoveFailure for: " .. actionDescription)
  157.     announce({{text="CRIT FAIL: Failed to " .. actionDescription .. ". ABORT DEFRAG.", c=COLORS.DARK_RED, b=true}})
  158.     if criticalMoveFailed and turtle.getFuelLevel() < 10 and turtle.getFuelLimit() ~= 0 and not fuelWarningSent then
  159.         announce({{text="HELP: Out of fuel ("..turtle.getFuelLevel()..") during critical move! Refuel & '@defrag unstuck'.", c=COLORS.RED, b=true}})
  160.         fuelWarningSent = true
  161.     end
  162.     isDefragRunning = false
  163.     botLog("!!! handleMoveFailure set isDefragRunning to false.")
  164. end
  165.  
  166. local function performDefragCycle()
  167.     botLog(">>> performDefragCycle STARTING. isDefragRunning should be true.")
  168.     isDefragRunning = true; fuelWarningSent = false
  169.    
  170.     local announce_success_initial, announce_error_initial = pcall(announce, {{text="Starting ME Drive Defragmentation Cycle...",c=COLORS.GOLD,b=true}})
  171.     if not announce_success_initial then botLog("!!! Initial announce in performDefragCycle FAILED: " .. tostring(announce_error_initial)) end
  172.    
  173.     botLog(">>> Current pos: X:"..tostring(current_pos.x).." Y:"..tostring(current_pos.y).." Z:"..tostring(current_pos.z).." Dir:"..(current_dir and DIR_NAMES[current_dir]or"Unk"))
  174.     botLog(">>> Target home: X:"..LOCATIONS.home.x.." Y:"..LOCATIONS.home.y.." Z:"..LOCATIONS.home.z.." Dir:"..LOCATIONS.home.dir)
  175.    
  176.     announce({{text="Stage 1: Initial setup...",c=COLORS.AQUA}})
  177.     if not ml.moveTo(LOCATIONS.home.x,LOCATIONS.home.y,LOCATIONS.home.z,LOCATIONS.home.dir)then botLog(">>> moveTo home FAILED.");handleMoveFailure("move to home",true);botLog(">>> pDC RETURN from home fail.");return end
  178.     botLog(">>> moveTo home OK.");ml.refuel()
  179.    
  180.     local initInvItems={};for i=1,15 do if turtle.getItemCount(i)>0 then local d=turtle.getItemDetail(i);table.insert(initInvItems,{slot=i,count=turtle.getItemCount(i),name=(d and d.name or "Unk")})end end
  181.     if #initInvItems>0 then
  182.         announce({{text="Pre-existing items found. Placing...",c=COLORS.YELLOW}});local allPlaced=true
  183.         for itemIdx=#initInvItems,1,-1 do local itemD=initInvItems[itemIdx];announce({{text="Placing "..itemD.count.."x "..itemD.name,c=COLORS.GRAY}});local curPlaced=false
  184.             for _,driveL in ipairs(ALL_ME_DRIVE_LOCATIONS)do local skipDrive=false
  185.                 if not isDefragRunning then announce({{text="Defrag stopped.",c=COLORS.YELLOW}});return end
  186.                 if turtle.getItemCount(itemD.slot)==0 then curPlaced=true;break end
  187.                 if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move to ME Drive low fuel",true);return end
  188.                 if not ml.moveTo(driveL.x,driveL.y,driveL.z,driveL.dir)then announce({{text="Fail move to ME Drive "..driveL.name..". Skip.",c=COLORS.RED}});skipDrive=true end
  189.                 if not skipDrive then turtle.select(itemD.slot);local dropAtt=turtle.getItemCount(itemD.slot)
  190.                     for _=1,dropAtt do if not turtle.drop()then moveLog("Fail drop itemD to "..driveL.name);break end;moveLog("Placed 1x "..itemD.name);if turtle.getItemCount(itemD.slot)==0 then curPlaced=true;break end;sleep(0.2)end
  191.                     if curPlaced then break end
  192.                 end
  193.             end
  194.             if not curPlaced then announce({{text="Could not place all "..itemD.name,c=COLORS.RED}});allPlaced=false end
  195.         end
  196.         if not allPlaced then announce({{text="WARN: Some pre-existing items not placed.",c=COLORS.YELLOW}})else announce({{text="All pre-existing items placed.",c=COLORS.GREEN}})end
  197.     end
  198.  
  199.     for _,locKey in ipairs({"io_port","hopper","chest_temp_storage"})do local skipClear=false
  200.         if not isDefragRunning then announce({{text="Defrag stopped.",c=COLORS.YELLOW}});return end;local loc=LOCATIONS[locKey];announce({{text="Clearing "..loc.name.."...",c=COLORS.GRAY}})
  201.         if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move to "..loc.name.." low fuel",true);return end
  202.         if not ml.moveTo(loc.x,loc.y,loc.z,loc.dir)then announce({{text="Fail move to "..loc.name..". Skip.",c=COLORS.RED}});skipClear=true end
  203.         if not skipClear then local suckedC=0;for _=1,16*64 do if turtle.suck()then suckedC=suckedC+1 else break end end
  204.             if suckedC>0 then announce({{text="Sucked "..suckedC.." from "..loc.name,c=COLORS.AQUA}})
  205.                 if locKey~="chest_temp_storage"then local tempL=LOCATIONS.chest_temp_storage
  206.                     if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move to temp chest low fuel",true);return end
  207.                     if not ml.moveTo(tempL.x,tempL.y,tempL.z,tempL.dir)then announce({{text="Fail move to temp chest. Items remain.",c=COLORS.RED}})else
  208.                         for slot=1,15 do if turtle.getItemCount(slot)>0 then turtle.select(slot);turtle.drop()end end;announce({{text="Moved items from "..loc.name.." to temp chest.",c=COLORS.GRAY}})
  209.                     end
  210.                 end
  211.             end
  212.         end
  213.     end
  214.     announce({{text="Stage 2: Processing ME Drive Bays...",c=COLORS.AQUA}})
  215.     for i,driveL in ipairs(ALL_ME_DRIVE_LOCATIONS)do local skipDriveLoop=false
  216.         if not isDefragRunning then announce({{text="Defrag stopped.",c=COLORS.YELLOW}});return end
  217.         announce({{text="Processing ME Drive "..i.."/"..#ALL_ME_DRIVE_LOCATIONS.." ("..driveL.name..")",c=COLORS.GOLD}})
  218.         if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move to ME Drive "..driveL.name.." low fuel",true);return end
  219.         if not ml.moveTo(driveL.x,driveL.y,driveL.z,driveL.dir)then announce({{text="Fail move to ME Drive. Skip.",c=COLORS.RED}});skipDriveLoop=true end
  220.         if not skipDriveLoop then local disksToProc={};announce({{text="Retrieving disks...",c=COLORS.GRAY}})
  221.             for slot=1,15 do turtle.select(slot)
  222.                 if turtle.getItemCount(slot)==0 then local suckAtt=0
  223.                     while suckAtt<64 do if not turtle.suck()then break end;suckAtt=suckAtt+1;if turtle.getItemCount(slot)>=64 then break end end
  224.                     if suckAtt>0 then local det=turtle.getItemDetail(slot);table.insert(disksToProc,{slot=slot,name=(det and det.name or "Unk"),count=suckAtt});announce({{text="Collected "..suckAtt.."x "..(det and det.name or "Disk").." in slot "..slot,c=COLORS.AQUA}})end
  225.                 end
  226.             end
  227.             if #disksToProc==0 then announce({{text="No disks in this ME Drive.",c=COLORS.YELLOW}});skipDriveLoop=true end
  228.             if not skipDriveLoop then local totalDisks=0;for _,d in ipairs(disksToProc)do totalDisks=totalDisks+d.count end;announce({{text="Collected "..totalDisks.." disk(s) total.",c=COLORS.GREEN}})
  229.                 local chestL=LOCATIONS.chest_temp_storage;if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move to temp chest low fuel",true);return end
  230.                 if not ml.moveTo(chestL.x,chestL.y,chestL.z,chestL.dir)then handleMoveFailure("move to temp chest",false);return end
  231.                 announce({{text="Dropping "..totalDisks.." disk(s) to chest...",c=COLORS.GRAY}});for _,diskD in ipairs(disksToProc)do turtle.select(diskD.slot);if not turtle.drop()then announce({{text="Fail drop from slot "..diskD.slot,c=COLORS.RED}})end end
  232.                 local ioL=LOCATIONS.io_port;if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move to IO port low fuel",true);return end
  233.                 if not ml.moveTo(ioL.x,ioL.y,ioL.z,ioL.dir)then handleMoveFailure("move to IO port",false);return end
  234.                 announce({{text="Waiting for "..totalDisks.." processed disk(s)...",c=COLORS.GRAY}});local procRet=0;local noDiskTries=0;local MAX_NO_DISK_TRIES=20
  235.                 while procRet<totalDisks and noDiskTries<MAX_NO_DISK_TRIES do
  236.                     if not isDefragRunning then announce({{text="Defrag stopped.",c=COLORS.YELLOW}});return end;local retrievedThis=false
  237.                     for slot=1,15 do if turtle.getItemCount(slot)==0 then turtle.select(slot)
  238.                         if turtle.suck()then procRet=procRet+1;announce({{text="Retrieved proc disk "..procRet.."/"..totalDisks,c=COLORS.AQUA}});noDiskTries=0;retrievedThis=true;break end
  239.                     end end
  240.                     if procRet>=totalDisks then break end
  241.                     if not retrievedThis then noDiskTries=noDiskTries+1;announce({{text="Waiting... (No disk try "..noDiskTries.."/"..MAX_NO_DISK_TRIES..")",c=COLORS.DARK_GRAY}})end;sleep(10)
  242.                 end
  243.                 if procRet<totalDisks then announce({{text="Timeout: Retrieved "..procRet.."/"..totalDisks..". Proceeding.",c=COLORS.RED}})end
  244.                 if procRet==0 and totalDisks>0 then announce({{text="No disks from IO Port. Skip return.",c=COLORS.RED}});skipDriveLoop=true end
  245.                 if not skipDriveLoop then
  246.                     if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move home refuel low fuel",true);return end
  247.                     if not ml.moveTo(LOCATIONS.home.x,LOCATIONS.home.y,LOCATIONS.home.z,LOCATIONS.home.dir)then announce({{text="Fail move home for refuel.",c=COLORS.RED}})end
  248.                     ml.refuel()
  249.                     if turtle.getFuelLevel()<10 and turtle.getFuelLimit()~=0 then handleMoveFailure("move back ME Drive low fuel",true);return end
  250.                     if not ml.moveTo(driveL.x,driveL.y,driveL.z,driveL.dir)then announce({{text="Fail return to ME Drive. Disks remain.",c=COLORS.RED}});skipDriveLoop=true end
  251.                     if not skipDriveLoop then announce({{text="Returning "..procRet.." processed disk(s)...",c=COLORS.GRAY}})
  252.                         for slot=1,15 do if turtle.getItemCount(slot)>0 then turtle.select(slot);local countIS=turtle.getItemCount(slot);local droppedC=0
  253.                             for _=1,countIS do if turtle.drop()then droppedC=droppedC+1 else break end end
  254.                             if droppedC>0 then announce({{text="Returned "..droppedC.." disk(s) from slot "..slot,c=COLORS.GREEN}})end
  255.                             if droppedC<countIS then announce({{text="Failed return all from slot "..slot,c=COLORS.RED}})end
  256.                         end end;announce({{text="Finished ME Drive "..driveL.name,c=COLORS.GOLD}})
  257.                     end
  258.                 end
  259.             end
  260.         end;sleep(1)
  261.     end
  262.     announce({{text="Defragmentation Cycle Complete!",c=COLORS.GOLD,b=true}});isDefragRunning=false
  263. end
  264.  
  265. commandHandlers.startdefrag=function(u,_)
  266.     botLog(">>> startdefrag command called. Current isDefragRunning: " .. tostring(isDefragRunning))
  267.     if isDefragRunning then announce({{text="Defrag already running.",c=COLORS.YELLOW}}); botLog(">>> startdefrag: Already running, exiting."); return end;
  268.     botLog(u.." started defrag.")
  269.     performDefragCycle() -- Run directly in the main thread
  270.     botLog(">>> startdefrag: performDefragCycle completed or errored.")
  271. end
  272. commandHandlers.stopdefrag=function(u,_)
  273.     if isDefragRunning then
  274.         announce({{text="Attempting to signal stop to defrag process...",c=COLORS.YELLOW}})
  275.         isDefragRunning=false; -- Set the flag
  276.         botLog(u.." requested stop. isDefragRunning set to false.")
  277.         -- Note: Since performDefragCycle is now in main thread, this flag only stops it at its next check.
  278.         -- A true immediate stop would require more complex inter-thread communication or event handling within loops.
  279.     else
  280.         announce({{text="Defrag not running.",c=COLORS.GRAY}})
  281.     end
  282. end
  283. --#endregion
  284.  
  285. --#region DefragBot Main Loop
  286. local function run()
  287.     if DEBUG_LOGGING_ENABLED then
  288.         local file, err = fs.open(LOG_FILE_NAME, "w")
  289.         if file then file.write(string.format("[%s] %s Log Initialized.\n", os.date("%Y-%m-%d %H:%M:%S"), CHAT_BOT_NAME)); file.write("============================================================\n"); file.close()
  290.         else print("LOGGING ERROR: Could not initialize "..LOG_FILE_NAME..": "..tostring(err)) end
  291.     end
  292.     term.clear();term.setCursorPos(1,1);
  293.    
  294.     chatBox = peripheral.find(CHAT_BOX_PERIPHERAL_NAME)
  295.     botLog("DefragBot initializing...")
  296.     botLog("Initial ChatBox peripheral check: " .. tostring(chatBox))
  297.     -- Determine chatBoxFunctional based on direct method check
  298.     if chatBox and chatBox.sendFormattedMessage and chatBox.sendFormattedMessageToPlayer then
  299.         chatBoxFunctional = true
  300.         botLog("ChatBox determined to be FUNCTIONAL (key methods exist via direct check).")
  301.     else
  302.         if chatBox then
  303.             botLog("ChatBox found, but MISSING ESSENTIAL sendFormattedMessage(ToPlayer) methods. Marking as non-functional.")
  304.         else
  305.             botLog("ChatBox peripheral IS NIL at startup. Marking as non-functional.")
  306.         end
  307.         chatBoxFunctional = false
  308.     end
  309.    
  310.     botLog(CHAT_BOT_NAME.." online. MoveLib init complete.")
  311.     print(CHAT_BOT_NAME.." online. '"..COMMAND_PREFIX.." help' or '@all'.")
  312.    
  313.     local initial_announce_success = announce({{text=CHAT_BOT_NAME.." online, defrag duties (v1.8)!",c=COLORS.GREEN}})
  314.     if not initial_announce_success then
  315.         botLog("Initial online announcement FAILED. Chat will be console/log only if ChatBox is non-functional.")
  316.     end
  317.    
  318.     while true do
  319.         -- If defrag is running in main thread, this loop won't process events until it finishes.
  320.         -- This is the trade-off for easier debugging of performDefragCycle.
  321.         local eventData={os.pullEvent()}
  322.         local eventType=eventData[1]
  323.  
  324.         if not isDefragRunning then -- Only process chat if defrag isn't blocking
  325.             if eventType=="chat"then local eU,eM=eventData[2],eventData[3]
  326.                 if eM then if string.lower(eM)=="@all"then botLog("@all from "..eU);announce({{text="Use '",c=COLORS.GREEN},{text=COMMAND_PREFIX.." help",c=COLORS.AQUA},{text="' for my commands.",c=COLORS.GREEN}})
  327.                     elseif string.sub(eM,1,#COMMAND_PREFIX)==COMMAND_PREFIX then botLog("Chat cmd from "..eU..": "..eM);local p={};for pt in string.gmatch(eM,"[^%s]+")do table.insert(p,pt)end;local cn="";if p[2]then cn=string.lower(p[2])end;local ca={};for i=3,#p do table.insert(ca,p[i])end
  328.                         if commandHandlers[cn]then commandHandlers[cn](eU,ca)elseif cn~=""then announce({{text="Unk Cmd: '",c=COLORS.RED},{text=cn,c=COLORS.YELLOW},{text="'.",c=COLORS.RED}})end
  329.                     end
  330.                 end
  331.             elseif eventType=="terminate"then
  332.                 botLog("Terminate. Shutdown.");isDefragRunning=false;
  333.                 if chatBoxFunctional then announce({{text=CHAT_BOT_NAME.." shut down.",c=COLORS.YELLOW}}) end;
  334.                 return
  335.             end
  336.         elseif eventType == "terminate" then -- Still handle terminate if defrag is running
  337.             botLog("Terminate received during defrag. Shutting down.");isDefragRunning=false;
  338.             if chatBoxFunctional then announce({{text=CHAT_BOT_NAME.." shut down (during defrag).",c=COLORS.YELLOW}}) end;
  339.             return
  340.         else
  341.             -- If defrag is running, we might just ignore other events or log them
  342.             botLog("Event received during defrag (ignored by main loop): " .. eventType)
  343.         end
  344.     end
  345. end
  346. run()
  347. --#endregion
Add Comment
Please, Sign In to add comment