main.lua

--[[

Copyright 2014-2015 The Luvit Authors. All Rights Reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS-IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

--]]

local bundle = require("luvi").bundle
loadstring(bundle.readfile("luvit-loader.lua"), "bundle:luvit-loader.lua")()

-- Upvalues
local uv = require("uv")
local version = require("./package").version
local log = require("log").log
require("snapshot")

-- Global setup
_G.p = require("pretty-print").prettyPrint

-- Settings
local EXIT_SUCCESS = 0
local EXIT_FAILURE = -1

local aliases = { ["-v"] = "version", ["-h"] = "help" }

local commandLine = {}

function commandLine.run()
  local command = commandLine.processArguments()
  local success, errorMessage = xpcall(function()
    coroutine.wrap(commandLine.executeCommand)(command)
    return uv.run()
  end, debug.traceback)

  if not success then
    return commandLine.reportFailure(errorMessage)
  end

  return commandLine.reportSuccess()
end

function commandLine.processArguments()
  local command = args[1] or "help"
  if command:sub(1, 2) == "--" then
    command = command:sub(3)
  end
  command = aliases[command] or command
  return command
end

function commandLine.executeCommand(command)
  commandLine.outputVersionInfo()

  if command == "version" then
    -- Since the version is always printed, there's nothing left to do
    return commandLine.exitWithCode(EXIT_SUCCESS)
  end

  local commandHandler = "./commands/" .. command .. ".lua"
  if commandLine.isValidCommand(commandHandler) then
    log("command", table.concat(args, " "), "highlight")
    commandLine.executeCommandHandler(commandHandler)
  else
    log("invalid command", command, "failure")
    commandLine.executeCommandHandler("./commands/help.lua")
    commandLine.reportFailure("Invalid Command: " .. command)
  end
end

function commandLine.reportSuccess()
  log("done", "success", "success")
  print()
  return commandLine.exitWithCode(EXIT_SUCCESS)
end

function commandLine.reportFailure(errorMessage)
  log("fail", errorMessage, "failure")
  print()
  return commandLine.exitWithCode(EXIT_FAILURE)
end

function commandLine.outputVersionInfo()
  log("lit version", version)
  log("luvi version", require("luvi").version)
end

function commandLine.exitWithCode(exitCode)
  uv.walk(function(handle)
    if handle then
      local function close()
        if not handle:is_closing() then
          handle:close()
        end
      end
      if handle.shutdown then
        handle:shutdown(close)
      else
        close()
      end
    end
  end)
  uv.run()
  return os.exit(exitCode)
end

function commandLine.isValidCommand(commandHandler)
  return bundle.stat(commandHandler:sub(3)) -- A command is valid if a script handler for it exists
end

function commandLine.executeCommandHandler(commandHandler)
  return require(commandHandler)()
end

commandLine.run()