free-range 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8cef8634d2f2d3e9e90289e4098b5bac3341872844764b0db4bca9e0b3bdaf0e
-  data.tar.gz: ca753e285e70bce1f8ca7d947c56d76683901666c103765de05a6f58a36c879b
+  metadata.gz: 8c353a0a3ea2e3a261a5cb6af72b3e5e4ac847be178a710e79927264e256393a
+  data.tar.gz: 771ead5640626c25b48b5639746246a08a44bb2f58db8cca3971581257de97e8
 SHA512:
-  metadata.gz: ac8287096aa861556171e3b7ad393640ff4193e16de0d062666f0b576b80b6768aaa96fa47e511ab8b94ce4e0473cec28c7f402b89b5c76f691c72d62e017934
-  data.tar.gz: 9a2778e40dd3a4e189ebe32bd5e56b0676d08be1bb05f13f9c3b08378d9c9a65638b5f0742bff33991f2e2521dd4549a2b9f112a526226f71889c1fde8116616
+  metadata.gz: ef260cde29dfee53e9ab895c14a232c0e023052cb2c614e2ce3e7d34ffbdefebabbe856c2fbaa143179b7fa387468dd32d89718da69ab22d669230fd2a164a76
+  data.tar.gz: 2477e3ea176e80ceb40869bf1e2835ffd63b4a8e733fb87d3f4497bcc70d27d74814f94586e946b2c5b24f8db9f7359f4bb60c7a3087078927d608d03234915d

data/lib/free-range.rb

@@ -3,7 +3,6 @@ require 'rmagick'
 require 'fileutils'
 
 module FreeRange
-
   # Перевірка наявності ImageMagick
   begin
     require 'rmagick'
@@ -20,16 +19,20 @@ module FreeRange
     end
 
     # Додаємо VLAN до списку (віртуальний метод, може бути перевизначений)
+    # @param vlan [Integer] VLAN ID to add
+    # @return [void]
     def add_vlan(vlan)
       @vlans << vlan
     end
 
     # Повертаємо масив VLAN
+    # @return [Array<Integer>] Sorted array of unique VLAN IDs
     def vlans
       @vlans.uniq.sort
     end
 
     # Створюємо хеш діапазонів із VLAN
+    # @return [Hash{Integer => Integer}] Hash mapping range start to range end
     def ranges
       return {} if @vlans.empty?
 
@@ -59,16 +62,23 @@ module FreeRange
     end
 
     # Додаємо діапазон до списку VLAN-ів
+    # @param start [Integer] Start of VLAN range
+    # @param finish [Integer] End of VLAN range
+    # @return [void]
     def add_range(start, finish)
       (start..finish).each { |vlan| @vlans << vlan }
     end
 
     # Додаємо діапазон до списку "інших" VLAN-ів
+    # @param start [Integer] Start of another VLAN range
+    # @param finish [Integer] End of another VLAN range
+    # @return [void]
     def add_another_range(start, finish)
       (start..finish).each { |vlan| @another_in_ranges << vlan }
     end
 
     # Повертаємо масив "інших" VLAN-ів
+    # @return [Array<Integer>] Sorted array of unique "another" VLAN IDs
     def another_in_ranges
       @another_in_ranges.uniq.sort
     end
@@ -84,6 +94,9 @@ module FreeRange
 
   # Клас для виводу даних
   class Print
+    # Виводить хеш діапазонів VLAN у порядку зростання
+    # @param ranges [Ranges] Object containing VLAN ranges
+    # @return [void]
     def self.ranged(ranges)
       puts "\nСформований хеш діапазонів (в порядку зростання):"
       if ranges.ranges.empty?
@@ -95,6 +108,9 @@ module FreeRange
       end
     end
 
+    # Виводить зайняті VLAN-и в межах діапазонів у порядку зростання
+    # @param vlans [Vlans] Object containing VLANs
+    # @return [void]
     def self.vlans(vlans)
       puts "\nЗайняті VLAN-и в межах діапазонів (в порядку зростання):"
       if vlans.vlans.empty?
@@ -104,6 +120,9 @@ module FreeRange
       end
     end
 
+    # Виводить діапазони VLAN-ів у порядку зростання
+    # @param vlans [Vlans] Object containing VLANs
+    # @return [void]
     def self.vlan_ranges(vlans)
       puts "\nДіапазони VLAN-ів (в порядку зростання):"
       vlan_ranges = vlans.vlan_ranges
@@ -116,6 +135,13 @@ module FreeRange
       end
     end
 
