idrac 0.1.40 → 0.1.60

data/lib/idrac/client.rb

@@ -10,9 +10,15 @@ require 'colorize'
 module IDRAC
   class Client
     attr_reader :host, :username, :password, :port, :use_ssl, :verify_ssl, :auto_delete_sessions, :session, :web
-    attr_accessor :direct_mode
+    attr_accessor :direct_mode, :verbosity
+    
+    include PowerMethods
+    include SessionMethods
+    include Debuggable
+    include JobMethods
+    include LifecycleMethods
 
-    def initialize(host:, username:, password:, port: 443, use_ssl: true, verify_ssl: true, direct_mode: false, auto_delete_sessions: true)
+    def initialize(host:, username:, password:, port: 443, use_ssl: true, verify_ssl: false, direct_mode: false, auto_delete_sessions: true)
       @host = host
       @username = username
       @password = password
@@ -21,6 +27,7 @@ module IDRAC
       @verify_ssl = verify_ssl
       @direct_mode = direct_mode
       @auto_delete_sessions = auto_delete_sessions
+      @verbosity = 0
       
       # Initialize the session and web classes
       @session = Session.new(self)
@@ -32,6 +39,13 @@ module IDRAC
         faraday.request :multipart
         faraday.request :url_encoded
         faraday.adapter Faraday.default_adapter
