discourse/lib/seed_data/categories.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

210 lines
5.9 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
module SeedData
class Categories
def self.with_default_locale
SeedData::Categories.new(SiteSetting.default_locale)
end
def initialize(locale)
@locale = locale
end
def create(site_setting_names: nil)
I18n.with_locale(@locale) do
categories(site_setting_names).each { |params| create_category(**params) }
end
end
def update(site_setting_names: nil, skip_changed: false)
I18n.with_locale(@locale) do
categories(site_setting_names).each do |params|
params.slice!(:site_setting_name, :name, :description)
params[:skip_changed] = skip_changed
2021-09-27 08:45:05 -04:00
update_category(**params)
end
end
end
def reseed_options
I18n.with_locale(@locale) do
categories
.map do |params|
category = find_category(params[:site_setting_name])
next unless category
{ id: params[:site_setting_name], name: category.name, selected: unchanged?(category) }
end
.compact
end
end
private
def categories(site_setting_names = nil)
categories = [
{
site_setting_name: "uncategorized_category_id",
name: I18n.t("uncategorized_category_name"),
description: nil,
position: 0,
color: "0088CC",
text_color: "FFFFFF",
permissions: {
everyone: :full,
},
force_permissions: true,
force_existence: true,
},
{
site_setting_name: "meta_category_id",
name: I18n.t("meta_category_name"),
description: I18n.t("meta_category_description"),
position: 1,
color: "808281",
text_color: "FFFFFF",
permissions: {
everyone: :full,
},
force_permissions: true,
sidebar: true,
},
{
site_setting_name: "staff_category_id",
name: I18n.t("staff_category_name"),
description: I18n.t("staff_category_description"),
position: 2,
color: "E45735",
text_color: "FFFFFF",
permissions: {
staff: :full,
},
force_permissions: true,
sidebar: true,
},
{
site_setting_name: "general_category_id",
name: I18n.t("general_category_name"),
description: I18n.t("general_category_description"),
position: 3,
color: "25AAE2",
text_color: "FFFFFF",
permissions: {
everyone: :full,
},
force_permissions: true,
sidebar: true,
default_composer_category: true,
},
]
if site_setting_names
categories.select! { |c| site_setting_names.include?(c[:site_setting_name]) }
end
categories
end
def create_category(
site_setting_name:,
name:,
description:,
position:,
color:,
text_color:,
permissions:,
force_permissions:,
force_existence: false,
sidebar: false,
default_composer_category: false
)
category_id = SiteSetting.get(site_setting_name)
if should_create_category?(category_id, force_existence)
category =
Category.new(
name: unused_category_name(category_id, name),
description: description,
user_id: Discourse::SYSTEM_USER_ID,
position: position,
color: color,
text_color: text_color,
)
category.skip_category_definition = true if description.blank?
category.set_permissions(permissions)
category.save!
SiteSetting.set(site_setting_name, category.id)
if sidebar
sidebar_categories = SiteSetting.default_sidebar_categories.split("|")
sidebar_categories << category.id
SiteSetting.set("default_sidebar_categories", sidebar_categories.join("|"))
end
SiteSetting.set("default_composer_category", category.id) if default_composer_category
elsif category = Category.find_by(id: category_id)
if description.present? && (category.topic_id.blank? || !Topic.exists?(category.topic_id))
category.description = description
category.create_category_definition
end
if force_permissions
category.set_permissions(permissions)
category.save!(validate: false) if category.changed?
end
end
end
def should_create_category?(category_id, force_existence)
return false if User.human_users.any?
if category_id > 0
force_existence ? !Category.exists?(category_id) : false
else
true
end
end
def unused_category_name(category_id, name)
category_exists =
Category.where(
"id <> :id AND LOWER(name) = :name",
id: category_id,
name: name.downcase,
).exists?
category_exists ? "#{name}#{SecureRandom.hex}" : name
end
def update_category(site_setting_name:, name:, description:, skip_changed:)
category = find_category(site_setting_name)
return if !category || (skip_changed && !unchanged?(category))
name = unused_category_name(category.id, name)
category.name = name
category.slug = Slug.for(name, "")
category.save!
if description.present? && description_post = category&.topic&.first_post
changes = { title: I18n.t("category.topic_prefix", category: name), raw: description }
description_post.revise(Discourse.system_user, changes, skip_validations: true)
end
end
def find_category(site_setting_name)
category_id = SiteSetting.get(site_setting_name)
Category.find_by(id: category_id) if category_id > 0
end
def unchanged?(category)
if description_post = category&.topic&.first_post
return description_post.last_editor_id == Discourse::SYSTEM_USER_ID
end
true
end
end
end