RadLine.lua

local win32 = require "lrwin32"
require "Utils"

-- function FindRegKey(s)

function DebugOutLn(s)
    win32.OutputDebugString(s .. "\n")
end

function CaseInsensitiveLess(s1, s2)
    return s1:lower() < s2:lower()
end

DebugOutLn("RadLine ".._VERSION)

internal = {
    "assoc", "call", "cd", "chdir", "cls", "color", "copy", "date", "del", "dir", "echo", "endlocal", "erase", "exit", "for",
    "ftype", "goto", "if", "md", "mkdir", "mklink", "move", "path", "popd", "prompt", "pushd", "rem", "ren", "rd", "rmdir",
    "set", "setlocal", "shift", "start", "time", "title", "type", "ver", "verify", "vol"
}

function FileNameIsDir(s)
    return s:sub(-1) == "\\";
end

function FileNameLess(s1, s2)
    if FileNameIsDir(s1) and not FileNameIsDir(s2) then
        return true;
    elseif not FileNameIsDir(s1) and FileNameIsDir(s2) then
        return false;
    else
        return CaseInsensitiveLess(s1, s2)
    end
end

FindFilesE = { All = 0, DirOnly = 1, FileOnly = 2 }
function FindFiles(s, e)
    assert(table.contains(FindFilesE, e), "e=" .. tostring(e) .. " is not in FindFilesE")
    s = s:gsub('"', '')

    if s:beginswith("~\\") and tonumber(win32.GetEnvironmentVariable("RADLINE_TILDE") or 0) ~= 0 then
        s = "%USERPROFILE%" .. s:sub(2, -1)
    end

    win32.SetEnvironmentVariable("CD", win32.GetCurrentDirectory())
    s = win32.ExpandEnvironmentStrings(s)
    win32.SetEnvironmentVariable("CD", nil)

    local files = {}
    local FindFileData = {}
    local hFind = win32.FindFirstFile(s, FindFileData)
    if hFind ~= win32.INVALID_HANDLE_VALUE then
        repeat
            if FindFileData.cFileName ~= "." and FindFileData.cFileName ~= ".." then
            --if FindFileData.cFileName ~= "." and FindFileData.cFileName ~= ".." and (FindFileData.dwFileAttributes & win32.FILE_ATTRIBUTE.HIDDEN) == 0 then
                local add = true;
                if     e == FindFilesE.FileOnly and (FindFileData.dwFileAttributes & win32.FILE_ATTRIBUTE.DIRECTORY) ~= 0 then add = false;
                elseif e == FindFilesE.DirOnly  and (FindFileData.dwFileAttributes & win32.FILE_ATTRIBUTE.DIRECTORY) == 0 then add = false;
                end
                if add then
                    if (FindFileData.dwFileAttributes & win32.FILE_ATTRIBUTE.DIRECTORY) ~= 0 then
                        FindFileData.cFileName = FindFileData.cFileName.."\\"
                    end
                    files[#files+1] = FindFileData.cFileName
                end
            end
        until not win32.FindNextFile(hFind, FindFileData)
        win32.FindClose(hFind);
    end

    table.sort(files, FileNameLess)

    return files;
end

function FindExeFiles(s)
    local pathext = win32.GetEnvironmentVariable("PATHEXT"):split(";")
    local dot = s:match'^.*()%.' -- Find last of '.'
    local slash = s:match'^.*()\\' or 0 -- Find last of '\'
    local ext = (dot and dot > slash) and s:sub(dot):upper() or nil

    local f = {}
    if not ext then
        for _,i in ipairs(pathext) do
            table.concat(f, FindFiles(s.."*"..i, FindFilesE.FileOnly))
        end
    else
        for _,i in ipairs(pathext) do
            local dot = i:match'^.*()%.' -- Find last of '.'
            local iext = (dot and dot > 1) and i:sub(dot) or nil
            if i:upper():beginswith(ext) then
                table.concat(f, FindFiles(s..i:sub(#ext + 1), FindFilesE.FileOnly))
            end
            if iext and iext:upper():beginswith(ext) then
                table.concat(f, FindFiles(s..i:sub(dot + #ext), FindFilesE.FileOnly))
            end
            --table.concat(f, FindFiles(s.."*"..i, FindFilesE.FileOnly))
        end
    end
    return f
end

function FindPathExeFiles(s)
    local path = win32.GetEnvironmentVariable("PATH"):split(";")
    local f = {}
    for _,i in ipairs(path) do
        table.concat(f, FindExeFiles(i.."\\"..s))
    end
    return f
end

function FindEnv(s, enclose)
    s = s:upper()
    local e = {}

    local env = win32.GetEnvironmentStrings();
    for k,v in pairs(env) do
        if k:upper():beginswith(s) then
            if enclose then
                e[#e+1] = "%"..k.."%"
            else
                e[#e+1] = k
            end
        end
    end

    return e;
end

function FindEnvBegin(s)
    local count = 0
    for c in s:gmatch("%%") do
        count = count + 1
    end
    if count%2 == 1 then
        return s:match'^.*()%%' -- Find last of '%'
    else
        return nil
    end
end

function FindAlias(s)
    s = s:lower()
    local a = {}
    local aliases = win32.GetConsoleAliases("cmd.exe")
    for k,v in pairs(aliases) do
        if k:lower():beginswith(s) then
            a[#a+1] = k
        end
    end
    return a;
end

function RegEnumKeyEx(k)
	local l = {}
	if k then
		local i = 0
		local n = win32.RegEnumKeyEx(k, i)
		while n do
			table.insert(l, n)
			i = i + 1
			n = win32.RegEnumKeyEx(k, i)
		end
	end
	return l
end

local reg_root_keys = {
	HKCU = win32.HKEY.CURRENT_USER,
	HKCR = win32.HKEY.CLASSES_ROOT,
	HKLM = win32.HKEY.LOCAL_MACHINE,
	HKCC = win32.HKEY.CURRENT_CONFIG,
	HKU = win32.HKEY.USERS,
	HKEY_CURRENT_USER = win32.HKEY.CURRENT_USER,
	HKEY_CLASSES_ROOT = win32.HKEY.CLASSES_ROOT,
	HKEY_LOCAL_MACHINE = win32.HKEY.LOCAL_MACHINE,
	HKEY_CURRENT_CONFIG = win32.HKEY.CURRENT_CONFIG,
	HKEY_USERS = win32.HKEY.USERS,
}

function findlast(s)
    local i = s:reverse():find("\\", 1, true)   -- TODO Can this be better?
    if i then
        i = #s - i + 1
    end
    return i
end

function FindRegKey(s)
    local i = findlast(s)
    local p
    if i then
        p = s:sub(i + 1)
        p = p:lower()
        s = s:sub(1, i - 1)
	else
        p = s:upper()
        s = nil
	end

    local r = {}
	if s then
        local k
        i = s:find("\\", 1, true)
        if i then
            local ks = s:sub(1, i - 1):upper()
            k = win32.RegOpenKey(reg_root_keys[ks], s:sub(i + 1))
        else
            k = reg_root_keys[s:upper()]
        end

        if k then
            local c = RegEnumKeyEx(k)
            for i,v in ipairs(c) do
                if v:lower():beginswith(p) then
                    r[#r+1] = v.."\\"
                end
            end
            win32.RegCloseKey(k)
        end
	else
        for k,v in pairs(reg_root_keys) do
            if k:beginswith(p) then
                r[#r+1] = k.."\\"
            end
        end
	end

    return r
end

command_sep = {
    ["&"] = true;
    ["|"] = true;
    ["&&"] = true;
    ["||"] = true;
}

redirect_sep = {
    [">"] = true;
    [">>"] = true;
    ["<"] = true;
    ["2>"] = true;
}

function GetParam(params, p)
    local s = params[p] or ""
    local i = s:match'^.*()[\\/]' -- Find last of '\' or '/'
    i = i and i + 1 or 1
    if #s > 0 and s:sub(1, 1) == '"' then
        s = s:sub(2)
        if i == 1 then
            i = i + 1
        end
    end
    if #s > 0 and s:sub(#s, #s) == '"' then
        s = s:sub(1, -2)
    end
    return s, i
end

function FindPotentialDefault(params, p)
    local s,i = GetParam(params, p)
    return FindFiles(s.."*", FindFilesE.All), i
end

function FindPotentialExe(params, p)    -- checks path, no alias
    local s,i = GetParam(params, p)
    local f = {}
    table.concat_if(f, s:lower(), internal)
    table.concat(f, FindFiles(s.."*", FindFilesE.DirOnly))
    table.concat(f, FindExeFiles(s))
    table.concat(f, FindPathExeFiles(s))
    return f, i
end

function FindPotentialDirs(params, p)
    local s,i = GetParam(params, p)
    return FindFiles(s.."*", FindFilesE.DirOnly), i
end

function FindPotentialDlls(params, p)
    local s,i = GetParam(params, p)
    return FindFiles(s.."*.dll", FindFilesE.FileOnly), i
end

function FindPotentialAlias(params, p)
    local s,i = GetParam(params, p)
    return FindAlias(s)
end

function FindPotentialEnv(params, p)
    local s,i = GetParam(params, p)
    return FindEnv(s, false)
end

function LookUpExt(t, key)
    local pathext = win32.GetEnvironmentVariable("PATHEXT"):split(";")
    for _,i in ipairs(pathext) do
        local f = rawget(t, (key..i):lower())
        if f then
            return f
        end
    end
    return rawget(t, "_default")
end

command_fn = setmetatable({
    _default = FindPotentialDefault,
    cd = FindPotentialDirs,
    chdir = FindPotentialDirs,
    md = FindPotentialDirs,
    mkdir = FindPotentialDirs,
    pushd = FindPotentialDirs,
}, { __index = LookUpExt })

command_options = setmetatable({}, { __index = LookUpExt })
command_command = setmetatable({}, { __index = LookUpExt }) -- first command line argument

other_env = {
    "%CD%", "%CMDCMDLINE%", "%CMDEXTVERSION%", "%DATE%", "%ERRORLEVEL%", "%RANDOM%", "%TIME%",
    "%FIRMWARE_TYPE%", "%HighestNumaNodeNumber%",
    "%__APPDIR__%", "%__CD__%", "%__PID__%", "%__TICK__%", "%__USERCD__%", "%RADLINE_LOADED%", "%RADLINE_DIR%",
    "%=ExitCode%", "%=ExitCodeAscii%", "%=C:%", "%=D:%", "%=E:%", "%=F:%"
}

function GetCmdOutput(cmd)
    local d = {}
    local f1 = io.popen(cmd, 'r')
    for line in f1:lines() do
        d[#d+1] = line
    end
    f1:close()
    return d
end

command_output = setmetatable({}, { __index = LookUpExt })

function FindPotentialCmdOutput(params, p)
    local s,i = GetParam(params, p)
    local command = params[1]
    local f = {}
    table.concat_if(f, s:lower(), GetCmdOutput(command_output[command]))
    return f
end

function FindPotential(params, p)
    local s,i = GetParam(params, p)
    local e = FindEnvBegin(s)

    if e then
        local f = {}
        table.concat(f, FindEnv(s:sub(e + 1), true))
        table.concat_if(f, s:sub(e):lower(), other_env)
        return f, (e + i - 1)
    elseif p == 1 or command_sep[params[p - 1]] then
        local f = {}
        if i == 1 then
            table.concat_if(f, s:lower(), internal)
            table.concat(f, FindFiles(s.."*", FindFilesE.DirOnly))
            table.concat(f, FindExeFiles(s))
            table.concat(f, FindPathExeFiles(s))
            table.concat(f, FindAlias(s))
        else
            table.concat(f, FindFiles(s.."*", FindFilesE.DirOnly))
            table.concat(f, FindExeFiles(s))
        end
        return f, i
    elseif p > 1 and redirect_sep[params[p - 1]] then
        return FindFiles(s.."*", FindFilesE.All), i
    else
        local command = params[1]
        local cmd = command_command[command]
        if s:sub(1, 1) == "/" or s:sub(1, 1) == "-" then
            local f = {}
            table.concat_if(f, s:lower(), command_options[command])
            return f
        elseif p == 2 and cmd then
            if type(cmd) == "function" then
                return cmd(params, p)
            else
                local f = {}
                table.concat_if(f, s:lower(), cmd)
                return f
            end
        else
            -- TODO Unquote and remove path ???
            local fn = command_fn[command]
            return fn(params, p)
        end
    end
end