local guide = require 'parser.guide'

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)
    if source.ref then
        for _, ref in ipairs(source.ref) do
            if ref.type == 'getlocal' then
                local parent = ref.parent
                if key == guide.getKeyName(parent) then
                    local tp     = parent.type
                    if     tp == 'setfield' then
                        callback(parent.field, 'set')
                    elseif tp == 'setmethod' then
                        callback(parent.method, 'set')
                    elseif tp == 'setindex' then
                        callback(parent.index, 'set')
                    elseif tp == 'getfield' then
                        callback(parent.field, 'get')
                    elseif tp == 'getmethod' then
                        callback(parent.method, 'get')
                    elseif tp == 'getindex' then
                        callback(parent.index, 'get')
                    end
                end
            elseif ref.type == 'setlocal' then
                self:eachField(ref.value, key, callback)
            elseif ref.type == 'getglobal' then
                -- _ENV.XXX
                if key == guide.getKeyName(ref) then
                    callback(ref, 'get')
                end
            elseif ref.type == 'setglobal' then
                -- _ENV.XXX = XXX
                if key == guide.getKeyName(ref) then
                    callback(ref, 'set')
                end
            end
        end
    end
    if source.tag == 'self' then
        local method = source.method
        local node = method.node
        self:eachField(node, key, callback)
    end
    if source.value then
        self:eachField(source.value, key, callback)
    end
end

return m