plugin.rb

# name: flexible-rate-limits
# version: 0.2.1
# author: Muhlis Budi Cahyono (muhlisbc@gmail.com)
# url: https://github.com/ryanerwin/discourse-flexible-rate-limits

enabled_site_setting :flexible_rate_limits_enabled

%i(common desktop mobile).each do |layout|
  register_asset "stylesheets/flexible-rate-limits/#{layout}.scss", layout
end

add_admin_route "flexible_rate_limits.admin.nav_label", "flexible-rate-limits"

require_relative "lib/flexible_rate_limits"

after_initialize {

  load File.expand_path("../lib/controllers/flexible_rate_limits_controller.rb", __FILE__)
  load File.expand_path("../lib/controllers/admin/flexible_rate_limits_controller.rb", __FILE__)

  Discourse::Application.routes.append {
    scope("flexible_rate_limits") {
      post "/:category_id/:limits_type" => "flexible_rate_limits#index"
    }

    scope "/admin/plugins/flexible-rate-limits", constraints: AdminConstraint.new do
      get ""       => "admin/flexible_rate_limits#index"
      post "save"  => "admin/flexible_rate_limits#save"
    end
  }

  require_dependency "topic"
  Topic.class_eval {
    alias_method :orig_limit_topics_per_day, :limit_topics_per_day

    def limit_topics_per_day
      if user && user.new_user_posting_on_first_day?
        limit_first_day_topics_per_day
      else
        if SiteSetting.flexible_rate_limits_enabled
          frl = FlexibleRateLimits.new(self.user, self.category_id)
          if frl.category_group_name
            RateLimiter.new(self.user, "cg-#{frl.category_group_name}-topic", frl.topic_limit, frl.cooldown.minutes.to_i)
          else
            apply_per_day_rate_limit_for("topics", :max_topics_per_day)
          end
        else
          apply_per_day_rate_limit_for("topics", :max_topics_per_day)
        end
      end
    end
  }

  require_dependency "post"
  Post.class_eval {
    alias_method :orig_limit_posts_per_day, :limit_posts_per_day

    def limit_posts_per_day
      return if !user
      return if (!post_number || post_number <= 1)
      return orig_limit_posts_per_day if !SiteSetting.flexible_rate_limits_enabled

      return default_limit if user.new_user_posting_on_first_day?

      frl = FlexibleRateLimits.new(self.user, self.topic&.category_id) # int or nil

      if frl.category_group_name
        RateLimiter.new(self.user, "cg-#{frl.category_group_name}-post", frl.post_limit, frl.cooldown.minutes.to_i)
      end
    end

    def default_limit
      RateLimiter.new(user, "first-day-replies-per-day", SiteSetting.max_replies_in_first_day, 1.day.to_i)
    end
  }
}