Files
danbooru/app/controllers/reports_controller.rb
evazion f73d2e3956 reports: add ability to group reports by column.
Add ability to group reports by various columns. For example, you can see
the posts by the top 10 uploaders over time, or posts grouped by rating
over time.
2022-10-22 04:05:10 -05:00

115 lines
4.6 KiB
Ruby

# frozen_string_literal: true
class ReportsController < ApplicationController
respond_to :html, :json, :xml
rate_limit :show, rate: 1.0/3.seconds, burst: 15
def index
end
def show
@report = params[:id]
@mode = params.dig(:search, :mode) || "chart"
@period = params.dig(:search, :period)&.downcase || "day"
@from = params.dig(:search, :from) || 1.month.ago
@to = params.dig(:search, :to) || Time.zone.now
@columns = params.dig(:search, :columns).to_s.split(/[[:space:],]/).map(&:to_sym)
@group = params.dig(:search, :group)&.downcase&.tr(" ", "_")
@group_limit = params.dig(:search, :group_limit) || 10
case @report
when "posts"
@model = Post
@title = "Posts Report"
@available_columns = { posts: "COUNT(*)", uploaders: "COUNT(distinct uploader_id)" }
@available_groups = %w[uploader approver rating is_deleted]
when "post_votes"
@model = PostVote
@title = "Post Votes Report"
@available_columns = { votes: "COUNT(*)", posts: "COUNT(distinct post_id)", voters: "COUNT(distinct user_id)" }
@available_groups = %w[]
when "pools"
@model = Pool
@title = "Pools Report"
@available_columns = { pools: "COUNT(*)" }
@available_groups = %w[category is_deleted]
when "comments"
@model = Comment
@title = "Comments Report"
@available_columns = { comments: "COUNT(*)", commenters: "COUNT(distinct creator_id)" }
@available_groups = %w[creator do_not_bump_post is_deleted is_sticky]
when "comment_votes"
@model = CommentVote
@title = "Comment Votes Report"
@available_columns = { votes: "COUNT(*)", comments: "COUNT(distinct comment_id)", voters: "COUNT(distinct user_id)" }
@available_groups = %w[]
when "forum_posts"
@model = ForumPost
@title = "Forum Posts Report"
@available_columns = { forum_posts: "COUNT(*)", posters: "COUNT(distinct creator_id)" }
@available_groups = %w[creator is_deleted]
when "bulk_update_requests"
@model = BulkUpdateRequest
@title = "Bulk Update Requests Report"
@available_columns = { requests: "COUNT(*)", requestors: "COUNT(distinct user_id)" }
@available_groups = %w[user approver status]
when "tag_aliases"
@model = TagAlias
@title = "Tag Aliases Report"
@available_columns = { aliases: "COUNT(*)" }
@available_groups = %w[status approver]
when "tag_implications"
@model = TagImplication
@title = "Tag Implications Report"
@available_columns = { aliases: "COUNT(*)" }
@available_groups = %w[status approver]
when "artist_versions"
@model = ArtistVersion
@title = "Artist Edits Report"
@available_columns = { artist_edits: "COUNT(*)", artists: "COUNT(distinct artist_id)", editors: "COUNT(distinct updater_id)" }
@available_groups = %w[updater]
when "note_versions"
@model = NoteVersion
@title = "Note Edits Report"
@available_columns = { note_edits: "COUNT(*)", posts: "COUNT(distinct post_id)", editors: "COUNT(distinct updater_id)" }
@available_groups = %w[updater]
when "wiki_page_versions"
@model = WikiPageVersion
@title = "Wiki Edits Report"
@available_columns = { wiki_edits: "COUNT(*)", editors: "COUNT(distinct updater_id)" }
@available_groups = %w[updater]
when "users"
@model = User
@title = "New Users Report"
@available_columns = { users: "COUNT(*)" }
@available_groups = %w[level]
when "bans"
@model = Ban
@title = "Bans Report"
@available_columns = { bans: "COUNT(*)", banners: "COUNT(DISTINCT banner_id)" }
@available_groups = %w[banner duration]
else
raise ActiveRecord::RecordNotFound
end
if CurrentUser.user.is_member? && CurrentUser.user.statement_timeout < 10_000
@statement_timeout = 10_000
else
@statement_timeout = CurrentUser.user.statement_timeout
end
ApplicationRecord.set_timeout(@statement_timeout) do
@group = nil unless @group&.in?(@available_groups)
@columns = @available_columns.slice(*@columns)
@columns = [@available_columns.first].to_h if @columns.blank?
@dataframe = @model.search(params[:search], CurrentUser.user).timeseries(period: @period, from: @from, to: @to, groups: [@group].compact_blank, group_limit: @group_limit, columns: @columns)
@dataframe["date"] = @dataframe["date"].map(&:to_date)
@dataframe[@group] = @dataframe[@group].map(&:pretty_name) if @group.in?(%w[creator updater uploader banner approver user])
@dataframe = @dataframe.crosstab("date", @group) if @group
end
respond_with(@dataframe)
end
end