+        # Add request/response logging based on verbosity
+        if @verbosity > 0
+          faraday.response :logger, Logger.new(STDOUT), bodies: @verbosity >= 2 do |logger|
+            logger.filter(/(Authorization: Basic )([^,\n]+)/, '\1[FILTERED]')
+            logger.filter(/(Password"=>"?)([^,"]+)/, '\1[FILTERED]')
+          end
+        end
       end
     end
 
@@ -70,6 +84,8 @@ module IDRAC
         raise Error, "Maximum retry count reached for authenticated request"
       end
       
+      debug "Authenticated request: #{method.to_s.upcase} #{path}", 1
+      
       # If we're in direct mode, use Basic Auth
       if @direct_mode
         # Create Basic Auth header
@@ -79,6 +95,8 @@ module IDRAC
         options[:headers] ||= {}
         options[:headers]['Authorization'] = auth_header
         
+        debug "Using Basic Auth for request", 2
+        
         # Make the request
         begin
           response = connection.send(method, path) do |req|
@@ -86,6 +104,10 @@ module IDRAC
             req.body = options[:body] if options[:body]
           end
           
+          debug "Response status: #{response.status}", 1
+          debug "Response headers: #{response.headers.inspect}", 2
+          debug "Response body: #{response.body}", 3 if response.body
+          
           return response
         rescue => e
           puts "Error during authenticated request (direct mode): #{e.message}".red.bold
@@ -98,6 +120,8 @@ module IDRAC
           options[:headers] ||= {}
           options[:headers]['X-Auth-Token'] = session.x_auth_token
           
+          debug "Using X-Auth-Token for request", 2
+          
           # Make the request
           begin
             response = connection.send(method, path) do |req|
@@ -105,6 +129,10 @@ module IDRAC
               req.body = options[:body] if options[:body]
             end
             
+            debug "Response status: #{response.status}", 1
+            debug "Response headers: #{response.headers.inspect}", 2
+            debug "Response body: #{response.body}", 3 if response.body
+            
             # Check if the session is still valid
             if response.status == 401 || response.status == 403
               puts "Session expired or invalid, attempting to create a new session...".light_yellow
@@ -154,6 +182,8 @@ module IDRAC
         web.login unless web.session_id
       end
       
+      debug "GET request to #{base_url}/#{path}", 1
+      
       headers_to_use = {
         "User-Agent" => "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36",
         "Accept-Encoding" => "deflate, gzip"
@@ -161,18 +191,29 @@ module IDRAC
       
       if web.cookies
         headers_to_use["Cookie"] = web.cookies
+        debug "Using WebUI cookies for request", 2
       elsif @direct_mode
         # In direct mode, use Basic Auth
         headers_to_use["Authorization"] = "Basic #{Base64.strict_encode64("#{username}:#{password}")}"
+        debug "Using Basic Auth for GET request", 2
       elsif session.x_auth_token
         headers_to_use["X-Auth-Token"] = session.x_auth_token
+        debug "Using X-Auth-Token for GET request", 2
       end
       
-      HTTParty.get(
+      debug "Request headers: #{headers_to_use.merge(headers).inspect}", 3
+      
+      response = HTTParty.get(
         "#{base_url}/#{path}",
         headers: headers_to_use.merge(headers),
         verify: false
       )
+      
+      debug "Response status: #{response.code}", 1
+      debug "Response headers: #{response.headers.inspect}", 2
+      debug "Response body: #{response.body.to_s[0..500]}#{response.body.to_s.length > 500 ? '...' : ''}", 3 if response.body
+      
+      response
     end
 
     def screenshot
@@ -183,5 +224,15 @@ module IDRAC
       protocol = use_ssl ? 'https' : 'http'
       "#{protocol}://#{host}:#{port}"
     end
+
+    def redfish_version
+      response = authenticated_request(:get, "/redfish/v1")
+      if response.status == 200
+        data = JSON.parse(response.body)
+        data["RedfishVersion"]
+      else
+        raise Error, "Failed to get Redfish version: #{response.status} - #{response.body}"
+      end
+    end
   end
 end

data/lib/idrac/firmware.rb

@@ -8,6 +8,8 @@ require 'securerandom'
 require 'set'
 require 'colorize'
 require_relative 'firmware_catalog'
+require 'faraday'
+require 'faraday/multipart'
 
 module IDRAC
   class Firmware
@@ -427,6 +429,24 @@ module IDRAC
       end
     end
 
+    def get_power_state
+      # Ensure we have a client
+      raise Error, "Client is required for power management" unless client
+      
+      # Login to iDRAC if needed
+      client.login unless client.instance_variable_get(:@session_id)
+      
+      # Get system information
+      response = client.authenticated_request(:get, "/redfish/v1/Systems/System.Embedded.1")
+      
+      if response.status == 200
+        system_data = JSON.parse(response.body)
+        return system_data["PowerState"]
+      else
+        raise Error, "Failed to get power state. Status code: #{response.status}"
+      end
+    end
+
     private
 
     def upload_firmware(firmware_path)
@@ -478,16 +498,20 @@ module IDRAC
         file_content = File.read(firmware_path)
         
         headers = {
-          'Content-Type' => 'application/octet-stream',
+          'Content-Type' => 'multipart/form-data',
           'If-Match' => etag
         }
         
+        # Create a temp file for multipart upload
+        upload_io = Faraday::UploadIO.new(firmware_path, 'application/octet-stream')
+        payload = { :file => upload_io }
+        
         upload_response = client.authenticated_request(
           :post,
           http_push_uri,
           {
             headers: headers,
-            body: file_content
+            body: payload,
           }
         )
         

data/lib/idrac/jobs.rb

@@ -0,0 +1,212 @@
+require 'json'
+require 'colorize'
+
+module IDRAC
+  module JobMethods
+    # Get a list of jobs
+    def jobs
+      response = authenticated_request(:get, '/redfish/v1/Managers/iDRAC.Embedded.1/Jobs?$expand=*($levels=1)')
+      
+      if response.status == 200
+        begin
+          jobs_data = JSON.parse(response.body)
+          puts "Jobs: #{jobs_data['Members'].count}"
+          if jobs_data['Members'].count > 0
+            puts "Job IDs:"
+            jobs_data["Members"].each do |job|
+              puts "  #{job['Id']}"
+            end
+          end
+          return jobs_data
+        rescue JSON::ParserError
+          raise Error, "Failed to parse jobs response: #{response.body}"
+        end
+      else
+        raise Error, "Failed to get jobs. Status code: #{response.status}"
+      end
+    end
+    
+    # Get detailed job information
+    def jobs_detail
+      response = authenticated_request(:get, '/redfish/v1/Managers/iDRAC.Embedded.1/Jobs?$expand=*($levels=1)')
+      
+      if response.status == 200
+        begin
+          jobs_data = JSON.parse(response.body)
+          jobs_data["Members"].each do |job| 
+            puts "#{job['Id']} : #{job['JobState']} > #{job['Message']}" 
+          end
+          return jobs_data
+        rescue JSON::ParserError
+          raise Error, "Failed to parse jobs detail response: #{response.body}"
+        end
+      else
+        raise Error, "Failed to get jobs detail. Status code: #{response.status}"
+      end
+    end
+    
+    # Clear all jobs from the job queue
+    def clear_jobs!
+      # Get list of jobs
+      jobs_response = authenticated_request(:get, '/redfish/v1/Managers/iDRAC.Embedded.1/Jobs?$expand=*($levels=1)')
+      
+      if jobs_response.status == 200
+        begin
+          jobs_data = JSON.parse(jobs_response.body)
+          members = jobs_data["Members"]
+          
+          # Delete each job individually
+          members.each.with_index do |job, i|
+            puts "Removing #{job['Id']} [#{i+1}/#{members.count}]"
+            delete_response = authenticated_request(:delete, "/redfish/v1/Managers/iDRAC.Embedded.1/Jobs/#{job['Id']}")
+            
+            unless delete_response.status.between?(200, 299)
+              puts "Warning: Failed to delete job #{job['Id']}. Status code: #{delete_response.status}".yellow
+            end
+          end
+          
+          puts "Successfully cleared all jobs".green
+          return true
+        rescue JSON::ParserError
+          raise Error, "Failed to parse jobs response: #{jobs_response.body}"
+        end
+      else
+        raise Error, "Failed to get jobs. Status code: #{jobs_response.status}"
+      end
+    end
+    
+    # Force clear the job queue
+    def force_clear_jobs!
+      # Clear the job queue using force option which will also clear any pending data and restart processes
+      path = '/redfish/v1/Dell/Managers/iDRAC.Embedded.1/DellJobService/Actions/DellJobService.DeleteJobQueue'
+      payload = { "JobID" => "JID_CLEARALL_FORCE" }
+      
+      response = authenticated_request(
+        :post, 
+        path, 
+        body: payload.to_json, 
+        headers: { 'Content-Type' => 'application/json' }
+      )
+      
+      if response.status.between?(200, 299)
+        puts "Successfully force-cleared job queue".green
+        
+        # Monitor LC status until it's Ready
+        puts "Waiting for LC status to be Ready..."
+        
+        retries = 12  # ~2 minutes with 10s sleep
+        while retries > 0
+          lc_response = authenticated_request(
+            :post, 
+            '/redfish/v1/Dell/Managers/iDRAC.Embedded.1/DellLCService/Actions/DellLCService.GetRemoteServicesAPIStatus',
+            body: {}.to_json, 
+            headers: { 'Content-Type': 'application/json' }
+          )
+          
+          if lc_response.status.between?(200, 299)
+            begin
+              lc_data = JSON.parse(lc_response.body)
+              status = lc_data["LCStatus"]
+              
+              if status == "Ready"
+                puts "LC Status is Ready".green
+                return true
+              end
+              
+              puts "Current LC Status: #{status}. Waiting..."
+            rescue JSON::ParserError
+              puts "Failed to parse LC status response, will retry...".yellow
+            end
+          end
+          
+          retries -= 1
+          sleep 10
+        end
+        
+        puts "Warning: LC status did not reach Ready state within timeout".yellow
+        return true
+      else
+        error_message = "Failed to force-clear job queue. Status code: #{response.status}"
+        
+        begin
+          error_data = JSON.parse(response.body)
+          error_message += ", Message: #{error_data['error']['message']}" if error_data['error'] && error_data['error']['message']
+        rescue
+          # Ignore JSON parsing errors
+        end
+        
+        raise Error, error_message
+      end
+    end
+    
+    # Wait for a job to complete
+    def wait_for_job(job_id)
+      # Job ID can be a job ID, path, or response hash from another request
+      job_path = if job_id.is_a?(Hash)
+                   if job_id['headers'] && job_id['headers']['location']
+                     job_id['headers']['location'].sub(/^\/redfish\/v1\//, '')
+                   else
+                     raise Error, "Invalid job hash, missing location header"
+                   end
+                 elsif job_id.to_s.start_with?('/redfish/v1/')
+                   job_id.sub(/^\/redfish\/v1\//, '')
+                 else
+                   "Managers/iDRAC.Embedded.1/Jobs/#{job_id}"
+                 end
+      
+      puts "Waiting for job to complete: #{job_id}".light_cyan
+      
+      retries = 36  # ~6 minutes with 10s sleep
+      while retries > 0
+        response = authenticated_request(:get, "/redfish/v1/#{job_path}")
+        
+        if response.status == 200
+          begin
+            job_data = JSON.parse(response.body)
+            job_state = job_data["JobState"]
+            
+            case job_state
+            when "Completed"
+              puts "Job completed successfully".green
+              return job_data
+            when "Failed"
+              puts "Job failed: #{job_data['Message']}".red
+              raise Error, "Job failed: #{job_data['Message']}"
+            when "CompletedWithErrors"
+              puts "Job completed with errors: #{job_data['Message']}".yellow
+              return job_data
+            end
+            
+            puts "Job state: #{job_state}. Waiting...".yellow
+          rescue JSON::ParserError
+            puts "Failed to parse job status response, will retry...".yellow
+          end
+        else
+          puts "Failed to get job status. Status code: #{response.status}".red
+        end
+        
+        retries -= 1
+        sleep 10
+      end
+      
+      raise Error, "Timeout waiting for job to complete"
+    end
+    
+    # Get system tasks
+    def tasks
+      response = authenticated_request(:get, '/redfish/v1/TaskService/Tasks')
+      
+      if response.status == 200
+        begin
+          tasks_data = JSON.parse(response.body)
+          # "Tasks: #{tasks_data['Members'].count}", 0 
+          return tasks_data['Members']
+        rescue JSON::ParserError
+          raise Error, "Failed to parse tasks response: #{response.body}"
+        end
+      else
+        raise Error, "Failed to get tasks. Status code: #{response.status}"
+      end
+    end
+  end
+end