local core = require 'core.rename'
local files = require 'files'

local function catch_target(script)
    local list = {}
    local cur = 1
    while true do
        local start, finish  = script:find('<[!?].-[!?]>', cur)
        if not start then
            break
        end
        list[#list+1] = {
            start  = start + 2,
            finish = finish - 2,
        }
        cur = finish + 1
    end
    return list
end

local function founded(targets, results)
    if #targets ~= #results then
        return false
    end
    for _, target in ipairs(targets) do
        for _, result in ipairs(results) do
            if target.start == result.start and target.finish == result.finish then
                goto NEXT
            end
        end
        do return false end
        ::NEXT::
    end
    return true
end

function TEST(newName)
    return function (script)
        files.removeAll()
        local target = catch_target(script)
        local start  = script:find('<?', 1, true)
        local finish = script:find('?>', 1, true)
        local pos = (start + finish) // 2 + 1
        local new_script = script:gsub('<[!?]', '  '):gsub('[!?]>', '  ')
        files.setText('', new_script)

        local positions = core('', pos, newName)
        if positions then
            assert(founded(target, positions))
        else
            assert(#target == 0)
        end
    end
end

TEST 'b' [[
local <?a?> = 1
]]

TEST 'b' [[
local <?a?> = 1
<!a!> = 2
<!a!> = <!a!>
]]

TEST 'b' [[
t.<?a?> = 1
a = t.<!a!>
]]

TEST 'b' [[
t[<!'a'!>] = 1
a = t.<?a?>
]]

TEST 'b' [[
:: <?a?> ::
goto <!a!>
]]

TEST 'b' [[
local function f(<!a!>)
    return <?a?>
end
]]