From a089d43dba7438532c56e1c582c5974713bd48f8 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Mon, 24 Jun 2024 21:57:34 +0200
Subject: [PATCH] feat: rewrite some known plugins to lazy specs instead of
 luarocks (plenary.nvim). Closes #1540

---
 lua/lazy/pkg/init.lua     |  3 ++-
 lua/lazy/pkg/rockspec.lua | 27 ++++++++++++++++++++++++++-
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/lua/lazy/pkg/init.lua b/lua/lazy/pkg/init.lua
index ee0daa4..a81ab9c 100644
--- a/lua/lazy/pkg/init.lua
+++ b/lua/lazy/pkg/init.lua
@@ -14,6 +14,7 @@ M.dirty = false
 
 ---@class LazyPkgSpec
 ---@field file string
+---@field source? string
 ---@field spec? LazySpec
 ---@field code? string
 
@@ -54,7 +55,7 @@ function M.update()
           local pkg = {
             name = plugin.name,
             dir = plugin.dir,
-            source = source.name,
+            source = spec.source or source.name,
             file = spec.file,
             spec = spec.spec or {},
           }
diff --git a/lua/lazy/pkg/rockspec.lua b/lua/lazy/pkg/rockspec.lua
index 72defae..0eba34c 100644
--- a/lua/lazy/pkg/rockspec.lua
+++ b/lua/lazy/pkg/rockspec.lua
@@ -7,6 +7,10 @@ local M = {}
 M.dev_suffix = "-1.rockspec"
 M.skip = { "lua" }
 
+M.rewrites = {
+  ["plenary.nvim"] = { "nvim-lua/plenary.nvim", lazy = true },
+}
+
 ---@param plugin LazyPlugin
 function M.deps(plugin)
   local root = Config.options.rocks.root .. "/" .. plugin.name
@@ -32,6 +36,14 @@ end
 ---@param plugin LazyPlugin
 ---@return LazyPkgSpec?
 function M.get(plugin)
+  if M.rewrites[plugin.name] then
+    return {
+      file = "rewrite",
+      source = "lazy",
+      spec = M.rewrites[plugin.name],
+    }
+  end
+
   local rockspec_file ---@type string?
   Util.ls(plugin.dir, function(path, name, t)
     if t == "file" and name:sub(-#M.dev_suffix) == M.dev_suffix then
@@ -59,9 +71,16 @@ function M.get(plugin)
 
   local has_lua = not not vim.uv.fs_stat(plugin.dir .. "/lua")
 
+  ---@type LazyPluginSpec
+  local rewrites = {}
+
   ---@param dep string
   local rocks = vim.tbl_filter(function(dep)
     local name = dep:gsub("%s.*", "")
+    if M.rewrites[name] then
+      table.insert(rewrites, M.rewrites[name])
+      return false
+    end
     return not vim.tbl_contains(M.skip, name)
   end, rockspec.dependencies or {})
 
@@ -75,6 +94,12 @@ function M.get(plugin)
     )
 
   if not use then
+    if #rewrites > 0 then
+      return {
+        file = vim.fn.fnamemodify(rockspec_file, ":t"),
+        spec = rewrites,
+      }
+    end
     return
   end
 
@@ -90,7 +115,7 @@ function M.get(plugin)
       build = "rockspec",
       lazy = lazy,
     },
-  } or nil
+  }
 end
 
 return M