SpaceVim/bundle/nvim-plug/lua/plug/installer.lua

--=============================================================================
-- installer.lua
-- Copyright 2025 Eric Wong
-- Author: Eric Wong < wsdjeg@outlook.com >
-- License: GPLv3
--=============================================================================

local M = {}

local job = require('spacevim.api.job')
local notify = require('spacevim.api.notify')
local jobs = {}
local config = require('plug.config')

local processes = 0

local installation_queue = {}
local building_queue = {}

--- @param plugSpec PluginSpec
local function build(plugSpec)
  if processes >= config.max_processes then
    table.insert(building_queue, plugSpec)
    return
  end
  local jobid = job.start(plugSpec.build, {
    on_stdout = function(id, data)
      for _, v in ipairs(data) do
        notify.notify(jobs['jobid_' .. id .. ':' .. v])
      end
    end,
    on_stderr = function(id, data)
      for _, v in ipairs(data) do
        notify.notify(jobs['jobid_' .. id .. ':' .. v])
      end
    end,
    on_exit = function(id, data, single)
      if data == 0 and single == 0 then
        notify.notify('Successfully build ' .. jobs['jobid_' .. id])
      else
        notify.notify('failed to build ' .. jobs['jobid_' .. id])
      end
      processes = processes - 1
      if #building_queue > 0 then
        build(table.remove(building_queue))
      end
    end,
  })
  processes = processes + 1
  jobs['jobid_' .. jobid] = plugSpec.name
end

--- @param plugSpec PluginSpec
local function install_plugin(plugSpec)
  if processes >= config.max_processes then
    table.insert(installation_queue, plugSpec)
    return
  end
  local cmd = { 'git', 'clone', '--depth', '1' }
  if plugSpec.branch then
    table.insert(cmd, '--branch')
    table.insert(cmd, plugSpec.branch)
  elseif plugSpec.tag then
    table.insert(cmd, '--branch')
    table.insert(cmd, plugSpec.tag)
  end

  table.insert(cmd, plugSpec.url)
  table.insert(cmd, plugSpec.path)
  local jobid = job.start(cmd, {
    on_stdout = function(id, data)
      for _, v in ipairs(data) do
        notify.notify(jobs['jobid_' .. id .. ':' .. v])
      end
    end,
    on_stderr = function(id, data)
      for _, v in ipairs(data) do
        notify.notify(jobs['jobid_' .. id .. ':' .. v])
      end
    end,
    on_exit = function(id, data, single)
      if data == 0 and single == 0 then
        notify.notify('Successfully installed ' .. jobs['jobid_' .. id])
        if plugSpec.build then
          build(plugSpec)
        end
      else
        notify.notify('failed to install ' .. jobs['jobid_' .. id])
      end
      processes = processes - 1
      if #installation_queue > 0 then
        install_plugin(table.remove(installation_queue, 1))
      end
    end,
  })
  processes = processes + 1
  jobs['jobid_' .. jobid] = plugSpec.name
end

M.install = function(plugSpecs)
  for _, v in ipairs(plugSpecs) do
    install_plugin(v)
  end
end

return M