83 lines
2.5 KiB
Ruby
83 lines
2.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module ::DiscourseDataExplorer
|
|
class ReportGenerator
|
|
def self.generate(query_id, query_params, recipients, opts = {})
|
|
query = DiscourseDataExplorer::Query.find(query_id)
|
|
return [] if !query || recipients.empty?
|
|
|
|
recipients = filter_recipients_by_query_access(recipients, query)
|
|
params = params_to_hash(query_params)
|
|
|
|
result = DataExplorer.run_query(query, params)[:pg_result]
|
|
query.update!(last_run_at: Time.now)
|
|
|
|
return [] if opts[:skip_empty] && result.values.empty?
|
|
table = ResultToMarkdown.convert(result)
|
|
|
|
build_report_pms(query, table, recipients)
|
|
end
|
|
|
|
private
|
|
|
|
def self.params_to_hash(query_params)
|
|
params = JSON.parse(query_params)
|
|
params_hash = {}
|
|
|
|
if !params.blank?
|
|
param_key, param_value = [], []
|
|
params.flatten.each.with_index do |data, i|
|
|
if i % 2 == 0
|
|
param_key << data
|
|
else
|
|
param_value << data
|
|
end
|
|
end
|
|
|
|
params_hash = Hash[param_key.zip(param_value)]
|
|
end
|
|
|
|
params_hash
|
|
end
|
|
|
|
def self.build_report_pms(query, table = "", targets = [])
|
|
pms = []
|
|
targets.each do |target|
|
|
name = target[0]
|
|
pm_type = "target_#{target[1]}s"
|
|
|
|
pm = {}
|
|
pm["title"] = "Scheduled Report for #{query.name}"
|
|
pm[pm_type] = Array(name)
|
|
pm["raw"] = "Hi #{name}, your data explorer report is ready.\n\n" +
|
|
"Query Name:\n#{query.name}\n\nHere are the results:\n#{table}\n\n" +
|
|
"<a href='#{Discourse.base_url}/admin/plugins/explorer?id=#{query.id}'>View query in Data Explorer</a>\n\n" +
|
|
"Report created at #{Time.zone.now.strftime("%Y-%m-%d at %H:%M:%S")} (#{Time.zone.name})"
|
|
pms << pm
|
|
end
|
|
pms
|
|
end
|
|
|
|
def self.filter_recipients_by_query_access(recipients, query)
|
|
users = User.where(username: recipients)
|
|
groups = Group.where(name: recipients)
|
|
emails = recipients - users.pluck(:username) - groups.pluck(:name)
|
|
result = []
|
|
|
|
users.each do |user|
|
|
result << [user.username, "username"] if Guardian.new(user).user_can_access_query?(query)
|
|
end
|
|
|
|
groups.each do |group|
|
|
if group.id == Group::AUTO_GROUPS[:admins] || query.query_groups.exists?(group_id: group.id)
|
|
result << [group.name, "group_name"]
|
|
end
|
|
end
|
|
|
|
emails.each { |email| result << [email, "email"] if Email.is_valid?(email) }
|
|
|
|
result
|
|
end
|
|
end
|
|
end
|