# frozen_string_literal: true
require "csv"
require File.expand_path(File.dirname(__FILE__) + "/base.rb")
# Отредактируйте константы и метод initialize для ваших данных импорта.
class ImportScripts::Muut < ImportScripts::Base
JSON_FILE_PATH = "/path/to/json/file"
CSV_FILE_PATH = "/path/to/csv/file"
def initialize
super
@imported_users = load_csv
@imported_json = load_json
end
def execute
puts "", "Импорт из Muut..."
import_users
import_categories
import_discussions
puts "", "Готово"
end
def load_json
JSON.parse(repair_json(File.read(JSON_FILE_PATH)))
end
def load_csv
CSV.parse(File.read(CSV_FILE_PATH))
end
def repair_json(arg)
arg.gsub!(/^\(/, "") # содержимое файла окружено скобками ( )
arg.gsub!(/\)$/, "")
arg.gsub!(/\]\]$/, "]") # в конце может быть лишняя ]
arg.gsub!(/\}\{/, "},{") # иногда не хватает запятых!
arg.gsub!("}]{", "},{") # неожиданные квадратные скобки
arg.gsub!("}[{", "},{") # :troll:
arg
end
def import_users
puts '', "Импорт пользователей"
create_users(@imported_users) do |u|
{
id: u[0],
email: u[1],
created_at: Time.now
}
end
end
def import_categories
puts "", "Импорт категорий"
create_categories(@imported_json['categories']) do |category|
{
id: category['path'], # в Muut нет id для категорий, поэтому используем путь
name: category['title'],
slug: category['path']
}
end
end
def import_discussions
puts "", "Импорт обсуждений"
topics = 0
posts = 0
@imported_json['categories'].each do |category|
@imported_json['threads'][category['path']].each do |thread|
next if thread["seed"]["key"] == "skip-this-topic"
mapped = {}
mapped[:id] = "#{thread["seed"]["key"]}-#{thread["seed"]["date"]}"
if thread["seed"]["author"] && user_id_from_imported_user_id(thread["seed"]["author"]["path"]) != ""
mapped[:user_id] = user_id_from_imported_user_id(thread["seed"]["author"]["path"]) || -1
else
mapped[:user_id] = -1
end
# обновить отображаемое имя пользователя
if thread["seed"]["author"] && thread["seed"]["author"]["displayname"] != "" && mapped[:user_id] != -1
user = User.find_by(id: mapped[:user_id])
if user
user.name = thread["seed"]["author"]["displayname"]
user.save!
end
end
mapped[:created_at] = Time.zone.at(thread["seed"]["date"])
mapped[:category] = category_id_from_imported_category_id(thread["seed"]["path"])
mapped[:title] = CGI.unescapeHTML(thread["seed"]["title"])
if thread["seed"]["body"] == ""
thread["seed"]["body"] = " ";
end
mapped[:raw] = process_muut_post_body(thread["seed"]["body"])
mapped[:raw] = CGI.unescapeHTML(thread["seed"]["title"]) if mapped[:raw] == ""
parent_post = create_post(mapped, mapped[:id])
unless parent_post.is_a?(Post)
puts "Ошибка создания темы #{mapped[:id]}. Пропуск."
puts parent_post.inspect
end
# раскомментируйте строку ниже для создания постоянной ссылки
# Permalink.create(url: "#{thread["seed"]["path"]}:#{thread["seed"]["key"]}", topic_id: parent_post.topic_id)
# создать ответы
if thread["replies"].present? && thread["replies"].count > 0
thread["replies"].reverse_each do |post|
if post_id_from_imported_post_id(post["id"])
next # этот пост уже импортирован
end
if post["body"] == ""
post["body"] = " "
end
new_post = create_post({
id: "#{post["key"]}-#{post["date"]}",
topic_id: parent_post.topic_id,
# Измените следующую строку, чтобы получить уникальный user_id из значения author/path.
user_id: user_id_from_imported_user_id(post["author"]["path"]) || -1,
raw: process_muut_post_body(post["body"]),
created_at: Time.zone.at(post["date"])
}, post["id"])
if new_post.is_a?(Post)
posts += 1
else
puts "Ошибка создания поста #{post["id"]}. Пропуск."
puts new_post.inspect
end
end
end
topics += 1
end
end
puts "", "Импортировано #{topics} тем с #{topics + posts} постами."
end
def process_muut_post_body(arg)
raw = arg.dup
raw = raw.to_s
raw = raw[0..-1]
# новая строка
if raw != nil
raw.gsub!(/\\n/, "\n")
# блок кода
raw.gsub!("---", "```\n")
# табуляция
raw.gsub!(/\\t/, ' ')
# двойная кавычка
raw.gsub!(/\\\"/, '"')
raw = CGI.unescapeHTML(raw)
end
raw
end
def file_full_path(relpath)
File.join JSON_FILES_DIR, relpath.split("?").first
end
end
if __FILE__ == $0
ImportScripts::Muut.new.perform
end