forked from e621ng/e621ng
98 lines
2.7 KiB
Ruby
98 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "digest/md5"
|
|
require "net/http"
|
|
require "tempfile"
|
|
|
|
unless Rails.env.test?
|
|
puts "== Creating elasticsearch indices ==\n"
|
|
|
|
Post.__elasticsearch__.create_index!
|
|
end
|
|
|
|
puts "== Seeding database with sample content ==\n"
|
|
|
|
# Uncomment to see detailed logs
|
|
# ActiveRecord::Base.logger = ActiveSupport::Logger.new($stdout)
|
|
|
|
admin = User.find_or_create_by!(name: "admin") do |user|
|
|
user.created_at = 2.weeks.ago
|
|
user.password = "e621test"
|
|
user.password_confirmation = "e621test"
|
|
user.password_hash = ""
|
|
user.email = "admin@e621.net"
|
|
user.can_upload_free = true
|
|
user.can_approve_posts = true
|
|
user.level = User::Levels::ADMIN
|
|
end
|
|
|
|
User.find_or_create_by!(name: Danbooru.config.system_user) do |user|
|
|
user.password = "ae3n4oie2n3oi4en23oie4noienaorshtaioresnt"
|
|
user.password_confirmation = "ae3n4oie2n3oi4en23oie4noienaorshtaioresnt"
|
|
user.password_hash = ""
|
|
user.email = "system@e621.net"
|
|
user.can_upload_free = true
|
|
user.can_approve_posts = true
|
|
user.level = User::Levels::JANITOR
|
|
end
|
|
|
|
ForumCategory.find_or_create_by!(id: Danbooru.config.alias_implication_forum_category) do |category|
|
|
category.name = "Tag Alias and Implication Suggestions"
|
|
category.can_view = 0
|
|
end
|
|
|
|
def api_request(path)
|
|
response = HTTParty.get("https://e621.net#{path}", {
|
|
headers: { "User-Agent" => "e621ng/seeding" },
|
|
})
|
|
JSON.parse(response.body)
|
|
end
|
|
|
|
def import_posts
|
|
resources = YAML.load_file Rails.root.join("db/seeds.yml")
|
|
json = api_request("/posts.json?limit=#{ENV.fetch('SEED_POST_COUNT', 100)}&tags=id:#{resources['post_ids'].join(',')}")
|
|
|
|
json["posts"].each do |post|
|
|
puts post["file"]["url"]
|
|
|
|
post["tags"].each do |category, tags|
|
|
Tag.find_or_create_by_name_list(tags.map { |tag| "#{category}:#{tag}" })
|
|
end
|
|
|
|
service = UploadService.new({
|
|
uploader: CurrentUser.user,
|
|
uploader_ip_addr: CurrentUser.ip_addr,
|
|
direct_url: post["file"]["url"],
|
|
tag_string: post["tags"].values.flatten.join(" "),
|
|
source: post["sources"].join("\n"),
|
|
description: post["description"],
|
|
rating: post["rating"],
|
|
})
|
|
|
|
service.start!
|
|
end
|
|
end
|
|
|
|
def import_mascots
|
|
api_request("/mascots.json").each do |mascot|
|
|
puts mascot["url_path"]
|
|
Mascot.create!(
|
|
creator: CurrentUser.user,
|
|
mascot_file: Downloads::File.new(mascot["url_path"]).download!,
|
|
display_name: mascot["display_name"],
|
|
background_color: mascot["background_color"],
|
|
artist_url: mascot["artist_url"],
|
|
artist_name: mascot["artist_name"],
|
|
safe_mode_only: mascot["safe_mode_only"],
|
|
active: mascot["active"],
|
|
)
|
|
end
|
|
end
|
|
|
|
unless Rails.env.test?
|
|
CurrentUser.user = admin
|
|
CurrentUser.ip_addr = "127.0.0.1"
|
|
import_posts
|
|
import_mascots
|
|
end
|