require "log"
require "sys"
require "utils"
require "patch"
cjson=require("cjson")
MQTT=require("mqtt_library")
base64=require("base64")
dyutils=require("dyutils") 
bit=require("bit") 
lpack=require("lua_pack") 
syslog=require("syslog")     
bcd=require("bcd") 
sqlite3=require("lsqlite3") 
bnot = bit.bnot 
band, bor, bxor = bit.band, bit.bor, bit.bxor 
lshift, rshift, rol = bit.lshift, bit.rshift, bit.rol

--MQTT.Utility.set_debug(true)

broker_addr = "localhost"
broker_port = 1883

block_size = 192
time_last = 0
fail_cnt = 0
mi = 0
function on_mqtt_msg(topic, payload)
  -- application specific code
  --log.info("receive ", topic, payload);
  payload_obj = cjson.decode(payload)
  if (payload_obj['src_identifier'] == '__transparent') then
    coroutine.yield(topic, payload_obj)
  end
end

function mqtt_listen()
    while(true)
    do
        if(time_last ~= 0 and os.time() >= (time_last + 10)) then
            coroutine.yield(nil, nil)
        end
        mqtt_client:handler()
    end
end

-- send data to dev_sn
function send_data_to_sn(mqtt_client, gw_sn, dev_sn, mi, data, obj_send)
	topic = obj_send['topic']
    obj_send['mi'] = mi
    obj_send['len'] = #data
    log.info("TX:",string.toHex(data," "),#data)
    obj_send['data_b64'] = base64.encode(data)
    json_str = cjson.encode(obj_send)
    --log.info(json_str)
    mqtt_client:publish(topic, json_str)
    time_last = os.time()
end

-- send data to dev_sn
function send_upgrade_process(mqtt_client, gw_sn, dev_sn, mi, Upgrade_Status, Upgrade_Progress)
	topic = string.format("ipc/%s/upgrade/device/%s/data_filtered/event/__Upgrade_Progress", gw_sn, dev_sn)
    local payload={}
	payload['identifier'] = '__Upgrade_Progress'
    payload['sn'] = dev_sn
    payload['mi'] = mi
    payload['time'] = os.time()
    local progress = {}
    progress['Upgrade_Status'] = Upgrade_Status
    progress['Upgrade_Progress'] = Upgrade_Progress
    payload['tags'] = progress
    json_str = cjson.encode(payload)
    --log.info(json_str)
    mqtt_client:publish(topic, json_str)
end

function upgrade_node(json_str, gw_sn, firmware, send_out_json)
    log.info("upgrade_node param",json_str, gw_sn, firmware, send_out_json)
    local obj=cjson.decode(json_str)
    local obj_send = cjson.decode(send_out_json)
    local ret = 0

    mqtt_client = MQTT.client.create(broker_addr, broker_port, on_mqtt_msg)
    mqtt_client:connect("upgrade_xieneng")
	topic = string.format("ipc/%s/+/device/%s/data/raw_data", gw_sn, obj['sn'])
    mqtt_client:subscribe({topic})
    mqtt = coroutine.create(mqtt_listen)

    -- 获取firmware的version信息
    obj.sn:match(".+%_(%w+)$")
    --/tmp/SeIbuV-Board205-16S-D0503-0x5024-23.60.1.6-20221021.bin
    firmware_version = firmware:match("SeIbuV.*%-(%d+%.%d+%.%d+%.%d+).*%.bin")--string.match(firmware,'[(Se)|(Auto)]lbuV(%d.%d.%d.%d).bin')
    log.info("firmware_version",firmware_version)
    addr = get_target_addr(obj)

    -- 进入升级模式
    while (false) do
        mi = 1000
        data = string.format("0x%04x EnterBoot", addr);
        --data = '0x1000 ShowVersion'
        send_data_to_sn(mqtt_client, gw_sn, obj['sn'], mi, data, obj_send)
        ret, topic, payload_obj = coroutine.resume(mqtt)
        if (payload_obj and payload_obj.sn == obj['sn'] and payload_obj.mi == mi) then
            local data = base64.decode(payload_obj['data_b64'])
            if (data == 'true') then
                log.info("enter boot")
            else
                log.info("EnterBoot failed")
                return -1
            end
            send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "start", 0);
            break
        end
    end

    -- get version
    while (true) do
        mi = 1000
        data = string.format("0x%04x ShowVersion", addr);
        --data = '0x1000 ShowVersion'
        log.info("data:", data,string.toHex(data))
        send_data_to_sn(mqtt_client, gw_sn, obj['sn'], mi, data, obj_send)
        ret, topic, payload_obj = coroutine.resume(mqtt)
        if (payload_obj and payload_obj.sn == obj['sn'] and payload_obj.mi == mi) then
            local data = base64.decode(payload_obj['data_b64'])
            pv = string.match(data, 'pv%s=%s(%d+)')
            mv = string.match(data, 'mv%s=%s(%d+)')
            sv = string.match(data, 'sv%s=%s(%d+)')
            cv = string.match(data, 'cv%s=%s(%d+)')
            cur_version = string.format('%d.%d.%d.%d', pv, mv, sv, cv)
            log.info("VERSION:",data)
            if (cur_version == firmware_version) then
                -- 版本号与当前版本一致,不进行升级
                log.info("same version")
                send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "success", 100)
                return 0
            end
            send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "ready", 0);
            break
        end
    end

    file = io.open(firmware, "r")
    local bin_buf = file:read("*a")
    local mod = math.fmod(#bin_buf, 4)    -- 取余数
    if mod ~= 0 then
        pad = 4 - mod
        for i=1,pad do
            -- bin_buf = bin_buf .. 0
            bin_buf=bpack("AC", bin_buf, 0)
        end
    end

    -- send data
	file:seek("set")
    local offset = 0
    local last_progress = 0
    while (true)
    do
        if (offset >= #bin_buf) then break end
        block_data = string.sub(bin_buf, offset+1, offset+block_size)
        if not block_data then
            log.info("read data failed")
            break
        end
        
        --log.info(dyutils.bin_str_to_hex(block_data))
        log.info("block_data",block_data)
        block_pkt = bpack("C<SCC<IA", 0xD0, addr, 0, #block_data+4, offset, block_data);
        local crc = dyutils.CRC16(block_pkt, #block_pkt)
        local bin_str=bpack("A>S",block_pkt,crc)
        --log.info(dyutils.bin_str_to_hex(bin_str))
        while (true)
        do
            mi = mi + 1
            send_data_to_sn(mqtt_client, gw_sn, obj['sn'], mi, bin_str, obj_send)
            ret, topic, payload_obj = coroutine.resume(mqtt)
            if (payload_obj and payload_obj.sn == obj['sn'] and payload_obj.mi == mi) then
                local data = base64.decode(payload_obj['data_b64'])
                log.info("RX:",string.toHex(data," "),#data)
                offset = offset + #block_data
                _,_,_,_,_,_,result = bunpack(data, "CSCCCI");
                --log.info("result:", result)
                if (result == offset) then
                    fail_cnt = 0
                    progress = math.modf((offset * 100)/#bin_buf)
                    if (progress ~= last_progress) then
                        send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "upgrading", progress);
                        last_progress = progress
                    end
                    break
                else
                    fail_cnt = fail_cnt + 1
                    if (fail_cnt > 10) then
                        send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "failed", 0)
                        data = string.format("0x%04x reset", addr);
                        send_data_to_sn(mqtt_client, gw_sn, obj['sn'], mi, data, obj_send)
                        return -1
                    end
                end
            else
                fail_cnt = fail_cnt + 1
                if (fail_cnt > 10) then
                    send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "failed", 0)
                    data = string.format("0x%04x reset", addr);
                    send_data_to_sn(mqtt_client, gw_sn, obj['sn'], mi, data, obj_send)
                    return -1
                end
            end
        end
    end 

    --发送校验
    --
	local file_check = dyutils.CalculateXorCrc(bin_buf, #bin_buf)
	--log.info(file_check)
    local check_data = bpack("C<SCC<I<I<I", 0xD0, addr, 0, 12, 0xFFFFFFFF, #bin_buf, file_check);
	local crc = dyutils.CRC16(check_data, #check_data)
	local bin_str=bpack(">A>S",check_data,crc)
	--log.info(dyutils.bin_str_to_hex(bin_str))
	while (true)
    do
        mi = mi + 1
        send_data_to_sn(mqtt_client, gw_sn, obj['sn'], mi, bin_str, obj_send)
        ret, topic, payload_obj = coroutine.resume(mqtt)
        if (payload_obj and payload_obj.sn == obj['sn'] and payload_obj.mi == mi) then
            local data = base64.decode(payload_obj['data_b64'])
            log.info("RX:",string.toHex(data," "),#data)
            offset = offset + #block_data
            _,_,_,_,_,_,result = bunpack(data, "CSCCCI");
            log.info("result:", result)
            ret = result
            break
        end
    end
    
    -- if (ret == 0) then -- 防止一次升级失败后面再不能升级的问题
    if (obj['Reset'] == 1) then
        data = string.format("0x%04x reset", addr);
        send_data_to_sn(mqtt_client, gw_sn, obj['sn'], mi, data, obj_send)
    end
    -- end
    mi = mi + 1
    if (ret == 0) then
        send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "success", 100)
    else
        send_upgrade_process(mqtt_client, gw_sn, obj['sn'], mi, "failed", 0)
    end

    return ret
end


function get_target_addr(obj)
    -- 先关心单播方式升级
    addr = 0
    addr = bit.bor(addr, bit.lshift(obj['L3_DevId'], 12)) -- (14 .. 12)
    addr = bit.bor(addr, bit.lshift(obj['L2_DevId'], 7)) -- (11 .. 7)
    addr = bit.bor(addr, obj['L1_DevId']) -- (6 .. 0)
    return addr
end



-- upgrade_node('{"identifier":"__Upgrade","Action":"start","L1_DevId":0,"sn":"21880B00016801","L3_DevId":1,"L2_DevId":0,"mi":34004323,"URL":"http://10.3.1.229:8001/SeMasterV2.5.7.8.bin","MD5":"bbaa3ea3bea9730dd4d33a6c0603cc92","timestamp":1584091544}', '22014C000234', '/tmp/SeMasterV-A0333-A0334-0x6AB0-14.1.1.6-20211221.bin')
-- upgrade_node('{"identifier":"__Upgrade","Action":"start","L1_DevId":0,"Dev_Type":0,"L3_DevId":1,"sn":"21880B00016801","time":1642397668,"L2_DevId":0,"Reset":1,"mi":41672023,"URL":"http://10.128.70.201/bms/download/JczmCsReDtCyA_47ZR9VDh67MY6mzF-nV9fUsPboXfSHPWxEEygfEA5AF-71QN2_fLO_Cb1DbvzYgWqYHBwRtw==/SeMasterV-A0333-A0334-0x6AB0-14.1.1.6-20211221.bin","MD5":"51F20D3CA4591BA028C0747430769559"}',"21880B000168","/tmp/SeMasterV-A0333-A0334-0x6AB0-14.1.1.6-20211221.bin",'{"len": 0,    "period": 0,    "port": "RS485_1",    "mi": 0,    "src_identifier": "__transparent",    "sn": "21880B00016801",    "dtu_sn": "",    "protocol": 3,    "communication_timeout": 0,    "term_addr": "0100000000000000",    "tcp_ip_addr": "",    "tcp_port": 0,    "topic": "ipc/21880B000168/RS485_1/device/21880B00016801/data/Set_Rglt_Raw"}')
--upgrade_node('{"identifier": "__Upgrade","Action": "start","L1_DevId": 1,"Dev_Type": 1,"L3_DevId": 1,"sn": "218812000008_Pack_1","L2_DevId": 1,"Reset": 1,"mi": 99033386,"URL": "http://minio.lnxall.com:8000/000000-camel/upload/20221021/8a1bf528bdc98992b6750bc79c7db9f0.bin","MD5": "A1056C90CB7D2BED86CEBE1D93964ABE","timestamp": 1666343985}',"218812000008","/tmp/SeIbuV-Board205-16S-D0503-0x5024-23.60.1.6-20221021.bin",'{"len": 0,    "period": 0,    "port": "RS485_1",    "mi": 0,    "src_identifier": "__transparent",    "sn": "218812000008_Pack_1",    "dtu_sn": "",    "protocol": 3,    "communication_timeout": 0,    "term_addr": "0100000000000000",    "tcp_ip_addr": "",    "tcp_port": 0,    "topic": "ipc/218812000008/RS485_1/device/218812000008_Pack_1/data/Set_Rglt_Raw"}')