fix(spacecat-shared-data-access/src/models/site/config.js): add support for AI generated prompts

Author: byteclimber

packages/spacecat-shared-data-access/src/models/site/config.js

@@ -61,6 +61,25 @@ const QUESTION_SCHEMA = Joi.object({
   importTime: Joi.string().isoDate().optional(),
 });
 
+// AI Topics configuration schemas
+const AI_PROMPT_SCHEMA = Joi.object({
+  prompt: Joi.string().min(1).required(),
+  regions: Joi.array().items(Joi.string().pattern(/^[a-z]{2}$/)).required(),
+  origin: Joi.string().allow('human', 'ai', '').optional(),
+  source: Joi.string().optional(), // Allows any source. Active: 'flow' (created by flow). Schema-defined: 'config', 'api'. Tests: 'manual', 'ahrefs'
+}).unknown(true);
+
+const AI_TOPIC_SCHEMA = Joi.object({
+  name: Joi.string().min(1).required(),
+  prompts: Joi.array().items(AI_PROMPT_SCHEMA).min(1).required(),
+  category: Joi.string().min(1).optional(),
+}).unknown(true);
+
+const AI_TOPICS_CONFIG_SCHEMA = Joi.object().pattern(
+  Joi.string().uuid(),
+  AI_TOPIC_SCHEMA,
+);
+
 const LLMO_URL_PATTERN_SCHEMA = {
   urlPattern: Joi.string().uri().required(),
   tags: Joi.array().items(LLMO_TAG).optional(),
@@ -311,6 +330,7 @@ export const configSchema = Joi.object({
   contentAiConfig: Joi.object({
     index: Joi.string().optional(),
   }).optional(),
+  aiTopics: AI_TOPICS_CONFIG_SCHEMA.optional(),
   handlers: Joi.object().pattern(Joi.string(), Joi.object({
     mentions: Joi.object().pattern(Joi.string(), Joi.array().items(Joi.string())),
     excludedURLs: Joi.array().items(Joi.string()),
@@ -342,6 +362,13 @@ export const DEFAULT_CONFIG = {
   handlers: {},
 };
 
+// Export AI Topics schemas for external validation
+export const AI_TOPICS_SCHEMAS = {
+  AI_PROMPT_SCHEMA,
+  AI_TOPIC_SCHEMA,
+  AI_TOPICS_CONFIG_SCHEMA,
+};
+
 // Function to validate incoming configuration
 export function validateConfiguration(config) {
   const { error, value } = configSchema.validate(config);
@@ -390,6 +417,7 @@ export const Config = (data = {}) => {
   self.getSlackMentions = (type) => state?.handlers?.[type]?.mentions?.slack;
   self.getHandlerConfig = (type) => state?.handlers?.[type];
   self.getContentAiConfig = () => state?.contentAiConfig;
+  self.getAiTopics = () => state?.aiTopics;
   self.getHandlers = () => state.handlers;
   self.getImports = () => state.imports;
   self.getExcludedURLs = (type) => state?.handlers?.[type]?.excludedURLs;
@@ -661,6 +689,47 @@ export const Config = (data = {}) => {
     state.cdnLogsConfig = cdnLogsConfig;
   };
 
+  self.updateAiTopics = (aiTopics) => {
+    state.aiTopics = aiTopics;
+    validateConfiguration(state);
+  };
+
+  self.addAiTopic = (topicId, topic) => {
+    // Validate the topic
+    const { error } = AI_TOPIC_SCHEMA.validate(topic);
+    if (error) {
+      throw new Error(`Invalid AI topic: ${error.message}`);
+    }
+
+    state.aiTopics = state.aiTopics || {};
+    state.aiTopics[topicId] = topic;
+    validateConfiguration(state);
+  };
+
+  self.removeAiTopic = (topicId) => {
+    if (!state.aiTopics) return;
+
+    delete state.aiTopics[topicId];
+    validateConfiguration(state);
+  };
+
+  self.updateAiTopic = (topicId, topicUpdate) => {
+    if (!state.aiTopics || !state.aiTopics[topicId]) {
+      throw new Error(`AI topic ${topicId} not found`);
+    }
+
+    const updatedTopic = { ...state.aiTopics[topicId], ...topicUpdate };
+
+    // Validate the updated topic
+    const { error } = AI_TOPIC_SCHEMA.validate(updatedTopic);
+    if (error) {
+      throw new Error(`Invalid AI topic update: ${error.message}`);
+    }
+
+    state.aiTopics[topicId] = updatedTopic;
+    validateConfiguration(state);
+  };
+
   self.updateTokowakaConfig = (tokowakaConfig) => {
     state.tokowakaConfig = tokowakaConfig;
   };
@@ -674,6 +743,7 @@ Config.toDynamoItem = (config) => ({
   slack: config.getSlackConfig(),
   handlers: config.getHandlers(),
   contentAiConfig: config.getContentAiConfig(),
+  aiTopics: config.getAiTopics(),
   imports: config.getImports(),
   fetchConfig: config.getFetchConfig(),
   brandConfig: config.getBrandConfig(),