makit 0.0.158 → 0.0.162

data/lib/makit/port.rb

@@ -1,32 +1,32 @@
-# frozen_string_literal: true
-
-# This module provides classes for the Makit gem.
-module Makit
-  # This class provide methods for working with ports
-  #
-  class Port
-    def self.is_port_available?(port)
-      socket = Socket.new(:INET, :STREAM)
-      socket.bind(Socket.sockaddr_in(port, ""))
-      socket.close
-      true # Port is available for use
-    rescue Errno::EADDRINUSE, Errno::EACCES
-      false # Port is not available for use
-    end
-
-    def self.get_available_port(start_port = 1024, end_port = 65_535)
-      (start_port..end_port).each do |port|
-        return port if is_port_available?(port)
-      end
-      nil # No available port found in the range
-    end
-
-    def self.get_random_port
-      rand(1024..65_535)
-    end
-
-    def self.get_random_available_port
-      get_available_port(get_random_port)
-    end
-  end
-end
+# frozen_string_literal: true
+
+# This module provides classes for the Makit gem.
+module Makit
+  # This class provide methods for working with ports
+  #
+  class Port
+    def self.is_port_available?(port)
+      socket = Socket.new(:INET, :STREAM)
+      socket.bind(Socket.sockaddr_in(port, ""))
+      socket.close
+      true # Port is available for use
+    rescue Errno::EADDRINUSE, Errno::EACCES
+      false # Port is not available for use
+    end
+
+    def self.get_available_port(start_port = 1024, end_port = 65_535)
+      (start_port..end_port).each do |port|
+        return port if is_port_available?(port)
+      end
+      nil # No available port found in the range
+    end
+
+    def self.get_random_port
+      rand(1024..65_535)
+    end
+
+    def self.get_random_available_port
+      get_available_port(get_random_port)
+    end
+  end
+end

data/lib/makit/port_utility.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+require "open3"
+
+module Makit
+  module PortUtility
+    class Error < StandardError; end
+
+    # Ensures a port is free by killing any process using it.
+    #
+    # Examples:
+    #   Makit::PortUtility.ensure_port_free(5261)
+    #   Makit::PortUtility.ensure_port_free(5261, verbose: true)
+    #
+    # Returns true if the port is now free, false if it couldn't be freed.
+    #
+    def self.ensure_port_free(port, verbose: false)
+      raise ArgumentError, "port must be a positive integer" unless port.is_a?(Integer) && port > 0
+
+      return true unless port_in_use?(port)
+
+      puts "Port #{port} is in use. Attempting to free it..." if verbose
+
+      killed = kill_process_on_port(port, verbose: verbose)
+
+      if killed
+        # Wait a moment for the port to be released
+        sleep(1)
+        # Verify port is now free
+        unless port_in_use?(port)
+          puts "Port #{port} is now free." if verbose
+          return true
+        else
+          puts "Warning: Port #{port} is still in use after killing process." if verbose
+          return false
+        end
+      else
+        puts "Warning: Could not kill process on port #{port}." if verbose
+        return false
+      end
+    rescue Error, ArgumentError => e
+      puts "Error freeing port #{port}: #{e.message}" if verbose
+      false
+    end
+
+    # -----------------------
+    # Internals
+    # -----------------------
+
+    def self.port_in_use?(port)
+      if RUBY_PLATFORM =~ /mswin|mingw|cygwin/
+        # Windows: Use netstat
+        stdout, _stderr, status = Open3.capture3("netstat", "-ano")
+        return false unless status.success?
+        stdout.lines.any? { |line| line.include?(":#{port}") && line.include?("LISTENING") }
+      else
+        # Unix/macOS: Use lsof
+        stdout, _stderr, status = Open3.capture3("lsof", "-ti:#{port}")
+        status.success? && !stdout.strip.empty?
+      end
+    rescue Errno::ENOENT
+      # Command not found - assume port is not in use
+      false
+    end
+    private_class_method :port_in_use?
+
+    def self.kill_process_on_port(port, verbose: false)
+      if RUBY_PLATFORM =~ /mswin|mingw|cygwin/
+        # Windows: Use netstat to find PID, then taskkill
+        stdout, _stderr, status = Open3.capture3("netstat", "-ano")
+        return false unless status.success?
+
+        pids = []
+        stdout.lines.each do |line|
+          if line.include?(":#{port}") && line.include?("LISTENING")
+            parts = line.split
+            pid = parts.last
+            pids << pid if pid && pid.match?(/^\d+$/)
+          end
+        end
+
+        killed_any = false
+        pids.uniq.each do |pid|
+          begin
+            stdout, _stderr, status = Open3.capture3("taskkill", "/F", "/PID", pid)
+            if status.success?
+              puts "Killed process #{pid} on port #{port}" if verbose
+              killed_any = true
+            end
+          rescue Errno::ENOENT
+            # taskkill not found
+            return false
+          end
+        end
+
+        killed_any
+      else
+        # Unix/macOS: Use lsof to find PID, then kill
+        stdout, _stderr, status = Open3.capture3("lsof", "-ti:#{port}")
+        return false unless status.success?
+        return true if stdout.strip.empty? # Port is already free
+
+        pids = stdout.strip.split("\n").map(&:strip).reject(&:empty?)
+        return false if pids.empty?
+
+        killed_any = false
+        pids.each do |pid|
+          begin
+            stdout, _stderr, status = Open3.capture3("kill", "-9", pid)
+            if status.success?
+              puts "Killed process #{pid} on port #{port}" if verbose
+              killed_any = true
+            end
+          rescue Errno::ENOENT
+            # kill not found
+            return false
+          end
+        end
+
+        killed_any
+      end
+    rescue Errno::ENOENT
+      # Command not found
+      false
+    end
+    private_class_method :kill_process_on_port
+  end
+end