doodle3d-firmware/src/rest/api/api_printer.lua

local lfs = require('lfs')
local log = require('util.logger')
local utils = require('util.utils')
local settings = require('util.settings')
local printDriver = require('print3d')
local printerUtils = require('util.printer')
local accessManager = require('util.access')

local M = {
	isApi = true
}

local SEQ_NUM_FILE = "/tmp/d3d-print-sequence-number.txt"

function M._global(request, response)
	-- TODO: list all printers (based on /dev/ttyACM* and /dev/ttyUSB*)
	response:setSuccess()
end

function M.temperature(request, response)
	local argId = request:get("id")
	local printer,msg = printerUtils.createPrinterOrFail(argId, response)
	if not printer then return false end

	local temperatures,msg = printer:getTemperatures()

	response:addData('id', argId)
	if temperatures then
		response:setSuccess()
		response:addData('hotend', temperatures.hotend)
		response:addData('hotend_target', temperatures.hotend_target)
		response:addData('bed', temperatures.bed)
		response:addData('bed_target', temperatures.bed_target)
	else
		response:setError(msg)
		return false;
	end

	return true;
end

function M.progress(request, response)
	local argId = request:get("id")
	local printer,msg = printerUtils.createPrinterOrFail(argId, response)
	if not printer then return false end

	-- NOTE: despite their names, `currentLine` is still the error indicator and `bufferedLines` the message in such case.
	local currentLine,bufferedLines,totalLines = printer:getProgress()

	response:addData('id', argId)
	if currentLine then
		response:setSuccess()
		response:addData('current_line', currentLine)
		response:addData('buffered_lines', bufferedLines)
		response:addData('total_lines', totalLines)
	else
		response:setError(bufferedLines)
		return false
	end

	return true;
end

-- NOTE: onlyReturnState is optional and prevents response from being modified
function M.state(request, response, onlyReturnState)
	local argId = request:get("id")
	if not onlyReturnState then response:addData('id', argId) end

	local printer,msg = printerUtils.createPrinterOrFail(argId, response)
	if not printer then
		local printerState = "disconnected"
		if not onlyReturnState then
			response:setSuccess()
			response:addData('state', printerState)
		end
		return true, printerState
	else
		local rv,msg = printer:getState()
		if rv then
			if not onlyReturnState then
				response:setSuccess()
				response:addData('state', rv)
			end
			return true, rv
		else
			if not onlyReturnState then response:setError(msg) end
			return false
		end
	end
	return true
end

function M.heatup_POST(request, response)

	if not accessManager.hasControl(request.remoteAddress) then
		response:setFail("No control access")
		return
	end

	local argId = request:get("id")
	local printer,msg = printerUtils.createPrinterOrFail(argId, response)
	if not printer then return false end

	local temperature = settings.get('printer.heatup.temperature')
	local rv,msg = printer:heatup(temperature)

	response:addData('id', argId)
	if rv then response:setSuccess()
	else response:setFail(msg)
	end
end

function M.stop_POST(request, response)

	log:info("API:printer/stop")

	if not accessManager.hasControl(request.remoteAddress) then
		response:setFail("No control access")
		return
	end

	local argId = request:get("id")
	local argGcode = request:get("gcode")
	local printer,msg = printerUtils.createPrinterOrFail(argId, response)
	if not printer then return end
	
	-- resetting sequence number
	setSequenceNumber(-1)
	
	
	
	
	--[[
	-- replacing variables in endgcode
	local printingTemperature  	  = settings.get('printer.temperature')
	local printingBedTemperature  = settings.get('printer.bed.temperature')
	local preheatTemperature      = settings.get('printer.heatup.temperature')
	local preheatBedTemperature   = settings.get('printer.heatup.bed.temperature')
	local endCode 				  = settings.get('printer.endcode.marlin')

	--log:info("  printingBedTemperature: "..utils.dump(printingBedTemperature))
	--log:info("  preheatBedTemperature: "..utils.dump(preheatBedTemperature))
	--log:info("  endCode : "..utils.dump(endCode))
	endCode = string.gsub(endCode,"{printingTemp}",printingTemperature)
	endCode = string.gsub(endCode,"{printingBedTemp}",printingBedTemperature)
	endCode = string.gsub(endCode,"{preheatTemp}",preheatTemperature)
	endCode = string.gsub(endCode,"{preheatBedTemp}",preheatBedTemperature)
	--log:info("  >endCode : "..utils.dump(endCode))
	]]--
	
	if(argGcode == nil) then
		argGcode = ""
	end
	local rv,msg = printer:stopPrint(argGcode)

	response:addData('id', argId)
	if rv then response:setSuccess()
	else response:setError(msg)
	end
