mirror of
https://github.com/discourse/discourse.git
synced 2024-12-16 09:56:39 +08:00
71a1d75d7e
The previous code resulted in NameError: undefined local variable or method `totp' for main:Object I now understand what @tgxworld meant about we should only disable totp when I submitted this before. This is the kind of Ruby stuff that I still don't understand well,(perhaps this isn't the most Ruby way to do this?) but this does what I think is supposed to happen. And it worked just now.
160 lines
4.8 KiB
Ruby
160 lines
4.8 KiB
Ruby
desc "Change topic/post ownership of all the topics/posts by a specific user (without creating new revision)"
|
|
task "users:change_post_ownership", [:old_username, :new_username, :archetype] => [:environment] do |_, args|
|
|
old_username = args[:old_username]
|
|
new_username = args[:new_username]
|
|
archetype = args[:archetype]
|
|
archetype = archetype.downcase if archetype
|
|
|
|
if !old_username || !new_username
|
|
puts "ERROR: Expecting rake users:change_post_ownership[old_username,new_username,archetype]"
|
|
exit 1
|
|
end
|
|
|
|
old_user = find_user(old_username)
|
|
new_user = find_user(new_username)
|
|
|
|
if archetype == "private"
|
|
posts = Post.private_posts.where(user_id: old_user.id)
|
|
elsif archetype == "public" || !archetype
|
|
posts = Post.public_posts.where(user_id: old_user.id)
|
|
else
|
|
puts "ERROR: Expecting rake users:change_post_ownership[old_username,new_username,archetype] where archetype is public or private"
|
|
exit 1
|
|
end
|
|
|
|
puts "Changing post ownership"
|
|
i = 0
|
|
posts.each do |p|
|
|
PostOwnerChanger.new(post_ids: [p.id], topic_id: p.topic.id, new_owner: User.find_by(username_lower: new_user.username_lower), acting_user: User.find_by(username_lower: "system"), skip_revision: true).change_owner!
|
|
putc "."
|
|
i += 1
|
|
end
|
|
puts "", "#{i} posts ownership changed!", ""
|
|
end
|
|
|
|
desc "Merge the source user into the target user"
|
|
task "users:merge", [:source_username, :target_username] => [:environment] do |_, args|
|
|
source_username = args[:source_username]
|
|
target_username = args[:target_username]
|
|
|
|
if !source_username || !target_username
|
|
puts "ERROR: Expecting rake users:merge[source_username,target_username]"
|
|
exit 1
|
|
end
|
|
|
|
source_user = find_user(source_username)
|
|
target_user = find_user(target_username)
|
|
|
|
UserMerger.new(source_user, target_user).merge!
|
|
puts "", "Users merged!", ""
|
|
end
|
|
|
|
desc "Rename a user"
|
|
task "users:rename", [:old_username, :new_username] => [:environment] do |_, args|
|
|
old_username = args[:old_username]
|
|
new_username = args[:new_username]
|
|
|
|
if !old_username || !new_username
|
|
puts "ERROR: Expecting rake users:rename[old_username,new_username]"
|
|
exit 1
|
|
end
|
|
|
|
changer = UsernameChanger.new(find_user(old_username), new_username)
|
|
changer.change(asynchronous: false)
|
|
puts "", "User renamed!", ""
|
|
end
|
|
|
|
desc "Update username in quotes and mentions. Use this if the user was renamed before proper renaming existed."
|
|
task "users:update_posts", [:old_username, :current_username] => [:environment] do |_, args|
|
|
old_username = args[:old_username]
|
|
current_username = args[:current_username]
|
|
|
|
if !old_username || !current_username
|
|
puts "ERROR: Expecting rake users:update_posts[old_username,current_username]"
|
|
exit 1
|
|
end
|
|
|
|
user = find_user(current_username)
|
|
UsernameChanger.update_username(user_id: user.id,
|
|
old_username: old_username,
|
|
new_username: user.username,
|
|
avatar_template: user.avatar_template,
|
|
asynchronous: false)
|
|
|
|
puts "", "Username updated!", ""
|
|
end
|
|
|
|
desc 'Recalculate post and topic counts in user stats'
|
|
task 'users:recalculate_post_counts' => :environment do
|
|
puts '', 'Updating user stats...'
|
|
|
|
filter_public_posts_and_topics = <<~SQL
|
|
p.deleted_at IS NULL
|
|
AND NOT COALESCE(p.hidden, 't')
|
|
AND p.post_type = 1
|
|
AND t.deleted_at IS NULL
|
|
AND COALESCE(t.visible, 't')
|
|
AND t.archetype <> 'private_message'
|
|
AND p.user_id > 0
|
|
SQL
|
|
|
|
puts 'post counts...'
|
|
|
|
# all public replies
|
|
DB.exec <<~SQL
|
|
WITH X AS (
|
|
SELECT p.user_id, COUNT(p.id) post_count
|
|
FROM posts p
|
|
JOIN topics t ON t.id = p.topic_id
|
|
WHERE #{filter_public_posts_and_topics}
|
|
AND p.post_number > 1
|
|
GROUP BY p.user_id
|
|
)
|
|
UPDATE user_stats
|
|
SET post_count = X.post_count
|
|
FROM X
|
|
WHERE user_stats.user_id = X.user_id
|
|
AND user_stats.post_count <> X.post_count
|
|
SQL
|
|
|
|
puts 'topic counts...'
|
|
|
|
# public topics
|
|
DB.exec <<~SQL
|
|
WITH X AS (
|
|
SELECT p.user_id, COUNT(p.id) topic_count
|
|
FROM posts p
|
|
JOIN topics t ON t.id = p.topic_id
|
|
WHERE #{filter_public_posts_and_topics}
|
|
AND p.post_number = 1
|
|
GROUP BY p.user_id
|
|
)
|
|
UPDATE user_stats
|
|
SET topic_count = X.topic_count
|
|
FROM X
|
|
WHERE user_stats.user_id = X.user_id
|
|
AND user_stats.topic_count <> X.topic_count
|
|
SQL
|
|
|
|
puts 'Done!', ''
|
|
end
|
|
|
|
desc "Disable 2FA for user with the given username"
|
|
task "users:disable_2fa", [:username] => [:environment] do |_, args|
|
|
username = args[:username]
|
|
user = find_user(username)
|
|
UserSecondFactor.where(user_id: user.id, method: UserSecondFactor.methods[:totp]).each(&:destroy!)
|
|
puts "2FA disabled for #{username}"
|
|
end
|
|
|
|
def find_user(username)
|
|
user = User.find_by_username(username)
|
|
|
|
if !user
|
|
puts "ERROR: User with username #{username} does not exist"
|
|
exit 1
|
|
end
|
|
|
|
user
|
|
end
|