mirror of
https://github.com/discourse/discourse.git
synced 2024-12-19 18:04:45 +08:00
25a226279a
The #pluck_first freedom patch, first introduced by @danielwaterworth has served us well, and is used widely throughout both core and plugins. It seems to have been a common enough use case that Rails 6 introduced it's own method #pick with the exact same implementation. This allows us to retire the freedom patch and switch over to the built-in ActiveRecord method. There is no replacement for #pluck_first!, but a quick search shows we are using this in a very limited capacity, and in some cases incorrectly (by assuming a nil return rather than an exception), which can quite easily be replaced with #pick plus some extra handling.
98 lines
3.1 KiB
Ruby
98 lines
3.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module ImportScripts
|
|
class LookupContainer
|
|
def initialize
|
|
puts "Loading existing groups..."
|
|
@groups = GroupCustomField.where(name: "import_id").pluck(:value, :group_id).to_h
|
|
|
|
puts "Loading existing users..."
|
|
@users = UserCustomField.where(name: "import_id").pluck(:value, :user_id).to_h
|
|
|
|
puts "Loading existing categories..."
|
|
@categories = CategoryCustomField.where(name: "import_id").pluck(:value, :category_id).to_h
|
|
|
|
puts "Loading existing posts..."
|
|
@posts = PostCustomField.where(name: "import_id").pluck(:value, :post_id).to_h
|
|
|
|
puts "Loading existing topics..."
|
|
@topics = {}
|
|
Post
|
|
.joins(:topic)
|
|
.pluck("posts.id, posts.topic_id, posts.post_number, topics.slug")
|
|
.each do |p|
|
|
@topics[p[0]] = { topic_id: p[1], post_number: p[2], url: Post.url(p[3], p[1], p[2]) }
|
|
end
|
|
end
|
|
|
|
# Get the Discourse Post id based on the id of the source record
|
|
def post_id_from_imported_post_id(import_id)
|
|
@posts[import_id] || @posts[import_id.to_s]
|
|
end
|
|
|
|
# Get the Discourse topic info (a hash) based on the id of the source record
|
|
def topic_lookup_from_imported_post_id(import_id)
|
|
post_id = post_id_from_imported_post_id(import_id)
|
|
post_id ? @topics[post_id] : nil
|
|
end
|
|
|
|
# Get the Discourse Group id based on the id of the source group
|
|
def group_id_from_imported_group_id(import_id)
|
|
@groups[import_id] || @groups[import_id.to_s]
|
|
end
|
|
|
|
# Get the Discourse Group based on the id of the source group
|
|
def find_group_by_import_id(import_id)
|
|
GroupCustomField.where(name: "import_id", value: import_id.to_s).first.try(:group)
|
|
end
|
|
|
|
# Get the Discourse User id based on the id of the source user
|
|
def user_id_from_imported_user_id(import_id)
|
|
@users[import_id] || @users[import_id.to_s]
|
|
end
|
|
|
|
# Get the Discourse User based on the id of the source user
|
|
def find_user_by_import_id(import_id)
|
|
UserCustomField.where(name: "import_id", value: import_id.to_s).first.try(:user)
|
|
end
|
|
|
|
def find_username_by_import_id(import_id)
|
|
user_id = user_id_from_imported_user_id(import_id)
|
|
User.where(id: user_id).pick(:username) if user_id.present?
|
|
end
|
|
|
|
# Get the Discourse Category id based on the id of the source category
|
|
def category_id_from_imported_category_id(import_id)
|
|
@categories[import_id] || @categories[import_id.to_s]
|
|
end
|
|
|
|
def add_group(import_id, group)
|
|
@groups[import_id.to_s] = group.id
|
|
end
|
|
|
|
def add_user(import_id, user)
|
|
@users[import_id.to_s] = user.id
|
|
end
|
|
|
|
def add_category(import_id, category)
|
|
@categories[import_id.to_s] = category.id
|
|
end
|
|
|
|
def add_post(import_id, post)
|
|
@posts[import_id.to_s] = post.id
|
|
end
|
|
|
|
def add_topic(post)
|
|
@topics[post.id] = { post_number: post.post_number, topic_id: post.topic_id, url: post.url }
|
|
end
|
|
|
|
def user_already_imported?(import_id)
|
|
@users.has_key?(import_id) || @users.has_key?(import_id.to_s)
|
|
end
|
|
|
|
def post_already_imported?(import_id)
|
|
@posts.has_key?(import_id) || @posts.has_key?(import_id.to_s)
|
|
end
|
|
end
|
|
end
|