+    # Виводить комбіновані діапазони VLAN зі статусами
+    # @param ranges [Ranges] Object containing VLAN ranges
+    # @param vlans [Vlans] Object containing VLANs
+    # @param use_color [Boolean] Enable colored output
+    # @param target [String] Target device hostname
+    # @param interface [String, nil] Interface name or nil
+    # @return [void]
     def self.combined_ranges(ranges, vlans, use_color, target, interface = nil)
       if ranges.ranges.empty?
         puts "Не знайдено діапазонів."
@@ -142,6 +168,13 @@ module FreeRange
       puts result.join(',')
     end
 
+    # Виводить таблицю розподілу VLAN
+    # @param ranges [Ranges] Object containing VLAN ranges
+    # @param vlans [Vlans] Object containing VLANs
+    # @param use_color [Boolean] Enable colored output
+    # @param target [String] Target device hostname
+    # @param interface [String, nil] Interface name or nil
+    # @return [void]
     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)
@@ -191,6 +224,13 @@ module FreeRange
       puts summary_text
     end
 
+    # Зберігає таблицю розподілу VLAN як PNG-зображення
+    # @param ranges [Ranges] Object containing VLAN ranges
+    # @param vlans [Vlans] Object containing VLANs
+    # @param path [String] Directory path to save the PNG
+    # @param target [String] Target device hostname
+    # @param interface [String, nil] Interface name or nil
+    # @return [void]
     def self.table_png(ranges, vlans, path, target, interface = nil)
       all_vlans, status_counts = build_vlan_statuses(ranges, vlans)
       cell_width = 12
@@ -307,6 +347,10 @@ module FreeRange
 
     private
 
+    # Будує хеш статусів VLAN і підраховує кількість кожного статусу
+    # @param ranges [Ranges] Object containing VLAN ranges
+    # @param vlans [Vlans] Object containing VLANs
+    # @return [Array<Hash, Hash>] Hash of VLAN statuses and status counts
     def self.build_vlan_statuses(ranges, vlans)
       all_vlans = {}
       ranges.ranges.each do |start, finish|
@@ -321,6 +365,12 @@ module FreeRange
       [all_vlans, status_counts]
     end
 
+    # Форматує діапазон VLAN зі статусом для виводу
+    # @param start [Integer] Start of VLAN range
+    # @param finish [Integer] End of VLAN range
+    # @param status [String] VLAN status ('f', 'b', 'e', 'c', 'a', 'u')
+    # @param use_color [Boolean] Enable colored output
+    # @return [String] Formatted range string
     def self.format_range(start, finish, status, use_color)
       range_text = start == finish ? "#{start}" : "#{start}-#{finish}"
       range_text_with_status = "#{range_text}(#{status})"
@@ -339,6 +389,10 @@ module FreeRange
       end
     end
 
+    # Форматує символ для таблиці VLAN
+    # @param status [String] VLAN status ('f', 'b', 'e', 'c', 'a', 'u', or ' ')
+    # @param use_color [Boolean] Enable colored output
+    # @return [String] Formatted character for table display
     def self.format_table_char(status, use_color)
       if use_color
         case status
@@ -357,6 +411,10 @@ module FreeRange
   end
 
   # Метод для заповнення Ranges для одного інтерфейсу
+  # @param ssh_command [String] SSH command to execute
+  # @param interface [String, nil] Interface name or nil for all interfaces
+  # @param ranges [Ranges] Object to store VLAN ranges
+  # @return [void]
   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'
@@ -404,6 +462,7 @@ module FreeRange
   end
 
   # Основна логіка виконання
+  # @return [void]
   def self.run
     options = {}
     OptionParser.new do |opts|
@@ -437,12 +496,6 @@ module FreeRange
     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]}"

data/lib/{free-range.rb.~ → free-range.rb~}

@@ -1,68 +1,18 @@
-module FreeRange
-
-  require 'optparse'
-  require 'rmagick'
-  require 'fileutils'
-
-  # Парсинг аргументів командного рядка
-  options = {}
-  OptionParser.new do |opts|
-    opts.banner = "Використання: ruby free-range.rb <IP-адреса або hostname> [опції]"
-
-    opts.on("-u", "--username USERNAME", "Ім'я користувача для SSH") do |u|
-      options[:username] = u
-    end
-
-    opts.on("-p", "--password PASSWORD", "Пароль для SSH") do |p|
-      options[:password] = p
-    end
-
-    opts.on("-n", "--no-color", "Вимкнути кольоровий вивід") do
-      options[:no_color] = true
-    end
+require 'optparse'
+require 'rmagick'
+require 'fileutils'
 
