Sha256: 53d704d9c70a7122673dca5ac53414eda40894c28e646c408816fbd157e7add2
Contents?: true
Size: 1.88 KB
Versions: 14
Compression:
Stored size: 1.88 KB
Contents
# frozen_string_literal: true
require "spyse"
module Mihari
module Analyzers
class Spyse < Base
param :query
option :type, default: proc { "domain" }
# @return [String, nil]
attr_reader :api_key
def initialize(*args, **kwargs)
super(*args, **kwargs)
@api_key = kwargs[:api_key] || Mihari.config.spyse_api_key
end
def artifacts
search || []
end
private
def search_params
@search_params ||= JSON.parse(query)
end
def configuration_keys
%w[spyse_api_key]
end
def api
@api ||= ::Spyse::API.new(api_key)
end
#
# Check whether a type is valid or not
#
# @return [Boolean]
#
def valid_type?
%w[ip domain cert].include? type
end
#
# Domain search
#
# @return [Array<Mihari::Artifact>]
#
def domain_search
res = api.domain.search(search_params, limit: 100)
items = res.dig("data", "items") || []
items.map do |item|
data = item["name"]
Artifact.new(data: data, source: source, metadata: item)
end
end
#
# IP search
#
# @return [Array<Mihari::Artifact>]
#
def ip_search
res = api.ip.search(search_params, limit: 100)
items = res.dig("data", "items") || []
items.map do |item|
data = item["ip"]
Artifact.new(data: data, source: source, metadata: item)
end
end
#
# IP/domain search
#
# @return [Array<Mihari::Artifact>]
#
def search
case type
when "domain"
domain_search
when "ip"
ip_search
else
raise InvalidInputError, "#{query}(type: #{type || "unknown"}) is not supported." unless valid_type?
end
end
end
end
end
Version data entries
14 entries across 14 versions & 1 rubygems