Sha256: a437d9c08a98abe24abff9bda35289c269da6201b93744e644235c0ffdec2f87
Contents?: true
Size: 1.57 KB
Versions: 5
Compression:
Stored size: 1.57 KB
Contents
# frozen_string_literal: true
module Mihari
module Clients
class UrlScan < Base
#
# @param [String] base_url
# @param [String, nil] api_key
# @param [Hash] headers
# @param [Interval, nil] interval
#
def initialize(base_url = "https://urlscan.io", api_key:, headers: {}, interval: nil)
raise(ArgumentError, "'api_key' argument is required") if api_key.nil?
headers["api-key"] = api_key
super(base_url, headers: headers, interval: interval)
end
#
# @param [String] q
# @param [Integer, nil] size
# @param [String, nil] search_after
#
# @return [Structs::Urlscan::Response]
#
def search(q, size: nil, search_after: nil)
params = { q: q, size: size, search_after: search_after }.compact
res = get("/api/v1/search/", params: params)
Structs::Urlscan::Response.from_dynamic! JSON.parse(res.body.to_s)
end
#
# @param [String] q
# @param [Integer, nil] size
# @param [Integer] pagination_limit
#
# @return [Enumerable<Structs::Urlscan::Response>]
#
def search_with_pagination(q, size: nil, pagination_limit: Mihari.config.pagination_limit)
search_after = nil
Enumerator.new do |y|
pagination_limit.times do
res = search(q, size: size, search_after: search_after)
y.yield res
break unless res.has_more
search_after = res.results.last.sort.join(",")
sleep_interval
end
end
end
end
end
end
Version data entries
5 entries across 5 versions & 1 rubygems