local helpers = require('runner.handlers.helpers')
local utils = require('runner.handlers.utils')

return function(buffer)
  utils.run_command(utils.script_path() .. 'get-bins.sh', function(output)
    local bins = {}

    for _, line in pairs(output) do
      for _, data in pairs(line) do
        if vim.trim(data) ~= '' then
          bins[#bins + 1] = data
        end
      end
    end

    local run_handlers = {}

    for _, bin in pairs(bins) do
      run_handlers['Run "' .. bin .. '"'] = helpers.shell_handler('cargo run --bin ' .. bin)
    end

    utils.run_command(utils.script_path() .. 'get-tests.sh', function(output)
      local bins = {}

      for _, line in pairs(output) do
        for _, data in pairs(line) do
          if vim.trim(data) ~= '' then
            bins[#bins + 1] = data
          end
        end
      end

      local handlers = {
        unpack(run_handlers),
        ['Custom'] = helpers.shell_handler('cargo ', true),
        ['Test all'] = helpers.shell_handler('cargo test'),
      }

      for _, bin in pairs(bins) do
        handlers['Test "' .. bin .. '"'] = helpers.shell_handler('cargo test --test ' .. bin)
      end

      helpers.choice(handlers)(buffer)
    end)
  end)
end