require 'utility'
local uni = require 'ffi.unicode'
local w3xparser = require 'w3xparser'
local lni = require 'lni-c'
local slk = w3xparser.slk
local txt = w3xparser.txt
local ini = w3xparser.ini
local pairs = pairs
local string_lower = string.lower

local mt = {}

local metadata
local keydata
local editstring
local default
local miscnames
local wts

function mt:parse_lni(...)
    return lni(...)
end

function mt:parse_slk(buf)
    return slk(buf)
end

function mt:parse_txt(...)
    return txt(...)
end

function mt:parse_ini(buf)
    return ini(buf)
end

function mt:metadata()
    if not metadata then
        metadata = lni(io.load(self.defined / 'metadata.ini'))
    end
    return metadata
end

function mt:keydata()
    if not keydata then
        keydata = lni(io.load(self.defined / 'keydata.ini'))
    end
    return keydata
end

function mt:miscnames()
    if not miscnames then
        miscnames = lni(io.load(self.defined / 'miscnames.ini'))
    end
    return miscnames
end

function mt:editstring(str)
    -- TODO: WESTRING不区分大小写，不过我们把WorldEditStrings.txt改了，暂时不会出现问题
    if not editstring then
        editstring = ini(io.load(self.mpq / 'ui' / 'WorldEditStrings.txt'))['WorldEditStrings']
    end
    if not editstring[str] then
        return str
    end
    repeat
        str = editstring[str]
    until not editstring[str]
    return str:gsub('%c+', '')
end

local function create_default(w2l)
    local default = {}
    for _, name in ipairs {'ability', 'buff', 'unit', 'item', 'upgrade', 'doodad', 'destructable', 'txt', 'misc'} do
        local str = io.load(w2l.default / (name .. '.ini'))
        if str then
            default[name] = lni(str)
        end
    end
    return default
end

function mt:get_default(create)
    if create then
        return create_default(self)
    end
    if not default then
        default = create_default(self)
    end
    return default
end

-- 同时有英文逗号和英文双引号的字符串存在txt里会解析出错
-- 包含右大括号的字符串存在wts里会解析出错
-- 超过256字节的字符串存在二进制里会崩溃

function mt:load_wts(wts, content, max, reason, fmter)
    if not wts then
        return content
    end
    return content:gsub('TRIGSTR_(%d+)', function(i)
        local str_data = wts[tonumber(i)]
        if not str_data then
            message('-report|9其他', '没有找到字符串定义:', ('TRIGSTR_%03d'):format(i))
            return
        end
        local text = str_data.text
        if fmter then
            text = fmter(text)
        end
        if max and #text > max then
            return self:save_wts(wts, text, reason)
        end
        return text
    end)
end

function mt:save_wts(wts, text, reason)
    message('-report|7保存到wts中的文本', reason)
    message('-tip', '文本保存在wts中会导致加载速度变慢: ', (text:sub(1, 1000):gsub('\r\n', ' ')))
    if text:find('}', 1, false) then
        message('-report|2警告', '文本中的"}"被修改为了"|"')
        message('-tip', (text:sub(1, 1000):gsub('\r\n', ' ')))
        text = text:gsub('}', '|')
    end
    local index = #wts.mark + 1
    wts.mark[index] = text
    return ('TRIGSTR_%03d'):format(index-1)
end

function mt:refresh_wts(wts)
    local lines = {}
    for index, text in ipairs(wts.mark) do
        lines[#lines+1] = ('STRING %d\r\n{\r\n%s\r\n}'):format(index-1, text)
    end
    return table.concat(lines, '\r\n\r\n')
end

local function get_exepath()
    return fs.path(uni.a2u(package.cpath:sub(1, (package.cpath:find(';') or 0)-6))):remove_filename():remove_filename()
end

function mt:initialize(root)
    if self.initialized then
        return
    end
    self.initialized = true
    self.root = root or get_exepath()
    self.template = self.root / 'template'
    self.mpq = self.root / 'script' / 'mpq'
    self.prebuilt = self.root / 'script' / 'prebuilt'
    self.default = self.prebuilt / 'default'
    self.defined = self.prebuilt / 'defined'
    self.info   = lni(assert(io.load(self.root / 'script' / 'info.ini')), 'info')
    self.config = lni(assert(io.load(self.root / 'config.ini')), 'config')
    local fmt = self.config.target_format
    self.config = self.config[fmt]
    self.config.target_format = fmt
end

-- 加载脚本
local slk_convertors = {
    ['frontend']             = true,
    ['frontend_wts']         = true,
    ['frontend_slk']         = true,
    ['frontend_lni']         = true,
    ['frontend_obj']         = true,
    ['frontend_misc']        = true,
    ['frontend_updateobj']   = true,
    ['frontend_updatelni']   = true,
    ['frontend_merge']       = true,
    ['backend']              = true,
    ['backend_mark']         = true,
    ['backend_lni']          = true,
    ['backend_slk']          = true,
    ['backend_txt']          = true,
    ['backend_obj']          = true,
    ['backend_searchjass']   = true,
    ['backend_convertjass']  = true,
    ['backend_convertwtg']   = true,
    ['backend_searchdoo']    = true,
    ['backend_computed']     = true,
    ['backend_extra_txt']    = true,
    ['backend_txtlni']       = true,
    ['backend_misc']         = true,
    ['backend_skin']         = true,
    ['backend_searchparent'] = true,
    ['backend_cleanobj']     = true,
    ['backend_imp']          = true,
    ['backend_optimizejass'] = true,
}

local other_convertors = {
    ['lni2w3i']         = true,
    ['read_w3i']        = true,
    ['w3i2lni']         = true,
    ['create_unitsdoo'] = true,
}

setmetatable(mt, {
    __index = function(self, name)
        if slk_convertors[name] then
            local res = require('slk.' .. name)
            self[name] = res
            return res
        elseif other_convertors[name] then
            local res = require('other.' .. name)
            self[name] = res
            return res
        end
        return nil
    end,
})

return mt