mirror of
https://github.com/discourse/discourse.git
synced 2024-11-28 15:35:22 +08:00
194 lines
4.8 KiB
Ruby
194 lines
4.8 KiB
Ruby
# `dropdb bbpress`
|
|
# `createdb bbpress`
|
|
# `bundle exec rake db:migrate`
|
|
|
|
BB_PRESS_DB = "import"
|
|
|
|
require 'mysql2'
|
|
|
|
@client = Mysql2::Client.new(
|
|
host: "localhost",
|
|
username: "root",
|
|
password: "password",
|
|
:database => BB_PRESS_DB
|
|
)
|
|
|
|
|
|
require File.expand_path(File.dirname(__FILE__) + "/../../config/environment")
|
|
SiteSetting.email_domains_blacklist = ''
|
|
RateLimiter.disable
|
|
|
|
def create_admin
|
|
User.new.tap { |admin|
|
|
admin.email = "sam.saffron@gmail.com"
|
|
admin.username = "sam"
|
|
admin.password = SecureRandom.uuid
|
|
admin.save
|
|
admin.grant_admin!
|
|
admin.change_trust_level!(:regular)
|
|
admin.email_tokens.update_all(confirmed: true)
|
|
}
|
|
end
|
|
|
|
def create_user(opts, import_id)
|
|
opts[:name] = User.suggest_name(opts[:name] || opts[:email])
|
|
opts[:username] = UserNameSuggester.suggest(opts[:username] || opts[:name] || opts[:email])
|
|
opts[:email] = opts[:email].downcase
|
|
|
|
u = User.new(opts)
|
|
u.custom_fields["import_id"] = import_id
|
|
|
|
u.save!
|
|
u
|
|
|
|
rescue
|
|
# try based on email
|
|
u = User.find_by(email: opts[:email].downcase)
|
|
u.custom_fields["import_id"] = import_id
|
|
u.save!
|
|
u
|
|
end
|
|
|
|
|
|
def create_post(opts)
|
|
|
|
user = User.find(opts[:user_id])
|
|
opts = opts.merge(skip_validations: true)
|
|
|
|
PostCreator.create(user, opts)
|
|
end
|
|
|
|
|
|
results = @client.query("
|
|
select ID,
|
|
user_login username,
|
|
display_name name,
|
|
user_url website,
|
|
user_email email,
|
|
user_registered created_at
|
|
from wp_users where spam = 0 and deleted = 0").to_a
|
|
|
|
|
|
users = {}
|
|
|
|
UserCustomField.where(name: 'import_id')
|
|
.pluck(:user_id, :value)
|
|
.each do |user_id, import_id|
|
|
users[import_id.to_i] = user_id
|
|
end
|
|
|
|
skipped = 0
|
|
results.delete_if do |u|
|
|
skipped+= 1 if users[u["ID"]]
|
|
end
|
|
|
|
puts "Importing #{results.length} users (skipped #{skipped})"
|
|
|
|
i = 0
|
|
results.each do |u|
|
|
putc "." if ((i+=1)%10) == 0
|
|
|
|
id = u.delete("ID")
|
|
users[id] = create_user(ActiveSupport::HashWithIndifferentAccess.new(u), id).id
|
|
end
|
|
|
|
|
|
|
|
results = @client.query("
|
|
select ID, post_name from wp_posts where post_type = 'forum'
|
|
").to_a
|
|
|
|
categories={}
|
|
|
|
CategoryCustomField.where(name: 'import_id')
|
|
.pluck(:category_id, :value)
|
|
.each do |category_id, import_id|
|
|
categories[import_id.to_i] = category_id
|
|
end
|
|
|
|
|
|
skipped = 0
|
|
results.delete_if do |u|
|
|
skipped+= 1 if categories[u["ID"]]
|
|
end
|
|
|
|
puts
|
|
puts "Importing #{results.length} categories (skipped #{skipped})"
|
|
|
|
results.each do |c|
|
|
c["post_name"] = "unknown" if c["post_name"].blank?
|
|
category = Category.new(name: c["post_name"], user_id: -1)
|
|
category.custom_fields["import_id"] = c["ID"]
|
|
category.save!
|
|
categories[c["ID"]] = category.id
|
|
end
|
|
|
|
results = @client.query("
|
|
select ID,
|
|
post_author,
|
|
post_date,
|
|
post_content,
|
|
post_title,
|
|
post_type,
|
|
post_parent
|
|
from wp_posts
|
|
where post_status <> 'spam'
|
|
and post_type in ('topic', 'reply')
|
|
order by ID
|
|
").to_a
|
|
|
|
posts={}
|
|
|
|
PostCustomField.where(name: 'import_id')
|
|
.pluck(:post_id, :value)
|
|
.each do |post_id, import_id|
|
|
posts[import_id.to_i] = post_id
|
|
end
|
|
|
|
|
|
skipped = 0
|
|
results.delete_if do |u|
|
|
skipped+= 1 if posts[u["ID"]]
|
|
end
|
|
|
|
puts "Importing #{results.length} posts (skipped #{skipped})"
|
|
|
|
topic_lookup = {}
|
|
Post.pluck(:id, :topic_id, :post_number).each do |p,t,n|
|
|
topic_lookup[p] = {topic_id: t, post_number: n}
|
|
end
|
|
|
|
i = 0
|
|
results.each do |post|
|
|
putc "." if ((i+=1)%10) == 0
|
|
|
|
mapped = {}
|
|
|
|
mapped[:user_id] = users[post["post_author"]]
|
|
mapped[:raw] = post["post_content"]
|
|
mapped[:created_at] = post["post_date"]
|
|
|
|
if post["post_type"] == "topic"
|
|
mapped[:category] = categories[post["post_parent"]]
|
|
mapped[:title] = CGI.unescapeHTML post["post_title"]
|
|
else
|
|
parent_id = posts[post["post_parent"]]
|
|
parent = topic_lookup[parent_id]
|
|
unless parent
|
|
puts; puts "Skipping #{post["ID"]}: #{post["post_content"][0..40]}"
|
|
next
|
|
end
|
|
mapped[:topic_id] = parent[:topic_id]
|
|
mapped[:reply_to_post_number] = parent[:post_number] if parent[:post_number] > 1
|
|
end
|
|
|
|
mapped[:custom_fields] = {import_id: post["ID"]}
|
|
|
|
d_post = create_post(mapped)
|
|
posts[post["ID"]] = d_post.id
|
|
topic_lookup[d_post.id] = {post_number: d_post.post_number, topic_id: d_post.topic_id}
|
|
|
|
end
|
|
|
|
Post.exec_sql("update topics t set bumped_at = (select max(created_at) from posts where topic_id = t.id)")
|