free-range 0.2.1 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ce3d41efcb7e6ee1906179227559f6d004b2d07816b3aa230756f4b64f6f0327
-  data.tar.gz: f61a1feddd95066226602b0540c54efeada8a2846c6eb8ba297655a76240454d
+  metadata.gz: ad5fa03c9d62cb4dc177607da76525b1c63876a9a0bd62a334121d9ff2f5e3a9
+  data.tar.gz: 751d1afc032a061e359d038e18350a6f661acb01c465c5e0c0fb6c92720c98f6
 SHA512:
-  metadata.gz: 2bb942063fff21ce9e3a80e1b6223986da7c2a6ff632d01eb05c2e9c2c1a00faaf6655cc544d5b74fcfe14e5dd64c933d581a016e78078c53d8308f98a1af488
-  data.tar.gz: ce98a667f77b4ee2cf0dccca0cba7655b18ae2251e4e5284051bb366f725f58fc55c47ccaeb01522e056e2d68658804ffb6aee6d5a19d28d4bd23d235c8e56c3
+  metadata.gz: 9d4ff514571c47ea9f504eafb441536b19bd25b43fe23e786244651827629fc757b222364eb08542790a5c06c7150f8249525ac76b8acff2a9f8f15251eaaf74
+  data.tar.gz: 72de1f7f68fd17f8a22f64cd73c0d9cb89780246983859c5ed4281a22d4a237bf544564c1e64e1087ac0c19d88d331ef26fce433ea0b088c1d39a2c3e6b571f8

data/lib/free-range.rb

@@ -2,7 +2,10 @@ require 'optparse'
 require 'rmagick'
 require 'fileutils'
 
-module FreeRange
+class FreeRange
+  # Змінні екземпляра для статичних параметрів
+  attr_reader :config, :subscribers_result, :target, :use_color, :debug, :table_mode, :table_png_mode
+
   # Перевірка наявності ImageMagick
   begin
     require 'rmagick'
@@ -465,14 +468,12 @@ module FreeRange
   end
 
   # Метод для заповнення Ranges для одного інтерфейсу
-  # @param config [Config] Configuration object with commands
   # @param interface [String, nil] Interface name or nil for all interfaces
   # @param ranges [Ranges] Object to store VLAN ranges
-  # @param debug [Boolean] Enable debug output
   # @return [void]
-  def self.process_interface(config, interface, ranges, debug = false)
-    full_cmd = "#{config.ssh_command} '#{config.command_ranges(interface)}'"
-    puts "[DEBUG] Executing command: #{full_cmd}" if debug
+  def process_interface(interface, ranges)
+    full_cmd = "#{@config.ssh_command} '#{@config.command_ranges(interface)}'"
+    puts "[DEBUG] Executing command: #{full_cmd}" if @debug
     result = `#{full_cmd}`.strip
     unless result.empty?
       result.each_line do |line|
@@ -484,8 +485,8 @@ module FreeRange
       end
     end
 
-    full_cmd = "#{config.ssh_command} '#{config.command_demux(interface)}'"
-    puts "[DEBUG] Executing command: #{full_cmd}" if debug
+    full_cmd = "#{@config.ssh_command} '#{@config.command_demux(interface)}'"
+    puts "[DEBUG] Executing command: #{full_cmd}" if @debug
     result = `#{full_cmd}`.strip
     unless result.empty?
       result.each_line do |line|
@@ -499,8 +500,8 @@ module FreeRange
       end
     end
 
-    full_cmd = "#{config.ssh_command} '#{config.command_another(interface)}'"
-    puts "[DEBUG] Executing command: #{full_cmd}" if debug
+    full_cmd = "#{@config.ssh_command} '#{@config.command_another(interface)}'"
+    puts "[DEBUG] Executing command: #{full_cmd}" if @debug
     result = `#{full_cmd}`.strip
     unless result.empty?
       result.each_line do |line|
@@ -516,20 +517,13 @@ module FreeRange
   end
 
   # Обробляє VLAN для інтерфейсу та виводить результати
-  # @param config [Config] Configuration object with commands
   # @param interface [String, nil] Interface name or nil for all interfaces
-  # @param subscribers_result [String] Result of subscribers command
-  # @param target [String] Target device hostname
-  # @param use_color [Boolean] Enable colored output
-  # @param debug [Boolean] Enable debug output
-  # @param table_mode [Boolean] Display VLAN distribution table
-  # @param table_png_mode [String, nil] Path to save PNG or nil
   # @return [void]
-  def self.process_and_output(config, interface, subscribers_result, target, use_color, debug, table_mode, table_png_mode)
+  def process_and_output(interface)
     ranges = Ranges.new
     vlans = Vlans.new
-    subscribers_result.each_line do |line|
-      if line.split.first =~ /dhcp(?:_[0-9a-fA-F.]+)?_([^:]+):(\d+)@#{Regexp.escape(target)}$/
+    @subscribers_result.each_line do |line|
+      if line.split.first =~ /dhcp(?:_[0-9a-fA-F.]+)?_([^:]+):(\d+)@#{Regexp.escape(@target)}$/
         subscriber_interface, vlan = $1, $2.to_i
         if interface
           vlans.add_vlan(vlan) if subscriber_interface == interface && vlan > 0
@@ -539,26 +533,33 @@ module FreeRange
       end
     end
 
-    process_interface(config, interface, ranges, debug)
-    if debug
+    process_interface(interface, ranges)
+    if @debug
       puts "\nІнтерфейс: #{interface}" if interface
       Print.ranged(ranges)
       Print.vlans(vlans)
       Print.vlan_ranges(vlans)
       puts
     end
-    if table_png_mode
-      Print.table_png(ranges, vlans, table_png_mode, target, interface)
-    elsif table_mode
-      Print.table(ranges, vlans, use_color, target, interface)
+    if @table_png_mode
+      Print.table_png(ranges, vlans, @table_png_mode, @target, interface)
+    elsif @table_mode
+      Print.table(ranges, vlans, @use_color, @target, interface)
     else
