discourse-ai/app/controllers/discourse_ai/admin/ai_features_controller.rb

# frozen_string_literal: true

module DiscourseAi
  module Admin
    class AiFeaturesController < ::Admin::AdminController
      requires_plugin ::DiscourseAi::PLUGIN_NAME

      def index
        render json: serialize_modules(DiscourseAi::Configuration::Module.all)
      end

      def edit
        raise Discourse::InvalidParameters.new(:id) if params[:id].blank?

        a_module = DiscourseAi::Configuration::Module.find_by(id: params[:id].to_i)

        render json: serialize_module(a_module)
      end

      private

      def serialize_modules(modules)
        modules.map { |a_module| serialize_module(a_module) }
      end

      def serialize_module(a_module)
        return nil if a_module.blank?

        {
          id: a_module.id,
          module_name: a_module.name,
          module_enabled: a_module.enabled?,
          features: a_module.features.map { |f| serialize_feature(f) },
        }
      end

      def serialize_feature(feature)
        {
          name: feature.name,
          persona: serialize_persona(persona_id_obj_hash[feature.persona_id]),
          enabled: feature.enabled?,
        }
      end

      def serialize_persona(persona)
        return nil if persona.blank?

        serialize_data(persona, AiFeaturesPersonaSerializer, root: false)
      end

      private

      def persona_id_obj_hash
        @persona_id_obj_hash ||=
          begin
            setting_names = DiscourseAi::Configuration::Feature.all_persona_setting_names
            ids = setting_names.map { |sn| SiteSetting.public_send(sn) }

            AiPersona.where(id: ids).index_by(&:id)
          end
      end
    end
  end
end