local w3xparser = require 'w3xparser'
local progress = require 'progress'

local math_floor = math.floor
local pairs = pairs
local ipairs = ipairs
local tostring = tostring
local wtonumber = w3xparser.tonumber
local next = next
local table_concat = table.concat
local string_lower = string.lower

local w2l
local metadata
local keydata
local slk_type

local function slk_to_type(tp, value)
    if tp == 0 then
        if not value then
            return 0
        end
        return math_floor(wtonumber(value))
    elseif tp == 1 or tp == 2 then
        if not value then
            return 0.0
        end
        return wtonumber(value) + 0.0
    elseif tp == 3 then
        if not value then
            return nil
        end
        if value == '' then
            return value
        end
        value = tostring(value)
        if not value:match '[^ %-%_]' then
            return nil
        end
        return value
    end
end

local function slk_read_data(obj, key, meta, data)
    if meta['repeat'] then
        local type = meta.type
        local t = {}
        if slk_type == 'doodad' then
            for i = 1, 10 do
                t[i] = slk_to_type(type, data[('%s%02d'):format(meta.field, i)]) or ''
            end
        else
            for i = 1, 4 do
                t[i] = slk_to_type(type, data[meta.field..i]) or ''
            end
        end
        obj[key] = t
    else
        obj[key] = slk_to_type(meta.type, data[meta.field])
    end
end

local function slk_read_obj(obj, name, data, keys, metas)
    if data.code then
        obj._code = data.code
    end
    if slk_type == 'unit' and not obj._name then
        obj._name = data.name  -- 单位的反slk可以用name作为线索
    end
    
    for i = 1, #keys do
        slk_read_data(obj, keys[i], metas[i], data)
    end

    local private = metadata[obj._code]
    if private then
        for key, meta in pairs(private) do
            slk_read_data(obj, key, meta, data)
        end
    end
end

local function slk_read(table, slk, keys, meta, update_level)
    for name, data in pairs(slk) do
        if not table[name] then
            table[name] = {
                _id = name,
                _type = slk_type,
            }
        end
        local obj = table[name]
        slk_read_obj(obj, name, data, keys, meta)
        if update_level then
            obj._max_level = obj[update_level]
            if not obj._max_level or obj._max_level == 0 then
                obj._max_level = 1
            elseif obj._max_level and obj._max_level > 10000 then
                message('-report|9其他', ('对象等级太高[%s][%d]'):format(name, obj._max_level))
                message('-tip', '可能会影响此工具的性能')
            end
        end
    end
end

local function txt_to_type(tp, value)
    if tp == 0 then
        if not value then
            return 0
        end
        return math_floor(wtonumber(value))
    elseif tp == 1 or tp == 2 then
        if not value then
            return 0.0
        end
        return wtonumber(value) + 0.0
    elseif tp == 3 then
        if not value then
            return nil
        end
        return tostring(value)
    end
end

local function txt_read_data(name, obj, key, meta, txt)
    if meta.index then
        local value = txt and txt[meta.key] or meta.default
        obj[key] = txt_to_type(meta.type, value and value[meta.index])
        return
    end

    if meta.appendindex then
        local t = {}
        if txt then
            local null
            local max_level = txt[key..'count'] and math.floor(txt[key..'count'][1]) or 1
            for i = max_level, 1, -1 do
                local new_key = i == 1 and key or (key .. (i-1))
                local value = txt[new_key]
                if value and #value > 0 then
                    t[i] = table_concat(value, ',')
                    null = ''
                else
                    t[i] = null
                end
            end
        end
        obj[key] = t
        return
    end

    local value = txt and txt[key] or meta.default
    if not value or #value == 0 then
        local value = txt_to_type(meta.type)
        if not value then
            if meta['repeat'] then
                obj[key] = {}
            end
            return
        end
        if meta['repeat'] then
            obj[key] = {value}
        else
            obj[key] = value
        end
        return
    end
    if meta.concat then
        if #value > 1 then
            obj[key] = table_concat(value, ',')
        else
            obj[key] = txt_to_type(meta.type, value[1])
        end
        return
    end
    if meta['repeat'] then
        obj[key] = {}
        for i = 1, #value do
            obj[key][i] = txt_to_type(meta.type, value[i])
        end
    else
        obj[key] = txt_to_type(meta.type, value[1])
    end
end

local function txt_read(table, txt, used, keys, meta)
    for name, obj in pairs(table) do
        local lname = string_lower(name)
        local txt_data = txt[lname] or used[lname]
        txt[lname] = nil
        used[lname] = txt_data
        for i = 1, #keys do
            txt_read_data(lname, obj, keys[i], meta[i], txt_data)
        end
    end
end

local function txt_set_level(txt)
    for _, obj in pairs(txt) do
        obj._max_level = 1
    end
end

return function(w2l_, loader)
    w2l = w2l_
    metadata = w2l:metadata()
    keydata = w2l:keydata()
    local datas = {}
    local txt = {}
    local used = {}
    local count = 0
    progress:start(0.3)
    for _, filename in pairs(w2l.info.txt) do
        w2l:parse_txt(loader(filename), filename, txt)
    end
    progress:finish()
    
    local count = 0
    progress:start(1)
    for type, names in pairs(w2l.info.slk) do
        local level_key = w2l.info.key.max_level[type]
        slk_type = type

        datas[type] = {}
        for i, filename in ipairs(names) do
            local update_level
            local keys = {}
            local meta = {}
            for _, key in ipairs(keydata[filename]) do
                keys[#keys+1] = key
                meta[#meta+1] = metadata[type][key]
                if key == level_key then
                    update_level = level_key
                end
            end
            slk_read(datas[type], w2l:parse_slk(loader(filename)), keys, meta, update_level)
        end
        if keydata[type] then
            local keys = {}
            local meta = {}
            for _, key in ipairs(keydata[type]) do
                keys[#keys+1] = key
                meta[#meta+1] = metadata[type][key]
            end
            txt_read(datas[type], txt, used, keys, meta)
        end
        count = count + 1
        progress(count / 7)
    end
    txt_set_level(txt)
    progress:finish()

    -- 此单位只在一张单位slk里定义,是无效单位
    datas.unit.nrmf = nil
    datas.txt = txt
    return datas
end