-      Print.combined_ranges(ranges, vlans, use_color, target, interface)
+      Print.combined_ranges(ranges, vlans, @use_color, @target, interface)
     end
   end
 
-  # Основна логіка виконання
-  # @return [void]
-  def self.run
+  # Ініціалізує об’єкт FreeRange із параметрами
+  def initialize
+    @config = nil
+    @subscribers_result = nil
+    @target = nil
+    @use_color = false
+    @debug = false
+    @table_mode = false
+    @table_png_mode = nil
+
     options = {}
     OptionParser.new do |opts|
       opts.banner = <<~BANNER
@@ -583,22 +584,22 @@ module FreeRange
       exit 1
     end
 
-    # Визначаємо змінні з опцій
-    use_color = !options[:no_color] && ENV['TERM'] && ENV['TERM'] != 'dumb'
-    debug = options[:debug]
-    table_mode = options[:table]
-    table_png_mode = options[:table_png]
+    # Ініціалізуємо змінні екземпляра
+    @use_color = !options[:no_color] && ENV['TERM'] && ENV['TERM'] != 'dumb'
+    @debug = options[:debug]
+    @table_mode = options[:table]
+    @table_png_mode = options[:table_png]
     interface = options[:interface]
     config_file = options[:config_file]
 
     # Ініціалізуємо config з порожнім login
-    config = Config.new({ target: ARGV[0], username: nil, password: nil })
+    @config = Config.new({ target: ARGV[0], username: nil, password: nil })
 
     # Завантажуємо конфігураційний файл, якщо він вказаний
     if config_file
       begin
         # Виконуємо конфігураційний файл у контексті існуючого об’єкта config
-        config.instance_eval(File.read(config_file), config_file)
+        @config.instance_eval(File.read(config_file), config_file)
       rescue LoadError, Errno::ENOENT
         puts "Помилка: неможливо завантажити конфігураційний файл '#{config_file}'."
         exit 1
@@ -612,8 +613,8 @@ module FreeRange
     end
 
     # Визначаємо username і password з пріоритетом: аргументи > config > ENV
-    username = options[:username] || config.username || ENV['WHOAMI']
-    password = options[:password] || config.password || ENV['WHATISMYPASSWD']
+    username = options[:username] || @config.username || ENV['WHOAMI']
+    password = options[:password] || @config.password || ENV['WHATISMYPASSWD']
 
     if username.nil? || password.nil?
       puts "Помилка: необхідно вказати ім'я користувача та пароль."
@@ -622,41 +623,41 @@ module FreeRange
     end
 
     login = { target: ARGV[0], username: username, password: password }
-    target = ARGV[0].split('.')[0]
+    @target = ARGV[0].split('.')[0]
     puts "Connecting to device: #{login[:target]}"
 
     # Оновлюємо config з актуальними login даними
