# frozen_string_literal: true require "csv" require File.expand_path(File.dirname(__FILE__) + "/base.rb") # Edit the constants and initialize method for your import data. class ImportScripts::JsonGeneric < ImportScripts::Base JSON_FILE_PATH = ENV["JSON_FILE"] BATCH_SIZE ||= 1000 def initialize super @imported_json = load_json end def execute puts "", "Importing from JSON file..." import_users import_discussions puts "", "Done" end def load_json JSON.parse(File.read(JSON_FILE_PATH)) end def username_for(name) result = name.downcase.gsub(/[^a-z0-9\-\_]/, "") result = Digest::SHA1.hexdigest(name)[0...10] if result.blank? result end def import_users puts "", "Importing users" users = [] @imported_json["topics"].each { |t| t["posts"].each { |p| users << p["author"].scrub } } users.uniq! create_users(users) do |u| { id: username_for(u), username: username_for(u), name: u, email: "#{username_for(u)}@example.com", created_at: Time.now, } end end def import_discussions puts "", "Importing discussions" topics = 0 posts = 0 @imported_json["topics"].each do |t| first_post = t["posts"][0] next unless first_post topic = { id: t["id"], user_id: user_id_from_imported_user_id(username_for(first_post["author"])) || -1, raw: first_post["body"], created_at: Time.zone.parse(first_post["date"]), cook_method: Post.cook_methods[:raw_html], title: t["title"], category: ENV["CATEGORY_ID"], custom_fields: { import_id: "pid:#{first_post["id"]}", }, } topic[:pinned_at] = Time.zone.parse(first_post["date"]) if t["pinned"] topics += 1 parent_post = create_post(topic, topic[:id]) t["posts"][1..-1].each do |p| create_post( { id: p["id"], topic_id: parent_post.topic_id, user_id: user_id_from_imported_user_id(username_for(p["author"])) || -1, raw: p["body"], created_at: Time.zone.parse(p["date"]), cook_method: Post.cook_methods[:raw_html], custom_fields: { import_id: "pid:#{p["id"]}", }, }, p["id"], ) posts += 1 end end puts "", "Imported #{topics} topics with #{topics + posts} posts." end end ImportScripts::JsonGeneric.new.perform if __FILE__ == $0