gem "snapapi"

require "snapapi"

client = SnapAPI::Client.new(api_key: "sk_live_...")

# Take a screenshot
png = client.screenshot(url: "https://example.com")
File.binwrite("screenshot.png", png)

# Save directly to a file
client.screenshot_to_file("https://example.com", "screenshot.png")

SnapAPI.configure do |config|
  config.api_key     = "sk_live_..."
  config.base_url    = "https://api.snapapi.pics"  # Default
  config.timeout     = 60                           # Seconds (default: 60)
  config.max_retries = 3                            # Auto-retry on 429 / 5xx (default: 3)
  config.retry_delay = 0.5                          # Initial backoff seconds (doubles each retry)
end

client = SnapAPI::Client.new

client = SnapAPI::Client.new(
  api_key:     "sk_live_...",
  base_url:    "https://api.snapapi.pics",  # Default
  timeout:     60,                           # Seconds (default: 60)
  max_retries: 3,                            # Auto-retry on 429 / 5xx (default: 3)
  retry_delay: 0.5,                          # Initial backoff seconds (doubles each retry)
)

# Basic PNG screenshot
png = client.screenshot(url: "https://example.com")

# Full-page dark-mode WebP with ad blocking
webp = client.screenshot(
  url:              "https://github.com",
  format:           "webp",
  full_page:        true,
  dark_mode:        true,
  block_ads:        true,
  block_trackers:   true,
  width:            1440,
  height:           900,
)

# Render raw HTML
png = client.screenshot(html: "<h1 style='color:red'>Hello!</h1>")

# Capture only a specific element
png = client.screenshot(
  url:      "https://example.com",
  selector: "#main-content",
)

# Save to file directly
client.screenshot_to_file("https://example.com", "./output/screenshot.png")

# With custom device emulation
png = client.screenshot(
  url:                  "https://example.com",
  device:               "iPhone 14",
  device_scale_factor:  2.0,
  is_mobile:            true,
  has_touch:            true,
)

# With cookies and HTTP auth
png = client.screenshot(
  url:         "https://example.com/protected",
  http_auth:   { username: "user", password: "pass" },
  cookies:     [{ name: "session", value: "abc123", domain: "example.com" }],
)

# Store result in SnapAPI cloud
result = client.screenshot(
  url:     "https://example.com",
  storage: { destination: "snapapi" },
)
puts result.url  # Public CDN URL

# Basic PDF
pdf_bytes = client.pdf(url: "https://example.com")
File.binwrite("output.pdf", pdf_bytes)

# Save directly to file
client.pdf_to_file("https://example.com", "./output.pdf")

# A3 landscape with margins
pdf_bytes = client.pdf(
  url:       "https://example.com",
  page_size: "a3",
  landscape: true,
  margins:   { top: "1cm", right: "1cm", bottom: "1cm", left: "1cm" },
)

# From HTML
pdf_bytes = client.pdf(
  html:       "<h1>Invoice #1234</h1><p>Amount: $99.00</p>",
  page_size:  "letter",
)

# With custom header and footer
pdf_bytes = client.pdf(
  url:                     "https://example.com",
  display_header_footer:   true,
  header_template:         "<div style='font-size:10px'>Report</div>",
  footer_template:         "<div style='font-size:10px'>Page <span class='pageNumber'></span></div>",
)

# Scrape text content
result = client.scrape(url: "https://example.com")
result.results.each { |page| puts page["data"] }

# Scrape links
result = client.scrape(url: "https://example.com", type: "links")

# Scrape multiple pages
result = client.scrape(url: "https://example.com", pages: 3)

# Extract as Markdown (default)
result = client.extract(url: "https://example.com")
puts result.content

# Extract main article
result = client.extract_article("https://example.com/blog/post")

# Extract plain text
result = client.extract_text("https://example.com")

# Extract all links
result = client.extract_links("https://example.com")

# Extract all images
result = client.extract_images("https://example.com")

# Extract page metadata
result = client.extract_metadata("https://example.com")
puts result.content["title"]
puts result.content["description"]

# Scoped extraction with CSS selector
result = client.extract(
  url:      "https://example.com",
  type:     "markdown",
  selector: "#article-content",
)

# Truncate output
result = client.extract(
  url:        "https://example.com",
  max_length: 5000,
  clean_output: true,
)

# Basic MP4 recording
video_bytes = client.video(url: "https://example.com")
File.binwrite("recording.mp4", video_bytes)

# With scroll animation
video_bytes = client.video(
  url:       "https://example.com",
  format:    "mp4",
  duration:  10,
  scrolling: true,
  width:     1280,
  height:    720,
)

# Dark mode GIF
gif_bytes = client.video(
  url:       "https://example.com",
  format:    "gif",
  duration:  5,
  dark_mode: true,
)

og_bytes = client.og_image(url: "https://example.com")
File.binwrite("og.png", og_bytes)

# Custom dimensions
og_bytes = client.og_image(
  url:    "https://example.com",
  format: "jpeg",
  width:  1200,
  height: 628,
)

# Analyze with OpenAI
result = client.analyze(
  url:      "https://example.com",
  prompt:   "Summarize this page in 3 bullet points.",
  provider: "openai",
  api_key:  "sk-...",
)
puts result.result

# Structured output with JSON schema
result = client.analyze(
  url:     "https://example.com/product",
  prompt:  "Extract product information",
  provider: "openai",
  api_key:  "sk-...",
  json_schema: {
    type: "object",
    properties: {
      name:  { type: "string" },
      price: { type: "number" },
    }
  },
)

usage = client.get_usage
puts "#{usage.used} / #{usage.limit} calls used"
puts "Resets at: #{usage.reset_at}"

# quota is an alias
usage = client.quota

result = client.ping
puts result["status"]  # => "ok"

begin
  png = client.screenshot(url: "https://example.com")
rescue SnapAPI::AuthenticationError => e
  puts "Bad API key: #{e.message}"
rescue SnapAPI::RateLimitError => e
  puts "Rate limited. Retry after: #{e.retry_after}s"
  sleep(e.retry_after)
  retry
rescue SnapAPI::QuotaExceededError => e
  puts "Quota exhausted. Upgrade your plan."
rescue SnapAPI::ValidationError => e
  puts "Invalid params: #{e.message}"
  puts e.fields.inspect
rescue SnapAPI::TimeoutError
  puts "Request timed out."
rescue SnapAPI::NetworkError => e
  puts "Network failure: #{e.message}"
rescue SnapAPI::Error => e
  puts "API error #{e.status_code}: #{e.message} (#{e.code})"
end

client = SnapAPI::Client.new(
  api_key:     "sk_live_...",
  max_retries: 5,     # Up to 5 retries
  retry_delay: 1.0,   # Start at 1s, then 2s, 4s, 8s, 16s
)

client = SnapAPI::Client.new(api_key: "sk_live_...", max_retries: 0)

result = client.extract_metadata("https://example.com")
result.content      # => Hash or String
result.url          # => "https://example.com"
result.type         # => "metadata"
result.to_h         # => Raw Hash
result.raw          # => Raw Hash

usage = client.get_usage
usage.used          # => 42
usage.limit         # => 1000
usage.remaining     # => 958