mihari 5.6.2 → 5.7.0

data/lib/mihari/structs/ipinfo.rb

@@ -4,46 +4,25 @@ module Mihari
   module Structs
     module IPInfo
       class Response < Dry::Struct
+        # @!attribute [r] ip
+        #   @return [String]
         attribute :ip, Types::String
-        attribute :hostname, Types::String.optional
-        attribute :loc, Types::String.optional
-        attribute :country_code, Types::String.optional
-        attribute :asn, Types::Integer.optional
 
-        #
-        # @return [String]
-        #
-        def ip
-          attributes[:ip]
-        end
-
-        #
-        # @return [String, nil]
-        #
-        def hostname
-          attributes[:hostname]
-        end
+        # @!attribute [r] hostname
+        #   @return [String, nil]
+        attribute :hostname, Types::String.optional
 
-        #
-        # @return [String, nil]
-        #
-        def loc
-          attributes[:loc]
-        end
+        # @!attribute [r] loc
+        #   @return [String, nil]
+        attribute :loc, Types::String.optional
 
-        #
-        # @return [String, nil]
-        #
-        def country_code
-          attributes[:country_code]
-        end
+        # @!attribute [r] country_code
+        #   @return [String, nil]
+        attribute :country_code, Types::String.optional
 
-        #
-        # @return [Integer, nil]
-        #
-        def asn
-          attributes[:asn]
-        end
+        # @!attribute [r] asn
+        #   @return [Integer, nil]
+        attribute :asn, Types::Int.optional
 
         class << self
           include Mixins::AutonomousSystem

data/lib/mihari/structs/onyphe.rb

@@ -6,38 +6,22 @@ module Mihari
       class Result < Dry::Struct
         include Mixins::AutonomousSystem
 
+        # @!attribute [r] asn
+        #   @return [String]
         attribute :asn, Types::String
+
+        # @!attribute [r] country_code
+        #   @return [String, nll]
         attribute :country_code, Types::String.optional
-        attribute :ip, Types::String
-        attribute :metadata, Types::Hash
 
-        #
-        # @return [String]
-        #
-        def asn
-          attributes[:asn]
-        end
+        # @!attribute [r] ip
+        #   @return [String]
 
-        #
-        # @return [String, nil]
-        #
-        def country_code
-          attributes[:country_code]
-        end
-
-        #
-        # @return [String]
-        #
-        def ip
-          attributes[:ip]
-        end
+        attribute :ip, Types::String
 
-        #
-        # @return [Hash]
-        #
-        def metadata
-          attributes[:metadata]
-        end
+        # @!attribute [r] metadata
+        #   @return [Hash]
+        attribute :metadata, Types::Hash
 
         #
         # @return [Mihari::Models::Artifact]
@@ -90,62 +74,33 @@ module Mihari
       end
 
       class Response < Dry::Struct
+        # @!attribute [r] count
+        #   @return [Integer]
         attribute :count, Types::Int
-        attribute :error, Types::Int
-        attribute :max_page, Types::Int
-        attribute :page, Types::Int
-        attribute :results, Types.Array(Result)
-        attribute :status, Types::String
-        attribute :total, Types::Int
-
-        #
-        # @return [Integer]
-        #
-        def count
-          attributes[:count]
-        end
 
-        #
-        # @return [Integer]
-        #
-        def error
-          attributes[:error]
-        end
+        # @!attribute [r] error
+        #   @return [Integer]
+        attribute :error, Types::Int
 
-        #
-        # @return [Integer]
-        #
-        def max_page
-          attributes[:max_page]
-        end
+        # @!attribute [r] max_page
+        #   @return [Integer]
+        attribute :max_page, Types::Int
 
-        #
-        # @return [Integer]
-        #
-        def page
-          attributes[:page]
-        end
+        # @!attribute [r] page
+        #   @return [Integer]
+        attribute :page, Types::Int
 
-        #
-        # @return [Array<Result>]
-        #
-        def results
-          attributes[:results]
-        end
+        # @!attribute [r] results
+        #   @return [Array<Result>]
+        attribute :results, Types.Array(Result)
 
-        #
-        # @return [String]
-        #
-        def status
-          attributes[:status]
-        end
+        # @!attribute [r] status
+        #   @return [String]
+        attribute :status, Types::String
 
