wyvern/readline.lua
2018-08-13 08:06:20 +01:00

461 lines
13 KiB
Lua

--- Additional readline
-- Stolen from MBS https://github.com/SquidDev-CC/mbs/blob/master/modules/readline.lua
-- License: https://github.com/SquidDev-CC/mbs/blob/master/LICENSE (MIT)
local complete_fg = colours.grey
local complete_bg = -1
local colour_table = {
["default"] = -1,
}
for k, v in pairs(colours) do
if type(v) == "number" then colour_table[k] = v end
end
for k, v in pairs(colors) do
if type(v) == "number" then colour_table[k] = v end
end
local function clamp(value, min, max)
if value < min then return min end
if value > max then return max end
return value
end
local function read(_sReplaceChar, _tHistory, _fnComplete, _sDefault)
if _sReplaceChar ~= nil and type(_sReplaceChar) ~= "string" then
error("bad argument #1 (expected string, got " .. type(_sReplaceChar) .. ")", 2)
end
if _tHistory ~= nil and type(_tHistory) ~= "table" then
error("bad argument #2 (expected table, got " .. type(_tHistory) .. ")", 2)
end
if _fnComplete ~= nil and type(_fnComplete) ~= "function" then
error("bad argument #3 (expected function, got " .. type(_fnComplete) .. ")", 2)
end
if _sDefault ~= nil and type(_sDefault) ~= "string" then
error("bad argument #4 (expected string, got " .. type(_sDefault) .. ")", 2)
end
term.setCursorBlink(true)
local w = term.getSize()
local sx = term.getCursorPos()
local sLine = _sDefault or ""
local nPos, nScroll = #sLine, 0
local tKillRing, nKillRing = {}, 0
local nHistoryPos
local tDown = {}
local nMod = 0
if _sReplaceChar then _sReplaceChar = _sReplaceChar:sub(1, 1) end
local tCompletions
local nCompletion
local function recomplete()
if _fnComplete and nPos == #sLine then
tCompletions = _fnComplete(sLine)
if tCompletions and #tCompletions > 0 then
nCompletion = 1
else
nCompletion = nil
end
else
tCompletions = nil
nCompletion = nil
end
end
local function uncomplete()
tCompletions = nil
nCompletion = nil
end
local function updateModifier()
nMod = 0
if tDown[keys.leftCtrl] or tDown[keys.rightCtrl] then nMod = nMod + 1 end
if tDown[keys.leftAlt] or tDown[keys.rightAlt] then nMod = nMod + 2 end
end
local function nextWord()
-- Attempt to find the position of the next word
local nOffset = sLine:find("%w%W", nPos + 1)
if nOffset then return nOffset else return #sLine end
end
local function prevWord()
-- Attempt to find the position of the previous word
local nOffset = 1
while nOffset <= #sLine do
local nNext = sLine:find("%W%w", nOffset)
if nNext and nNext < nPos then
nOffset = nNext + 1
else
break
end
end
return nOffset - 1
end
local function redraw(_bClear)
local cursor_pos = nPos - nScroll
if sx + cursor_pos >= w then
-- We've moved beyond the RHS, ensure we're on the edge.
nScroll = sx + nPos - w
elseif cursor_pos < 0 then
-- We've moved beyond the LHS, ensure we're on the edge.
nScroll = nPos
end
local _, cy = term.getCursorPos()
term.setCursorPos(sx, cy)
local sReplace = (_bClear and " ") or _sReplaceChar
if sReplace then
term.write(string.rep(sReplace, math.max(#sLine - nScroll, 0)))
else
term.write(string.sub(sLine, nScroll + 1))
end
if nCompletion then
local sCompletion = tCompletions[ nCompletion ]
local oldText, oldBg
if not _bClear then
oldText = term.getTextColor()
oldBg = term.getBackgroundColor()
if complete_fg >= 0 then term.setTextColor(complete_fg) end
if complete_bg >= 0 then term.setBackgroundColor(complete_bg) end
end
if sReplace then
term.write(string.rep(sReplace, #sCompletion))
else
term.write(sCompletion)
end
if not _bClear then
term.setTextColor(oldText)
term.setBackgroundColor(oldBg)
end
end
term.setCursorPos(sx + nPos - nScroll, cy)
end
local function nsub(start, fin)
if start < 1 or fin < start then return "" end
return sLine:sub(start, fin)
end
local function clear()
redraw(true)
end
local function kill(text)
if #text == "" then return end
nKillRing = nKillRing + 1
tKillRing[nKillRing] = text
end
recomplete()
redraw()
local function acceptCompletion()
if nCompletion then
-- Clear
clear()
-- Find the common prefix of all the other suggestions which start with the same letter as the current one
local sCompletion = tCompletions[ nCompletion ]
sLine = sLine .. sCompletion
nPos = #sLine
-- Redraw
recomplete()
redraw()
end
end
while true do
local sEvent, param, param1, param2 = os.pullEvent()
if nMod == 0 and sEvent == "char" then
-- Typed key
clear()
sLine = string.sub(sLine, 1, nPos) .. param .. string.sub(sLine, nPos + 1)
nPos = nPos + 1
recomplete()
redraw()
elseif sEvent == "paste" then
-- Pasted text
clear()
sLine = string.sub(sLine, 1, nPos) .. param .. string.sub(sLine, nPos + 1)
nPos = nPos + #param
recomplete()
redraw()
elseif sEvent == "key" then
if param == keys.leftCtrl or param == keys.rightCtrl or param == keys.leftAlt or param == keys.rightAlt then
tDown[param] = true
updateModifier()
elseif param == keys.enter then
-- Enter
if nCompletion then
clear()
uncomplete()
redraw()
end
break
-- Moving through text/completions
elseif nMod == 1 and param == keys.d then
-- End of stream, abort
if nCompletion then
clear()
uncomplete()
redraw()
end
sLine = nil
nPos = 0
break
elseif (nMod == 0 and param == keys.left) or (nMod == 1 and param == keys.b) then
-- Left
if nPos > 0 then
clear()
nPos = nPos - 1
recomplete()
redraw()
end
elseif (nMod == 0 and param == keys.right) or (nMod == 1 and param == keys.f) then
-- Right
if nPos < #sLine then
-- Move right
clear()
nPos = nPos + 1
recomplete()
redraw()
else
-- Accept autocomplete
acceptCompletion()
end
elseif nMod == 2 and param == keys.b then
-- Word left
local nNewPos = prevWord()
if nNewPos ~= nPos then
clear()
nPos = nNewPos
recomplete()
redraw()
end
elseif nMod == 2 and param == keys.f then
-- Word right
local nNewPos = nextWord()
if nNewPos ~= nPos then
clear()
nPos = nNewPos
recomplete()
redraw()
end
elseif (nMod == 0 and (param == keys.up or param == keys.down)) or (nMod == 1 and (param == keys.p or param == keys.n)) then
-- Up or down
if nCompletion then
-- Cycle completions
clear()
if param == keys.up or param == keys.p then
nCompletion = nCompletion - 1
if nCompletion < 1 then
nCompletion = #tCompletions
end
elseif param == keys.down or param == keys.n then
nCompletion = nCompletion + 1
if nCompletion > #tCompletions then
nCompletion = 1
end
end
redraw()
elseif _tHistory then
-- Cycle history
clear()
if param == keys.up or param == keys.p then
-- Up
if nHistoryPos == nil then
if #_tHistory > 0 then
nHistoryPos = #_tHistory
end
elseif nHistoryPos > 1 then
nHistoryPos = nHistoryPos - 1
end
elseif param == keys.down or param == keys.n then
-- Down
if nHistoryPos == #_tHistory then
nHistoryPos = nil
elseif nHistoryPos ~= nil then
nHistoryPos = nHistoryPos + 1
end
end
if nHistoryPos then
sLine = _tHistory[nHistoryPos]
nPos, nScroll = #sLine, 0
else
sLine = ""
nPos, nScroll = 0, 0
end
uncomplete()
redraw()
end
elseif (nMod == 0 and param == keys.home) or (nMod == 1 and param == keys.a) then
-- Home
if nPos > 0 then
clear()
nPos = 0
recomplete()
redraw()
end
elseif (nMod == 0 and param == keys["end"]) or (nMod == 1 and param == keys.e) then
-- End
if nPos < #sLine then
clear()
nPos = #sLine
recomplete()
redraw()
end
-- Changing text
elseif nMod == 1 and param == keys.t then
-- Transpose char
local prev, cur
if nPos == #sLine then prev, cur = nPos - 1, nPos
elseif nPos == 0 then prev, cur = 1, 2
else prev, cur = nPos, nPos + 1
end
sLine = nsub(1, prev - 1) .. nsub(cur, cur) .. nsub(prev, prev) .. nsub(cur + 1, #sLine)
nPos = math.min(#sLine, cur)
-- We need the clear to remove the completion
clear(); recomplete(); redraw()
elseif nMod == 2 and param == keys.u then
-- Upcase word
if nPos < #sLine then
local nNext = nextWord()
sLine = nsub(1, nPos) .. nsub(nPos + 1, nNext):upper() .. nsub(nNext + 1, #sLine)
nPos = nNext
clear(); recomplete(); redraw()
end
elseif nMod == 2 and param == keys.l then
-- Lowercase word
if nPos < #sLine then
local nNext = nextWord()
sLine = nsub(1, nPos) .. nsub(nPos + 1, nNext):lower() .. nsub(nNext + 1, #sLine)
nPos = nNext
clear(); recomplete(); redraw()
end
elseif nMod == 2 and param == keys.c then
-- Capitalize word
if nPos < #sLine then
local nNext = nextWord()
sLine = nsub(1, nPos) .. nsub(nPos + 1, nPos + 1):upper() .. nsub(nPos + 2, nNext):lower() .. nsub(nNext + 1, #sLine)
nPos = nNext
clear(); recomplete(); redraw()
end
-- Killing text
elseif nMod == 0 and param == keys.backspace then
-- Backspace
if nPos > 0 then
clear()
sLine = string.sub(sLine, 1, nPos - 1) .. string.sub(sLine, nPos + 1)
nPos = nPos - 1
if nScroll > 0 then nScroll = nScroll - 1 end
recomplete()
redraw()
end
elseif nMod == 0 and param == keys.delete then
-- Delete
if nPos < #sLine then
clear()
sLine = string.sub(sLine, 1, nPos) .. string.sub(sLine, nPos + 2)
recomplete()
redraw()
end
elseif nMod == 1 and param == keys.u then
-- Delete from cursor to beginning of line
if nPos > 0 then
clear()
kill(sLine:sub(1, nPos))
sLine = sLine:sub(nPos + 1)
nPos = 0
recomplete(); redraw()
end
elseif nMod == 1 and param == keys.k then
-- Delete from cursor to end of line
if nPos < #sLine then
clear()
kill(sLine:sub(nPos + 1))
sLine = sLine:sub(1, nPos)
nPos = #sLine
recomplete(); redraw()
end
elseif nMod == 2 and param == keys.d then
-- Delete from cursor to end of next word
if nPos < #sLine then
local nNext = nextWord()
if nNext ~= nPos then
clear()
kill(sLine:sub(nPos + 1, nNext))
sLine = sLine:sub(1, nPos) .. sLine:sub(nNext + 1)
recomplete(); redraw()
end
end
elseif nMod == 1 and param == keys.w then
-- Delete from cursor to beginning of previous word
if nPos > 0 then
local nPrev = prevWord(nPos)
if nPrev ~= nPos then
clear()
kill(sLine:sub(nPrev + 1, nPos))
sLine = sLine:sub(1, nPrev) .. sLine:sub(nPos + 1)
nPos = nPrev
recomplete(); redraw()
end
end
elseif nMod == 1 and param == keys.y then
local insert = tKillRing[nKillRing]
if insert then
clear()
sLine = sLine:sub(1, nPos) .. insert .. sLine:sub(nPos + 1)
nPos = nPos + #insert
recomplete(); redraw()
end
-- Misc
elseif nMod == 0 and param == keys.tab then
-- Tab (accept autocomplete)
acceptCompletion()
end
elseif sEvent == "key_up" then
-- Update the status of the modifier flag
if param == keys.leftCtrl or param == keys.rightCtrl or param == keys.leftAlt or param == keys.rightAlt then
tDown[param] = false
updateModifier()
end
elseif sEvent == "mouse_click" or sEvent == "mouse_drag" and param == 1 then
local _, cy = term.getCursorPos()
if param2 == cy then
-- We first clamp the x position with in the start and end points
-- to ensure we don't scroll beyond the visible region.
local x = clamp(param1, sx, w)
-- Then ensure we don't scroll beyond the current line
nPos = clamp(nScroll + x - sx, 0, #sLine)
redraw()
end
elseif sEvent == "term_resize" then
-- Terminal resized
w = term.getSize()
redraw()
end
end
local _, cy = term.getCursorPos()
term.setCursorBlink(false)
term.setCursorPos(w + 1, cy)
print()
return sLine
end
return read