summaryrefslogtreecommitdiff
path: root/server/src/core/references.lua
blob: 1d1915f0b14fb39a10d018632d913e24008c6f47 (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
local findSource = require 'core.find_source'

local function parseResult(vm, result, declarat, callback)
    local tp = result.type
    if     tp == 'local' then
        vm:eachInfo(result, function (info)
            if info.source.uri == '' or not info.source.uri then
                return
            end
            if declarat or info.type == 'get' then
                callback(info.source)
            end
        end)
        result.value:eachInfo(function (info)
            if info.source.uri == '' or not info.source.uri then
                return
            end
            if declarat or info.type == 'get' then
                callback(info.source)
            end
        end)
    elseif tp == 'field' then
        vm:eachInfo(result, function (info)
            if info.source.uri == '' or not info.source.uri then
                return
            end
            if declarat or info.type == 'get' then
                callback(info.source)
            end
        end)
        result.value:eachInfo(function (info)
            if info.source.uri == '' or not info.source.uri then
                return
            end
            if declarat or info.type == 'get' then
                callback(info.source)
            end
        end)
    elseif tp == 'label' then
        vm:eachInfo(result, function (info)
            if declarat or info.type == 'goto' then
                callback(info.source)
            end
        end)
    end
end

return function (vm, pos, declarat)
    local result = findSource(vm, pos)
    if not result then
        return nil
    end
    local positions = {}
    local mark = {}
    parseResult(vm, result, declarat, function (source)
        if mark[source] then
            return
        end
        mark[source] = true
        positions[#positions+1] = {
            source.start,
            source.finish,
            source.uri,
        }
    end)
    return positions
end