local files   = require 'files'
local vm      = require 'vm'
local lang    = require 'language'
local config  = require 'config'
local searcher   = require 'core.searcher'
local define  = require 'proto.define'

return function (uri, callback)
    local ast = files.getAst(uri)
    if not ast then
        return
    end

    local cache = vm.getCache 'undefined-field'

    local function getAllDocClassFromInfer(src)
        local infers = cache[src]
        if cache[src] == nil then
            tracy.ZoneBeginN('undefined-field getInfers')
            infers = vm.getInfers(src, 0) or false
            local refs = vm.getRefs(src, 0)
            for _, ref in ipairs(refs) do
                cache[ref] = infers
            end
            tracy.ZoneEnd()
        end

        if not infers then
            return nil
        end

        local mark = {}
        local function addTo(allDocClass, src)
            if not mark[src] then
                allDocClass[#allDocClass+1] = src
                mark[src] = true
            end
        end

        local allDocClass = {}
        for i = 1, #infers do
            local infer = infers[i]
            if infer.type ~= '_G' and infer.type ~= 'any' and infer.type ~= 'table' then
                local inferSource = infer.source
                if inferSource.type == 'doc.class' then
                    addTo(allDocClass, inferSource)
                elseif inferSource.type == 'doc.class.name' then
                    addTo(allDocClass, inferSource.parent)
                elseif inferSource.type == 'doc.type.name' then
                    local docTypes = vm.getDocTypes(inferSource[1])
                    for _, docType in ipairs(docTypes) do
                        if docType.type == 'doc.class.name' then
                            addTo(allDocClass, docType.parent)
                        end
                    end
                end
            end
        end

        return allDocClass
    end

    local function getAllFieldsFromAllDocClass(allDocClass)
        local fields = {}
        local empty = true
        for _, docClass in ipairs(allDocClass) do
            tracy.ZoneBeginN('undefined-field getDefFields')
            local refs = vm.getDefFields(docClass)
            tracy.ZoneEnd()

            for _, ref in ipairs(refs) do
                local name = vm.getKeyName(ref)
                if not name or vm.getKeyType(ref) ~= 'string' then
                    goto CONTINUE
                end
                fields[name] = true
                empty = false
                ::CONTINUE::
            end
        end

        if empty then
            return nil
        else
            return fields
        end
    end

    local function checkUndefinedField(src)
        local fieldName = searcher.getKeyName(src)

        local allDocClass = getAllDocClassFromInfer(src.node)
        if (not allDocClass) or (#allDocClass == 0) then
            return
        end

        local fields = getAllFieldsFromAllDocClass(allDocClass)

        -- 没找到任何 field，跳过检查
        if not fields then
            return
        end

        if not fields[fieldName] then
            local message = lang.script('DIAG_UNDEF_FIELD', fieldName)
            if src.type == 'getfield' and src.field then
                callback {
                    start   = src.field.start,
                    finish  = src.field.finish,
                    message = message,
                }
            elseif src.type == 'getmethod' and src.method then
                callback {
                    start   = src.method.start,
                    finish  = src.method.finish,
                    message = message,
                }
            end
        end
    end
    searcher.eachSourceType(ast.ast, 'getfield', checkUndefinedField);
    searcher.eachSourceType(ast.ast, 'getmethod', checkUndefinedField);
end