Contents

# frozen_string_literal: true

module Mihari
  class Rule < ActiveRecord::Base
    has_many :alerts, foreign_key: :source

    #
    # Returns a hash representative
    #
    # @return [Hash]
    #
    def to_h
      symbolized_data = data.deep_symbolize_keys
      h = { id: id, created_at: created_at, yaml: data.to_yaml }
      h.merge symbolized_data
    end

    class << self
      #
      # Search rules
      #
      # @param [Structs::Rule::SearchFilterWithPagination] filter
      #
      # @return [Array<Rule>]
      #
      def search(filter)
        limit = filter.limit.to_i
        raise ArgumentError, "limit should be bigger than zero" unless limit.positive?

        page = filter.page.to_i
        raise ArgumentError, "page should be bigger than zero" unless page.positive?

        offset = (page - 1) * limit

        relation = build_relation(filter.without_pagination)

        # TODO: improve queires
        rule_ids = relation.limit(limit).offset(offset).order(created_at: :desc).pluck(:id).uniq
        where(id: [rule_ids]).order(created_at: :desc)
      end

      #
      # Count alerts
      #
      # @param [Structs::Rule::SearchFilterWithPagination] filter
      #
      # @return [Integer]
      #
      def count(filter)
        relation = build_relation(filter)
        relation.distinct("rules.id").count
      end

      private

      def build_relation(filter)
        relation = self
        relation = relation.includes(alerts: :tags)

        relation = relation.where(alerts: { tags: { name: filter.tag_name } }) if filter.tag_name

        relation = relation.where(title: filter.title) if filter.title

        relation = relation.where("rules.description LIKE ?", "%#{filter.description}%") if filter.description

        relation = relation.where("rules.created_at >= ?", filter.from_at) if filter.from_at
        relation = relation.where("rules.created_at <= ?", filter.to_at) if filter.to_at

        relation
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
mihari-4.3.0	lib/mihari/models/rule.rb
mihari-4.2.0	lib/mihari/models/rule.rb
mihari-4.1.2	lib/mihari/models/rule.rb
mihari-4.1.1	lib/mihari/models/rule.rb
mihari-4.1.0	lib/mihari/models/rule.rb