feat: add provider interface (#601)

1 files changed, 77 insertions, 0 deletions

diff --git a/lua/mason-core/providers/init.lua b/lua/mason-core/providers/init.lua
new file mode 100644
index 00000000..76bd431b
--- /dev/null
+++ b/lua/mason-core/providers/init.lua
@@ -0,0 +1,77 @@
+local settings = require "mason.settings"
+local log = require "mason-core.log"
+local Result = require "mason-core.result"
+
+---@alias GitHubReleaseAsset {url: string, id: integer, name: string, browser_download_url: string, created_at: string, updated_at: string, size: integer, download_count: integer}
+---@alias GitHubRelease { tag_name: string, prerelease: boolean, draft: boolean, assets:GitHubReleaseAsset[] }
+---@alias GitHubTag { name: string }
+
+---@class GitHubProvider
+---@field get_latest_release? async fun(repo: string, opts?: { include_prerelease?: boolean }): Result # Result<GitHubRelease>
+---@field get_all_release_versions? async fun(repo: string): Result # Result<string[]>
+---@field get_latest_tag? async fun(repo: string): Result # Result<GitHubTag>
+---@field get_all_tags? async fun(repo: string): Result # Result<string[]>
+
+---@alias NpmPackage { name: string, version: string }
+
+---@class NpmProvider
+---@field get_latest_version? async fun(pkg: string): Result # Result<NpmPackage>
+---@field get_all_versions? async fun(pkg: string): Result # Result<string[]>
+
+---@alias PyPiPackage { name: string, version: string }
+
+---@class PyPiProvider
+---@field get_latest_version? async fun(pkg: string): Result # Result<PyPiPackage>
+---@field get_all_versions? async fun(pkg: string): Result # Result<string[]> # Sorting should not be relied upon due to "proprietary" sorting algo in pip that is difficult to replicate in mason-registry-api.
+
+---@class Provider
+---@field github? GitHubProvider
+---@field npm? NpmProvider
+---@field pypi? PyPiProvider
+
+local function service_mt(service)
+    return setmetatable({}, {
+        __index = function(_, method)
+            return function(...)
+                if #settings.current.providers < 1 then
+                    log.error "No providers configured."
+                    return Result.failure "1 or more providers are required."
+                end
+                for _, provider_module in ipairs(settings.current.providers) do
+                    local ok, provider = pcall(require, provider_module)
+                    if ok and provider then
+                        local impl = provider[service] and provider[service][method]
+                        if impl then
+                            ---@type boolean, Result
+                            local ok, result = pcall(impl, ...)
+                            if ok and result:is_success() then
+                                return result
+                            else
+                                if getmetatable(result) == Result then
+                                    log.fmt_error("Provider %s %s failed: %s", service, method, result:err_or_nil())
+                                else
+                                    log.fmt_error("Provider %s %s errored: %s", service, method, result)
+                                end
+                            end
+                        end
+                    else
+                        log.fmt_error("Unable to find provider %s is not registered. %s", provider_module, provider)
+                    end
+                end
+                local err = ("No provider implementation found for %s.%s"):format(service, method)
+                log.error(err)
+                return Result.failure(err)
+            end
+        end,
+    })
+end
+
+---@type Provider
+local providers = setmetatable({}, {
+    __index = function(tbl, service)
+        tbl[service] = service_mt(service)
+        return tbl[service]
+    end,
+})
+
+return providers