-    config = Config.new(login) { |c|
-      c.username = config.username if config.username
-      c.password = config.password if config.password
+    @config = Config.new(login) { |c|
+      c.username = @config.username if @config.username
+      c.password = @config.password if @config.password
     }
 
-    if debug
+    if @debug
       puts "[DEBUG] Values:"
-      puts "[DEBUG] use_color: #{use_color}"
-      puts "[DEBUG] table_mode: #{table_mode}"
-      puts "[DEBUG] table_png_mode: #{table_png_mode}"
+      puts "[DEBUG] use_color: #{@use_color}"
+      puts "[DEBUG] table_mode: #{@table_mode}"
+      puts "[DEBUG] table_png_mode: #{@table_png_mode}"
       puts "[DEBUG] interface: #{interface}"
       puts "[DEBUG] ARGV[0]: #{ARGV[0]}"
-      puts "[DEBUG] target: #{target}"
+      puts "[DEBUG] target: #{@target}"
       puts "[DEBUG] login: #{login}"
       puts "[DEBUG] config_file: #{config_file}"
-      puts "[DEBUG] config.username: #{config.username}"
-      puts "[DEBUG] config.password: #{config.password}"
-      puts "[DEBUG] config.ssh_command: #{config.ssh_command}"
-      puts "[DEBUG] config.subscribers_command: #{config.subscribers_command}"
-      puts "[DEBUG] config.command_interfaces: #{config.command_interfaces}"
+      puts "[DEBUG] config.username: #{@config.username}"
+      puts "[DEBUG] config.password: #{@config.password}"
+      puts "[DEBUG] config.ssh_command: #{@config.ssh_command}"
+      puts "[DEBUG] config.subscribers_command: #{@config.subscribers_command}"
+      puts "[DEBUG] config.command_interfaces: #{@config.command_interfaces}"
     end
 
-    subscribers_result = `#{config.subscribers_command}`.strip
-    if subscribers_result.empty?
+    @subscribers_result = `#{@config.subscribers_command}`.strip
+    if @subscribers_result.empty?
       puts "Помилка: результат subscribers_command порожній. Перевір шлях або доступ."
       exit 1
     end
 
     if interface == "all"
-      full_cmd = "#{config.ssh_command} '#{config.command_interfaces}'"
-      puts "[DEBUG] Executing command: #{full_cmd}" if debug
+      full_cmd = "#{@config.ssh_command} '#{@config.command_interfaces}'"
+      puts "[DEBUG] Executing command: #{full_cmd}" if @debug
       result = `#{full_cmd}`.strip
       if result.empty?
         puts "Помилка: результат команди порожній. Перевір підключення або команду."
@@ -670,10 +671,16 @@ module FreeRange
       end
 
       interfaces.each do |intf|
-        process_and_output(config, intf, subscribers_result, target, use_color, debug, table_mode, table_png_mode)
+        process_and_output(intf)
       end
     else
-      process_and_output(config, interface, subscribers_result, target, use_color, debug, table_mode, table_png_mode)
+      process_and_output(interface)
     end
   end
+
+  # Основна логіка виконання
+  # @return [void]
+  def self.run
+    new
+  end
 end

data/lib/free-range.rb.~

@@ -14,19 +14,30 @@ module FreeRange
 
   # Клас для зберігання конфігураційних команд
   class Config
-    attr_accessor :ssh_command, :subscribers_command
+    attr_accessor :username, :password
 
-    # Ініціалізує об’єкт конфігурації з значеннями за замовчуванням
+    # Ініціалізує об’єкт конфігурації
     # @param login [Hash] Hash with target, username, and password
     # @yield [self] Yields self for block-based configuration
     def initialize(login)
       @login = login
-      # Значення за замовчуванням
-      @ssh_command = "sshpass -p \"#{@login[:password]}\" ssh -C -x -4 -o StrictHostKeyChecking=no #{@login[:username]}@#{@login[:target]}"
-      @subscribers_command = "ssh -C -x roffice /usr/local/share/noc/bin/radius-subscribers"
+      @username = nil
+      @password = nil
       yield self if block_given?
     end
 
+    # Повертає команду SSH
+    # @return [String] SSH command string
+    def ssh_command
+      "sshpass -p \"#{@login[:password]}\" ssh -C -x -4 -o StrictHostKeyChecking=no #{@login[:username]}@#{@login[:target]}"
+    end
+
+    # Повертає команду для отримання даних про передплатників
+    # @return [String] Subscribers command string
+    def subscribers_command
+      "ssh -C -x roffice /usr/local/share/noc/bin/radius-subscribers"
+    end
+
     # Повертає команду для отримання списку інтерфейсів
     # @return [String] Command to fetch interfaces
     def command_interfaces
@@ -36,21 +47,21 @@ module FreeRange
     # Повертає команду для отримання діапазонів VLAN
     # @param interface [String, nil] Interface name or nil for all interfaces
     # @return [String] Command to fetch VLAN ranges
-    def command_ranges(interface)
+    def command_ranges(interface = nil)
       interface ? "show configuration interfaces #{interface} | no-more | display set | match dynamic-profile | match \"ranges ([0-9]+(-[0-9]+)?)\"" : 'show configuration interfaces | no-more | display set | match dynamic-profile | match "ranges ([0-9]+(-[0-9]+)?)"'
     end
 
     # Повертає команду для отримання демультиплексорних VLAN
     # @param interface [String, nil] Interface name or nil for all interfaces
     # @return [String] Command to fetch demux VLANs
-    def command_demux(interface)
+    def command_demux(interface = nil)
       interface ? "show configuration interfaces #{interface} | display set | match unnumbered-address" : 'show configuration interfaces | display set | match unnumbered-address'
     end
 
     # Повертає команду для отримання інших VLAN
     # @param interface [String, nil] Interface name or nil for all interfaces
     # @return [String] Command to fetch other VLANs
-    def command_another(interface)
+    def command_another(interface = nil)
       interface ? "show configuration interfaces #{interface} | display set | match vlan-id" : 'show configuration interfaces | display set | match vlan-id'
     end
   end
@@ -457,9 +468,11 @@ module FreeRange
   # @param config [Config] Configuration object with commands
   # @param interface [String, nil] Interface name or nil for all interfaces
   # @param ranges [Ranges] Object to store VLAN ranges
+  # @param debug [Boolean] Enable debug output
   # @return [void]
-  def self.process_interface(config, interface, ranges)
+  def self.process_interface(config, interface, ranges, debug = false)
     full_cmd = "#{config.ssh_command} '#{config.command_ranges(interface)}'"
+    puts "[DEBUG] Executing command: #{full_cmd}" if debug
     result = `#{full_cmd}`.strip
     unless result.empty?
       result.each_line do |line|
@@ -472,6 +485,7 @@ module FreeRange
     end
 
     full_cmd = "#{config.ssh_command} '#{config.command_demux(interface)}'"
+    puts "[DEBUG] Executing command: #{full_cmd}" if debug
     result = `#{full_cmd}`.strip
     unless result.empty?
       result.each_line do |line|
@@ -486,6 +500,7 @@ module FreeRange
     end
 
     full_cmd = "#{config.ssh_command} '#{config.command_another(interface)}'"
+    puts "[DEBUG] Executing command: #{full_cmd}" if debug
     result = `#{full_cmd}`.strip
     unless result.empty?
       result.each_line do |line|
@@ -500,6 +515,47 @@ module FreeRange
     end
   end
 
+  # Обробляє VLAN для інтерфейсу та виводить результати
+  # @param config [Config] Configuration object with commands
+  # @param interface [String, nil] Interface name or nil for all interfaces
+  # @param subscribers_result [String] Result of subscribers command
+  # @param target [String] Target device hostname
+  # @param use_color [Boolean] Enable colored output
+  # @param debug [Boolean] Enable debug output
+  # @param table_mode [Boolean] Display VLAN distribution table
+  # @param table_png_mode [String, nil] Path to save PNG or nil
+  # @return [void]
+  def self.process_and_output(config, interface, subscribers_result, target, use_color, debug, table_mode, table_png_mode)
+    ranges = Ranges.new
+    vlans = Vlans.new
+    subscribers_result.each_line do |line|
+      if line.split.first =~ /dhcp(?:_[0-9a-fA-F.]+)?_([^:]+):(\d+)@#{Regexp.escape(target)}$/
+        subscriber_interface, vlan = $1, $2.to_i
+        if interface
+          vlans.add_vlan(vlan) if subscriber_interface == interface && vlan > 0
+        else
+          vlans.add_vlan(vlan) if vlan > 0
+        end
+      end
+    end
+
+    process_interface(config, interface, ranges, debug)
+    if debug
+      puts "\nІнтерфейс: #{interface}" if interface
+      Print.ranged(ranges)
+      Print.vlans(vlans)
+      Print.vlan_ranges(vlans)
+      puts
+    end
+    if table_png_mode
+      Print.table_png(ranges, vlans, table_png_mode, target, interface)
+    elsif table_mode
+      Print.table(ranges, vlans, use_color, target, interface)
+    else
+      Print.combined_ranges(ranges, vlans, use_color, target, interface)
+    end
+  end
+
   # Основна логіка виконання
   # @return [void]
   def self.run
@@ -527,14 +583,7 @@ module FreeRange
       exit 1
     end
 
-    username = options[:username] || ENV['WHOAMI']
-    password = options[:password] || ENV['WHATISMYPASSWD']
-    if username.nil? || password.nil?
-      puts "Помилка: необхідно вказати ім'я користувача та пароль."
-      puts "Використовуйте опції -u/--username і -p/--password або змінні оточення WHOAMI і WHATISMYPASSWD."
-      exit 1
-    end
-
+    # Визначаємо змінні з опцій
     use_color = !options[:no_color] && ENV['TERM'] && ENV['TERM'] != 'dumb'
     debug = options[:debug]
     table_mode = options[:table]
@@ -542,25 +591,63 @@ module FreeRange
     interface = options[:interface]
     config_file = options[:config_file]
 
-    login = { target: ARGV[0], username: username, password: password }
-    target = ARGV[0].split('.')[0]
-    puts "Connecting to device: #{login[:target]}"
+    # Ініціалізуємо config з порожнім login
+    config = Config.new({ target: ARGV[0], username: nil, password: nil })
 
-    # Ініціалізація конфігурації
-    config = Config.new(login)
+    # Завантажуємо конфігураційний файл, якщо він вказаний
     if config_file
       begin
-        load config_file
-        config = FreeRange::Config.new(login) { |c| eval(File.read(config_file), binding, config_file) }
+        # Виконуємо конфігураційний файл у контексті існуючого об’єкта config
+        config.instance_eval(File.read(config_file), config_file)
       rescue LoadError, Errno::ENOENT
         puts "Помилка: неможливо завантажити конфігураційний файл '#{config_file}'."
         exit 1
+      rescue ArgumentError => e
+        puts "Помилка в аргументах конфігураційного файлу '#{config_file}': #{e.message}"
+        exit 1
       rescue StandardError => e
         puts "Помилка в конфігураційному файлі '#{config_file}': #{e.message}"
         exit 1
       end
     end
 
+    # Визначаємо username і password з пріоритетом: аргументи > config > ENV
+    username = options[:username] || config.username || ENV['WHOAMI']
+    password = options[:password] || config.password || ENV['WHATISMYPASSWD']
+
+    if username.nil? || password.nil?
+      puts "Помилка: необхідно вказати ім'я користувача та пароль."
+      puts "Використовуйте опції -u/--username і -p/--password, конфігураційний файл або змінні оточення WHOAMI і WHATISMYPASSWD."
+      exit 1
+    end
+
+    login = { target: ARGV[0], username: username, password: password }
+    target = ARGV[0].split('.')[0]
+    puts "Connecting to device: #{login[:target]}"
+
+    # Оновлюємо config з актуальними login даними
+    config = Config.new(login) { |c|
+      c.username = config.username if config.username
+      c.password = config.password if config.password
+    }
+
+    if debug
+      puts "[DEBUG] Values:"
+      puts "[DEBUG] use_color: #{use_color}"
+      puts "[DEBUG] table_mode: #{table_mode}"
+      puts "[DEBUG] table_png_mode: #{table_png_mode}"
+      puts "[DEBUG] interface: #{interface}"
+      puts "[DEBUG] ARGV[0]: #{ARGV[0]}"
+      puts "[DEBUG] target: #{target}"
+      puts "[DEBUG] login: #{login}"
+      puts "[DEBUG] config_file: #{config_file}"
+      puts "[DEBUG] config.username: #{config.username}"
+      puts "[DEBUG] config.password: #{config.password}"
+      puts "[DEBUG] config.ssh_command: #{config.ssh_command}"
+      puts "[DEBUG] config.subscribers_command: #{config.subscribers_command}"
+      puts "[DEBUG] config.command_interfaces: #{config.command_interfaces}"
+    end
+
     subscribers_result = `#{config.subscribers_command}`.strip
     if subscribers_result.empty?
       puts "Помилка: результат subscribers_command порожній. Перевір шлях або доступ."
@@ -569,6 +656,7 @@ module FreeRange
 
     if interface == "all"
       full_cmd = "#{config.ssh_command} '#{config.command_interfaces}'"
+      puts "[DEBUG] Executing command: #{full_cmd}" if debug
       result = `#{full_cmd}`.strip
       if result.empty?
         puts "Помилка: результат команди порожній. Перевір підключення або команду."
@@ -582,60 +670,10 @@ module FreeRange
       end
 
       interfaces.each do |intf|
-        ranges = Ranges.new
-        vlans = Vlans.new
-        subscribers_result.each_line do |line|
-          if line.split.first =~ /dhcp(?:_[0-9a-fA-F.]+)?_([^:]+):(\d+)@#{Regexp.escape(target)}$/
-            subscriber_interface, vlan = $1, $2.to_i
-            vlans.add_vlan(vlan) if subscriber_interface == intf && vlan > 0
-          end
-        end
-
-        process_interface(config, intf, ranges)
-        if debug
-          puts "\nІнтерфейс: #{intf}"
-          Print.ranged(ranges)
-          Print.vlans(vlans)
-          Print.vlan_ranges(vlans)
-          puts
-        end
-        if table_png_mode
-          Print.table_png(ranges, vlans, table_png_mode, target, intf)
-        elsif table_mode
-          Print.table(ranges, vlans, use_color, target, intf)
-        else
-          Print.combined_ranges(ranges, vlans, use_color, target, intf)
-        end
+        process_and_output(config, intf, subscribers_result, target, use_color, debug, table_mode, table_png_mode)
       end
     else
-      ranges = Ranges.new
-      vlans = Vlans.new
-      subscribers_result.each_line do |line|
-        if line.split.first =~ /dhcp(?:_[0-9a-fA-F.]+)?_([^:]+):(\d+)@#{Regexp.escape(target)}$/
-          subscriber_interface, vlan = $1, $2.to_i
-          if interface
-            vlans.add_vlan(vlan) if subscriber_interface == interface && vlan > 0
-          else
-            vlans.add_vlan(vlan) if vlan > 0
-          end
-        end
-      end
-
-      process_interface(config, interface, ranges)
-      if debug
-        puts "\nІнтерфейс: #{interface}" if interface
-        Print.ranged(ranges)
-        Print.vlans(vlans)
-        Print.vlan_ranges(vlans)
-        puts
-      end
-      if table_png_mode
-        Print.table_png(ranges, vlans, table_png_mode, target, interface)
-      elsif table_mode
-        Print.table(ranges, vlans, use_color, target, interface)
-      else
-        Print.combined_ranges(ranges, vlans, use_color, target, interface)
-      end
+      process_and_output(config, interface, subscribers_result, target, use_color, debug, table_mode, table_png_mode)
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: free-range
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
 platform: ruby
 authors:
 - Oleksandr Russkikh //aka Olden Gremlin
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-09-29 00:00:00.000000000 Z
+date: 2025-09-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rmagick
@@ -77,7 +77,6 @@ files:
 - bin/free-range
 - lib/free-range.rb
 - lib/free-range.rb.~
-- lib/free-range.rb~
 homepage: https://github.com/oldengremlin/free-range
 licenses:
 - Apache-2.0

data/lib/free-range.rb~

@@ -1,531 +0,0 @@
-require 'optparse'
-require 'rmagick'
-require 'fileutils'
-
-module FreeRange
-
-  # Перевірка наявності ImageMagick
-  begin
-    require 'rmagick'
-  rescue LoadError
-    puts "Помилка: бібліотека rmagick не встановлена або ImageMagick недоступний."
-    puts "Встановіть ImageMagick і виконайте: gem install rmagick"
-    exit 1
-  end
-
-  # Абстрактний клас для роботи з VLAN
-  class VlanContainer
-    def initialize
-      @vlans = []
-    end
-
-    # Додаємо VLAN до списку (віртуальний метод, може бути перевизначений)
-    def add_vlan(vlan)
-      @vlans << vlan
-    end
-
-    # Повертаємо масив VLAN
-    def vlans
-      @vlans.uniq.sort
-    end
-
-    # Створюємо хеш діапазонів із VLAN
-    def ranges
-      return {} if @vlans.empty?
-
-      vlan_ranges_hash = {}
-      sorted_vlans = vlans
-      start = sorted_vlans.first
-      prev = start
-
-      sorted_vlans[1..-1].each do |vlan|
-        unless vlan == prev + 1
-          vlan_ranges_hash[start] = prev
-          start = vlan
-        end
-        prev = vlan
-      end
-      vlan_ranges_hash[start] = prev
-
-      vlan_ranges_hash
-    end
-  end
-
-  # Клас для зберігання та обробки діапазонів
-  class Ranges < VlanContainer
-    def initialize
-      super
-      @another_in_ranges = []
-    end
-
-    # Додаємо діапазон до списку VLAN-ів
-    def add_range(start, finish)
-      (start..finish).each { |vlan| @vlans << vlan }
-    end
-
-    # Додаємо діапазон до списку "інших" VLAN-ів
-    def add_another_range(start, finish)
-      (start..finish).each { |vlan| @another_in_ranges << vlan }
-    end
-
-    # Повертаємо масив "інших" VLAN-ів
-    def another_in_ranges
-      @another_in_ranges.uniq.sort
-    end
-  end
-
-  # Клас для зберігання та обробки VLAN-ів
-  class Vlans < VlanContainer
-    # Метод add_vlan уже успадковано від VlanContainer
-    # Метод vlans уже успадковано
-    # Метод ranges уже успадковано, але перейменуємо для зрозумілості
-    alias vlan_ranges ranges
-  end
-
-  # Клас для виводу даних
-  class Print
-    def self.ranged(ranges)
-      puts "\nСформований хеш діапазонів (в порядку зростання):"
-      if ranges.ranges.empty?
-        puts "Не знайдено діапазонів."
-      else
-        ranges.ranges.sort_by { |k, _| k }.each do |start, end_val|
-          puts "range[#{start}]=#{end_val}"
-        end
-      end
-    end
-
-    def self.vlans(vlans)
-      puts "\nЗайняті VLAN-и в межах діапазонів (в порядку зростання):"
-      if vlans.vlans.empty?
-        puts "Не знайдено VLAN-ів у межах діапазонів."
-      else
-        puts vlans.vlans.uniq.sort.join(", ")
-      end
-    end
-
-    def self.vlan_ranges(vlans)
-      puts "\nДіапазони VLAN-ів (в порядку зростання):"
-      vlan_ranges = vlans.vlan_ranges
-      if vlan_ranges.empty?
-        puts "Не знайдено діапазонів VLAN-ів."
-      else
-        vlan_ranges.sort_by { |k, _| k }.each do |start, end_val|
-          puts "range[#{start}]=#{end_val}"
-        end
-      end
-    end
-
-    def self.combined_ranges(ranges, vlans, use_color, target, interface = nil)
-      if ranges.ranges.empty?
-        puts "Не знайдено діапазонів."
-        return
-      end
-
-      all_vlans, _status_counts = build_vlan_statuses(ranges, vlans)
-      result = []
-      sorted_vlans = all_vlans.keys.sort
-      start = sorted_vlans.first
-      prev = start
-      status = all_vlans[start]
-
-      sorted_vlans[1..-1].each do |vlan|
-        unless vlan == prev + 1 && all_vlans[vlan] == status
-          result << format_range(start, prev, status, use_color)
-          start = vlan
-          status = all_vlans[vlan]
-        end
-        prev = vlan
-      end
-      result << format_range(start, prev, status, use_color)
-
-      puts result.join(',')
-    end
-
-    def self.table(ranges, vlans, use_color, target, interface = nil)
-      puts "VLAN Distribution for #{target}#{interface ? " (#{interface})" : ''}"
-      all_vlans, status_counts = build_vlan_statuses(ranges, vlans)
-
-      puts "     0         1         2         3         4         5         6         7         8         9         "
-      (0..40).each do |h|
-        start_vlan = h * 100
-        end_vlan = [start_vlan + 99, 4094].min
-        row = (start_vlan..end_vlan).map { |vlan| format_table_char(all_vlans[vlan] || ' ', use_color) }.join
-        puts "#{format("%4d", start_vlan)} #{row}"
-      end
-
-      legend_parts = [
-        ["Legend: ", nil],
-        ["f", 'f'], ["=free", nil], [", ", nil],
-        ["b", 'b'], ["=busy", nil], [", ", nil],
-        ["e", 'e'], ["=error", nil], [", ", nil],
-        ["c", 'c'], ["=configured", nil], [", ", nil],
-        ["a", 'a'], ["=another", nil], [", ", nil],
-        ["u", 'u'], ["=unused", nil]
-      ]
-      legend_text = legend_parts.map do |text, status|
-        if status && use_color
-          format_table_char(status, use_color)
-        else
-          text
-        end
-      end.join
-      puts "\n#{legend_text}"
-
-      summary_parts = [
-        ["Total: ", nil],
-        ["f", 'f'], ["=#{status_counts['f']}", nil], [", ", nil],
-        ["b", 'b'], ["=#{status_counts['b']}", nil], [", ", nil],
-        ["e", 'e'], ["=#{status_counts['e']}", nil], [", ", nil],
-        ["c", 'c'], ["=#{status_counts['c']}", nil], [", ", nil],
-        ["a", 'a'], ["=#{status_counts['a']}", nil], [", ", nil],
-        ["u", 'u'], ["=#{status_counts['u']}", nil]
-      ]
-      summary_text = summary_parts.map do |text, status|
-        if status && use_color
-          format_table_char(status, use_color)
-        else
-          text
-        end
-      end.join
-      puts summary_text
-    end
-
-    def self.table_png(ranges, vlans, path, target, interface = nil)
-      all_vlans, status_counts = build_vlan_statuses(ranges, vlans)
-      cell_width = 12
-      cell_height = 20
-      rows = 41
-      cols = 100
-      header_height = 60
-      label_width = 50
-      width = label_width + cols * cell_width + 10
-      height = header_height + rows * cell_height + 20 + 50
-      font_size = 14
-      title_font_size = 18
-      font = 'Courier'
-
-      canvas = Magick::Image.new(width, height) { |options| options.background_color = 'white' }
-      gc = Magick::Draw.new
-      gc.font = font
-      gc.pointsize = font_size
-      gc.text_antialias = true
-
-      gc.fill('black')
-      gc.pointsize = title_font_size
-      gc.text(10, 25, "VLAN Distribution for #{target}#{interface ? " (#{interface})" : ''}")
-      gc.pointsize = font_size
-
-      (0..9).each do |i|
-        x = label_width + i * 10 * cell_width - 3
-        gc.fill('black')
-        gc.text(x + 5, header_height - 5, i.to_s)
-      end
-
-      (0..40).each do |h|
-        start_vlan = h * 100
-        end_vlan = [start_vlan + 99, 4094].min
-        y = header_height + h * cell_height
-        gc.fill('black')
-        gc.text(5, y + font_size, format("%4d", start_vlan))
-
-        (start_vlan..end_vlan).each_with_index do |vlan, i|
-          status = all_vlans[vlan] || ' '
-          x = label_width + i * cell_width
-          color = case status
-                  when 'f' then '#00FF00'  # Зелений
-                  when 'b' then '#FFFF00'  # Жовтий
-                  when 'e' then '#FF0000'  # Червоний
-                  when 'c' then '#FF00FF'  # Фіолетовий
-                  when 'a' then '#0000FF'  # Синій
-                  when 'u' then '#555555'  # Темно-сірий
-                  else 'white'  # Пробіл
-                  end
-          gc.fill(color)
-          gc.rectangle(x, y, x + cell_width - 1, y + cell_height - 1)
-          gc.fill('black')
-          gc.text(x + 2, y + font_size, status) unless status == ' '
-        end
-      end
-
-      legend_y = height - 50
-      x = 10
-      legend_parts = [
-        ["Legend: ", nil],
-        ["f", '#00FF00'], ["=free", nil], [", ", nil],
-        ["b", '#FFFF00'], ["=busy", nil], [", ", nil],
-        ["e", '#FF0000'], ["=error", nil], [", ", nil],
-        ["c", '#FF00FF'], ["=configured", nil], [", ", nil],
-        ["a", '#0000FF'], ["=another", nil], [", ", nil],
-        ["u", '#555555'], ["=unused", nil]
-      ]
-      legend_parts.each do |text, color|
-        if color
-          gc.fill(color)
-          gc.rectangle(x, legend_y - font_size + 2, x + 10, legend_y + 2)
-          gc.fill('black')
-          gc.text(x + 2, legend_y, text)
-          x += 12
-        else
-          gc.fill('black')
-          gc.text(x, legend_y, text)
-          x += text.length * 8
-        end
-      end
-
-      summary_y = height - 30
-      x = 10
-      summary_parts = [
-        ["Total: ", nil],
-        ["f", '#00FF00'], ["=#{status_counts['f']}", nil], [", ", nil],
-        ["b", '#FFFF00'], ["=#{status_counts['b']}", nil], [", ", nil],
-        ["e", '#FF0000'], ["=#{status_counts['e']}", nil], [", ", nil],
-        ["c", '#FF00FF'], ["=#{status_counts['c']}", nil], [", ", nil],
-        ["a", '#0000FF'], ["=#{status_counts['a']}", nil], [", ", nil],
-        ["u", '#555555'], ["=#{status_counts['u']}", nil]
-      ]
-      summary_parts.each do |text, color|
-        if color
-          gc.fill(color)
-          gc.rectangle(x, summary_y - font_size + 2, x + 10, summary_y + 2)
-          gc.fill('black')
-          gc.text(x + 2, summary_y, text)
-          x += 12
-        else
-          gc.fill('black')
-          gc.text(x, summary_y, text)
-          x += text.length * 8
-        end
-      end
-
-      gc.draw(canvas)
-      FileUtils.mkdir_p(path) unless Dir.exist?(path)
-      filename = File.join(path, "free-range-#{target}#{interface ? "-#{interface.tr('/', '-')}" : ''}.png")
-      canvas.write(filename)
-      puts "Зображення збережено: #{filename}"
-    end
-
-    private
-
-    def self.build_vlan_statuses(ranges, vlans)
-      all_vlans = {}
-      ranges.ranges.each do |start, finish|
-        (start..finish).each { |vlan| all_vlans[vlan] = 'f' }
-      end
-      vlans.vlans.uniq.each { |vlan| all_vlans[vlan] = all_vlans.key?(vlan) ? 'b' : 'e' }
-      (1..4094).each { |vlan| all_vlans[vlan] = 'u' unless all_vlans.key?(vlan) }
-      ranges.another_in_ranges.each { |vlan| all_vlans[vlan] = all_vlans.key?(vlan) && all_vlans[vlan] != 'u' ? 'c' : 'a' }
-      status_counts = { 'f' => 0, 'b' => 0, 'e' => 0, 'c' => 0, 'a' => 0, 'u' => 0 }
-      all_vlans.each_value { |status| status_counts[status] += 1 }
-
-      [all_vlans, status_counts]
-    end
-
-    def self.format_range(start, finish, status, use_color)
-      range_text = start == finish ? "#{start}" : "#{start}-#{finish}"
-      range_text_with_status = "#{range_text}(#{status})"
-      if use_color
-        case status
-        when 'f' then "\e[32m#{range_text}\e[0m"  # Зелений для free
-        when 'b' then "\e[33m#{range_text}\e[0m"  # Жовтий для busy
-        when 'e' then "\e[31m#{range_text}\e[0m"  # Червоний для error
-        when 'c' then "\e[35m#{range_text}\e[0m"  # Фіолетовий для configured
-        when 'a' then "\e[34m#{range_text}\e[0m"  # Синій для another
-        when 'u' then "\e[90m#{range_text}\e[0m"  # Темно-сірий для unused
-        else range_text  # Без кольору для інших статусів
-        end
-      else
-        range_text_with_status  # Текстовий вивід зі статусами
-      end
-    end
-
-    def self.format_table_char(status, use_color)
-      if use_color
-        case status
-        when 'f' then "\e[48;5;2m\e[30m#{status}\e[0m"  # Зелений фон, чорний текст
-        when 'b' then "\e[48;5;3m\e[30m#{status}\e[0m"  # Жовтий фон, чорний текст
-        when 'e' then "\e[48;5;1m\e[30m#{status}\e[0m"  # Червоний фон, чорний текст
-        when 'c' then "\e[48;5;5m\e[30m#{status}\e[0m"  # Фіолетовий фон, чорний текст
-        when 'a' then "\e[48;5;4m\e[30m#{status}\e[0m"  # Синій фон, чорний текст
-        when 'u' then "\e[48;5;8m\e[30m#{status}\e[0m"  # Темно-сірий фон, чорний текст
-        else status  # Без кольору для інших статусів
-        end
-      else
-        status  # Текстовий вивід без кольорів
-      end
-    end
-  end
-
-  # Метод для заповнення Ranges для одного інтерфейсу
-  def self.process_interface(ssh_command, interface, ranges)
-    command_ranges = interface ? "show configuration interfaces #{interface} | no-more | display set | match dynamic-profile | match \"ranges ([0-9]+(-[0-9]+)?)\"" : 'show configuration interfaces | no-more | display set | match dynamic-profile | match "ranges ([0-9]+(-[0-9]+)?)"'
-    command_demux = interface ? "show configuration interfaces #{interface} | display set | match unnumbered-address" : 'show configuration interfaces | display set | match unnumbered-address'
-    command_another = interface ? "show configuration interfaces #{interface} | display set | match vlan-id" : 'show configuration interfaces | display set | match vlan-id'
-
-    full_cmd = "#{ssh_command} '#{command_ranges}'"
-    result = `#{full_cmd}`.strip
-    unless result.empty?
-      result.each_line do |line|
-        if line =~ /ranges (\d+)(?:-(\d+))?/
-          start_range = $1.to_i
-          end_range = $2 ? $2.to_i : $1.to_i
-          ranges.add_range(start_range, end_range)
-        end
-      end
-    end
-
-    full_cmd = "#{ssh_command} '#{command_demux}'"
-    result = `#{full_cmd}`.strip
-    unless result.empty?
-      result.each_line do |line|
-        if line =~ /unit (\d+)/
-          start_range = $1.to_i
-          if start_range > 0
-            end_range = start_range
-            ranges.add_range(start_range, end_range)
-          end
-        end
-      end
-    end
-
-    full_cmd = "#{ssh_command} '#{command_another}'"
-    result = `#{full_cmd}`.strip
-    unless result.empty?
-      result.each_line do |line|
-        if line =~ /vlan-id (\d+)/
-          start_range = $1.to_i
-          if start_range > 0
-            end_range = start_range
-            ranges.add_another_range(start_range, end_range)
-          end
-        end
-      end
-    end
-  end
-
-  # Основна логіка виконання
-  def self.run
-    options = {}
-    OptionParser.new do |opts|
-      opts.banner = "Використання: free-range <IP-адреса або hostname> [опції]"
-      opts.on("-u", "--username USERNAME", "Ім'я користувача для SSH") { |u| options[:username] = u }
-      opts.on("-p", "--password PASSWORD", "Пароль для SSH") { |p| options[:password] = p }
-      opts.on("-n", "--no-color", "Вимкнути кольоровий вивід") { options[:no_color] = true }
-      opts.on("-d", "--debug", "Увімкнути дебаг-режим") { options[:debug] = true }
-      opts.on("-t", "--table", "Вивести діаграму розподілу VLAN-ів") { options[:table] = true }
-      opts.on("-g", "--table-png PATH", "Зберегти діаграму розподілу VLAN-ів як PNG") { |path| options[:table_png] = path }
-      opts.on("-i", "--interface INTERFACE", "Назва інтерфейсу або 'all'") { |i| options[:interface] = i }
-    end.parse!
-
-    if ARGV.empty?
-      puts "Помилка: потрібно вказати IP-адресу або hostname роутера."
-      puts "Використання: free-range <IP-адреса або hostname> [-u|--username USERNAME] [-p|--password PASSWORD] [-n|--no-color] [-d|--debug] [-t|--table] [--table-png PATH] [-i|--interface INTERFACE]"
-      exit 1
-    end
-
-    username = options[:username] || ENV['WHOAMI']
-    password = options[:password] || ENV['WHATISMYPASSWD']
-    if username.nil? || password.nil?
-      puts "Помилка: необхідно вказати ім'я користувача та пароль."
-      puts "Використовуйте опції -u|--username і -p|--password або змінні оточення WHOAMI і WHATISMYPASSWD."
-      exit 1
-    end
-
-    use_color = !options[:no_color] && ENV['TERM'] && ENV['TERM'] != 'dumb'
-    debug = options[:debug]
-    table_mode = options[:table]
-    table_png_mode = options[:table_png]
-    interface = options[:interface]
-
-#    # Перевірка формату інтерфейсу
-#    if interface && interface != "all" && interface !~ /^[a-z]+-\d+\/\d+\/\d+$/
-#      puts "Помилка: некоректна назва інтерфейсу. Використовуйте формат 'xe-0/0/2' або 'all'."
-#      exit 1
-#    end
-
-    login = { target: ARGV[0], username: username, password: password }
-    target = ARGV[0].split('.')[0]
-    puts "Connecting to device: #{login[:target]}"
-
-    ssh_command = "sshpass -p \"#{login[:password]}\" ssh -C -x -4 -o StrictHostKeyChecking=no #{login[:username]}@#{login[:target]}"
-    subscribers_command = "ssh -C -x roffice /usr/local/share/noc/bin/radius-subscribers"
-
-    subscribers_result = `#{subscribers_command}`.strip
-    if subscribers_result.empty?
-      puts "Помилка: результат subscribers_command порожній. Перевір шлях або доступ."
-      exit 1
-    end
-
-    if interface == "all"
-      command_interfaces = 'show configuration interfaces | no-more | display set | match dynamic-profile | match "ranges ([0-9]+(-[0-9]+)?)"'
-      full_cmd = "#{ssh_command} '#{command_interfaces}'"
-      result = `#{full_cmd}`.strip
-      if result.empty?
-        puts "Помилка: результат команди порожній. Перевір підключення або команду."
-        exit 1
-      end
-
-      interfaces = result.each_line.map { |line| line.split[2] }.uniq
-      if interfaces.empty?
-        puts "Помилка: не знайдено інтерфейсів із діапазонами."
-        exit 1
-      end
-
-      interfaces.each do |intf|
-        ranges = Ranges.new
-        vlans = Vlans.new
-        subscribers_result.each_line do |line|
-          if line.split.first =~ /dhcp(?:_[0-9a-fA-F.]+)?_([^:]+):(\d+)@#{Regexp.escape(target)}$/
-            subscriber_interface, vlan = $1, $2.to_i
-            vlans.add_vlan(vlan) if subscriber_interface == intf && vlan > 0
-          end
-        end
-
-        process_interface(ssh_command, intf, ranges)
-        if debug
-          puts "\nІнтерфейс: #{intf}"
-          Print.ranged(ranges)
-          Print.vlans(vlans)
-          Print.vlan_ranges(vlans)
-          puts
-        end
-        if table_png_mode
-          Print.table_png(ranges, vlans, table_png_mode, target, intf)
-        elsif table_mode
-          Print.table(ranges, vlans, use_color, target, intf)
-        else
-          Print.combined_ranges(ranges, vlans, use_color, target, intf)
-        end
-      end
-    else
-      ranges = Ranges.new
-      vlans = Vlans.new
-      subscribers_result.each_line do |line|
-        if line.split.first =~ /dhcp(?:_[0-9a-fA-F.]+)?_([^:]+):(\d+)@#{Regexp.escape(target)}$/
-          subscriber_interface, vlan = $1, $2.to_i
-          if interface
-            vlans.add_vlan(vlan) if subscriber_interface == interface && vlan > 0
-          else
-            vlans.add_vlan(vlan) if vlan > 0
-          end
-        end
-      end
-
-      process_interface(ssh_command, interface, ranges)
-      if debug
-        puts "\nІнтерфейс: #{interface}" if interface
-        Print.ranged(ranges)
-        Print.vlans(vlans)
-        Print.vlan_ranges(vlans)
-        puts
-      end
-      if table_png_mode
-        Print.table_png(ranges, vlans, table_png_mode, target, interface)
-      elsif table_mode
-        Print.table(ranges, vlans, use_color, target, interface)
-      else
-        Print.combined_ranges(ranges, vlans, use_color, target, interface)
-      end
-    end
-  end
-end