idrac 0.1.29 → 0.1.30

data/lib/idrac/firmware.rb

@@ -6,6 +6,7 @@ require 'nokogiri'
 require 'fileutils'
 require 'securerandom'
 require 'set'
+require 'colorize'
 require_relative 'firmware_catalog'
 
 module IDRAC
@@ -126,30 +127,30 @@ module IDRAC
       system_model = inventory[:system][:model]
       service_tag = inventory[:system][:service_tag]
       
-      puts "Checking updates for system with service tag: #{service_tag}"
-      puts "Searching for updates for model: #{system_model}"
+      puts "Checking updates for system with service tag: #{service_tag}".light_cyan
+      puts "Searching for updates for model: #{system_model}".light_cyan
       
       # Find system models in the catalog
       models = catalog.find_system_models(system_model)
       
       if models.empty?
-        puts "No matching system model found in catalog"
+        puts "No matching system model found in catalog".yellow
         return []
       end
       
       # Use the first matching model
       model = models.first
-      puts "Found system IDs for #{model[:name]}: #{model[:id]}"
+      puts "Found system IDs for #{model[:name]}: #{model[:id]}".green
       
       # Find updates for this system
       catalog_updates = catalog.find_updates_for_system(model[:id])
-      puts "Found #{catalog_updates.size} firmware updates for #{model[:name]}"
+      puts "Found #{catalog_updates.size} firmware updates for #{model[:name]}".green
       
       # Compare current firmware with available updates
       updates = []
       
       # Print header for firmware comparison table
-      puts "\nFirmware Version Comparison:"
+      puts "\nFirmware Version Comparison:".green.bold
       puts "=" * 100
       puts "%-30s %-20s %-20s %-10s %-15s %s" % ["Component", "Current Version", "Available Version", "Updateable", "Category", "Status"]
       puts "-" * 100
@@ -205,25 +206,25 @@ module IDRAC
               fw[:name].to_s[0..29],
               fw[:version],
               update[:version],
-              fw[:updateable] ? "Yes" : "No",
+              fw[:updateable] ? "Yes".light_green : "No".light_red,
               update[:category] || "N/A",
-              "UPDATE AVAILABLE"
+              "UPDATE AVAILABLE".light_green.bold
             ]
           else
             # Print row with no update needed
             status = if !needs_update
-                       "Current"
+                       "Current".light_blue
                      elsif !fw[:updateable]
-                       "Not updateable"
+                       "Not updateable".light_red
                      else
-                       "No update needed"
+                       "No update needed".light_yellow
                      end
             
             puts "%-30s %-20s %-20s %-10s %-15s %s" % [
               fw[:name].to_s[0..29],
               fw[:version],
               update[:version] || "N/A",
-              fw[:updateable] ? "Yes" : "No",
+              fw[:updateable] ? "Yes".light_green : "No".light_red,
               update[:category] || "N/A",
               status
             ]
@@ -234,9 +235,9 @@ module IDRAC
             fw[:name].to_s[0..29],
             fw[:version],
             "N/A",
-            fw[:updateable] ? "Yes" : "No",
+            fw[:updateable] ? "Yes".light_green : "No".light_red,
             "N/A",
-            "No update available"
+            "No update available".light_yellow
           ]
         end
       end
@@ -261,11 +262,11 @@ module IDRAC
         
         puts "%-30s %-20s %-20s %-10s %-15s %s" % [
           update_name.to_s[0..29],
-          "Not Installed",
+          "Not Installed".light_red,
           update[:version] || "Unknown",
           "N/A",
           update[:category] || "N/A",
-          "NEW COMPONENT"
+          "NEW COMPONENT".light_blue
         ]
       end
       
@@ -278,13 +279,13 @@ module IDRAC
       updates = selected_updates || check_updates(catalog_path)
       
       if updates.empty?
-        puts "No updates available for your system."
+        puts "No updates available for your system.".yellow
         return
       end
       
-      puts "\nAvailable updates:"
+      puts "\nAvailable Updates:".green.bold
       updates.each_with_index do |update, index|
-        puts "#{index + 1}. #{update[:name]}: #{update[:current_version]} -> #{update[:available_version]}"
+        puts "#{index + 1}. #{update[:name]}: #{update[:current_version]} -> #{update[:available_version]}".light_cyan
       end
       
       # If selected_updates is provided, use those directly
@@ -292,7 +293,7 @@ module IDRAC
         selected = selected_updates
       else
         # Otherwise prompt the user for selection
-        puts "\nEnter the number of the update to install (or 'all' for all updates, 'q' to quit):"
+        puts "\nEnter the number of the update to install (or 'all' for all updates, 'q' to quit):".light_yellow
         choice = STDIN.gets.chomp
         
         return if choice.downcase == 'q'
@@ -304,14 +305,14 @@ module IDRAC
                       if index >= 0 && index < updates.size
                         [updates[index]]
                       else
-                        puts "Invalid selection."
+                        puts "Invalid selection.".red
                         return
                       end
                     end
       end
       
       selected.each do |update|
