postal/app/models/webhook_request.rb

# frozen_string_literal: true

# == Schema Information
#
# Table name: webhook_requests
#
#  id          :integer          not null, primary key
#  server_id   :integer
#  webhook_id  :integer
#  url         :string(255)
#  event       :string(255)
#  uuid        :string(255)
#  payload     :text(65535)
#  attempts    :integer          default(0)
#  retry_after :datetime
#  error       :text(65535)
#  created_at  :datetime
#

class WebhookRequest < ApplicationRecord

  include HasUUID

  RETRIES = { 1 => 2.minutes, 2 => 3.minutes, 3 => 6.minutes, 4 => 10.minutes, 5 => 15.minutes }

  belongs_to :server
  belongs_to :webhook, optional: true

  validates :url, presence: true
  validates :event, presence: true

  serialize :payload, Hash

  after_commit :queue, on: :create

  def self.trigger(server, event, payload = {})
    unless server.is_a?(Server)
      server = Server.find(server.to_i)
    end

    webhooks = server.webhooks.enabled.includes(:webhook_events).references(:webhook_events).where("webhooks.all_events = ? OR webhook_events.event = ?", true, event)
    webhooks.each do |webhook|
      server.webhook_requests.create!(event: event, payload: payload, webhook: webhook, url: webhook.url)
    end
  end

  def self.requeue_all
    where("retry_after < ?", Time.now).find_each(&:queue)
  end

  def queue
    WebhookDeliveryJob.queue(:main, id: id)
  end

  def deliver
    logger = Postal.logger_for(:webhooks)
    payload = { event: event, timestamp: created_at.to_f, payload: self.payload, uuid: uuid }.to_json
    logger.info "[#{id}] Sending webhook request to `#{url}`"
    result = Postal::HTTP.post(url, sign: true, json: payload, timeout: 5)
    self.attempts += 1
    self.retry_after = RETRIES[self.attempts]&.from_now
    server.message_db.webhooks.record(
      event: event,
      url: url,
      webhook_id: webhook_id,
      attempt: self.attempts,
      timestamp: Time.now.to_f,
      payload: self.payload.to_json,
      uuid: uuid,
      status_code: result[:code],
      body: result[:body],
      will_retry: (retry_after ? 0 : 1)
    )

    if result[:code] >= 200 && result[:code] < 300
      logger.info "[#{id}]   -> Received #{result[:code]} status code. That's OK."
      destroy
      webhook&.update_column(:last_used_at, Time.now)
      true
    else
      logger.error "[#{id}]   -> Received #{result[:code]} status code. That's not OK."
      self.error = "Couldn't send to URL. Code received was #{result[:code]}"
      if retry_after
        logger.info "[#{id}]   -> Will retry #{retry_after} (this was attempt #{self.attempts})"
        save
      else
        logger.info "[#{id}]   -> Have tried #{self.attempts} times. Giving up."
        destroy
      end
      false
    end
  end

end