Add post regenerations

app/controllers/post_regenerations_controller.rb

@@ -0,0 +1,18 @@
+class PostRegenerationsController < ApplicationController
+  respond_to :html, :xml, :json, :js
+
+  def create
+    @post_regeneration = authorize PostRegeneration.new(creator: CurrentUser.user, **permitted_attributes(PostRegeneration))
+    @post_regeneration.execute_category_action!
+    @post_regeneration.save
+
+    respond_with(@post_regeneration, location: @post_regeneration.post)
+  end
+
+  def index
+    @post_regenerations = authorize PostRegeneration.paginated_search(params)
+    @post_regenerations = @post_regenerations.includes(:creator, :post) if request.format.html?
+
+    respond_with(@post_regenerations)
+  end
+end

app/logical/upload_service/utils.rb

@@ -32,6 +32,22 @@ class UploadService
       [preview_file, crop_file, sample_file]
     end
 
+    def process_resizes(upload, file, original_post_id, media_file: nil)
+      media_file ||= upload.media_file
+      preview_file, crop_file, sample_file = Utils.generate_resizes(media_file)
+
+      begin
+        Utils.distribute_files(file, upload, :original, original_post_id: original_post_id) if file.present?
+        Utils.distribute_files(sample_file, upload, :large, original_post_id: original_post_id) if sample_file.present?
+        Utils.distribute_files(preview_file, upload, :preview, original_post_id: original_post_id) if preview_file.present?
+        Utils.distribute_files(crop_file, upload, :crop, original_post_id: original_post_id) if crop_file.present?
+      ensure
+        preview_file.try(:close!)
+        crop_file.try(:close!)
+        sample_file.try(:close!)
+      end
+    end
+
     def process_file(upload, file, original_post_id: nil)
       upload.file = file
       media_file = upload.media_file
@@ -45,18 +61,7 @@ class UploadService
       upload.validate!(:file)
       upload.tag_string = "#{upload.tag_string} #{Utils.automatic_tags(media_file)}"
 
-      preview_file, crop_file, sample_file = Utils.generate_resizes(media_file)
-
-      begin
-        Utils.distribute_files(file, upload, :original, original_post_id: original_post_id)
-        Utils.distribute_files(sample_file, upload, :large, original_post_id: original_post_id) if sample_file.present?
-        Utils.distribute_files(preview_file, upload, :preview, original_post_id: original_post_id) if preview_file.present?
-        Utils.distribute_files(crop_file, upload, :crop, original_post_id: original_post_id) if crop_file.present?
-      ensure
-        preview_file.try(:close!)
-        crop_file.try(:close!)
-        sample_file.try(:close!)
-      end
+      process_resizes(upload, file, original_post_id)
     end
 
     def automatic_tags(media_file)

app/models/post_regeneration.rb

@@ -0,0 +1,31 @@
+class PostRegeneration < ApplicationRecord
+  belongs_to :creator, :class_name => "User"
+  belongs_to :post
+
+  validates :category, inclusion: %w[iqdb resizes]
+
+  module SearchMethods
+    def search(params)
+      q = search_attributes(params, :id, :created_at, :updated_at, :category, :creator, :post)
+      q.apply_default_order(params)
+    end
+  end
+
+  extend SearchMethods
+
+  def execute_category_action!
+    if category == "iqdb"
+      post.update_iqdb_async
+    elsif category == "resizes"
+      media_file = MediaFile.open(post.file, frame_data: post.pixiv_ugoira_frame_data)
+      UploadService::Utils.process_resizes(post, nil, post.id, media_file: media_file)
+    else
+      # should never happen
+      raise Error, "Unknown category: #{category}"
+    end
+  end
+
+  def self.searchable_includes
+    [:creator, :post]
+  end
+end

app/policies/post_regeneration_policy.rb

@@ -0,0 +1,9 @@
+class PostRegenerationPolicy < ApplicationPolicy
+  def create?
+    user.is_moderator?
+  end
+
+  def permitted_attributes_for_create
+    [:post_id, :category]
+  end
+end

app/views/post_regenerations/create.js.erb

@@ -0,0 +1,5 @@
+<% if @post_regeneration.errors.any? %>
+  Danbooru.error("<%= j @post_regeneration.errors.full_messages.join(',') %>");
+<% else %>
+  Danbooru.notice("Post regenerated");
+<% end %>

app/views/post_regenerations/index.html.erb

@@ -0,0 +1,17 @@
+<div id="c-post-regenerations">
+  <div id="a-index">
+    <h1>Post regenerations</h1>
+    <%= table_for @post_regenerations, width: "100%" do |t| %>
+      <% t.column "Post", width: "1%" do |regeneration| %>
+        <%= PostPresenter.preview(regeneration.post, show_deleted: true) %>
+      <% end %>
+      <% t.column :category %>
+      <% t.column "Creator", width: "10%" do |regeneration| %>
+        <%= compact_time regeneration.created_at %>
+        <br> by <%= link_to_user regeneration.creator %>
+      <% end %>
+    <% end %>
+
+    <%= numbered_paginator(@post_regenerations) %>
+  </div>
+</div>

app/views/posts/partials/show/_options.html.erb

@@ -86,4 +86,9 @@
         <li id="post-option-replace-image"><%= link_to "Replace image", new_post_replacement_path(post_id: post.id), remote: true %></li>
       <% end %>
     <% end %>
+
+    <% if policy(PostRegeneration).create? %>
+      <li id="post-option-regenerate-iqdb"><%= link_to "Regenerate IQDB", post_regenerations_path(post_regeneration: {post_id: post.id, category: "iqdb"}), remote: true, method: :post %></li>
+      <li id="post-option-regenerate-preview"><%= link_to "Regenerate image sizes", post_regenerations_path(post_regeneration: {post_id: post.id, category: "resizes"}), remote: true, method: :post %></li>
+    <% end %>
 </ul>

app/views/posts/show.html.erb

@@ -33,6 +33,7 @@
       <li id="post-history-moderation"><%= link_to "Moderation", post_events_path(@post.id) %></li>
       <li id="post-history-commentary"><%= link_to "Commentary", artist_commentary_versions_path(search: { post_id: @post.id }) %></li>
       <li id="post-history-replacements"><%= link_to "Replacements", post_replacements_path(search: {post_id: @post.id }) %></li>
+      <li id="post-history-regenerations"><%= link_to "Regenerations", post_regenerations_path(search: {post_id: @post.id }) %></li>
     </ul>
   </section>
 <% end %>