end

--accepts: first(bool) (chunks will be concatenated but output file will be cleared first if this argument is true)
--accepts: start(bool) (only when this argument is true will printing be started)
function M.print_POST(request, response)

	-- controll access check
	local controllerIP = accessManager.getController()
	local hasControl = false
	if controllerIP == "" then
		accessManager.setController(request.remoteAddress)
		hasControl = true
	elseif controllerIP == request.remoteAddress then
		hasControl = true
	end
	
	log:info("  hasControl: "..utils.dump(hasControl))
	if not hasControl then
		response:setFail("No control access")
		return
	end

	-- arguments
	local argId = request:get("id")
	local argGcode = request:get("gcode")
	local argIsFirst = utils.toboolean(request:get("first"))
	local argStart = utils.toboolean(request:get("start"))
	local argSeqNum = tonumber(request:get("seqNum"))
	
	-- reset sequence number?
	if argIsFirst == true then
		setSequenceNumber(-1)
	end
	
	-- check sequence number
	if(argSeqNum ~= nil and argSeqNum ~= "") then 
		local prevSeqNum = getSequenceNumber()
		
		if(prevSeqNum+1 ~= argSeqNum) then
			response:setError(" part isn't in sequence / has wrong sequence number")
			return
		end
		setSequenceNumber(prevSeqNum+1)
		
		log:debug("  seqNum: "..utils.dump(prevSeqNum).." > "..(prevSeqNum+1))
	end
	
	local printer,msg = printerUtils.createPrinterOrFail(argId, response)
	if not printer then return end

	response:addData('id', argId)

	if argGcode == nil or argGcode == '' then
		response:setError("missing gcode argument")
		return
	end

	if argIsFirst == true then
		log:debug("clearing all gcode for " .. printer:getId())
		response:addData('gcode_clear',true)
		local rv,msg = printer:clearGcode()

		if not rv then
			response:setError(msg)
			return
		end
	end

	local rv,msg

	-- TODO: return errors with a separate argument like here in the rest of the code (this is how we designed the API right?)
	local startTime = os.clock()
	rv,msg = printer:appendGcode(argGcode)
	local duration = os.clock()-startTime
	log:debug("  duration: "..utils.dump(duration))
	
	if rv then
		--NOTE: this does not report the number of lines, but only the block which has just been added
		response:addData('gcode_append',argGcode:len())
	else
		response:setError("could not add gcode")
		response:addData('msg', msg)
		return
	end

	if argStart == true then
		rv,msg = printer:startPrint()

		if rv then
			response:setSuccess()
			response:addData('gcode_print',true)
		else
			response:setError("could not send gcode")
			response:addData('msg', msg)
		end
	else
		response:setSuccess()
	end
end

function getSequenceNumber() 
	local file, error = io.open(SEQ_NUM_FILE,'r')
	if file == nil then
		return -1
	else
		local seqNum = file:read('*a')
		seqNum = tonumber(seqNum)
		file:close()
		return seqNum
	end
end

function setSequenceNumber(num) 
	log:info("API:printer:setSequenceNumber: "..utils.dump(num))
	local file = io.open(SEQ_NUM_FILE,'w')
	file:write(num)
	file:flush()
	file:close()
end

return M