60 lines
1.9 KiB
Ruby
60 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
include HasSanitizableFields
|
|
|
|
module ::DiscourseDataExplorer
|
|
class ResultToMarkdown
|
|
def self.convert(pg_result, render_url_columns = false)
|
|
relations, colrender = DataExplorer.add_extra_data(pg_result)
|
|
result_data = []
|
|
|
|
# column names to search in place of id columns (topic_id, user_id etc)
|
|
cols = %w[name title username]
|
|
|
|
# find values from extra data, based on result id
|
|
pg_result.values.each do |row|
|
|
row_data = []
|
|
|
|
row.each_with_index do |col, col_index|
|
|
col_name = pg_result.fields[col_index]
|
|
col_render = colrender[col_index]
|
|
related = relations.dig(col_render.to_sym) if col_render.present?
|
|
|
|
if related.is_a?(ActiveModel::ArraySerializer)
|
|
related_row = related.object.find_by(id: col)
|
|
if col_name.include?("_id")
|
|
column = cols.find { |c| related_row.try c }
|
|
else
|
|
column = related_row.try(col_name)
|
|
end
|
|
|
|
if column.nil?
|
|
row_data[col_index] = col
|
|
else
|
|
row_data[col_index] = "#{related_row[column]} (#{col})"
|
|
end
|
|
elsif col_render == "url" && render_url_columns
|
|
url, text = guess_url(col)
|
|
row_data[col_index] = "[#{text}](#{url})"
|
|
else
|
|
row_data[col_index] = col
|
|
end
|
|
end
|
|
|
|
result_data << row_data.map { |c| "| #{sanitize_field(c.to_s)} " }.join + "|\n"
|
|
end
|
|
|
|
table_headers = pg_result.fields.map { |c| " #{c.gsub("_id", "")} |" }.join
|
|
table_body = pg_result.fields.size.times.map { " :----- |" }.join
|
|
|
|
"|#{table_headers}\n|#{table_body}\n#{result_data.join}"
|
|
end
|
|
|
|
def self.guess_url(column_value)
|
|
text, url = column_value.split(/,(.+)/)
|
|
|
|
[url || column_value, text || column_value]
|
|
end
|
|
end
|
|
end
|