summaryrefslogtreecommitdiff
path: root/script/core
diff options
context:
space:
mode:
Diffstat (limited to 'script/core')
-rw-r--r--script/core/diagnostics/await-in-sync.lua6
-rw-r--r--script/core/diagnostics/not-yieldable.lua54
2 files changed, 58 insertions, 2 deletions
diff --git a/script/core/diagnostics/await-in-sync.lua b/script/core/diagnostics/await-in-sync.lua
index 5fb6467d..bfb4af38 100644
--- a/script/core/diagnostics/await-in-sync.lua
+++ b/script/core/diagnostics/await-in-sync.lua
@@ -2,6 +2,7 @@ local files = require 'files'
local guide = require 'parser.guide'
local vm = require 'vm'
local lang = require 'language'
+local await = require 'await'
return function (uri, callback)
local state = files.getState(uri)
@@ -9,11 +10,12 @@ return function (uri, callback)
return
end
- guide.eachSourceType(state.ast, 'call', function (source)
+ guide.eachSourceType(state.ast, 'call', function (source) ---@async
local currentFunc = guide.getParentFunction(source)
- if currentFunc and vm.isAsync(currentFunc) then
+ if currentFunc and vm.isAsync(currentFunc, false) then
return
end
+ await.delay()
if not vm.isAsync(source.node, true) then
return
end
diff --git a/script/core/diagnostics/not-yieldable.lua b/script/core/diagnostics/not-yieldable.lua
new file mode 100644
index 00000000..2f80e4d0
--- /dev/null
+++ b/script/core/diagnostics/not-yieldable.lua
@@ -0,0 +1,54 @@
+local files = require 'files'
+local await = require 'await'
+local guide = require 'parser.guide'
+local vm = require 'vm'
+local lang = require 'language'
+
+local function isYieldAble(defs, i)
+ local hasFuncDef
+ for _, def in ipairs(defs) do
+ if def.type == 'function' then
+ hasFuncDef = true
+ local arg = def.args and def.args[i]
+ if arg and vm.isAsync(arg, true) then
+ return true
+ end
+ end
+ if def.type == 'doc.type.function' then
+ hasFuncDef = true
+ local arg = def.args and def.args[i]
+ if arg and vm.isAsync(arg.extends, true) then
+ return true
+ end
+ end
+ end
+ return not hasFuncDef
+end
+
+return function (uri, callback)
+ local state = files.getState(uri)
+ if not state then
+ return
+ end
+
+ guide.eachSourceType(state.ast, 'call', function (source) ---@async
+ if not source.args then
+ return
+ end
+ await.delay()
+ local defs = vm.getDefs(source.node)
+ if #defs == 0 then
+ return
+ end
+ for i, arg in ipairs(source.args) do
+ if vm.isAsync(arg, true)
+ and not isYieldAble(defs, i) then
+ callback {
+ start = source.node.start,
+ finish = source.node.finish,
+ message = lang.script('DIAG_NOT_YIELDABLE', i),
+ }
+ end
+ end
+ end)
+end