Air724ugLora/lib/console.lua

--- 模块功能：Luat控制台.
-- 使用说明参考demo/console下的《console功能使用说明.docx》
-- @module console
-- @author openLuat
-- @license MIT
-- @copyright openLuat
-- @release 2017.9.15

require"ril"
module(..., package.seeall)
local uart_id
local console_task

local function read_line()
    while true do
        local s = uart.read(uart_id, "*l")
        if s ~= "" then
            return s
        end
        coroutine.yield()
    end
end

local function write(s)
    uart.write(uart_id, s)
end

local function on_wait_event_timeout()
    coroutine.resume(console_task, "TIMEOUT")
end

local function wait_event(event, timeout)
    if timeout then
        sys.timerStart(on_wait_event_timeout, timeout)
    end

    while true do
        local receive_event = coroutine.yield()
        if receive_event == event then
            sys.timerStop(on_wait_event_timeout)
            return
        elseif receive_event == "TIMEOUT" then
            write("WAIT EVENT " .. event .. "TIMEOUT\r\n")
            return
        end
    end
end

local function main_loop()
    local cache_data = ""
    local wait_event_flag

    -- 定义执行环境，命令行下输入的脚本的print重写到命令行的write
    local execute_env = {
        print = function(...)
			local arg = { ... }
            for i, v in ipairs(arg) do
                arg[i] = type(v) == "nil" and "nil" or tostring(v)
            end
            write(table.concat(arg, "\t"))
            write("\r\n")
        end,
        sendat = function(cmd, data)
            ril.request(cmd, data, function(cmd, success, response, intermediate)
                if intermediate then
                    write("\r\n" .. intermediate .. "\r\n")
                end
                if response then
                    write("\r\n" .. response .. "\r\n")
                end
                coroutine.resume(console_task, "WAIT_AT_RESPONSE")
            end, nil)
            wait_event_flag = "WAIT_AT_RESPONSE"
        end,
    }
    setmetatable(execute_env, { __index = _G })

    -- 输出提示语
    write("\r\nWelcome to Luat Console\r\n")
    write("\r\n> ")

    while true do
        -- 读取输入
        local new_data = read_line("*l")
        -- 输出回显
        write(new_data)
        -- 拼接之前未成行的剩余数据
        cache_data = cache_data .. new_data
        -- 去掉回车换行
        local line = string.match(cache_data, "(.-)\r?\n")
        if line then
            -- 收到一整行的数据 清除缓冲数据
            cache_data = ""
            -- 输出新行
            write("\n")
            -- 用xpcall执行用户输入的脚本，可以捕捉脚本的错误
            xpcall(function()
                -- 执行用户输入的脚本
                local f = assert(loadstring(line.." "))
                setfenv(f, execute_env)
                f()
            end,
                function(err) -- 错误输出
                    write(err .. '\r\n')
                    write(debug.traceback())
                end)
            if wait_event_flag then
                wait_event(wait_event_flag, 3000)
                wait_event_flag = nil
            end
            -- 输出输入提示符
            write("\r\n> ")
        end
    end
end

--- 配置控制台使用的串口参数，创建控制台协程
-- @number id 控制台使用的串口ID：1表示串口1，2表示串口2
-- @number[opt=115200] baudrate 控制台使用的串口波特率
-- 支持1200,2400,4800,9600,10400，14400,19200,28800,38400,57600,76800,115200,230400,460800,576000,921600,1152000,4000000
-- @return nil
-- @usage console.setup(1, 115200)
function setup(id, baudrate)
    -- 默认串口1
    uart_id = id or 1
    -- 默认波特率115200
    baudrate = baudrate or 115200
    -- 创建console处理的协程
    console_task = coroutine.create(main_loop)
    -- 初始化串口
    uart.setup(uart_id, baudrate, 8, uart.PAR_NONE, uart.STOP_1)
    -- 串口收到数据时唤醒console协程
    uart.on(uart_id, "receive", function()
        coroutine.resume(console_task)
    end)
    coroutine.resume(console_task)
end