-    opts.on("-d", "--debug", "Увімкнути дебаг-режим") do
-      options[:debug] = true
-    end
-
-    opts.on("-t", "--table", "Вивести діаграму розподілу VLAN-ів") do
-      options[:table] = true
-    end
-
-    opts.on("-g", "--table-png PATH", "Зберегти діаграму розподілу VLAN-ів як PNG у вказаний каталог") do |path|
-      options[:table_png] = path
-    end
-
-    opts.on("-i", "--interface INTERFACE", "Назва інтерфейсу або 'all'") do |i|
-      options[:interface] = i
-    end
-  end.parse!
-
-  # Перевіряємо, чи передано IP-адресу або hostname
-  if ARGV.empty?
-    puts "Помилка: потрібно вказати IP-адресу або hostname роутера як аргумент командного рядка."
-    puts "Використання: ruby free-range.rb <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']
+module FreeRange
 
-  # Перевіряємо наявність облікових даних
-  if username.nil? || password.nil?
-    puts "Помилка: необхідно вказати ім'я користувача та пароль."
-    puts "Використовуйте опції -u|--username і -p|--password або змінні оточення WHOAMI і WHATISMYPASSWD."
+  # Перевірка наявності ImageMagick
+  begin
+    require 'rmagick'
+  rescue LoadError
+    puts "Помилка: бібліотека rmagick не встановлена або ImageMagick недоступний."
+    puts "Встановіть ImageMagick і виконайте: gem install rmagick"
     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]
-
   # Абстрактний клас для роботи з VLAN
   class VlanContainer
     def initialize
@@ -173,8 +123,6 @@ module FreeRange
       end
 
       all_vlans, _status_counts = build_vlan_statuses(ranges, vlans)
-
-      # Формуємо діапазони з урахуванням статусів
       result = []
       sorted_vlans = all_vlans.keys.sort
       start = sorted_vlans.first
@@ -183,14 +131,12 @@ module FreeRange
 
       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(',')
@@ -200,7 +146,6 @@ module FreeRange
       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
@@ -209,7 +154,6 @@ module FreeRange
         puts "#{format("%4d", start_vlan)} #{row}"
       end
 
