local guide = require 'parser.guide'
local checkSMT = require 'core.setmetatable'

local m = {}

function m:def(source, callback)
    if source.tag ~= 'self' then
        callback(source, 'local')
    end
    if source.ref then
        for _, ref in ipairs(source.ref) do
            if ref.type == 'setlocal' then
                callback(ref, 'set')
            end
        end
    end
    if source.tag == 'self' then
        local method = source.method
        local node = method.node
        self:eachDef(node, callback)
    end
end

function m:ref(source, callback)
    if source.tag ~= 'self' then
        callback(source, 'local')
    end
    if source.ref then
        for _, ref in ipairs(source.ref) do
            if ref.type == 'setlocal' then
                callback(ref, 'set')
            elseif ref.type == 'getlocal' then
                callback(ref, 'get')
            end
        end
    end
    if source.tag == 'self' then
        local method = source.method
        local node = method.node
        self:eachRef(node, callback)
    end
end

function m:field(source, key, callback)
    local used = {}
    used[source] = true
    local found = false
    local refs = source.ref
    if refs then
        for i = 1, #refs do
            local ref = refs[i]
            if ref.type == 'getlocal' then
                used[ref] = true
                local parent = ref.parent
                if key == guide.getKeyName(parent) then
                    self:childRef(parent, function (src, mode)
                        callback(src, mode)
                        if mode == 'set' then
                            found = true
                        end
                    end)
                end
            elseif ref.type == 'getglobal' then
                used[ref] = true
                if key == guide.getKeyName(ref) then
                    -- _ENV.XXX
                    callback(ref, 'get')
                end
            elseif ref.type == 'setglobal' then
                used[ref] = true
                -- _ENV.XXX = XXX
                if key == guide.getKeyName(ref) then
                    callback(ref, 'set')
                    found = true
                end
            end
        end
    end
    if source.tag == 'self' then
        local method = source.method
        local node = method.node
        self:eachField(node, key, function (src, mode)
            callback(src, mode)
            if mode == 'set' then
                found = true
            end
        end)
    end
    self:eachValue(source, function (src)
        if source ~= src then
            self:eachField(src, key, function (src, mode)
                callback(src, mode)
                if mode == 'set' then
                    found = true
                end
            end)
        end
    end)
    checkSMT(self, key, used, found, callback)
end

function m:value(source, callback)
    callback(source)
    local refs = source.ref
    if refs then
        for i = 1, #refs do
            local ref = refs[i]
            if ref.type == 'setlocal' then
                self:eachValue(ref.value, callback)
            elseif ref.type == 'getlocal' then
                local parent = ref.parent
                if parent.type == 'setmethod' then
                    local func = parent.value
                    if func and func.locals then
                        callback(func.locals[1])
                    end
                end
            end
        end
    end
    if source.value then
        self:eachValue(source.value, callback)
    end
end

return m