LeonOS/data/computercraft/lua/rom/modules/main/cc/completion.lua

-- cc.completion

local expect = require("cc.expect").expect
local settings = require("settings")
local peripheral = require("peripheral")

local c = {}

-- choices and options!
-- extension: add_space can be a table of booleans
function c.choice(text, choices, add_space)
  expect(1, text, "string")
  expect(2, choices, "table")
  expect(3, add_space, "boolean", "table", "nil")

  local options = {}

  for i=1, #choices, 1 do
    local add = add_space
    if type(add_space) == "table" then
      add = add_space[i] or add_space[1]
    end

    if choices[i]:sub(0, #text) == text then
      options[#options+1] = choices[i]:sub(#text+1) .. (add and " " or "")
    end
  end

  return options
end

function c.peripheral(text, add_space)
  return c.choice(text, peripheral.getNames(), add_space)
end

local sides = {"front", "back", "top", "bottom", "left", "right"}
function c.side(text, add_space)
  return c.choice(text, sides, add_space)
end

function c.setting(text, add_space)
  return c.choice(text, settings.getNames(), add_space)
end

return c