feat: add support for socket.dev firewall client (#2088)

1 files changed, 22 insertions, 1 deletions

diff --git a/lua/mason-core/installer/InstallRunner.lua b/lua/mason-core/installer/InstallRunner.lua
index 3eba879a..336f0c93 100644
--- a/lua/mason-core/installer/InstallRunner.lua
+++ b/lua/mason-core/installer/InstallRunner.lua
@@ -39,7 +39,14 @@ function InstallRunner:execute(opts, callback)
     local handle = self.handle
     log.fmt_info("Executing installer for %s %s", handle.package, opts)
 
-    local context = InstallContext:new(handle, opts)
+    local context = InstallContext:new(handle, opts, {
+        suspend = function()
+            self:suspend()
+        end,
+        resume = function()
+            self:resume()
+        end,
+    })
 
     local tailed_output = {}
 
@@ -212,6 +219,20 @@ function InstallRunner:acquire_permit()
     return Result.success(channel)
 end
 
+function InstallRunner:suspend()
+    if self.global_permit then
+        self.global_permit:forget()
+        self.global_permit = nil
+    end
+end
+
+---@async
+function InstallRunner:resume()
+    self.handle:set_state "QUEUED"
+    self.global_permit = self.global_semaphore:acquire()
+    self.handle:set_state "ACTIVE"
+end
+
 ---@private
 function InstallRunner:release_permit()
     if self.global_permit then