cc-libs/cube.lua

local _VERSION = '1.0.0';
local CUBE_CHANNEL = 64;

local net = require('/apis/net')();

local cubeCommand, firstArg, secondArg = ...;

local function isFlag(name)
  return function(arg)
    return arg == '-' .. name or arg == '--' .. name;
  end
end

local isHelpFlag = isFlag('help');
local isVersionFlag = isFlag('version');

local function writeFile(path, content)
  local file = fs.open(path, "w");

  if not file then
    return false;
  end

  file.write(content)
  file.close();

  return true;
end

local function isConfigFileExists()
  return fs.exists('.cuberc');
end

local function printUsage()
  print('cube usage:')
  print();
  print('\t\t\tcube init');
  print('\t\t\tcube ls');
  print('\t\t\tcube configure');
  print('\t\t\tcube set-startup <machineId> [command]')
  print('\t\t\tcube reboot <machineId>')
  print('\t\t\tcube deploy')
  print('\t\t\tcube version')
  print('\t\t\tcube help <command>')
end

local function printUsageCommand(commandName)
  local USAGES = {
    init = function()
      print('\t\t\tcube init');
      print('Init the master cube directory.')
    end,
    ls = function()
      print('\t\t\tcube ls');
      print('Print all available cubes in the cluster.')
    end,
    configure = function()
      print('\t\t\tcube configure');
      print('Setup remote slave cubes.')
    end,
    ["set-startup"] = function()
      print('\t\t\tcube set-startup <machineId> [command]')
      print('Setup a startup shell command on a remote cube.')
    end,
    reboot = function()
      print('\t\t\tcube reboot <machineId>')
      print('Reboot a cube machine.');
    end,
    deploy = function()
      print('\t\t\tcube deploy')
      print('Transfer files on all slave cubes.')
    end,
    version = function()
      print('\t\t\tcube version')
      print('Print the program version.')
    end,
    help = function()
      print('\t\t\tcube help <command>')
      print('Print help on commands.')
    end,
  }

  local usageFn = USAGES[commandName]

  if not usageFn then
    return printUsage();
  end

  return usageFn();
end

if cubeCommand == nil or cubeCommand == '' or isHelpFlag(cubeCommand) then
  printUsage();
  return;
end

local COMMANDS = {
  init = function()
    if isConfigFileExists() then
      print('cube is already initialized.');
    else
      local ok = writeFile('.cuberc', '');
      if ok then
        print('.cuberc file created');
      else
        error('Cannot create .cuberc file');
      end
    end
  end,
  ls = function()
    print('TODO: ls');
  end,
  configure = function()
    if not isConfigFileExists() then
      print('Error: unable to configure because \'.cuberc\' file is missing\nTry: \'cube init\' command')
      return;
    end
    print('TODO: configure');
  end,
  ["set-startup"] = function()
    if not isConfigFileExists() then
      print('Error: unable to deploy because \'.cuberc\' file is missing\nTry: \'cube init\' command')
      return;
    end

    local machineId = firstArg;
    local shellCommand = secondArg;

    if not machineId then
      printUsageCommand('set-startup');
      return;
    end

    print('changed startup script on machine \'' ..
      tostring(machineId) .. '\' by \'' .. tostring(shellCommand or '') .. '\'');
  end,
  reboot = function()
    if not isConfigFileExists() then
      print('Error: unable to deploy because \'.cuberc\' file is missing\nTry: \'cube init\' command')
      return;
    end

    local machineId = firstArg;

    if not machineId or machineId == '' then
      printUsageCommand('reboot');
      return;
    end

    if tonumber(machineId) then
      local ok, result, packet = net.sendRequest(CUBE_CHANNEL, 'reboot', true, tonumber(machineId))

      if not ok then
        error(result)
      end

      print('reboot machine \'' .. tostring(packet.sourceId) .. '\'.');
      return
    end

    local ok, results, packets = net.sendMultipleRequests(CUBE_CHANNEL, 'reboot', true, machineId);

    if not ok then
      error(results);
    end

    for k in ipairs(results) do
      local packet = packets[k];

      print('reboot machine \'' .. tostring(packet.sourceId) .. '\'.');
    end
  end,
  deploy = function()
    if not isConfigFileExists() then
      print('Error: unable to deploy because \'.cuberc\' file is missing\nTry: \'cube init\' command')
      return;
    end

    print('TODO: deploy.');
  end,
  version = function()
    print('cube client v' .. _VERSION);
  end,
  help = function()
    local commandName = firstArg;
    printUsageCommand(commandName);
  end
}

local cmd;
if isVersionFlag(cubeCommand) then
  cmd = COMMANDS.version;
else
  cmd = COMMANDS[cubeCommand];
end

if not cmd then
  printUsage();
  return;
end

if (isHelpFlag(firstArg)) then
  printUsageCommand(cubeCommand);
  return;
end

cmd();