class IDRAC::Screenshot

def capture
  # Login to get the forward URL and cookies
  forward_url = client.login
  
  # Extract the key-value pairs from the forward URL (format: index?ST1=ABC,ST2=DEF)
  tokens = forward_url.split("?").last.split(",").inject({}) do |acc, kv| 
    k, v = kv.split("=")
    acc[k] = v
    acc
  end
  
  # Generate a timestamp for the request
  timestamp_ms = (Time.now.to_f * 1000).to_i
  
  # First request to trigger the screenshot capture
  path = "data?get=consolepreview[manual%20#{timestamp_ms}]"
  res = client.get(path: path, headers: tokens)
  raise Error, "Failed to trigger screenshot capture." unless res.code.between?(200, 299)
  
  # Wait for the screenshot to be generated
  sleep 2
  
  # Second request to get the actual screenshot image
  path = "capconsole/scapture0.png?#{timestamp_ms}"
  res = client.get(path: path, headers: tokens)
  raise Error, "Failed to retrieve screenshot image." unless res.code.between?(200, 299)
  
  # Save the screenshot to a file
  filename = "idrac_screenshot_#{timestamp_ms}.png"
  File.open(filename, "wb") { |f| f.write(res.body) }
  
  # Return the filename
  filename
end

def initialize(client)
  @client = client
end