-        puts "Downloading #{update[:name]} version #{update[:available_version]}..."
+        puts "Downloading #{update[:name]} version #{update[:available_version]}...".light_cyan.bold
         
         # Create temp directory
         temp_dir = Dir.mktmpdir
@@ -332,33 +333,33 @@ module IDRAC
                   end
                 end
               else
-                puts "Failed to download update: #{response.code} #{response.message}"
+                puts "Failed to download update: #{response.code} #{response.message}".red
                 next
               end
             end
           end
           
-          puts "Installing #{update[:name]} version #{update[:available_version]}..."
+          puts "Installing #{update[:name]} version #{update[:available_version]}...".light_cyan.bold
           
           begin
             job_id = update(update_path, wait: true)
-            puts "Update completed with job ID: #{job_id}"
+            puts "Update completed with job ID: #{job_id}".green.bold
           rescue IDRAC::Error => e
             if e.message.include?("already in progress")
-              puts "Error: #{e.message}"
-              puts "\nTips for resolving this issue:"
-              puts "1. Wait for any existing firmware updates to complete (check iDRAC web interface)"
-              puts "2. Restart the iDRAC if no updates appear to be in progress (Settings > iDRAC Settings > Reset iDRAC)"
-              puts "3. Try again after a few minutes"
+              puts "Error: #{e.message}".red.bold
+              puts "\nTips for resolving this issue:".yellow.bold
+              puts "1. Wait for any existing firmware updates to complete (check iDRAC web interface)".light_yellow
+              puts "2. Restart the iDRAC if no updates appear to be in progress (Settings > iDRAC Settings > Reset iDRAC)".light_yellow
+              puts "3. Try again after a few minutes".light_yellow
             elsif e.message.include?("job ID") || e.message.include?("job not found")
-              puts "Error: #{e.message}"
-              puts "\nThe job ID could not be found or monitored. This could be because:"
-              puts "1. The iDRAC is busy processing other requests"
-              puts "2. The job was created but not properly tracked"
-              puts "3. The iDRAC firmware may need to be updated first"
-              puts "\nCheck the iDRAC web interface to see if the update was actually initiated."
+              puts "Error: #{e.message}".red.bold
+              puts "\nThe job ID could not be found or monitored. This could be because:".yellow
+              puts "1. The iDRAC is busy processing other requests".light_yellow
+              puts "2. The job was created but not properly tracked".light_yellow
+              puts "3. The iDRAC firmware may need to be updated first".light_yellow
+              puts "\nCheck the iDRAC web interface to see if the update was actually initiated.".light_cyan
             else
-              puts "Error during firmware update: #{e.message}"
+              puts "Error during firmware update: #{e.message}".red.bold
             end
           end
           

data/lib/idrac/firmware_catalog.rb

@@ -2,10 +2,13 @@ require 'net/http'
 require 'uri'
 require 'fileutils'
 require 'nokogiri'
+require 'open-uri'
+require 'colorize'
 
 module IDRAC
   class FirmwareCatalog
-    CATALOG_URL = "https://downloads.dell.com/catalog/Catalog.xml.gz"
+    DELL_CATALOG_BASE = 'https://downloads.dell.com'
+    DELL_CATALOG_URL = "#{DELL_CATALOG_BASE}/catalog/Catalog.xml.gz"
     
     attr_reader :catalog_path
     
@@ -14,40 +17,38 @@ module IDRAC
     end
     
     def download(output_dir = nil)
-      # Use ~/.idrac as the default directory
-      output_dir ||= File.expand_path("~/.idrac")
+      # Default to ~/.idrac directory
+      output_dir ||= File.expand_path('~/.idrac')
       FileUtils.mkdir_p(output_dir) unless Dir.exist?(output_dir)
       
-      catalog_gz_path = File.join(output_dir, "Catalog.xml.gz")
-      catalog_path = File.join(output_dir, "Catalog.xml")
-      
-      puts "Downloading Dell catalog from #{CATALOG_URL}..."
-      
-      uri = URI.parse(CATALOG_URL)
-      Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
-        request = Net::HTTP::Get.new(uri)
-        http.request(request) do |response|
-          if response.code == "200"
-            File.open(catalog_gz_path, 'wb') do |file|
-              response.read_body do |chunk|
-                file.write(chunk)
-              end
-            end
-          else
-            raise Error, "Failed to download catalog: #{response.code} #{response.message}"
-          end
-        end
-      end
+      catalog_gz = File.join(output_dir, 'Catalog.xml.gz')
+      catalog_xml = File.join(output_dir, 'Catalog.xml')
       
-      puts "Extracting catalog..."
-      system("gunzip -f #{catalog_gz_path}")
+      puts "Downloading Dell catalog from #{DELL_CATALOG_URL}...".light_cyan
       
