# frozen_string_literal: true

module Mihari
  module Schemas
    #
    # Analyzer schemas
    #
    module Analyzers
      extend Concerns::Orrable

      # Analyzer with API key and pagination
      [
        Mihari::Analyzers::BinaryEdge.class_keys,
        Mihari::Analyzers::GreyNoise.class_keys,
        Mihari::Analyzers::Onyphe.class_keys,
        Mihari::Analyzers::Shodan.class_keys,
        Mihari::Analyzers::Urlscan.class_keys,
        Mihari::Analyzers::VirusTotalIntelligence.class_keys
      ].each do |keys|
        key = keys.first
        const_set(key.upcase, Dry::Schema.Params do
          required(:analyzer).value(Types::String.enum(*keys))
          required(:query).value(:string)
          optional(:api_key).value(:string)
          optional(:options).hash(AnalyzerPaginationOptions)
        end)
      end

      # Analyzer with API key
      [
        Mihari::Analyzers::OTX.class_keys,
        Mihari::Analyzers::Pulsedive.class_keys,
        Mihari::Analyzers::VirusTotal.class_keys,
        Mihari::Analyzers::SecurityTrails.class_keys
      ].each do |keys|
        key = keys.first
        const_set(key.upcase, Dry::Schema.Params do
          required(:analyzer).value(Types::String.enum(*keys))
          required(:query).value(:string)
          optional(:api_key).value(:string)
          optional(:options).hash(AnalyzerOptions)
        end)
      end

      DNSTwister = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::DNSTwister.class_keys))
        required(:query).value(:string)
        optional(:options).hash(AnalyzerOptions)
      end

      Censys = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::Censys.class_keys))
        required(:query).value(:string)
        optional(:id).value(:string)
        optional(:secret).value(:string)
        optional(:options).hash(AnalyzerPaginationOptions)
      end

      CIRCL = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::CIRCL.class_keys))
        required(:query).value(:string)
        optional(:username).value(:string)
        optional(:password).value(:string)
        optional(:options).hash(AnalyzerOptions)
      end

      Fofa = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::Fofa.class_keys))
        required(:query).value(:string)
        optional(:api_key).value(:string)
        optional(:email).value(:string)
        optional(:options).hash(AnalyzerPaginationOptions)
      end

      PassiveTotal = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::PassiveTotal.class_keys))
        required(:query).value(:string)
        optional(:username).value(:string)
        optional(:api_key).value(:string)
        optional(:options).hash(AnalyzerOptions)
      end

      ZoomEye = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::ZoomEye.class_keys))
        required(:query).value(:string)
        required(:type).value(Types::String.enum("host", "web"))
        optional(:options).hash(AnalyzerPaginationOptions)
      end

      Crtsh = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::Crtsh.class_keys))
        required(:query).value(:string)
        optional(:exclude_expired).value(:bool).default(true)
        optional(:options).hash(AnalyzerOptions)
      end

      HunterHow = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::HunterHow.class_keys))
        required(:query).value(:string)
        required(:start_time).value(:date)
        required(:end_time).value(:date)
        optional(:api_key).value(:string)
        optional(:options).hash(AnalyzerPaginationOptions)
      end

      Feed = Dry::Schema.Params do
        required(:analyzer).value(Types::String.enum(*Mihari::Analyzers::Feed.class_keys))
        required(:query).value(:string)
        required(:selector).value(:string)
        optional(:method).value(Types::HTTPRequestMethods).default("GET")
        optional(:headers).value(:hash).default({})
        optional(:params).value(:hash)
        optional(:form).value(:hash)
        optional(:json).value(:hash)
        optional(:options).hash(AnalyzerOptions)
      end
    end

    Analyzer = Schemas::Analyzers.get_or_composition
  end
end