summaryrefslogtreecommitdiff
path: root/script/core/completion/postfix.lua
diff options
context:
space:
mode:
Diffstat (limited to 'script/core/completion/postfix.lua')
-rw-r--r--script/core/completion/postfix.lua74
1 files changed, 74 insertions, 0 deletions
diff --git a/script/core/completion/postfix.lua b/script/core/completion/postfix.lua
index 71d3c616..a04b5e65 100644
--- a/script/core/completion/postfix.lua
+++ b/script/core/completion/postfix.lua
@@ -16,8 +16,72 @@ local function register(key)
end
end
+local function hasNonFieldInNode(source)
+ local block = guide.getParentBlock(source)
+ while source ~= block do
+ if source.type == 'call'
+ or source.type == 'getindex'
+ or source.type == 'getmethod' then
+ return true
+ end
+ source = source.parent
+ end
+ return false
+end
+
+register 'function' {
+ function (state, source, callback)
+ if source.type ~= 'getglobal'
+ and source.type ~= 'getfield'
+ and source.type ~= 'getlocal'
+ and source.type ~= 'local' then
+ return
+ end
+ if hasNonFieldInNode(source) then
+ return
+ end
+ local subber = subString(state)
+ callback(string.format('function %s($1)\n\t$0\nend'
+ , subber(source.start + 1, source.finish)
+ ))
+ end
+}
+
+register 'method' {
+ function (state, source, callback)
+ if source.type == 'getfield' then
+ if hasNonFieldInNode(source) then
+ return
+ end
+ local subber = subString(state)
+ callback(string.format('function %s:%s($1)\n\t$0\nend'
+ , subber(source.start + 1, source.dot.start)
+ , subber(source.dot .finish + 1, source.finish)
+ ))
+ end
+ if source.type == 'getmethod' then
+ if hasNonFieldInNode(source.parent) then
+ return
+ end
+ local subber = subString(state)
+ callback(string.format('function %s:%s($1)\n\t$0\nend'
+ , subber(source.start + 1, source.colon.start)
+ , subber(source.colon.finish + 1, source.finish)
+ ))
+ end
+ end
+}
+
register 'pcall' {
function (state, source, callback)
+ if source.type ~= 'getglobal'
+ and source.type ~= 'getfield'
+ and source.type ~= 'getmethod'
+ and source.type ~= 'getindex'
+ and source.type ~= 'getlocal'
+ and source.type ~= 'call' then
+ return
+ end
local subber = subString(state)
if source.type == 'call' then
if source.args and #source.args > 0 then
@@ -40,6 +104,14 @@ register 'pcall' {
register 'xpcall' {
function (state, source, callback)
+ if source.type ~= 'getglobal'
+ and source.type ~= 'getfield'
+ and source.type ~= 'getmethod'
+ and source.type ~= 'getindex'
+ and source.type ~= 'getlocal'
+ and source.type ~= 'call' then
+ return
+ end
local subber = subString(state)
if source.type == 'call' then
if source.args and #source.args > 0 then
@@ -61,6 +133,8 @@ register 'xpcall' {
}
local accepts = {
+ ['local'] = true,
+ ['getlocal'] = true,
['getglobal'] = true,
['getfield'] = true,
['getindex'] = true,