discourse/plugins/poll/lib/tasks/migrate_old_polls.rake

49 lines
1.7 KiB
Ruby

desc "Migrate old polls to new syntax"
task "poll:migrate_old_polls" => :environment do
require "timecop"
# iterate over all polls
PluginStoreRow.where(plugin_name: "poll")
.where("key LIKE 'poll_options_%'")
.pluck(:key)
.each do |poll_options_key|
# extract the post_id
post_id = poll_options_key["poll_options_".length..-1].to_i
# load the post from the db
if post = Post.find_by(id: post_id)
putc "."
# go back in time
Timecop.freeze(post.created_at + 1.minute) do
# fix the RAW when needed
post.raw << "\n\n"
if post.raw !~ /\[poll\]/
lists = /^[ ]*- .+?$\n\n/m.match(post.raw)
next if lists.blank? || lists.length == 0
first_list = lists[0]
post.raw = post.raw.sub(first_list, "\n[poll]\n#{first_list.strip}\n[/poll]\n")
end
# save the poll
post.save
# retrieve the new options
options = post.custom_fields["polls"]["poll"]["options"]
# iterate over all votes
PluginStoreRow.where(plugin_name: "poll")
.where("key LIKE 'poll_vote_#{post_id}_%'")
.pluck(:key, :value)
.each do |poll_vote_key, vote|
# extract the user_id
user_id = poll_vote_key["poll_vote_#{post_id}_%".length..-1].to_i
# find the selected option
vote = vote.strip
selected_option = options.detect { |o| o["html"].strip === vote }
# make sure we have a match
next if selected_option.blank?
# submit vote
DiscoursePoll::Poll.vote(post_id, "poll", [selected_option["id"]], user_id) rescue nil
end
end
end
end
puts "", "Done!"
end