mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-02-10 05:24:44 +00:00
bab5e52e38
This commit uses a new plugin modifier introduced in https://github.com/discourse/discourse/pull/26508 to mark all uploads as _not_ secure in shared PM AI conversations. This is so images created by the AI bot (or uploaded by the user) do not end up as broken URLs because of the security requirements around them. This relies on the UpdateTopicUploadSecurity job in core as well, which is fired when an AI conversation is shared or deleted.
52 lines
1.8 KiB
Ruby
52 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module ::Jobs
|
|
class SharedConversationAdjustUploadSecurity < ::Jobs::Base
|
|
def execute(args)
|
|
if args[:conversation_id].present?
|
|
# The conversation context includes post cooked content so this
|
|
# must be updated when target uploads security changes.
|
|
update_conversation(args[:conversation_id])
|
|
elsif args[:target_id].present? && args[:target_type].present?
|
|
# If we deleted the conversation then we just need to update the target's
|
|
# uploads security, no need to update the conversation.
|
|
update_target(args[:target_id], args[:target_type])
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def update_conversation(conversation_id)
|
|
conversation = SharedAiConversation.find_by(id: conversation_id)
|
|
return if conversation.blank?
|
|
|
|
# NOTE: Only Topics are supported for now, in future we will need a more flexible
|
|
# way of doing this.
|
|
if conversation.target_type == "Topic"
|
|
rebaked_posts = TopicUploadSecurityManager.new(conversation.target).run
|
|
|
|
if rebaked_posts.any?
|
|
new_context =
|
|
conversation.context.map do |context_post|
|
|
rebaked_post = rebaked_posts.find { |p| p.id == context_post["id"] }
|
|
context_post["cooked"] = rebaked_post.cooked if rebaked_post
|
|
context_post
|
|
end
|
|
|
|
conversation.update(context: new_context)
|
|
end
|
|
end
|
|
end
|
|
|
|
def update_target(target_id, target_type)
|
|
# NOTE: Only Topics are supported for now, in future we will need a more flexible
|
|
# way of doing this.
|
|
if target_type == "Topic"
|
|
topic = target_type.constantize.find_by(id: target_id)
|
|
return if topic.blank?
|
|
TopicUploadSecurityManager.new(topic).run
|
|
end
|
|
end
|
|
end
|
|
end
|