-        #
-        # @return [Integer]
-        #
-        def total
-          attributes[:total]
-        end
+        # @!attribute [r] total
+        #   @return [Integer]
+        attribute :total, Types::Int
 
         #
         # @return [Array<Mihari::Models::Artifact>]

data/lib/mihari/structs/shodan.rb

@@ -4,22 +4,13 @@ module Mihari
   module Structs
     module Shodan
       class Location < Dry::Struct
+        # @!attribute [r] country_code
+        #   @return [String, nil]
         attribute :country_code, Types::String.optional
-        attribute :country_name, Types::String.optional
-
-        #
-        # @return [String, nil]
-        #
-        def country_code
-          attributes[:country_code]
-        end
 
-        #
-        # @return [String, nil]
-        #
-        def country_name
-          attributes[:country_name]
-        end
+        # @!attribute [r] country_name
+        #   @return [String, nil]
+        attribute :country_name, Types::String.optional
 
         #
         # @return [Mihari::Geolocation, nil]
@@ -52,55 +43,33 @@ module Mihari
       class Match < Dry::Struct
         include Mixins::AutonomousSystem
 
+        # @!attribute [r] asn
+        #   @return [String, nil]
         attribute :asn, Types::String.optional
-        attribute :hostnames, Types.Array(Types::String)
-        attribute :location, Location
-        attribute :domains, Types.Array(Types::String)
-        attribute :ip_str, Types::String
-        attribute :port, Types::Integer
-        attribute :metadata, Types::Hash
 
-        #
-        # @return [String, nil]
-        #
-        def asn
-          attributes[:asn]
-        end
+        # @!attribute [r] hostname
+        #   @return [Array<String>]
+        attribute :hostnames, Types.Array(Types::String)
 
-        #
-        # @return [Array<String>]
-        #
-        def hostnames
-          attributes[:hostnames]
-        end
+        # @!attribute [r] location
+        #   @return [Location]
+        attribute :location, Location
 
-        #
-        # @return [Location]
-        #
-        def location
-          attributes[:location]
-        end
+        # @!attribute [r] domains
+        #   @return [Array<String>]
+        attribute :domains, Types.Array(Types::String)
 
-        #
-        # @return [String]
-        #
-        def ip_str
-          attributes[:ip_str]
-        end
+        # @!attribute [r] ip_str
+        #   @return [String]
+        attribute :ip_str, Types::String
 
-        #
-        # @return [Integer]
-        #
-        def port
-          attributes[:port]
-        end
+        # @!attribute [r] port
+        #   @return [Integer]
+        attribute :port, Types::Int
 
-        #
-        # @return [Hash]
-        #
-        def metadata
-          attributes[:metadata]
-        end
+        # @!attribute [r] metadata
+        #   @return [Hash]
+        attribute :metadata, Types::Hash
 
         #
         # @return [Mihari::AutonomousSystem, nil]
@@ -141,22 +110,13 @@ module Mihari
       end
 
       class Response < Dry::Struct
+        # @!attribute [r] matches
+        #   @return [Array<Match>]
         attribute :matches, Types.Array(Match)
-        attribute :total, Types::Int
-
-        #
-        # @return [Array<Match>]
-        #
-        def matches
-          attributes[:matches]
-        end
 
-        #
-        # @return [Integer]
-        #
-        def total
-          attributes[:total]
-        end
+        # @!attribute [r] total
+        #   @return [Integer]
+        attribute :total, Types::Int
 
         #
         # Collect metadata from matches
@@ -233,54 +193,29 @@ module Mihari
       end
 
       class InternetDBResponse < Dry::Struct
+        # @!attribute [r] ip
+        #   @return [String]
         attribute :ip, Types::String
-        attribute :ports, Types.Array(Types::Int)
-        attribute :cpes, Types.Array(Types::String)
-        attribute :hostnames, Types.Array(Types::String)
-        attribute :tags, Types.Array(Types::String)
-        attribute :vulns, Types.Array(Types::String)
-
-        #
-        # @return [String]
-        #
-        def ip
-          attributes[:ip]
-        end
 