-      # Виводимо легенду
       legend_parts = [
         ["Legend: ", nil],
         ["f", 'f'], ["=free", nil], [", ", nil],
@@ -228,7 +172,6 @@ module FreeRange
       end.join
       puts "\n#{legend_text}"
 
-      # Виводимо підсумок
       summary_parts = [
         ["Total: ", nil],
         ["f", 'f'], ["=#{status_counts['f']}", nil], [", ", nil],
@@ -250,8 +193,6 @@ module FreeRange
 
     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
@@ -259,42 +200,35 @@ module FreeRange
       header_height = 60
       label_width = 50
       width = label_width + cols * cell_width + 10
-      height = header_height + rows * cell_height + 20 + 50  # Вистачає для легенди і підсумку
+      height = header_height + rows * cell_height + 20 + 50
       font_size = 14
-      title_font_size = 18  # Більший шрифт для заголовка
+      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  # Повертаємо стандартний розмір шрифту
+      gc.pointsize = font_size
 
-      # Малюємо заголовок (0 1 2 ... 9)
       (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
@@ -314,7 +248,6 @@ module FreeRange
         end
       end
 
-      # Малюємо легенду з кольоровими фонами
       legend_y = height - 50
       x = 10
       legend_parts = [
@@ -336,11 +269,10 @@ module FreeRange
         else
           gc.fill('black')
           gc.text(x, legend_y, text)
-          x += text.length * 8  # Приблизно 8 пікселів на символ
+          x += text.length * 8
         end
       end
 
-      # Малюємо підсумок VLAN-ів з кольоровими фонами
       summary_y = height - 30
       x = 10
       summary_parts = [
@@ -362,11 +294,10 @@ module FreeRange
         else
           gc.fill('black')
           gc.text(x, summary_y, text)
-          x += text.length * 8  # Приблизно 8 пікселів на символ
+          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")
@@ -378,17 +309,12 @@ module FreeRange
 
     def self.build_vlan_statuses(ranges, vlans)
       all_vlans = {}
-      # Позначаємо VLAN із діапазонів як free
       ranges.ranges.each do |start, finish|
         (start..finish).each { |vlan| all_vlans[vlan] = 'f' }
       end
-      # Оновлюємо статуси для зайнятих VLAN
       vlans.vlans.uniq.each { |vlan| all_vlans[vlan] = all_vlans.key?(vlan) ? 'b' : 'e' }
-      # Позначаємо всі інші VLAN як unused
       (1..4094).each { |vlan| all_vlans[vlan] = 'u' unless all_vlans.key?(vlan) }
-      # Оновлюємо статуси для "інших" 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 }
 
@@ -398,23 +324,15 @@ module FreeRange
     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'
-          "\e[32m#{range_text}\e[0m"  # Зелений для free
-        when 'b'
-          "\e[33m#{range_text}\e[0m"  # Жовтий для busy
-        when 'e'
-          "\e[31m#{range_text}\e[0m"  # Червоний для error
-        when 'c'
-          "\e[35m#{range_text}\e[0m"  # Фіолетовий для configured
-        when 'a'
-          "\e[34m#{range_text}\e[0m"  # Синій для another
-        when 'u'
-          "\e[90m#{range_text}\e[0m"  # Темно-сірий для unused
-        else
-          range_text  # Без кольору для інших статусів
+        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  # Текстовий вивід зі статусами
@@ -424,20 +342,13 @@ module FreeRange
     def self.format_table_char(status, use_color)
       if use_color
         case status
-        when 'f'
-          "\e[48;5;2m\e[30m#{status}\e[0m"  # Зелений фон, чорний текст
-        when 'b'
-          "\e[48;5;3m\e[30m#{status}\e[0m"  # Жовтий фон, чорний текст
-        when 'e'
-          "\e[48;5;1m\e[30m#{status}\e[0m"  # Червоний фон, чорний текст
-        when 'c'
-          "\e[48;5;5m\e[30m#{status}\e[0m"  # Фіолетовий фон, чорний текст
-        when 'a'
-          "\e[48;5;4m\e[30m#{status}\e[0m"  # Синій фон, чорний текст
-        when 'u'
-          "\e[48;5;8m\e[30m#{status}\e[0m"  # Темно-сірий фон, чорний текст
-        else
-          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  # Текстовий вивід без кольорів
@@ -445,25 +356,14 @@ module FreeRange
     end
   end
 
-  # Основна логіка
-  login = { target: ARGV[0], username: username, password: password }
-  target = ARGV[0].split('.')[0] # Обрізаємо суфікс (наприклад, rhoh15-1.ukrhub.net → rhoh15-1)
-
-  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"
-
-  # Функція для заповнення Ranges для одного інтерфейсу
-  def process_interface(ssh_command, interface, ranges)
+  # Метод для заповнення 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+))?/
@@ -474,10 +374,8 @@ module FreeRange
       end
     end
 
-    # Виконуємо команду для отримання unnumbered-address інтерфейсів (demux-source)
     full_cmd = "#{ssh_command} '#{command_demux}'"
     result = `#{full_cmd}`.strip
-
     unless result.empty?
       result.each_line do |line|
         if line =~ /unit (\d+)/
@@ -490,10 +388,8 @@ module FreeRange
       end
     end
 
-    # Виконуємо команду для отримання vlan-ів усіх наявних інтерфейсів
     full_cmd = "#{ssh_command} '#{command_another}'"
     result = `#{full_cmd}`.strip
-
     unless result.empty?
       result.each_line do |line|
         if line =~ /vlan-id (\d+)/
@@ -507,94 +403,129 @@ module FreeRange
     end
   end
 
-  # Виконуємо команду для отримання списку абонентів
-  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
+  # Основна логіка виконання
+  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
 
-    if result.empty?
-      puts "Помилка: результат команди порожній. Перевір підключення або команду."
+    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
 
-    interfaces = result.each_line.map { |line| line.split[2] }.uniq
-    if interfaces.empty?
-      puts "Помилка: не знайдено інтерфейсів із діапазонами."
+    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
 
-    interfaces.each do |intf|
+    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
-
-      # Заповнюємо VLAN-и, фільтруючи за інтерфейсом
       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
+          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, intf, ranges)
-
-      # Виводимо результати
+      process_interface(ssh_command, interface, ranges)
       if debug
-        puts "\nІнтерфейс: #{intf}"
+        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, intf)
+        Print.table_png(ranges, vlans, table_png_mode, target, interface)
       elsif table_mode
-        Print.table(ranges, vlans, use_color, target, intf)
+        Print.table(ranges, vlans, use_color, target, interface)
       else
-        Print.combined_ranges(ranges, vlans, use_color, target, intf)
+        Print.combined_ranges(ranges, vlans, use_color, target, interface)
       end
     end
-  else
-    ranges = Ranges.new
-    vlans = Vlans.new
-
-    # Заповнюємо VLAN-и, фільтруючи за інтерфейсом, якщо задано
-    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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: free-range
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Oleksandr Russkikh //aka Olden Gremlin
@@ -61,9 +61,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/free-range
-- 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/bin/free-range.~

@@ -1,6 +0,0 @@
-#!/usr/bin/env ruby
-require 'free-range'
-
-# Виклик основної логіки, якщо потрібно (зазвичай код уже запускається в lib/free_range.rb)
-
-FreeRange.main if defined?(FreeRange.main)