discourse/plugins/discourse-ai/lib/summarization/entry_point.rb

# frozen_string_literal: true

module DiscourseAi
  module Summarization
    class EntryPoint
      def inject_into(plugin)
        plugin.add_to_serializer(:current_user, :can_summarize) do
          return false if !SiteSetting.ai_summarization_enabled

          if (ai_persona = AiPersona.find_by(id: SiteSetting.ai_summarization_persona)).blank?
            return false
          end
          scope.user.in_any_groups?(ai_persona.allowed_group_ids.to_a)
        end

        plugin.add_to_serializer(:topic_view, :summarizable) do
          scope.can_see_summary?(object.topic)
        end

        plugin.add_to_serializer(:web_hook_topic_view, :summarizable) do
          scope.can_see_summary?(object.topic)
        end

        plugin.add_to_serializer(
          :topic_view,
          :ai_summary_record,
          include_condition: -> { false },
        ) do
          return @ai_summary_record if defined?(@ai_summary_record)
          @ai_summary_record =
            object.topic.ai_summaries.find_by(summary_type: AiSummary.summary_types[:complete])
        end

        plugin.add_to_serializer(
          :topic_view,
          :ai_summary,
          include_condition: -> do
            DiscoursePluginRegistry.apply_modifier(:serialize_ai_summary, false) &&
              scope.can_see_summary?(object.topic) && ai_summary_record.present?
          end,
        ) do
          {
            id: ai_summary_record.id,
            summarized_text: ai_summary_record.summarized_text,
            algorithm: ai_summary_record.algorithm,
            outdated: ai_summary_record.outdated,
            created_at: ai_summary_record.created_at,
            updated_at: ai_summary_record.updated_at,
          }
        end

        # Don't add gists to the following topic lists.
        gist_skipped_lists = %i[suggested semantic_related]

        plugin.register_modifier(:topic_query_create_list_topics) do |topics, options|
          if SiteSetting.ai_summarization_enabled && SiteSetting.ai_summary_gists_enabled &&
               !gist_skipped_lists.include?(options[:filter])
            topics.includes(:ai_gist_summary)
          else
            topics
          end
        end

        plugin.add_to_serializer(
          :topic_list_item,
          :ai_topic_gist,
          include_condition: -> { scope.can_see_gists? },
        ) do
          return if gist_skipped_lists.include?(options[:filter])
          object.ai_gist_summary&.summarized_text
        end

        # As this event can be triggered quite often, let's be overly cautious enqueueing
        # jobs if the feature is disabled.
        plugin.on(:post_created) do |post|
          if SiteSetting.discourse_ai_enabled && SiteSetting.ai_summarization_enabled &&
               SiteSetting.ai_summary_gists_enabled && post.topic
            Jobs.enqueue(:fast_track_topic_gist, topic_id: post&.topic_id)
          end
        end
      end
    end
  end
end