-      if File.exist?(catalog_path)
-        puts "Catalog downloaded and extracted to #{catalog_path}"
-        @catalog_path = catalog_path
-        return catalog_path
-      else
-        raise Error, "Failed to extract catalog"
+      begin
+        # Download the catalog
+        URI.open(DELL_CATALOG_URL) do |remote_file|
+          File.open(catalog_gz, 'wb') do |local_file|
+            local_file.write(remote_file.read)
+          end
+        end
+        
+        puts "Extracting catalog...".light_cyan
+        
+        # Extract the catalog
+        system("gunzip -f #{catalog_gz}")
+        
+        if File.exist?(catalog_xml)
+          puts "Catalog downloaded and extracted to #{catalog_xml}".green
+          @catalog_path = catalog_xml
+          return catalog_xml
+        else
+          raise Error, "Failed to extract catalog"
+        end
+      rescue => e
+        puts "Error downloading catalog: #{e.message}".red.bold
+        raise Error, "Failed to download Dell catalog: #{e.message}"
       end
     end
     

data/lib/idrac/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module IDRAC
-  VERSION = "0.1.29"
+  VERSION = "0.1.30"
 end

data/test_firmware_update.rb

@@ -1,6 +1,7 @@
 #!/usr/bin/env ruby
 
-require_relative 'lib/idrac'
+require 'lib/idrac'
+require 'colorize'
 
 # Create a client
 client = IDRAC::Client.new(
@@ -12,59 +13,56 @@ client = IDRAC::Client.new(
 
 begin
   # Login to iDRAC
+  puts "Logging in to iDRAC...".light_cyan
   client.login
-  puts "Logged in successfully"
+  puts "Logged in successfully".green
   
   # Create a firmware instance
   firmware = IDRAC::Firmware.new(client)
   
   # Get system inventory
-  puts "Getting system inventory..."
+  puts "Getting system inventory...".light_cyan
   inventory = firmware.get_system_inventory
   
-  puts "System Information:"
-  puts "  Model: #{inventory[:system][:model]}"
-  puts "  Manufacturer: #{inventory[:system][:manufacturer]}"
-  puts "  Service Tag: #{inventory[:system][:service_tag]}"
-  puts "  BIOS Version: #{inventory[:system][:bios_version]}"
+  puts "System Information:".green.bold
+  puts "  Model: #{inventory[:system][:model]}".light_cyan
+  puts "  Manufacturer: #{inventory[:system][:manufacturer]}".light_cyan
+  puts "  Service Tag: #{inventory[:system][:service_tag]}".light_cyan
+  puts "  BIOS Version: #{inventory[:system][:bios_version]}".light_cyan
   
-  puts "\nInstalled Firmware:"
+  puts "\nInstalled Firmware:".green.bold
   inventory[:firmware].each do |fw|
-    puts "  #{fw[:name]}: #{fw[:version]} (#{fw[:updateable] ? 'Updateable' : 'Not Updateable'})"
+    puts "  #{fw[:name]}: #{fw[:version]} (#{fw[:updateable] ? 'Updateable'.light_green : 'Not Updateable'.light_red})".light_cyan
   end
   
   # Check for updates
   catalog_path = File.expand_path("~/.idrac/Catalog.xml")
   if File.exist?(catalog_path)
-    puts "\nChecking for updates using catalog: #{catalog_path}"
+    puts "\nChecking for updates using catalog: #{catalog_path}".light_cyan
     updates = firmware.check_updates(catalog_path)
     
-    if updates.empty?
-      puts "No updates available."
-    else
-      puts "\nAvailable Updates:"
+    if updates.any?
+      puts "\nAvailable Updates:".green.bold
       updates.each_with_index do |update, index|
-        puts "#{index + 1}. #{update[:name]}: #{update[:current_version]} -> #{update[:available_version]}"
+        puts "#{index + 1}. #{update[:name]}: #{update[:current_version]} -> #{update[:available_version]}".light_cyan
       end
       
-      # For testing, select the first update
-      if updates.any?
-        selected_update = updates.first
-        puts "\nSelected update: #{selected_update[:name]}: #{selected_update[:current_version]} -> #{selected_update[:available_version]}"
-        
-        # Download and install the update
-        puts "Starting interactive update for selected component..."
-        firmware.interactive_update(catalog_path, [selected_update])
-      end
+      # Interactive update for the first update
+      puts "\nSelected update: #{updates.first[:name]}".light_yellow
+      puts "Starting interactive update for selected component...".light_cyan.bold
+      
+      firmware.interactive_update(catalog_path, [updates.first])
+    else
+      puts "No updates available for your system.".yellow
     end
   else
-    puts "\nCatalog file not found at #{catalog_path}. Please download the catalog first."
+    puts "\nCatalog not found at #{catalog_path}. Run 'idrac firmware:catalog' to download it.".yellow
   end
   
 rescue IDRAC::Error => e
-  puts "Error: #{e.message}"
+  puts "Error: #{e.message}".red.bold
 ensure
   # Logout
-  client.logout if client.instance_variable_get(:@session_id)
-  puts "Logged out"
+  client.logout if client
+  puts "Logged out".light_cyan
 end