opus/sys/extensions/os.lua

require = requireInjector(getfenv(1))
local Config = require('config')
local SHA1 = require('sha1')

local REGISTRY_DIR = 'usr/.registry'

local config = {
  enable = false,
  pocketId = 10,
  distance = 8,
}

local lockId

function lockScreen()
  require = requireInjector(getfenv(1))
  local UI = require('ui')
  local Event = require('event')

  local center = math.floor(UI.term.width / 2)

  local page = UI.Page({
    backgroundColor = colors.blue,
    prompt = UI.Text({
      x = center - 9,
      y = math.floor(UI.term.height / 2),
      value = 'Password',
    }),
    password = UI.TextEntry({
      x = center,
      y = math.floor(UI.term.height / 2),
      width = 8,
      limit = 8 
    }),
    statusBar = UI.StatusBar(),
    accelerators = {
      q = 'back',
    },
  })

  function page:eventHandler(event)
    if event.type == 'key' and event.key == 'enter' then
      Config.load('os', config)
      if SHA1.sha1(self.password.value) == config.password then
        os.locked = false
        Event.exitPullEvents()
        lockId = false
        return true
      else
        self.statusBar:timedStatus('Invalid Password', 3)
      end
    end
    UI.Page.eventHandler(self, event)
  end

  UI:setPage(page)
  Event.pullEvents()
end

function os.verifyPassword(password)
  Config.load('os', config)
  return config.password and password == config.password
end

function os.getSecretKey()
  Config.load('os', config)
  if not config.secretKey then
    config.secretKey = math.random(100000, 999999)
    Config.update('os', config)
  end
  return config.secretKey
end

function os.getPublicKey()

  local exchange = {
    base = 11,
    primeMod = 625210769
  }

  local function modexp(base, exponent, modulo)
    local remainder = base

    for i = 1, exponent-1 do
      remainder = remainder * remainder
      if remainder >= modulo then
        remainder = remainder % modulo
      end
    end

    return remainder
  end

  local secretKey = os.getSecretKey()
  return modexp(exchange.base, secretKey, exchange.primeMod)
end

function os.updatePassword(password)
  Config.load('os', config)
  config.password = password
  Config.update('os', config)
end

function os.getPassword()
  Config.load('os', config)
  return config.password
end

os.lock = function()
  --os.locked = true

  if not lockId then
    lockId = multishell.openTab({
      title = 'Lock',
      env = getfenv(1),
      fn = lockScreen,
      focused = true,
    })
  end
end

os.unlock = function()
  os.locked = false

  if lockId then
    multishell.terminate(lockId)
    lockId = nil
  end
end

function os.isTurtle()
  return not not turtle
end

function os.isAdvanced()
  return term.native().isColor()
end

function os.isPocket()
  return not not pocket
end

function os.getVersion()
  local version

  if _CC_VERSION then
    version = tonumber(_CC_VERSION:gmatch('[%d]+%.?[%d][%d]', '%1')())
  end
  if not version and _HOST then
    version = tonumber(_HOST:gmatch('[%d]+%.?[%d][%d]', '%1')())
  end

  return version or 1.7
end

-- move completely into overview
-- just post event from appstore
function os.registerApp(app, key)

  app.key = SHA1.sha1(key)
  Util.writeTable(fs.combine(REGISTRY_DIR, app.key), app)
  os.queueEvent('os_register_app')
end

function os.unregisterApp(key)

  local filename = fs.combine(REGISTRY_DIR, SHA1.sha1(key))
  if fs.exists(filename) then
    fs.delete(filename)
    os.queueEvent('os_register_app')
  end
end