discourse-assign/lib/random_assign_utils.rb

# frozen_string_literal: true

class RandomAssignUtils
  def self.raise_error(automation, message)
    raise("[discourse-automation id=#{automation.id}] #{message}.")
  end

  def self.log_info(automation, message)
    Rails.logger.info("[discourse-automation id=#{automation.id}] #{message}.")
  end

  def self.automation_script!(context, fields, automation)
    unless SiteSetting.assign_enabled?
      raise_error(automation, "discourse-assign is not enabled")
    end

    unless topic_id = fields.dig('assigned_topic', 'value')
      raise_error(automation, "`assigned_topic` not provided")
    end

    unless topic = Topic.find_by(id: topic_id)
      raise_error(automation, "Topic(#{topic_id}) not found")
    end

    min_hours = fields.dig('minimum_time_between_assignments', 'value').presence
    if min_hours && TopicCustomField
        .where(name: 'assigned_to_id', topic_id: topic_id)
        .where('created_at < ?', min_hours.to_i.hours.ago)
        .exists?
      log_info(automation, "Topic(#{topic_id}) has already been assigned recently")
      return
    end

    unless group_id = fields.dig('assignees_group', 'value')
      raise_error(automation, "`assignees_group` not provided")
    end

    unless group = Group.find_by(id: group_id)
      raise_error(automation, "Group(#{group_id}) not found")
    end

    users_on_holiday = Set.new(
      User
        .where(id:
          UserCustomField
          .where(name: 'on_holiday', value: 't')
          .select(:user_id)
        ).pluck(:id)
    )

    group_users_ids = group
      .group_users
      .joins(:user)
      .pluck('users.id')
      .reject { |user_id| users_on_holiday.include?(user_id) }

    if group_users_ids.empty?
      RandomAssignUtils.no_one!(topic_id, group.name)
      return
    end

    max_recently_assigned_days = (fields.dig('max_recently_assigned_days', 'value').presence || 180).to_i.days.ago
    last_assignees_ids = RandomAssignUtils.recently_assigned_users_ids(topic_id, max_recently_assigned_days)
    users_ids = group_users_ids - last_assignees_ids
    if users_ids.blank?
      min_recently_assigned_days = (fields.dig('min_recently_assigned_days', 'value').presence || 14).to_i.days.ago
      recently_assigned_users_ids = RandomAssignUtils.recently_assigned_users_ids(topic_id, min_recently_assigned_days)
      users_ids = group_users_ids - recently_assigned_users_ids
    end

    if users_ids.blank?
      RandomAssignUtils.no_one!(topic_id, group.name)
      return
    end

    if fields.dig('in_working_hours', 'value')
      assign_to_user_id = users_ids.shuffle.find do |user_id|
        RandomAssignUtils.in_working_hours?(user_id)
      end
    end

    assign_to_user_id ||= users_ids.sample
    if assign_to_user_id.blank?
      RandomAssignUtils.no_one!(topic_id, group.name)
      return
    end

    assign_to = User.find(assign_to_user_id)
    result = nil
    if raw = fields.dig('post_template', 'value').presence
      post = PostCreator.new(
        Discourse.system_user,
        raw: raw,
        skip_validations: true,
        topic_id: topic.id
      ).create!

      result = Assigner.new(post, Discourse.system_user).assign(assign_to)

      if !result[:success]
        PostDestroyer.new(Discourse.system_user, post).destroy
      end
    else
      result = Assigner.new(topic, Discourse.system_user).assign(assign_to)
    end

    if !result[:success]
      RandomAssignUtils.no_one!(topic_id, group.name)
    end
  end

  def self.recently_assigned_users_ids(topic_id, from)
    posts = Post
      .joins(:user)
      .where(topic_id: topic_id, action_code: ['assigned', 'reassigned'])
      .where('posts.created_at > ?', from)
      .order(created_at: :desc)
    usernames = Post.custom_fields_for_ids(posts, [:action_code_who]).map { |_, v| v['action_code_who'] }.uniq
    User.where(username: usernames).limit(100).pluck(:id)
  end

  def self.user_tzinfo(user_id)
    timezone = UserOption.where(user_id: user_id).pluck(:timezone).first || "UTC"

    tzinfo = nil
    begin
      tzinfo = ActiveSupport::TimeZone.find_tzinfo(timezone)
    rescue TZInfo::InvalidTimezoneIdentifier
      Rails.logger.warn("#{User.find_by(id: user_id)&.username} has the timezone #{timezone} set, we do not know how to parse it in Rails (assuming UTC)")
      timezone = "UTC"
      tzinfo = ActiveSupport::TimeZone.find_tzinfo(timezone)
    end

    tzinfo
  end

  def self.no_one!(topic_id, group)
    PostCreator.create!(
      Discourse.system_user,
      topic_id: topic_id,
      raw: I18n.t("discourse_automation.scriptables.random_assign.no_one", group: group),
      validate: false
    )
  end

  def self.in_working_hours?(user_id)
    tzinfo = RandomAssignUtils.user_tzinfo(user_id)
    tztime = tzinfo.now

    !tztime.saturday? &&
    !tztime.sunday? &&
    tztime.hour > 7 &&
    tztime.hour < 11
  end
end