From e5010bebf9c3c822bfb3e15b8f5bfe31fe847d24 Mon Sep 17 00:00:00 2001
From: Albert Yi <albert@carbonfive.com>
Date: Wed, 5 Sep 2018 13:49:56 -0700
Subject: [PATCH] allow reportbooru to update post counts

---
 app/controllers/related_tags_controller.rb | 1 +
 app/models/tag.rb                          | 1 +
 2 files changed, 2 insertions(+)

diff --git a/app/controllers/related_tags_controller.rb b/app/controllers/related_tags_controller.rb
index 30f436b61..8acd30756 100644
--- a/app/controllers/related_tags_controller.rb
+++ b/app/controllers/related_tags_controller.rb
@@ -16,6 +16,7 @@ class RelatedTagsController < ApplicationController
     @tag = Tag.find_by_name(params[:name])
     @tag.related_tags = params[:related_tags]
     @tag.related_tags_updated_at = Time.now
+    @tag.post_count = params[:post_count] if params[:post_count].present?
     @tag.save
     head :ok
   end
diff --git a/app/models/tag.rb b/app/models/tag.rb
index 97e32b061..891ce3db8 100644
--- a/app/models/tag.rb
+++ b/app/models/tag.rb
@@ -814,6 +814,7 @@ class Tag < ApplicationRecord
         self.related_tags = RelatedTagCalculator.calculate_from_sample_to_array(name).join(" ")
       end
       self.related_tags_updated_at = Time.now
+      fix_post_count if rand(post_count) <= 1
       save
     rescue ActiveRecord::StatementInvalid
     end