local w3xparser = require 'w3xparser'
local lang = require 'share.lang'
local w2l
local slk = w3xparser.slk
local fixer
local codemapped

local concat_types = {
    abilCode = false,
    abilityList = true,
    aiBuffer = false,
    armorType = false,
    attackBits = false,
    attackTable = true,
    attackType = false,
    attributeType = false,
    bool = false,
    buffList = true,
    channelFlags = false,
    channelType = false,
    combatSound = false,
    deathType = false,
    defenseTable = true,
    defenseType = false,
    defenseTypeInt = false,
    detectionType = false,
    effectList = true,
    fullFlags = false,
    heroAbilityList = true,
    icon = false,
    int = false,
    intList = true,
    interactionFlags = false,
    itemClass = false,
    itemList = true,
    lightningEffect = false,
    lightningList = true,
    model = false,
    modelList = true,
    morphFlags = false,
    moveType = false,
    orderString = false,
    pathingListPrevent = true,
    pathingListRequire = true,
    pathingTexture = false,
    pickFlags = false,
    real = false,
    regenType = false,
    shadowImage = false,
    shadowTexture = false,
    silenceFlags = false,
    soundLabel = false,
    spellDetail = false,
    stackFlags = false,
    string = false,
    stringList = true,
    targetList = true,
    teamColor = false,
    techList = true,
    tilesetList = true,
    uberSplat = false,
    unitClass = true,
    unitCode = false,
    unitList = true,
    unitRace = false,
    unitSound = false,
    unreal = false,
    unrealList = true,
    upgradeClass = false,
    upgradeCode = false,
    upgradeEffect = false,
    upgradeList = true,
    versionFlags = false,
    weaponType = false,
}

local cant_empty = {
    unam = true,
    utub = true,
    ua1z = true,
    gnam = true,
    aart = true,
    atp1 = true,
    fart = true,
    gub1 = true,
    gtp1 = true,
    gar1 = true,
}

local typedefine

