summaryrefslogtreecommitdiff
path: root/script/core/diagnostics/missing-fields.lua
blob: 6e02f638dcdfeaa668d94cb85df779758a0d182d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
local vm    = require 'vm'
local files = require 'files'
local guide = require 'parser.guide'
local await = require 'await'
local lang  = require 'language'

---@async
return function (uri, callback)
    local state = files.getState(uri)
    if not state then
        return
    end

    ---@async
    guide.eachSourceType(state.ast, 'table', function (src)
        await.delay()

        local defs = vm.getDefs(src)
        for _, def in ipairs(defs) do
            if def.type == 'doc.class' and def.bindSource then
                if guide.isInRange(def.bindSource, src.start) then
                    return
                end
            end
            if def.type == 'doc.type.array'
            or def.type == 'doc.type.table' then
                return
            end
        end
        for _, def in ipairs(defs) do
            if def.type == 'doc.class' then
                if not def.fields then
                    return
                end

                local requiresKeys = {}
                for _, field in ipairs(def.fields) do
                    if  not field.optional
                    and not vm.compileNode(field):isNullable() then
                        local key = vm.getKeyName(field)
                        if key and not requiresKeys[key] then
                            requiresKeys[key] = true
                            requiresKeys[#requiresKeys+1] = key
                        end
                    end
                end

                if #requiresKeys == 0 then
                    return
                end
                local myKeys = {}
                for _, field in ipairs(src) do
                    local key = vm.getKeyName(field)
                    if key then
                        myKeys[key] = true
                    end
                end

                local missedKeys = {}
                for _, key in ipairs(requiresKeys) do
                    if not myKeys[key] then
                        missedKeys[#missedKeys+1] = ('`%s`'):format(key)
                    end
                end

                if #missedKeys == 0 then
                    return
                end

                callback {
                    start   = src.start,
                    finish  = src.finish,
                    message = lang.script('DIAG_MISSING_FIELDS', table.concat(missedKeys, ', ')),
                }
            end
        end
    end)
end