-        #
-        # @return [Array<Integer>]
-        #
-        def ports
-          attributes[:ports]
-        end
+        # @!attribute [r] ports
+        #   @return [Array<Integer>]
+        attribute :ports, Types.Array(Types::Int)
 
-        #
-        # @return [Array<String>]
-        #
-        def cpes
-          attributes[:cpes]
-        end
+        # @!attribute [r] cpes
+        #   @return [Array<String>]
+        attribute :cpes, Types.Array(Types::String)
 
-        #
-        # @return [Array<String>]
-        #
-        def hostnames
-          attributes[:hostnames]
-        end
+        # @!attribute [r] hostnames
+        #   @return [Array<String>]
+        attribute :hostnames, Types.Array(Types::String)
 
-        #
-        # @return [Array<String>]
-        #
-        def tags
-          attributes[:tags]
-        end
+        # @!attribute [r] tags
+        #   @return [Array<String>]
+        attribute :tags, Types.Array(Types::String)
 
-        #
-        # @return [Array<String>]
-        #
-        def vulns
-          attributes[:vulns]
-        end
+        # @!attribute [r] vulns
+        #   @return [Array<String>]
+        attribute :vulns, Types.Array(Types::String)
 
         class << self
           #

data/lib/mihari/structs/urlscan.rb

@@ -4,30 +4,17 @@ module Mihari
   module Structs
     module Urlscan
       class Page < Dry::Struct
+        # @!attribute [r] domain
+        #   @return [String]
         attribute :domain, Types::String.optional
-        attribute :ip, Types::String.optional
-        attribute :url, Types::String
-
-        #
-        # @return [String, nil]
-        #
-        def domain
-          attributes[:domain]
-        end
 
-        #
-        # @return [String, nil]
-        #
-        def ip
-          attributes[:ip]
-        end
+        # @!attribute [r] ip
+        #   @return [String, nil]
+        attribute :ip, Types::String.optional
 
-        #
-        # @return [String]
-        #
-        def url
-          attributes[:url]
-        end
+        # @!attribute [r] url
+        #   @return [String]
+        attribute :url, Types::String
 
         class << self
           #
@@ -47,38 +34,21 @@ module Mihari
       end
 
       class Result < Dry::Struct
+        # @!attribute [r] page
+        #   @return [Page]
         attribute :page, Page
-        attribute :id, Types::String
-        attribute :sort, Types.Array(Types::String | Types::Integer)
-        attribute :metadata, Types::Hash
-
-        #
-        # @return [Page]
-        #
-        def page
-          attributes[:page]
-        end
 
-        #
-        # @return [String]
-        #
-        def id
-          attributes[:id]
-        end
+        # @!attribute [r] pid
+        #   @return [String]
+        attribute :id, Types::String
 
-        #
-        # @return [Array<String, Integer>]
-        #
-        def sort
-          attributes[:sort]
-        end
+        # @!attribute [r] sort
+        #   @return [Array<String, Integer>]
+        attribute :sort, Types.Array(Types::String | Types::Int)
 
-        #
-        # @return [Array<String, Integer>]
-        #
-        def metadata
-          attributes[:metadata]
-        end
+        # @!attribute [r] metadata
+        #   @return [Hash]
+        attribute :metadata, Types::Hash
 
         #
         # @return [Array<Mihari::Models::Artifact>]
@@ -107,22 +77,13 @@ module Mihari
       end
 
       class Response < Dry::Struct
+        # @!attribute [r] results
+        #   @return [Array<Result>]
         attribute :results, Types.Array(Result)
-        attribute :has_more, Types::Bool
-
-        #
-        # @return [Array<Result>]
-        #
-        def results
-          attributes[:results]
-        end
 
-        #
-        # @return [Boolean]
-        #
-        def has_more
-          attributes[:has_more]
-        end
+        # @!attribute [r] has_more
+        #   @return [Boolean]
+        attribute :has_more, Types::Bool
 
         #
         # @return [Array<Mihari::Models::Artifact>]

data/lib/mihari/structs/virustotal_intelligence.rb

@@ -4,15 +4,10 @@ module Mihari
   module Structs
     module VirusTotalIntelligence
       class ContextAttributes < Dry::Struct