local function sortpairs(t)
    local sort = {}
    for k, v in pairs(t) do
        sort[#sort+1] = {k, v}
    end
    table.sort(sort, function (a, b)
        return a[1] < b[1]
    end)
    local n = 1
    return function()
        local v = sort[n]
        if not v then
            return
        end
        n = n + 1
        return v[1], v[2]
    end
end

local function fmtstring(s)
    if s:find '[^%w_]' then
        return ('%q'):format(s)
    end
    return s
end

local function stringify2(inf, outf)
    for name, obj in sortpairs(inf) do
        if type(obj) == 'table' then
            outf[#outf+1] = ('[.%s]'):format(fmtstring(name))
            for k, v in sortpairs(obj) do
                outf[#outf+1] = ('%s = %s'):format(fmtstring(k), v)
            end
        end
    end
end

local function stringify(inf, outf)
    for name, obj in sortpairs(inf) do
        if next(obj) then
            outf[#outf+1] = ('[%s]'):format(fmtstring(name))
            if obj.type == 'misc' then
                outf[#outf+1] = ('type = %s'):format(obj.type)
            end
            stringify2(obj, outf)
            outf[#outf+1] = ''
        end
    end
end

local function stringify_ex(inf)
    local f = {}
    for _, type in ipairs {'ability', 'buff', 'unit', 'item', 'upgrade', 'doodad', 'destructable', 'misc'} do
        stringify({[type]=inf[type]}, f)
        inf[type] = nil
    end
    stringify(inf, f)
    return table.concat(f, '\r\n')
end

local function is_enable(meta, type)
    if type == 'unit' then
        if meta.useHero == 1 or meta.useUnit == 1 or meta.useBuilding == 1 or meta.useCreep == 1 then
            return true
        else
            return false
        end
    end
    if type == 'item' then
        if meta['useItem'] == 1 then
            return true
        else
            return false
        end
    end
    return true
end

local characters = {'A','B','C','D','E','F','G','H','I'}

local function parse_id(w2l, metadata, id, meta, type, has_level)
    local key = meta.field
    local num  = meta.data
    local objs = meta.useSpecific or meta.section
    if num and num ~= 0 then
        key = key .. characters[num]
    end
    if meta._has_index then
        key = key .. '_' .. (meta.index + 1)
    end
    local data = {
        ['id'] = id,
        ['key'] = meta.slk == 'Profile' and meta.field:lower() or meta.field,
        ['type'] = typedefine[meta.type:lower()] or 3,
        ['field'] = key,
        ['appendindex'] = meta.appendIndex == 1 and true or nil,
        ['displayname'] = meta.displayName,
    }
    if concat_types[meta.type] then
        data.concat = true
    end
    if has_level and meta['repeat'] > 0 then
        data['repeat'] = meta['repeat']
    end
    if meta.index == -1 and data.type == 3 and not data.concat then
        data.splite = true
    end
    if num and num ~= 0 then
        data.data = num
    end
    if meta.slk == 'Profile' then
        data.profile = true
    end
    if meta._has_index then
        data.index = meta.index + 1
    end
    if cant_empty[id] then
        data.cantempty = true
    end
    local lkey = key:lower()
    if objs then
        for name in objs:gmatch '%w+' do
            local code = codemapped[name] or name
            if not metadata[code] then
                metadata[code] = { type = type }
            end
            if metadata[code][lkey] and metadata[code][lkey].id ~= data.id then
                if fixer[code][lkey] then
                    if fixer[code][lkey].id == data.id then
                        metadata[code][lkey] = data
                    end
                else
                    w2l.messager.report(lang.report.OTHER, 9, lang.raw.CREATED_ID_CONFLICT:format(name, code))
                end
            else
                metadata[code][lkey] = data
            end
        end
    else
        metadata[type][lkey] = data
    end
end

local function add_user_metadata(meta, type)
    meta['w2lobject'] = {
        ['id'] = 'W2lo',
        ['key'] = 'w2lobject',
        ['type'] = 3,
        ['field'] = 'W2LObject',
        ['profile'] = true,
    }
end

local function add_special(meta, type)
    if type == 'unit' then
        meta['missilespeed_1'].default = "{1500,1500}"
        meta['missilespeed_2'].default = "{1500,1500}"
    end
end

local function create_metadata(w2l, type, metadata, loader)
    metadata[type] = {}
    local has_level = w2l.info.key.max_level[type]
    local tbl = slk(loader('units\\' .. w2l.info.metadata[type]) or loader('doodads\\' .. w2l.info.metadata[type]), w2l.info.metadata[type], true)
    tbl.Ytip = nil
    local has_index = {}
    for k, v in pairs(tbl) do
        -- 进行部分预处理
        local name  = v['field']
        local index = v['index']
        if index and index >= 1 then
            has_index[name] = true
        end
    end
    for k, v in pairs(tbl) do
        local name = v['field']
        if has_index[name] then
            v._has_index = true
        end
    end
    for id, meta in pairs(tbl) do
        if is_enable(meta, type) then
            parse_id(w2l, metadata, id, meta, type, has_level)
        end
    end
    add_user_metadata(metadata[type], type)
    add_special(metadata[type], type)
end

local function copy_code(t, template)
    for name, d in pairs(template) do
        local code = d.code
        local data = t[name]
        if data then
            t[name] = nil
            if t[code] then
                for k, v in pairs(data) do
                    local dest = t[code][k]
                    if dest then
                        if v.id ~= dest.id then
                            w2l.messager.report(lang.report.OTHER, 9, lang.raw.COPIED_ID_CONFLICT:format(k, name, v.id, code, dest.id))
                        end
                    else
                        t[code][k] = v
                    end
                end
            else
                t[code] = {}
                for k, v in pairs(data) do
                    t[code][k] = v
                end
            end
        end
    end
end

local function get_codemapped(w2l, loader)
    local template = w2l:parse_slk(loader 'units\\abilitydata.slk')
    local t = {}
    for id, d in pairs(template) do
        t[id] = d.code
    end
    return t
end

local function get_typedefine(w2l, loader)
    local uniteditordata = w2l:parse_txt(loader 'ui\\uniteditordata.txt')
    local t = {
        int    = 0,
        bool   = 0,
        real   = 1,
        unreal = 2,
    }
    for key, data in pairs(uniteditordata) do
        local value = data['00'][1]
        local tp
        if tonumber(value) then
            tp = 0
        else
            tp = 3
        end
        t[key] = tp
    end
    return t
end

return function(w2l_, fixer_, loader)
    w2l = w2l_
    fixer = fixer_

    codemapped = get_codemapped(w2l, loader)
    typedefine = get_typedefine(w2l, loader)
    
    local metadata = {}
    for _, type in ipairs {'ability', 'buff', 'unit', 'item', 'upgrade', 'doodad', 'destructable', 'misc'} do
        create_metadata(w2l, type, metadata, loader)
    end
    return stringify_ex(metadata)
end
