discourse/lib/topic_creator.rb

131 lines
3.5 KiB
Ruby

class TopicCreator
attr_accessor :errors
def self.create(user, guardian, opts)
self.new(user, guardian, opts).create
end
def initialize(user, guardian, opts)
@user = user
@guardian = guardian
@opts = opts
end
def create
@topic = Topic.new(setup_topic_params)
setup_auto_close_time
process_private_message
save_topic
watch_topic
@topic
end
private
def watch_topic
unless @opts[:auto_track] == false
@topic.notifier.watch_topic!(@topic.user_id)
end
user_ids = @topic.topic_allowed_users(true).pluck(:user_id)
user_ids += @topic.topic_allowed_groups(true).map { |t| t.group.users.pluck(:id) }.flatten
user_ids.uniq.reject{ |id| id == @topic.user_id }.each do |user_id|
@topic.notifier.watch_topic!(user_id, nil) unless user_id == -1
end
CategoryUser.auto_watch_new_topic(@topic)
CategoryUser.auto_track_new_topic(@topic)
end
def setup_topic_params
topic_params = {
title: @opts[:title],
user_id: @user.id,
last_post_user_id: @user.id
}
[:subtype, :archetype, :meta_data].each do |key|
topic_params[key] = @opts[key] if @opts[key].present?
end
category = find_category
@guardian.ensure_can_create!(Topic, category)
topic_params[:category_id] = category.id if category.present?
topic_params[:created_at] = Time.zone.parse(@opts[:created_at].to_s) if @opts[:created_at].present?
topic_params
end
def find_category
# PM can't have a category
@opts.delete(:category) if @opts[:archetype].present? && @opts[:archetype] == Archetype.private_message
# Temporary fix to allow older clients to create topics.
# When all clients are updated the category variable should
# be set directly to the contents of the if statement.
if (@opts[:category].is_a? Integer) || (@opts[:category] =~ /^\d+$/)
Category.find_by(id: @opts[:category])
else
Category.find_by(name: @opts[:category])
end
end
def setup_auto_close_time
return unless @opts[:auto_close_time].present?
return unless @guardian.can_moderate?(@topic)
@topic.set_auto_close(@opts[:auto_close_time], @user)
end
def process_private_message
return unless @opts[:archetype] == Archetype.private_message
@topic.subtype = TopicSubtype.user_to_user unless @topic.subtype
unless @opts[:target_usernames].present? || @opts[:target_group_names].present?
@topic.errors.add(:archetype, :cant_send_pm)
@errors = @topic.errors
raise ActiveRecord::Rollback.new
end
add_users(@topic,@opts[:target_usernames])
add_groups(@topic,@opts[:target_group_names])
@topic.topic_allowed_users.build(user_id: @user.id)
end
def save_topic
unless @topic.save(validate: !@opts[:skip_validations])
@errors = @topic.errors
raise ActiveRecord::Rollback.new
end
end
def add_users(topic, usernames)
return unless usernames
User.where(username: usernames.split(',')).each do |user|
check_can_send_permission!(topic,user)
topic.topic_allowed_users.build(user_id: user.id)
end
end
def add_groups(topic, groups)
return unless groups
Group.where(name: groups.split(',')).each do |group|
check_can_send_permission!(topic,group)
topic.topic_allowed_groups.build(group_id: group.id)
end
end
def check_can_send_permission!(topic,item)
unless @guardian.can_send_private_message?(item)
topic.errors.add(:archetype, :cant_send_pm)
@errors = topic.errors
raise ActiveRecord::Rollback.new
end
end
end