local _VERSION = '1.4.0';

local REPO_BASE = 'https://raw.githubusercontent.com/guillaumearm/cc-libs/';
local LOCAL_MANIFEST_PATH = '/trapos/manifest.json';

local function printUsage()
  print('upgrade usage:');
  print();
  print('\t\tupgrade');
  print('\t\tupgrade --beta');
  print('\t\tupgrade --stable');
  print('\t\tupgrade version');
  print('\t\tupgrade help');
end

local function readLocalBranch()
  if not fs.exists(LOCAL_MANIFEST_PATH) then return nil end
  local f = fs.open(LOCAL_MANIFEST_PATH, 'r');
  if not f then return nil end
  local data = f.readAll();
  f.close();
  if not data or data == '' then return nil end
  local manifest = textutils.unserializeJSON(data);
  return manifest and manifest.branch or nil;
end

local command = ...;

if command == 'version' or command == '-version' or command == '--version' then
  print('upgrade v' .. _VERSION);
  return;
end

if command == 'help' or command == '-help' or command == '--help' then
  printUsage();
  return;
end

local branch;
local extraFlag;

if command == '--beta' or command == '-beta' then
  branch = 'next';
  extraFlag = '--beta';
elseif command == '--stable' or command == '-stable' then
  branch = 'master';
  extraFlag = '--stable';
elseif command ~= nil and command ~= '' then
  printUsage();
  return;
else
  branch = readLocalBranch() or 'master';
end

local installUrl = REPO_BASE .. branch .. '/install.lua';

if extraFlag then
  shell.execute('wget', 'run', installUrl, extraFlag);
else
  shell.execute('wget', 'run', installUrl);
end