mirror of
https://github.com/whippyshou/mastodon
synced 2024-12-30 22:48:33 +09:00
fcc4c9b34a
* Change domain block CSV parsing to be more robust and handle more lists * Add some tests * Improve domain block import validation and reporting
78 lines
2.7 KiB
Ruby
78 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'csv'
|
|
|
|
module Admin
|
|
class ExportDomainBlocksController < BaseController
|
|
include AdminExportControllerConcern
|
|
|
|
before_action :set_dummy_import!, only: [:new]
|
|
|
|
def new
|
|
authorize :domain_block, :create?
|
|
end
|
|
|
|
def export
|
|
authorize :instance, :index?
|
|
send_export_file
|
|
end
|
|
|
|
def import
|
|
authorize :domain_block, :create?
|
|
|
|
@import = Admin::Import.new(import_params)
|
|
return render :new unless @import.validate
|
|
|
|
@global_private_comment = I18n.t('admin.export_domain_blocks.import.private_comment_template', source: @import.data_file_name, date: I18n.l(Time.now.utc))
|
|
|
|
@form = Form::DomainBlockBatch.new
|
|
@domain_blocks = @import.csv_rows.filter_map do |row|
|
|
domain = row['#domain'].strip
|
|
next if DomainBlock.rule_for(domain).present?
|
|
|
|
domain_block = DomainBlock.new(domain: domain,
|
|
severity: row.fetch('#severity', :suspend),
|
|
reject_media: row.fetch('#reject_media', false),
|
|
reject_reports: row.fetch('#reject_reports', false),
|
|
private_comment: @global_private_comment,
|
|
public_comment: row['#public_comment'],
|
|
obfuscate: row.fetch('#obfuscate', false))
|
|
|
|
if domain_block.invalid?
|
|
flash.now[:alert] = I18n.t('admin.export_domain_blocks.invalid_domain_block', error: domain_block.errors.full_messages.join(', '))
|
|
next
|
|
end
|
|
|
|
domain_block
|
|
rescue ArgumentError => e
|
|
flash.now[:alert] = I18n.t('admin.export_domain_blocks.invalid_domain_block', error: e.message)
|
|
next
|
|
end
|
|
|
|
@warning_domains = Instance.where(domain: @domain_blocks.map(&:domain)).where('EXISTS (SELECT 1 FROM follows JOIN accounts ON follows.account_id = accounts.id OR follows.target_account_id = accounts.id WHERE accounts.domain = instances.domain)').pluck(:domain)
|
|
rescue ActionController::ParameterMissing
|
|
flash.now[:alert] = I18n.t('admin.export_domain_blocks.no_file')
|
|
set_dummy_import!
|
|
render :new
|
|
end
|
|
|
|
private
|
|
|
|
def export_filename
|
|
'domain_blocks.csv'
|
|
end
|
|
|
|
def export_headers
|
|
%w(#domain #severity #reject_media #reject_reports #public_comment #obfuscate)
|
|
end
|
|
|
|
def export_data
|
|
CSV.generate(headers: export_headers, write_headers: true) do |content|
|
|
DomainBlock.with_limitations.each do |instance|
|
|
content << [instance.domain, instance.severity, instance.reject_media, instance.reject_reports, instance.public_comment, instance.obfuscate]
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|