+        # @!attribute [r] url
+        #   @return [String, nil]
         attribute :url, Types::String.optional
 
-        #
-        # @return [String, nil]
-        #
-        def url
-          attributes[:url]
-        end
-
         class << self
           #
           # @param [Hash] d
@@ -27,38 +22,21 @@ module Mihari
       end
 
       class Datum < Dry::Struct
+        # @!attribute [r] type
+        #   @return [String]
         attribute :type, Types::String
-        attribute :id, Types::String
-        attribute :context_attributes, ContextAttributes.optional
-        attribute :metadata, Types::Hash
 
-        #
-        # @return [String]
-        #
-        def type
-          attributes[:type]
-        end
-
-        #
-        # @return [String]
-        #
-        def id
-          attributes[:id]
-        end
+        # @!attribute [r] id
+        #   @return [String]
+        attribute :id, Types::String
 
-        #
-        # @return [ContextAttributes, nil]
-        #
-        def context_attributes
-          attributes[:context_attributes]
-        end
+        # @!attribute [r] context_attributes
+        #   @return [ContextAttributes, nil]
+        attribute :context_attributes, ContextAttributes.optional
 
-        #
-        # @return [Hash, nil]
-        #
-        def metadata
-          attributes[:metadata]
-        end
+        # @!attribute [r] metadata
+        #   @return [Hash]
+        attribute :metadata, Types::Hash
 
         #
         # @return [String, nil]
@@ -108,15 +86,10 @@ module Mihari
       end
 
       class Meta < Dry::Struct
+        # @!attribute [r] cursor
+        #   @return [String, nil]
         attribute :cursor, Types::String.optional
 
-        #
-        # @return [String, nil]
-        #
-        def cursor
-          attributes[:cursor]
-        end
-
         class << self
           #
           # @param [Hash] d
@@ -133,22 +106,13 @@ module Mihari
       end
 
       class Response < Dry::Struct
+        # @!attribute [r] meta
+        #   @return [Meta]
         attribute :meta, Meta
-        attribute :data, Types.Array(Datum)
-
-        #
-        # @return [Meta]
-        #
-        def meta
-          attributes[:meta]
-        end
 
-        #
-        # @return [Array<Datum>]
-        #
-        def data
-          attributes[:data]
-        end
+        # @!attribute [r] data
+        #   @return [Array<Datum>]
+        attribute :data, Types.Array(Datum)
 
         #
         # @return [Array<Mihari::Models::Artifact>]

data/lib/mihari/type_checker.rb

@@ -1,6 +1,9 @@
 # frozen_string_literal: true
 
 module Mihari
+  #
+  # Artifact type checker
+  #
   class TypeChecker
     # @return [String]
     attr_reader :data
@@ -54,6 +57,7 @@ module Mihari
       return "ip" if ip?
       return "domain" if domain?
       return "url" if url?
+
       "mail" if mail?
     end
 

data/lib/mihari/types.rb

@@ -1,6 +1,9 @@
 # frozen_string_literal: true
 
 module Mihari
+  #
+  # dry-type based types
+  #
   module Types
     include Dry.Types()
 

data/lib/mihari/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Mihari
-  VERSION = "5.6.2"
+  VERSION = "5.7.0"
 end

data/lib/mihari/web/api.rb

@@ -9,17 +9,22 @@ require "mihari/web/endpoints/rules"
 require "mihari/web/endpoints/tags"
 
 module Mihari
-  class API < Grape::API
-    prefix "api"
-    format :json
+  module Web
+    #
+    # Grape API
+    #
+    class API < Grape::API
+      prefix "api"
+      format :json
 
-    mount Endpoints::Alerts
-    mount Endpoints::Artifacts
-    mount Endpoints::Configs
-    mount Endpoints::IPAddresses
-    mount Endpoints::Rules
-    mount Endpoints::Tags
+      mount Endpoints::Alerts
+      mount Endpoints::Artifacts
+      mount Endpoints::Configs
+      mount Endpoints::IPAddresses
+      mount Endpoints::Rules
+      mount Endpoints::Tags
 
-    add_swagger_documentation(api_version: "v1", info: { title: "Mihari API" })
+      add_swagger_documentation(api_version: "v1", info: { title: "Mihari API" })
+    end
   end
 end