@akbarsafari00/bonyan-content-type-builder 5.52.1-fa.2

package/dist/admin/components/AIChat/lib/transforms/schemas/toCTB.mjs

@@ -0,0 +1,202 @@
+import isEqual from 'lodash/isEqual';
+import omit from 'lodash/omit';
+import pluralize from 'pluralize';
+
+const isPluginContentTypeUid = (uid)=>uid.startsWith('plugin::');
+const isContentTypeKind = (kind)=>{
+    return kind === 'collectionType' || kind === 'singleType';
+};
+/**
+ * Plugin / extension content-types use server-derived identity (globalId, collectionName, …).
+ * The AI chat uses a simplified shape that would otherwise overwrite those fields incorrectly.
+ */ const isPluginContentType = (schema, oldSchema)=>{
+    if (schema.plugin || isPluginContentTypeUid(schema.uid)) {
+        return true;
+    }
+    if (oldSchema && 'modelType' in oldSchema && oldSchema.modelType === 'contentType') {
+        const ct = oldSchema;
+        return Boolean(ct.plugin) || isPluginContentTypeUid(String(ct.uid));
+    }
+    return false;
+};
+const ACTION_TO_STATUS = {
+    create: 'NEW',
+    remove: 'REMOVED',
+    update: 'CHANGED'
+};
+/**
+ * Creates a new attribute with the specified status
+ */ const createAttributeWithStatus = (name, attributeData, status)=>({
+        ...attributeData,
+        name,
+        status
+    });
+/**
+ * Determines the status of an attribute by comparing new and old versions
+ */ const determineAttributeStatus = (newAttr, oldAttr, oldSchema)=>{
+    if (!oldAttr) {
+        return 'NEW';
+    }
+    // If the schema was already new, don't mark attributes as changed, keep them as new.
+    if (oldSchema?.status === 'NEW') {
+        return 'NEW';
+    }
+    // Compare attributes without the status field to determine if they've changed
+    const newAttrWithoutStatus = omit(newAttr, [
+        'status'
+    ]);
+    const oldAttrWithoutStatus = omit(oldAttr, [
+        'status'
+    ]);
+    if (!isEqual(newAttrWithoutStatus, oldAttrWithoutStatus)) {
+        return 'CHANGED';
+    }
+    // If unchanged, keep the previous status
+    return oldAttr.status;
+};
+/**
+ * Determines the status of a schema by comparing action and checking if oldSchema exists
+ */ const transformStatusFromChatToCTB = (schema, oldSchema)=>{
+    // If schema has an action, use the mapped status
+    if (schema.action) {
+        return ACTION_TO_STATUS[schema.action];
+    }
+    // If oldSchema doesn't exist, it's a new schema
+    if (!oldSchema) {
+        return 'NEW';
+    }
+    // If no action is specified and oldSchema exists, keep the existing status
+    return oldSchema.status;
+};
+/**
+ * Transform attributes from Chat format to CTB format
+ * Also performs a diff to determine the status of each attribute
+ */ const transformAttributesFromChatToCTB = ({ action, attributes }, oldSchema)=>{
+    // If it's a new schema or no oldAttributes provided, all attributes are NEW
+    if (action === 'create' || !oldSchema) {
+        return Object.entries(attributes).map(([name, attribute])=>createAttributeWithStatus(name, attribute, 'NEW'));
+    }
+    // Convert old attributes array to a lookup map for faster access
+    const oldAttributesMap = oldSchema.attributes.reduce((acc, attr)=>({
+            ...acc,
+            [attr.name]: attr
+        }), {});
+    // Process current attributes (new and changed)
+    const processedAttributes = Object.entries(attributes).map(([name, attr])=>{
+        const oldAttr = oldAttributesMap[name];
+        const status = determineAttributeStatus({
+            ...attr,
+            name
+        }, oldAttr, oldSchema);
+        return createAttributeWithStatus(name, attr, status);
+    });
+    // No need to mark removed attributes if the old schema is new, just remove it from the list
+    // TODO: Else a validation error occurs on the backend side.
+    if (oldSchema?.status === 'NEW') {
+        return processedAttributes;
+    }
+    // Find removed attributes (exist in old but not in new)
+    const removedAttributes = Object.entries(oldAttributesMap).filter(([name])=>!attributes[name]).map(([name, oldAttr])=>createAttributeWithStatus(name, oldAttr, 'REMOVED'));
+    // Combine both sets of attributes
+    return [
+        ...processedAttributes,
+        ...removedAttributes
+    ];
+};
+/**
+ * Transform schema format
+ *  AI chat -> CTB
+ *
+ * The AI chat returns a simplified format, and this layer transforms it to be compatible with the CTB reducer.
+ *
+ * We need to keep track of which changes have been made
+ */ const transformChatToCTB = (schema, oldSchema)=>{
+    const singularName = pluralize.singular(schema.name).toLowerCase().replace(/ /g, '-');
+    const pluralName = pluralize.plural(schema.name).toLowerCase().replace(/ /g, '-');
+    if (schema.modelType === 'component') {
+        return {
+            category: schema.category || 'default',
+            modelName: singularName,
+            attributes: transformAttributesFromChatToCTB(schema, oldSchema),
+            info: {
+                displayName: schema.name,
+                description: schema.description
+            },
+            modelType: schema.modelType,
+            uid: schema.uid,
+            collectionName: pluralName,
+            status: transformStatusFromChatToCTB(schema, oldSchema),
+            globalId: singularName
+        };
+    }
+    const previousContentType = oldSchema?.modelType === 'contentType' ? oldSchema : undefined;
+    const kind = isContentTypeKind(schema.kind) ? schema.kind : 'collectionType';
+    const contentTypeBase = {
+        uid: schema.uid,
+        modelType: 'contentType',
+        modelName: singularName,
+        kind,
+        info: {
+            displayName: schema.name.charAt(0).toUpperCase() + schema.name.slice(1),
+            // Always keep the old by default
+            singularName: previousContentType?.info.singularName ?? singularName,
+            // Always keep the old by default
+            pluralName: previousContentType?.info.pluralName ?? pluralName
+        },
+        collectionName: pluralName,
+        attributes: transformAttributesFromChatToCTB(schema, oldSchema),
+        options: {
+            draftAndPublish: schema.options?.draftAndPublish ?? true
+        },
+        pluginOptions: {
+            i18n: {
+                localized: schema.options?.localized ?? false
+            }
+        },
+        visible: true,
+        status: transformStatusFromChatToCTB(schema, oldSchema),
+        globalId: singularName,
+        restrictRelationsTo: null
+    };
+    if (isPluginContentType(schema, oldSchema) && oldSchema && oldSchema.modelType === 'contentType') {
+        const prev = oldSchema;
+        return {
+            ...contentTypeBase,
+            plugin: prev.plugin ?? schema.plugin,
+            globalId: prev.globalId,
+            modelName: prev.modelName,
+            collectionName: prev.collectionName,
+            info: {
+                ...contentTypeBase.info,
+                singularName: prev.info.singularName,
+                pluralName: prev.info.pluralName
+            },
+            options: {
+                ...prev.options,
+                ...contentTypeBase.options,
+                draftAndPublish: schema.options?.draftAndPublish ?? prev.options?.draftAndPublish ?? true
+            },
+            pluginOptions: {
+                ...prev.pluginOptions,
+                ...contentTypeBase.pluginOptions,
+                i18n: {
+                    ...prev.pluginOptions?.i18n,
+                    ...contentTypeBase.pluginOptions?.i18n,
+                    localized: schema.options?.localized ?? prev.pluginOptions?.i18n?.localized ?? false
+                }
+            },
+            visible: prev.visible,
+            restrictRelationsTo: prev.restrictRelationsTo
+        };
+    }
+    if (isPluginContentType(schema, oldSchema) && schema.plugin) {
+        return {
+            ...contentTypeBase,
+            plugin: schema.plugin
+        };
+    }
+    return contentTypeBase;
+};
+
+export { transformAttributesFromChatToCTB, transformChatToCTB };
+//# sourceMappingURL=toCTB.mjs.map

package/dist/admin/components/AIChat/lib/transforms/schemas/toCTB.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"toCTB.mjs","sources":["../../../../../../../admin/src/components/AIChat/lib/transforms/schemas/toCTB.ts"],"sourcesContent":["import isEqual from 'lodash/isEqual';\r\nimport omit from 'lodash/omit';\r\nimport pluralize from 'pluralize';\r\n\r\nimport type { ContentType, Component, AnyAttribute } from '../../../../../types';\r\nimport type { Schema, SchemaAttribute } from '../../types/schema';\r\nimport type { Struct, UID } from '@strapi/types';\r\n\r\nconst isPluginContentTypeUid = (uid: string) => uid.startsWith('plugin::');\r\n\r\nconst isContentTypeKind = (kind: Schema['kind']): kind is Struct.ContentTypeKind => {\r\n  return kind === 'collectionType' || kind === 'singleType';\r\n};\r\n\r\n/**\r\n * Plugin / extension content-types use server-derived identity (globalId, collectionName, …).\r\n * The AI chat uses a simplified shape that would otherwise overwrite those fields incorrectly.\r\n */\r\nconst isPluginContentType = (schema: Schema, oldSchema?: ContentType | Component): boolean => {\r\n  if (schema.plugin || isPluginContentTypeUid(schema.uid)) {\r\n    return true;\r\n  }\r\n  if (oldSchema && 'modelType' in oldSchema && oldSchema.modelType === 'contentType') {\r\n    const ct = oldSchema as ContentType;\r\n    return Boolean(ct.plugin) || isPluginContentTypeUid(String(ct.uid));\r\n  }\r\n  return false;\r\n};\r\n\r\nconst ACTION_TO_STATUS: Record<Schema['action'], ContentType['status']> = {\r\n  create: 'NEW',\r\n  remove: 'REMOVED',\r\n  update: 'CHANGED',\r\n};\r\n\r\n/**\r\n * Creates a new attribute with the specified status\r\n */\r\nconst createAttributeWithStatus = (\r\n  name: string,\r\n  attributeData: SchemaAttribute | AnyAttribute,\r\n  status: AnyAttribute['status']\r\n): AnyAttribute =>\r\n  ({\r\n    ...attributeData,\r\n    name,\r\n    status,\r\n  }) as AnyAttribute;\r\n\r\n/**\r\n * Determines the status of an attribute by comparing new and old versions\r\n */\r\nconst determineAttributeStatus = (\r\n  newAttr: Record<string, unknown>,\r\n  oldAttr?: AnyAttribute,\r\n  oldSchema?: ContentType | Component\r\n): AnyAttribute['status'] => {\r\n  if (!oldAttr) {\r\n    return 'NEW';\r\n  }\r\n\r\n  // If the schema was already new, don't mark attributes as changed, keep them as new.\r\n  if (oldSchema?.status === 'NEW') {\r\n    return 'NEW';\r\n  }\r\n\r\n  // Compare attributes without the status field to determine if they've changed\r\n  const newAttrWithoutStatus = omit(newAttr, ['status']);\r\n  const oldAttrWithoutStatus = omit(oldAttr, ['status']);\r\n\r\n  if (!isEqual(newAttrWithoutStatus, oldAttrWithoutStatus)) {\r\n    return 'CHANGED';\r\n  }\r\n\r\n  // If unchanged, keep the previous status\r\n  return oldAttr.status;\r\n};\r\n\r\n/**\r\n * Determines the status of a schema by comparing action and checking if oldSchema exists\r\n */\r\nconst transformStatusFromChatToCTB = (\r\n  schema: Schema,\r\n  oldSchema?: ContentType | Component\r\n): ContentType['status'] => {\r\n  // If schema has an action, use the mapped status\r\n  if (schema.action) {\r\n    return ACTION_TO_STATUS[schema.action];\r\n  }\r\n\r\n  // If oldSchema doesn't exist, it's a new schema\r\n  if (!oldSchema) {\r\n    return 'NEW';\r\n  }\r\n\r\n  // If no action is specified and oldSchema exists, keep the existing status\r\n  return oldSchema.status;\r\n};\r\n\r\n/**\r\n * Transform attributes from Chat format to CTB format\r\n * Also performs a diff to determine the status of each attribute\r\n */\r\nexport const transformAttributesFromChatToCTB = (\r\n  { action, attributes }: Schema,\r\n  oldSchema?: ContentType | Component\r\n): AnyAttribute[] => {\r\n  // If it's a new schema or no oldAttributes provided, all attributes are NEW\r\n  if (action === 'create' || !oldSchema) {\r\n    return Object.entries(attributes).map(([name, attribute]) =>\r\n      createAttributeWithStatus(name, attribute, 'NEW')\r\n    );\r\n  }\r\n\r\n  // Convert old attributes array to a lookup map for faster access\r\n  const oldAttributesMap = oldSchema.attributes.reduce(\r\n    (acc, attr) => ({ ...acc, [attr.name]: attr }),\r\n    {} as Record<string, AnyAttribute>\r\n  );\r\n\r\n  // Process current attributes (new and changed)\r\n  const processedAttributes = Object.entries(attributes).map(([name, attr]) => {\r\n    const oldAttr = oldAttributesMap[name];\r\n    const status = determineAttributeStatus({ ...attr, name }, oldAttr, oldSchema);\r\n\r\n    return createAttributeWithStatus(name, attr, status);\r\n  });\r\n\r\n  // No need to mark removed attributes if the old schema is new, just remove it from the list\r\n  // TODO: Else a validation error occurs on the backend side.\r\n  if (oldSchema?.status === 'NEW') {\r\n    return processedAttributes;\r\n  }\r\n\r\n  // Find removed attributes (exist in old but not in new)\r\n  const removedAttributes = Object.entries(oldAttributesMap)\r\n    .filter(([name]) => !attributes[name])\r\n    .map(([name, oldAttr]) => createAttributeWithStatus(name, oldAttr, 'REMOVED'));\r\n\r\n  // Combine both sets of attributes\r\n  return [...processedAttributes, ...removedAttributes];\r\n};\r\n\r\n/**\r\n * Transform schema format\r\n *  AI chat -> CTB\r\n *\r\n * The AI chat returns a simplified format, and this layer transforms it to be compatible with the CTB reducer.\r\n *\r\n * We need to keep track of which changes have been made\r\n */\r\nexport const transformChatToCTB = (\r\n  schema: Schema,\r\n  oldSchema?: ContentType | Component\r\n): ContentType | Component => {\r\n  const singularName = pluralize.singular(schema.name).toLowerCase().replace(/ /g, '-');\r\n  const pluralName = pluralize.plural(schema.name).toLowerCase().replace(/ /g, '-');\r\n\r\n  if (schema.modelType === 'component') {\r\n    return {\r\n      category: schema.category || 'default',\r\n      modelName: singularName,\r\n      attributes: transformAttributesFromChatToCTB(schema, oldSchema),\r\n      info: {\r\n        displayName: schema.name,\r\n        description: schema.description,\r\n        // TODO\r\n        // icon: schema.icon,\r\n      },\r\n      modelType: schema.modelType,\r\n      uid: schema.uid as UID.Component,\r\n      collectionName: pluralName,\r\n      status: transformStatusFromChatToCTB(schema, oldSchema),\r\n      globalId: singularName,\r\n    } satisfies Component;\r\n  }\r\n\r\n  const previousContentType = oldSchema?.modelType === 'contentType' ? oldSchema : undefined;\r\n  const kind = isContentTypeKind(schema.kind) ? schema.kind : 'collectionType';\r\n\r\n  const contentTypeBase = {\r\n    uid: schema.uid as UID.ContentType,\r\n    modelType: 'contentType',\r\n    modelName: singularName,\r\n    kind,\r\n    info: {\r\n      displayName: schema.name.charAt(0).toUpperCase() + schema.name.slice(1),\r\n      // Always keep the old by default\r\n      singularName: previousContentType?.info.singularName ?? singularName,\r\n      // Always keep the old by default\r\n      pluralName: previousContentType?.info.pluralName ?? pluralName,\r\n    },\r\n    collectionName: pluralName,\r\n    attributes: transformAttributesFromChatToCTB(schema, oldSchema),\r\n    options: {\r\n      draftAndPublish: schema.options?.draftAndPublish ?? true,\r\n    },\r\n    pluginOptions: {\r\n      i18n: {\r\n        localized: schema.options?.localized ?? false,\r\n      },\r\n    },\r\n    visible: true,\r\n    status: transformStatusFromChatToCTB(schema, oldSchema),\r\n    globalId: singularName,\r\n    restrictRelationsTo: null, // TODO: not sure what this is about\r\n  } satisfies ContentType;\r\n\r\n  if (\r\n    isPluginContentType(schema, oldSchema) &&\r\n    oldSchema &&\r\n    oldSchema.modelType === 'contentType'\r\n  ) {\r\n    const prev = oldSchema as ContentType;\r\n    return {\r\n      ...contentTypeBase,\r\n      plugin: prev.plugin ?? schema.plugin,\r\n      globalId: prev.globalId,\r\n      modelName: prev.modelName,\r\n      collectionName: prev.collectionName,\r\n      info: {\r\n        ...contentTypeBase.info,\r\n        singularName: prev.info.singularName,\r\n        pluralName: prev.info.pluralName,\r\n      },\r\n      options: {\r\n        ...prev.options,\r\n        ...contentTypeBase.options,\r\n        draftAndPublish: schema.options?.draftAndPublish ?? prev.options?.draftAndPublish ?? true,\r\n      },\r\n      pluginOptions: {\r\n        ...prev.pluginOptions,\r\n        ...contentTypeBase.pluginOptions,\r\n        i18n: {\r\n          ...(prev.pluginOptions?.i18n as Record<string, unknown> | undefined),\r\n          ...(contentTypeBase.pluginOptions?.i18n as Record<string, unknown> | undefined),\r\n          localized:\r\n            schema.options?.localized ??\r\n            (prev.pluginOptions?.i18n as { localized?: boolean } | undefined)?.localized ??\r\n            false,\r\n        },\r\n      },\r\n      visible: prev.visible,\r\n      restrictRelationsTo: prev.restrictRelationsTo,\r\n    } satisfies ContentType;\r\n  }\r\n\r\n  if (isPluginContentType(schema, oldSchema) && schema.plugin) {\r\n    return {\r\n      ...contentTypeBase,\r\n      plugin: schema.plugin,\r\n    } satisfies ContentType;\r\n  }\r\n\r\n  return contentTypeBase;\r\n};\r\n"],"names":["isPluginContentTypeUid","uid","startsWith","isContentTypeKind","kind","isPluginContentType","schema","oldSchema","plugin","modelType","ct","Boolean","String","ACTION_TO_STATUS","create","remove","update","createAttributeWithStatus","name","attributeData","status","determineAttributeStatus","newAttr","oldAttr","newAttrWithoutStatus","omit","oldAttrWithoutStatus","isEqual","transformStatusFromChatToCTB","action","transformAttributesFromChatToCTB","attributes","Object","entries","map","attribute","oldAttributesMap","reduce","acc","attr","processedAttributes","removedAttributes","filter","transformChatToCTB","singularName","pluralize","singular","toLowerCase","replace","pluralName","plural","category","modelName","info","displayName","description","collectionName","globalId","previousContentType","undefined","contentTypeBase","charAt","toUpperCase","slice","options","draftAndPublish","pluginOptions","i18n","localized","visible","restrictRelationsTo","prev"],"mappings":";;;;AAQA,MAAMA,sBAAAA,GAAyB,CAACC,GAAAA,GAAgBA,GAAAA,CAAIC,UAAU,CAAC,UAAA,CAAA;AAE/D,MAAMC,oBAAoB,CAACC,IAAAA,GAAAA;IACzB,OAAOA,IAAAA,KAAS,oBAAoBA,IAAAA,KAAS,YAAA;AAC/C,CAAA;AAEA;;;IAIA,MAAMC,mBAAAA,GAAsB,CAACC,MAAAA,EAAgBC,SAAAA,GAAAA;AAC3C,IAAA,IAAID,OAAOE,MAAM,IAAIR,sBAAAA,CAAuBM,MAAAA,CAAOL,GAAG,CAAA,EAAG;QACvD,OAAO,IAAA;AACT,IAAA;AACA,IAAA,IAAIM,aAAa,WAAA,IAAeA,SAAAA,IAAaA,SAAAA,CAAUE,SAAS,KAAK,aAAA,EAAe;AAClF,QAAA,MAAMC,EAAAA,GAAKH,SAAAA;AACX,QAAA,OAAOI,QAAQD,EAAAA,CAAGF,MAAM,KAAKR,sBAAAA,CAAuBY,MAAAA,CAAOF,GAAGT,GAAG,CAAA,CAAA;AACnE,IAAA;IACA,OAAO,KAAA;AACT,CAAA;AAEA,MAAMY,gBAAAA,GAAoE;IACxEC,MAAAA,EAAQ,KAAA;IACRC,MAAAA,EAAQ,SAAA;IACRC,MAAAA,EAAQ;AACV,CAAA;AAEA;;AAEC,IACD,MAAMC,yBAAAA,GAA4B,CAChCC,IAAAA,EACAC,aAAAA,EACAC,UAEC;AACC,QAAA,GAAGD,aAAa;AAChBD,QAAAA,IAAAA;AACAE,QAAAA;KACF,CAAA;AAEF;;AAEC,IACD,MAAMC,wBAAAA,GAA2B,CAC/BC,OAAAA,EACAC,OAAAA,EACAhB,SAAAA,GAAAA;AAEA,IAAA,IAAI,CAACgB,OAAAA,EAAS;QACZ,OAAO,KAAA;AACT,IAAA;;IAGA,IAAIhB,SAAAA,EAAWa,WAAW,KAAA,EAAO;QAC/B,OAAO,KAAA;AACT,IAAA;;IAGA,MAAMI,oBAAAA,GAAuBC,KAAKH,OAAAA,EAAS;AAAC,QAAA;AAAS,KAAA,CAAA;IACrD,MAAMI,oBAAAA,GAAuBD,KAAKF,OAAAA,EAAS;AAAC,QAAA;AAAS,KAAA,CAAA;IAErD,IAAI,CAACI,OAAAA,CAAQH,oBAAAA,EAAsBE,oBAAAA,CAAAA,EAAuB;QACxD,OAAO,SAAA;AACT,IAAA;;AAGA,IAAA,OAAOH,QAAQH,MAAM;AACvB,CAAA;AAEA;;IAGA,MAAMQ,4BAAAA,GAA+B,CACnCtB,MAAAA,EACAC,SAAAA,GAAAA;;IAGA,IAAID,MAAAA,CAAOuB,MAAM,EAAE;AACjB,QAAA,OAAOhB,gBAAgB,CAACP,MAAAA,CAAOuB,MAAM,CAAC;AACxC,IAAA;;AAGA,IAAA,IAAI,CAACtB,SAAAA,EAAW;QACd,OAAO,KAAA;AACT,IAAA;;AAGA,IAAA,OAAOA,UAAUa,MAAM;AACzB,CAAA;AAEA;;;UAIaU,gCAAAA,GAAmC,CAC9C,EAAED,MAAM,EAAEE,UAAU,EAAU,EAC9BxB,SAAAA,GAAAA;;IAGA,IAAIsB,MAAAA,KAAW,QAAA,IAAY,CAACtB,SAAAA,EAAW;AACrC,QAAA,OAAOyB,MAAAA,CAAOC,OAAO,CAACF,UAAAA,CAAAA,CAAYG,GAAG,CAAC,CAAC,CAAChB,IAAAA,EAAMiB,SAAAA,CAAU,GACtDlB,yBAAAA,CAA0BC,MAAMiB,SAAAA,EAAW,KAAA,CAAA,CAAA;AAE/C,IAAA;;IAGA,MAAMC,gBAAAA,GAAmB7B,UAAUwB,UAAU,CAACM,MAAM,CAClD,CAACC,GAAAA,EAAKC,IAAAA,IAAU;AAAE,YAAA,GAAGD,GAAG;YAAE,CAACC,IAAAA,CAAKrB,IAAI,GAAGqB;AAAK,SAAA,GAC5C,EAAC,CAAA;;IAIH,MAAMC,mBAAAA,GAAsBR,MAAAA,CAAOC,OAAO,CAACF,UAAAA,CAAAA,CAAYG,GAAG,CAAC,CAAC,CAAChB,IAAAA,EAAMqB,IAAAA,CAAK,GAAA;QACtE,MAAMhB,OAAAA,GAAUa,gBAAgB,CAAClB,IAAAA,CAAK;AACtC,QAAA,MAAME,SAASC,wBAAAA,CAAyB;AAAE,YAAA,GAAGkB,IAAI;AAAErB,YAAAA;AAAK,SAAA,EAAGK,OAAAA,EAAShB,SAAAA,CAAAA;QAEpE,OAAOU,yBAAAA,CAA0BC,MAAMqB,IAAAA,EAAMnB,MAAAA,CAAAA;AAC/C,IAAA,CAAA,CAAA;;;IAIA,IAAIb,SAAAA,EAAWa,WAAW,KAAA,EAAO;QAC/B,OAAOoB,mBAAAA;AACT,IAAA;;IAGA,MAAMC,iBAAAA,GAAoBT,MAAAA,CAAOC,OAAO,CAACG,gBAAAA,CAAAA,CACtCM,MAAM,CAAC,CAAC,CAACxB,IAAAA,CAAK,GAAK,CAACa,UAAU,CAACb,IAAAA,CAAK,CAAA,CACpCgB,GAAG,CAAC,CAAC,CAAChB,IAAAA,EAAMK,OAAAA,CAAQ,GAAKN,yBAAAA,CAA0BC,IAAAA,EAAMK,OAAAA,EAAS,SAAA,CAAA,CAAA;;IAGrE,OAAO;AAAIiB,QAAAA,GAAAA,mBAAAA;AAAwBC,QAAAA,GAAAA;AAAkB,KAAA;AACvD;AAEA;;;;;;;AAOC,IACM,MAAME,kBAAAA,GAAqB,CAChCrC,MAAAA,EACAC,SAAAA,GAAAA;IAEA,MAAMqC,YAAAA,GAAeC,SAAAA,CAAUC,QAAQ,CAACxC,MAAAA,CAAOY,IAAI,CAAA,CAAE6B,WAAW,EAAA,CAAGC,OAAO,CAAC,IAAA,EAAM,GAAA,CAAA;IACjF,MAAMC,UAAAA,GAAaJ,SAAAA,CAAUK,MAAM,CAAC5C,MAAAA,CAAOY,IAAI,CAAA,CAAE6B,WAAW,EAAA,CAAGC,OAAO,CAAC,IAAA,EAAM,GAAA,CAAA;IAE7E,IAAI1C,MAAAA,CAAOG,SAAS,KAAK,WAAA,EAAa;QACpC,OAAO;YACL0C,QAAAA,EAAU7C,MAAAA,CAAO6C,QAAQ,IAAI,SAAA;YAC7BC,SAAAA,EAAWR,YAAAA;AACXb,YAAAA,UAAAA,EAAYD,iCAAiCxB,MAAAA,EAAQC,SAAAA,CAAAA;YACrD8C,IAAAA,EAAM;AACJC,gBAAAA,WAAAA,EAAahD,OAAOY,IAAI;AACxBqC,gBAAAA,WAAAA,EAAajD,OAAOiD;AAGtB,aAAA;AACA9C,YAAAA,SAAAA,EAAWH,OAAOG,SAAS;AAC3BR,YAAAA,GAAAA,EAAKK,OAAOL,GAAG;YACfuD,cAAAA,EAAgBP,UAAAA;AAChB7B,YAAAA,MAAAA,EAAQQ,6BAA6BtB,MAAAA,EAAQC,SAAAA,CAAAA;YAC7CkD,QAAAA,EAAUb;AACZ,SAAA;AACF,IAAA;AAEA,IAAA,MAAMc,mBAAAA,GAAsBnD,SAAAA,EAAWE,SAAAA,KAAc,aAAA,GAAgBF,SAAAA,GAAYoD,SAAAA;AACjF,IAAA,MAAMvD,OAAOD,iBAAAA,CAAkBG,MAAAA,CAAOF,IAAI,CAAA,GAAIE,MAAAA,CAAOF,IAAI,GAAG,gBAAA;AAE5D,IAAA,MAAMwD,eAAAA,GAAkB;AACtB3D,QAAAA,GAAAA,EAAKK,OAAOL,GAAG;QACfQ,SAAAA,EAAW,aAAA;QACX2C,SAAAA,EAAWR,YAAAA;AACXxC,QAAAA,IAAAA;QACAiD,IAAAA,EAAM;AACJC,YAAAA,WAAAA,EAAahD,MAAAA,CAAOY,IAAI,CAAC2C,MAAM,CAAC,CAAA,CAAA,CAAGC,WAAW,EAAA,GAAKxD,MAAAA,CAAOY,IAAI,CAAC6C,KAAK,CAAC,CAAA,CAAA;;YAErEnB,YAAAA,EAAcc,mBAAAA,EAAqBL,KAAKT,YAAAA,IAAgBA,YAAAA;;YAExDK,UAAAA,EAAYS,mBAAAA,EAAqBL,KAAKJ,UAAAA,IAAcA;AACtD,SAAA;QACAO,cAAAA,EAAgBP,UAAAA;AAChBlB,QAAAA,UAAAA,EAAYD,iCAAiCxB,MAAAA,EAAQC,SAAAA,CAAAA;QACrDyD,OAAAA,EAAS;YACPC,eAAAA,EAAiB3D,MAAAA,CAAO0D,OAAO,EAAEC,eAAAA,IAAmB;AACtD,SAAA;QACAC,aAAAA,EAAe;YACbC,IAAAA,EAAM;gBACJC,SAAAA,EAAW9D,MAAAA,CAAO0D,OAAO,EAAEI,SAAAA,IAAa;AAC1C;AACF,SAAA;QACAC,OAAAA,EAAS,IAAA;AACTjD,QAAAA,MAAAA,EAAQQ,6BAA6BtB,MAAAA,EAAQC,SAAAA,CAAAA;QAC7CkD,QAAAA,EAAUb,YAAAA;QACV0B,mBAAAA,EAAqB;AACvB,KAAA;AAEA,IAAA,IACEjE,oBAAoBC,MAAAA,EAAQC,SAAAA,CAAAA,IAC5BA,aACAA,SAAAA,CAAUE,SAAS,KAAK,aAAA,EACxB;AACA,QAAA,MAAM8D,IAAAA,GAAOhE,SAAAA;QACb,OAAO;AACL,YAAA,GAAGqD,eAAe;AAClBpD,YAAAA,MAAAA,EAAQ+D,IAAAA,CAAK/D,MAAM,IAAIF,MAAAA,CAAOE,MAAM;AACpCiD,YAAAA,QAAAA,EAAUc,KAAKd,QAAQ;AACvBL,YAAAA,SAAAA,EAAWmB,KAAKnB,SAAS;AACzBI,YAAAA,cAAAA,EAAgBe,KAAKf,cAAc;YACnCH,IAAAA,EAAM;AACJ,gBAAA,GAAGO,gBAAgBP,IAAI;gBACvBT,YAAAA,EAAc2B,IAAAA,CAAKlB,IAAI,CAACT,YAAY;gBACpCK,UAAAA,EAAYsB,IAAAA,CAAKlB,IAAI,CAACJ;AACxB,aAAA;YACAe,OAAAA,EAAS;AACP,gBAAA,GAAGO,KAAKP,OAAO;AACf,gBAAA,GAAGJ,gBAAgBI,OAAO;AAC1BC,gBAAAA,eAAAA,EAAiB3D,OAAO0D,OAAO,EAAEC,mBAAmBM,IAAAA,CAAKP,OAAO,EAAEC,eAAAA,IAAmB;AACvF,aAAA;YACAC,aAAAA,EAAe;AACb,gBAAA,GAAGK,KAAKL,aAAa;AACrB,gBAAA,GAAGN,gBAAgBM,aAAa;gBAChCC,IAAAA,EAAM;oBACJ,GAAII,IAAAA,CAAKL,aAAa,EAAEC,IAAI;oBAC5B,GAAIP,eAAAA,CAAgBM,aAAa,EAAEC,IAAI;oBACvCC,SAAAA,EACE9D,MAAAA,CAAO0D,OAAO,EAAEI,SAAAA,IACfG,KAAKL,aAAa,EAAEC,MAA8CC,SAAAA,IACnE;AACJ;AACF,aAAA;AACAC,YAAAA,OAAAA,EAASE,KAAKF,OAAO;AACrBC,YAAAA,mBAAAA,EAAqBC,KAAKD;AAC5B,SAAA;AACF,IAAA;AAEA,IAAA,IAAIjE,mBAAAA,CAAoBC,MAAAA,EAAQC,SAAAA,CAAAA,IAAcD,MAAAA,CAAOE,MAAM,EAAE;QAC3D,OAAO;AACL,YAAA,GAAGoD,eAAe;AAClBpD,YAAAA,MAAAA,EAAQF,OAAOE;AACjB,SAAA;AACF,IAAA;IAEA,OAAOoD,eAAAA;AACT;;;;"}

package/dist/admin/components/AIChat/providers/ChatProvider.js

@@ -0,0 +1,192 @@
+'use strict';
+
+var jsxRuntime = require('react/jsx-runtime');
+var React = require('react');
+var ee = require('@strapi/admin/strapi-admin/ee');
+require('../../CTBSession/sessionContext.js');
+var useCTBTracking = require('../../CTBSession/useCTBTracking.js');
+var useDataManager = require('../../DataManager/useDataManager.js');
+var FeedbackModal = require('../FeedbackModal.js');
+var useAIFetch = require('../hooks/useAIFetch.js');
+var useChatTitle = require('../hooks/useChatTitle.js');
+var useLastSeenSchemas = require('../hooks/useLastSeenSchemas.js');
+var fromCTB = require('../lib/transforms/schemas/fromCTB.js');
+var UploadCodeModal = require('../UploadCodeModal.js');
+var UploadFigmaModal = require('../UploadFigmaModal.js');
+var SchemaProvider = require('./SchemaProvider.js');
+
+const ChatContext = /*#__PURE__*/ React.createContext(undefined);
+const generateRandomId = ()=>{
+    return Math.random().toString(36).substring(2, 15);
+};
+const BaseChatProvider = ({ children, defaultOpen = false })=>{
+    const [chatId, setChatId] = React.useState(generateRandomId());
+    const [isChatOpen, setIsChatOpen] = React.useState(defaultOpen);
+    const [openCount, setOpenCount] = React.useState(0);
+    const [input, setInput] = React.useState('');
+    // Files
+    const [attachments, setAttachments] = React.useState([]);
+    const { trackUsage } = useCTBTracking.useCTBTracking();
+    // DataManager
+    const { components, contentTypes } = useDataManager.useDataManager();
+    // Last user seen schemas
+    const { lastSeenSchemas } = useLastSeenSchemas.useLastSeenSchemas();
+    // Schemas to be sent to the chat
+    const schemas = React.useMemo(()=>{
+        return [
+            ...Object.values(contentTypes).filter((schema)=>schema.status !== 'REMOVED')// Probably we should still keep this one, not sure
+            .filter((schema)=>schema.uid !== 'plugin::users-permissions.user').filter((schema)=>schema.visible).map(fromCTB.transformCTBToChat),
+            ...Object.values(components).filter((schema)=>schema.status !== 'REMOVED').map(fromCTB.transformCTBToChat)
+        ];
+    }, [
+        contentTypes,
+        components
+    ]);
+    const { id, messages, sendMessage: _sendMessage, status, stop, ...chat } = useAIFetch.useAIChat({
+        id: chatId?.toString(),
+        experimental_throttle: 100
+    });
+    /* -------------------------------------------------------------------------------------------------
+   * AI SDK chat overrides
+   * -----------------------------------------------------------------------------------------------*/ // NOTE: body is using state variables, so they can not be passed as a prop in useChat
+    const sendMessage = async (message, options)=>{
+        if (status === 'streaming' || status === 'submitted') {
+            return;
+        }
+        return _sendMessage(message, {
+            ...options,
+            body: {
+                ...options?.body,
+                schemas,
+                metadata: {
+                    lastSeenSchemas: lastSeenSchemas.map((schema)=>schema.uid)
+                }
+            }
+        });
+    };
+    const handleSubmit = (e)=>{
+        e.preventDefault();
+        if (status === 'streaming' || status === 'submitted') {
+            return;
+        }
+        const readyAttachments = attachments.filter((a)=>a.status !== 'loading');
+        if (input.trim().length === 0 && attachments.length === 0) {
+            return;
+        }
+        const files = readyAttachments.map((attachment)=>({
+                type: 'file',
+                filename: attachment.filename,
+                mediaType: attachment.mediaType,
+                url: attachment.url
+            }));
+        sendMessage({
+            text: input,
+            files
+        });
+        setInput('');
+        setAttachments([]);
+    };
+    /* -------------------------------------------------------------------------------------------------
+   * Chat title
+   * -----------------------------------------------------------------------------------------------*/ const { title, generateTitle, resetTitle } = useChatTitle.useChatTitle({
+        chatId: id,
+        messages
+    });
+    // Automatically generate title when we have at least 1 message (user query)
+    React.useEffect(()=>{
+        if (messages.length >= 1 && !title) {
+            generateTitle();
+        }
+    }, [
+        messages.length,
+        title,
+        generateTitle
+    ]);
+    React.useEffect(()=>{
+        if (status === 'error') {
+            trackUsage('didAnswerMessage', {
+                successful: false
+            });
+        } else if (status === 'ready' && messages.length > 0 && messages[messages.length - 1]?.role === 'assistant') {
+            trackUsage('didAnswerMessage', {
+                successful: true
+            });
+        }
+    }, [
+        status,
+        messages,
+        trackUsage
+    ]);
+    const isChatAvailable = ee.useAIAvailability();
+    return /*#__PURE__*/ jsxRuntime.jsx(ChatContext.Provider, {
+        value: {
+            isChatEnabled: isChatAvailable,
+            id,
+            status,
+            stop,
+            sendMessage,
+            ...chat,
+            messages,
+            handleSubmit,
+            input,
+            setInput,
+            handleInputChange: (e)=>setInput(e.target.value),
+            reset: ()=>{
+                stop();
+                setChatId(generateRandomId());
+                trackUsage('didStartNewChat');
+                resetTitle();
+            },
+            schemas,
+            // Chat
+            title,
+            isChatOpen,
+            openChat: ()=>{
+                setIsChatOpen(true);
+                // if this is the first open, it's a new chat
+                if (openCount === 0) {
+                    trackUsage('didStartNewChat');
+                }
+                setOpenCount((prev)=>prev + 1);
+            },
+            closeChat: ()=>setIsChatOpen(false),
+            // Attachments
+            attachments,
+            setAttachments
+        },
+        children: children
+    });
+};
+const ChatProvider = ({ children, defaultOpen = false })=>{
+    return /*#__PURE__*/ jsxRuntime.jsx(BaseChatProvider, {
+        defaultOpen: defaultOpen,
+        children: /*#__PURE__*/ jsxRuntime.jsx(SchemaProvider.SchemaChatProvider, {
+            children: /*#__PURE__*/ jsxRuntime.jsx(UploadCodeModal.UploadProjectToChatProvider, {
+                children: /*#__PURE__*/ jsxRuntime.jsx(UploadFigmaModal.UploadFigmaToChatProvider, {
+                    children: /*#__PURE__*/ jsxRuntime.jsx(FeedbackModal.FeedbackProvider, {
+                        children: children
+                    })
+                })
+            })
+        })
+    });
+};
+function useStrapiChat() {
+    const context = React.useContext(ChatContext);
+    if (process.env.NODE_ENV === 'development') {
+        // In development, provide a fallback to prevent crashes during hot reloading
+        // This allows the app to continue working during HMR updates
+        if (context === undefined) {
+            console.warn('useStrapiChat was called outside of ChatProvider. This is only allowed in development mode.');
+            return {};
+        }
+    } else if (context === undefined) {
+        throw new Error('useStrapiChat must be used within a ChatProvider');
+    }
+    return context;
+}
+
+exports.BaseChatProvider = BaseChatProvider;
+exports.ChatProvider = ChatProvider;
+exports.useStrapiChat = useStrapiChat;
+//# sourceMappingURL=ChatProvider.js.map

package/dist/admin/components/AIChat/providers/ChatProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChatProvider.js","sources":["../../../../../admin/src/components/AIChat/providers/ChatProvider.tsx"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\r\nimport {\r\n  createContext,\r\n  useContext,\r\n  useMemo,\r\n  useState,\r\n  useEffect,\r\n  type ReactNode,\r\n  type ChangeEvent,\r\n} from 'react';\r\n\r\nimport { UIMessage, useChat } from '@ai-sdk/react';\r\nimport { useAIAvailability } from '@strapi/admin/strapi-admin/ee';\r\n\r\nimport { useCTBTracking } from '../../CTBSession/ctbSession';\r\nimport { useDataManager } from '../../DataManager/useDataManager';\r\nimport { FeedbackProvider } from '../FeedbackModal';\r\nimport { useAIChat } from '../hooks/useAIFetch';\r\nimport { useChatTitle } from '../hooks/useChatTitle';\r\nimport { useLastSeenSchemas } from '../hooks/useLastSeenSchemas';\r\nimport { transformCTBToChat } from '../lib/transforms/schemas/fromCTB';\r\nimport { Attachment } from '../lib/types/attachments';\r\nimport { Schema } from '../lib/types/schema';\r\nimport { UploadProjectToChatProvider } from '../UploadCodeModal';\r\nimport { UploadFigmaToChatProvider } from '../UploadFigmaModal';\r\n\r\nimport { SchemaChatProvider } from './SchemaProvider';\r\n\r\ninterface ChatContextType extends Omit<ReturnType<typeof useChat>, 'messages'> {\r\n  isChatEnabled: boolean;\r\n  title?: string;\r\n  messages: UIMessage[];\r\n  handleSubmit: (event: any) => void;\r\n  input: string;\r\n  setInput: React.Dispatch<React.SetStateAction<string>>;\r\n  handleInputChange: (e: ChangeEvent<HTMLInputElement> | ChangeEvent<HTMLTextAreaElement>) => void;\r\n  reset: () => void;\r\n  schemas: Schema[];\r\n  // Chat window\r\n  isChatOpen: boolean;\r\n  openChat: () => void;\r\n  closeChat: () => void;\r\n  // Attachments\r\n  attachments: Attachment[];\r\n  setAttachments: React.Dispatch<React.SetStateAction<Attachment[]>>;\r\n}\r\n\r\nconst ChatContext = createContext<ChatContextType | undefined>(undefined);\r\n\r\nconst generateRandomId = () => {\r\n  return Math.random().toString(36).substring(2, 15);\r\n};\r\n\r\nexport const BaseChatProvider = ({\r\n  children,\r\n  defaultOpen = false,\r\n}: {\r\n  children: ReactNode;\r\n  defaultOpen?: boolean;\r\n}) => {\r\n  const [chatId, setChatId] = useState<string | undefined>(generateRandomId());\r\n  const [isChatOpen, setIsChatOpen] = useState(defaultOpen);\r\n  const [openCount, setOpenCount] = useState(0);\r\n  const [input, setInput] = useState('');\r\n\r\n  // Files\r\n  const [attachments, setAttachments] = useState<Attachment[]>([]);\r\n\r\n  const { trackUsage } = useCTBTracking();\r\n\r\n  // DataManager\r\n  const { components, contentTypes } = useDataManager();\r\n\r\n  // Last user seen schemas\r\n  const { lastSeenSchemas } = useLastSeenSchemas();\r\n\r\n  // Schemas to be sent to the chat\r\n  const schemas = useMemo(() => {\r\n    return [\r\n      ...Object.values(contentTypes)\r\n        .filter((schema) => schema.status !== 'REMOVED')\r\n        // Probably we should still keep this one, not sure\r\n        .filter((schema) => schema.uid !== 'plugin::users-permissions.user')\r\n        .filter((schema) => schema.visible)\r\n        .map(transformCTBToChat as any),\r\n      ...Object.values(components)\r\n        .filter((schema) => schema.status !== 'REMOVED')\r\n        .map(transformCTBToChat as any),\r\n    ] as Schema[];\r\n  }, [contentTypes, components]);\r\n\r\n  const {\r\n    id,\r\n    messages,\r\n    sendMessage: _sendMessage,\r\n    status,\r\n    stop,\r\n    ...chat\r\n  } = useAIChat({\r\n    id: chatId?.toString(),\r\n    experimental_throttle: 100,\r\n  });\r\n\r\n  /* -------------------------------------------------------------------------------------------------\r\n   * AI SDK chat overrides\r\n   * -----------------------------------------------------------------------------------------------*/\r\n\r\n  // NOTE: body is using state variables, so they can not be passed as a prop in useChat\r\n  const sendMessage: typeof _sendMessage = async (message, options) => {\r\n    if (status === 'streaming' || status === 'submitted') {\r\n      return;\r\n    }\r\n\r\n    return _sendMessage(message, {\r\n      ...options,\r\n      body: {\r\n        ...options?.body,\r\n        schemas,\r\n        metadata: {\r\n          lastSeenSchemas: lastSeenSchemas.map((schema) => schema.uid),\r\n        },\r\n      },\r\n    });\r\n  };\r\n\r\n  const handleSubmit = (e: React.FormEvent<HTMLFormElement>) => {\r\n    e.preventDefault();\r\n\r\n    if (status === 'streaming' || status === 'submitted') {\r\n      return;\r\n    }\r\n\r\n    const readyAttachments = attachments.filter((a) => a.status !== 'loading');\r\n    if (input.trim().length === 0 && attachments.length === 0) {\r\n      return;\r\n    }\r\n\r\n    const files = readyAttachments.map(\r\n      (attachment) =>\r\n        ({\r\n          type: 'file',\r\n          filename: attachment.filename,\r\n          mediaType: attachment.mediaType,\r\n          url: attachment.url,\r\n        }) as const\r\n    );\r\n    sendMessage({ text: input, files });\r\n    setInput('');\r\n    setAttachments([]);\r\n  };\r\n\r\n  /* -------------------------------------------------------------------------------------------------\r\n   * Chat title\r\n   * -----------------------------------------------------------------------------------------------*/\r\n  const { title, generateTitle, resetTitle } = useChatTitle({ chatId: id, messages });\r\n\r\n  // Automatically generate title when we have at least 1 message (user query)\r\n  useEffect(() => {\r\n    if (messages.length >= 1 && !title) {\r\n      generateTitle();\r\n    }\r\n  }, [messages.length, title, generateTitle]);\r\n\r\n  useEffect(() => {\r\n    if (status === 'error') {\r\n      trackUsage('didAnswerMessage', {\r\n        successful: false,\r\n      });\r\n    } else if (\r\n      status === 'ready' &&\r\n      messages.length > 0 &&\r\n      messages[messages.length - 1]?.role === 'assistant'\r\n    ) {\r\n      trackUsage('didAnswerMessage', {\r\n        successful: true,\r\n      });\r\n    }\r\n  }, [status, messages, trackUsage]);\r\n\r\n  const isChatAvailable = useAIAvailability();\r\n\r\n  return (\r\n    <ChatContext.Provider\r\n      value={{\r\n        isChatEnabled: isChatAvailable,\r\n        id,\r\n        status,\r\n        stop,\r\n        sendMessage,\r\n        ...chat,\r\n        messages,\r\n        handleSubmit,\r\n        input,\r\n        setInput,\r\n        handleInputChange: (e) => setInput(e.target.value),\r\n        reset: () => {\r\n          stop();\r\n          setChatId(generateRandomId());\r\n          trackUsage('didStartNewChat');\r\n          resetTitle();\r\n        },\r\n        schemas,\r\n        // Chat\r\n        title,\r\n        isChatOpen,\r\n        openChat: () => {\r\n          setIsChatOpen(true);\r\n          // if this is the first open, it's a new chat\r\n          if (openCount === 0) {\r\n            trackUsage('didStartNewChat');\r\n          }\r\n          setOpenCount((prev) => prev + 1);\r\n        },\r\n        closeChat: () => setIsChatOpen(false),\r\n        // Attachments\r\n        attachments,\r\n        setAttachments,\r\n      }}\r\n    >\r\n      {children}\r\n    </ChatContext.Provider>\r\n  );\r\n};\r\n\r\nexport const ChatProvider = ({\r\n  children,\r\n  defaultOpen = false,\r\n}: {\r\n  children: React.ReactNode;\r\n  defaultOpen?: boolean;\r\n}) => {\r\n  return (\r\n    <BaseChatProvider defaultOpen={defaultOpen}>\r\n      <SchemaChatProvider>\r\n        <UploadProjectToChatProvider>\r\n          <UploadFigmaToChatProvider>\r\n            <FeedbackProvider>{children}</FeedbackProvider>\r\n          </UploadFigmaToChatProvider>\r\n        </UploadProjectToChatProvider>\r\n      </SchemaChatProvider>\r\n    </BaseChatProvider>\r\n  );\r\n};\r\n\r\nexport function useStrapiChat() {\r\n  const context = useContext(ChatContext);\r\n\r\n  if (process.env.NODE_ENV === 'development') {\r\n    // In development, provide a fallback to prevent crashes during hot reloading\r\n    // This allows the app to continue working during HMR updates\r\n    if (context === undefined) {\r\n      console.warn(\r\n        'useStrapiChat was called outside of ChatProvider. This is only allowed in development mode.'\r\n      );\r\n      return {} as ChatContextType;\r\n    }\r\n  } else if (context === undefined) {\r\n    throw new Error('useStrapiChat must be used within a ChatProvider');\r\n  }\r\n  return context;\r\n}\r\n"],"names":["ChatContext","createContext","undefined","generateRandomId","Math","random","toString","substring","BaseChatProvider","children","defaultOpen","chatId","setChatId","useState","isChatOpen","setIsChatOpen","openCount","setOpenCount","input","setInput","attachments","setAttachments","trackUsage","useCTBTracking","components","contentTypes","useDataManager","lastSeenSchemas","useLastSeenSchemas","schemas","useMemo","Object","values","filter","schema","status","uid","visible","map","transformCTBToChat","id","messages","sendMessage","_sendMessage","stop","chat","useAIChat","experimental_throttle","message","options","body","metadata","handleSubmit","e","preventDefault","readyAttachments","a","trim","length","files","attachment","type","filename","mediaType","url","text","title","generateTitle","resetTitle","useChatTitle","useEffect","successful","role","isChatAvailable","useAIAvailability","_jsx","Provider","value","isChatEnabled","handleInputChange","target","reset","openChat","prev","closeChat","ChatProvider","SchemaChatProvider","UploadProjectToChatProvider","UploadFigmaToChatProvider","FeedbackProvider","useStrapiChat","context","useContext","process","env","NODE_ENV","console","warn","Error"],"mappings":";;;;;;;;;;;;;;;;;AA+CA,MAAMA,4BAAcC,mBAAAA,CAA2CC,SAAAA,CAAAA;AAE/D,MAAMC,gBAAAA,GAAmB,IAAA;IACvB,OAAOC,IAAAA,CAAKC,MAAM,EAAA,CAAGC,QAAQ,CAAC,EAAA,CAAA,CAAIC,SAAS,CAAC,CAAA,EAAG,EAAA,CAAA;AACjD,CAAA;AAEO,MAAMC,mBAAmB,CAAC,EAC/BC,QAAQ,EACRC,WAAAA,GAAc,KAAK,EAIpB,GAAA;AACC,IAAA,MAAM,CAACC,MAAAA,EAAQC,SAAAA,CAAU,GAAGC,cAAAA,CAA6BV,gBAAAA,EAAAA,CAAAA;AACzD,IAAA,MAAM,CAACW,UAAAA,EAAYC,aAAAA,CAAc,GAAGF,cAAAA,CAASH,WAAAA,CAAAA;AAC7C,IAAA,MAAM,CAACM,SAAAA,EAAWC,YAAAA,CAAa,GAAGJ,cAAAA,CAAS,CAAA,CAAA;AAC3C,IAAA,MAAM,CAACK,KAAAA,EAAOC,QAAAA,CAAS,GAAGN,cAAAA,CAAS,EAAA,CAAA;;AAGnC,IAAA,MAAM,CAACO,WAAAA,EAAaC,cAAAA,CAAe,GAAGR,eAAuB,EAAE,CAAA;IAE/D,MAAM,EAAES,UAAU,EAAE,GAAGC,6BAAAA,EAAAA;;AAGvB,IAAA,MAAM,EAAEC,UAAU,EAAEC,YAAY,EAAE,GAAGC,6BAAAA,EAAAA;;IAGrC,MAAM,EAAEC,eAAe,EAAE,GAAGC,qCAAAA,EAAAA;;AAG5B,IAAA,MAAMC,UAAUC,aAAAA,CAAQ,IAAA;QACtB,OAAO;eACFC,MAAAA,CAAOC,MAAM,CAACP,YAAAA,CAAAA,CACdQ,MAAM,CAAC,CAACC,MAAAA,GAAWA,MAAAA,CAAOC,MAAM,KAAK,SAAA,CACtC;AACCF,aAAAA,MAAM,CAAC,CAACC,MAAAA,GAAWA,MAAAA,CAAOE,GAAG,KAAK,gCAAA,CAAA,CAClCH,MAAM,CAAC,CAACC,MAAAA,GAAWA,MAAAA,CAAOG,OAAO,CAAA,CACjCC,GAAG,CAACC,0BAAAA,CAAAA;AACJR,YAAAA,GAAAA,MAAAA,CAAOC,MAAM,CAACR,UAAAA,CAAAA,CACdS,MAAM,CAAC,CAACC,MAAAA,GAAWA,MAAAA,CAAOC,MAAM,KAAK,SAAA,CAAA,CACrCG,GAAG,CAACC,0BAAAA;AACR,SAAA;IACH,CAAA,EAAG;AAACd,QAAAA,YAAAA;AAAcD,QAAAA;AAAW,KAAA,CAAA;AAE7B,IAAA,MAAM,EACJgB,EAAE,EACFC,QAAQ,EACRC,WAAAA,EAAaC,YAAY,EACzBR,MAAM,EACNS,IAAI,EACJ,GAAGC,IAAAA,EACJ,GAAGC,oBAAAA,CAAU;AACZN,QAAAA,EAAAA,EAAI7B,MAAAA,EAAQL,QAAAA,EAAAA;QACZyC,qBAAAA,EAAuB;AACzB,KAAA,CAAA;AAEA;;AAEkG;IAGlG,MAAML,WAAAA,GAAmC,OAAOM,OAAAA,EAASC,OAAAA,GAAAA;QACvD,IAAId,MAAAA,KAAW,WAAA,IAAeA,MAAAA,KAAW,WAAA,EAAa;AACpD,YAAA;AACF,QAAA;AAEA,QAAA,OAAOQ,aAAaK,OAAAA,EAAS;AAC3B,YAAA,GAAGC,OAAO;YACVC,IAAAA,EAAM;AACJ,gBAAA,GAAGD,SAASC,IAAI;AAChBrB,gBAAAA,OAAAA;gBACAsB,QAAAA,EAAU;AACRxB,oBAAAA,eAAAA,EAAiBA,gBAAgBW,GAAG,CAAC,CAACJ,MAAAA,GAAWA,OAAOE,GAAG;AAC7D;AACF;AACF,SAAA,CAAA;AACF,IAAA,CAAA;AAEA,IAAA,MAAMgB,eAAe,CAACC,CAAAA,GAAAA;AACpBA,QAAAA,CAAAA,CAAEC,cAAc,EAAA;QAEhB,IAAInB,MAAAA,KAAW,WAAA,IAAeA,MAAAA,KAAW,WAAA,EAAa;AACpD,YAAA;AACF,QAAA;QAEA,MAAMoB,gBAAAA,GAAmBnC,YAAYa,MAAM,CAAC,CAACuB,CAAAA,GAAMA,CAAAA,CAAErB,MAAM,KAAK,SAAA,CAAA;QAChE,IAAIjB,KAAAA,CAAMuC,IAAI,EAAA,CAAGC,MAAM,KAAK,CAAA,IAAKtC,WAAAA,CAAYsC,MAAM,KAAK,CAAA,EAAG;AACzD,YAAA;AACF,QAAA;AAEA,QAAA,MAAMC,QAAQJ,gBAAAA,CAAiBjB,GAAG,CAChC,CAACsB,cACE;gBACCC,IAAAA,EAAM,MAAA;AACNC,gBAAAA,QAAAA,EAAUF,WAAWE,QAAQ;AAC7BC,gBAAAA,SAAAA,EAAWH,WAAWG,SAAS;AAC/BC,gBAAAA,GAAAA,EAAKJ,WAAWI;aAClB,CAAA,CAAA;QAEJtB,WAAAA,CAAY;YAAEuB,IAAAA,EAAM/C,KAAAA;AAAOyC,YAAAA;AAAM,SAAA,CAAA;QACjCxC,QAAAA,CAAS,EAAA,CAAA;AACTE,QAAAA,cAAAA,CAAe,EAAE,CAAA;AACnB,IAAA,CAAA;AAEA;;uGAGA,MAAM,EAAE6C,KAAK,EAAEC,aAAa,EAAEC,UAAU,EAAE,GAAGC,yBAAAA,CAAa;QAAE1D,MAAAA,EAAQ6B,EAAAA;AAAIC,QAAAA;AAAS,KAAA,CAAA;;IAGjF6B,eAAAA,CAAU,IAAA;AACR,QAAA,IAAI7B,QAAAA,CAASiB,MAAM,IAAI,CAAA,IAAK,CAACQ,KAAAA,EAAO;AAClCC,YAAAA,aAAAA,EAAAA;AACF,QAAA;IACF,CAAA,EAAG;AAAC1B,QAAAA,QAAAA,CAASiB,MAAM;AAAEQ,QAAAA,KAAAA;AAAOC,QAAAA;AAAc,KAAA,CAAA;IAE1CG,eAAAA,CAAU,IAAA;AACR,QAAA,IAAInC,WAAW,OAAA,EAAS;AACtBb,YAAAA,UAAAA,CAAW,kBAAA,EAAoB;gBAC7BiD,UAAAA,EAAY;AACd,aAAA,CAAA;AACF,QAAA,CAAA,MAAO,IACLpC,MAAAA,KAAW,OAAA,IACXM,QAAAA,CAASiB,MAAM,GAAG,CAAA,IAClBjB,QAAQ,CAACA,SAASiB,MAAM,GAAG,CAAA,CAAE,EAAEc,SAAS,WAAA,EACxC;AACAlD,YAAAA,UAAAA,CAAW,kBAAA,EAAoB;gBAC7BiD,UAAAA,EAAY;AACd,aAAA,CAAA;AACF,QAAA;IACF,CAAA,EAAG;AAACpC,QAAAA,MAAAA;AAAQM,QAAAA,QAAAA;AAAUnB,QAAAA;AAAW,KAAA,CAAA;AAEjC,IAAA,MAAMmD,eAAAA,GAAkBC,oBAAAA,EAAAA;IAExB,qBACEC,cAAA,CAAC3E,YAAY4E,QAAQ,EAAA;QACnBC,KAAAA,EAAO;YACLC,aAAAA,EAAeL,eAAAA;AACfjC,YAAAA,EAAAA;AACAL,YAAAA,MAAAA;AACAS,YAAAA,IAAAA;AACAF,YAAAA,WAAAA;AACA,YAAA,GAAGG,IAAI;AACPJ,YAAAA,QAAAA;AACAW,YAAAA,YAAAA;AACAlC,YAAAA,KAAAA;AACAC,YAAAA,QAAAA;AACA4D,YAAAA,iBAAAA,EAAmB,CAAC1B,CAAAA,GAAMlC,QAAAA,CAASkC,CAAAA,CAAE2B,MAAM,CAACH,KAAK,CAAA;YACjDI,KAAAA,EAAO,IAAA;AACLrC,gBAAAA,IAAAA,EAAAA;gBACAhC,SAAAA,CAAUT,gBAAAA,EAAAA,CAAAA;gBACVmB,UAAAA,CAAW,iBAAA,CAAA;AACX8C,gBAAAA,UAAAA,EAAAA;AACF,YAAA,CAAA;AACAvC,YAAAA,OAAAA;;AAEAqC,YAAAA,KAAAA;AACApD,YAAAA,UAAAA;YACAoE,QAAAA,EAAU,IAAA;gBACRnE,aAAAA,CAAc,IAAA,CAAA;;AAEd,gBAAA,IAAIC,cAAc,CAAA,EAAG;oBACnBM,UAAAA,CAAW,iBAAA,CAAA;AACb,gBAAA;gBACAL,YAAAA,CAAa,CAACkE,OAASA,IAAAA,GAAO,CAAA,CAAA;AAChC,YAAA,CAAA;AACAC,YAAAA,SAAAA,EAAW,IAAMrE,aAAAA,CAAc,KAAA,CAAA;;AAE/BK,YAAAA,WAAAA;AACAC,YAAAA;AACF,SAAA;AAECZ,QAAAA,QAAAA,EAAAA;;AAGP;AAEO,MAAM4E,eAAe,CAAC,EAC3B5E,QAAQ,EACRC,WAAAA,GAAc,KAAK,EAIpB,GAAA;AACC,IAAA,qBACEiE,cAAA,CAACnE,gBAAAA,EAAAA;QAAiBE,WAAAA,EAAaA,WAAAA;AAC7B,QAAA,QAAA,gBAAAiE,cAAA,CAACW,iCAAAA,EAAAA;AACC,YAAA,QAAA,gBAAAX,cAAA,CAACY,2CAAAA,EAAAA;AACC,gBAAA,QAAA,gBAAAZ,cAAA,CAACa,0CAAAA,EAAAA;AACC,oBAAA,QAAA,gBAAAb,cAAA,CAACc,8BAAAA,EAAAA;AAAkBhF,wBAAAA,QAAAA,EAAAA;;;;;;AAM/B;AAEO,SAASiF,aAAAA,GAAAA;AACd,IAAA,MAAMC,UAAUC,gBAAAA,CAAW5F,WAAAA,CAAAA;AAE3B,IAAA,IAAI6F,OAAAA,CAAQC,GAAG,CAACC,QAAQ,KAAK,aAAA,EAAe;;;AAG1C,QAAA,IAAIJ,YAAYzF,SAAAA,EAAW;AACzB8F,YAAAA,OAAAA,CAAQC,IAAI,CACV,6FAAA,CAAA;AAEF,YAAA,OAAO,EAAC;AACV,QAAA;IACF,CAAA,MAAO,IAAIN,YAAYzF,SAAAA,EAAW;AAChC,QAAA,MAAM,IAAIgG,KAAAA,CAAM,kDAAA,CAAA;AAClB,IAAA;IACA,OAAOP,OAAAA;AACT;;;;;;"}

package/dist/admin/components/AIChat/providers/ChatProvider.mjs

@@ -0,0 +1,188 @@
+import { jsx } from 'react/jsx-runtime';
+import { useState, useMemo, useEffect, useContext, createContext } from 'react';
+import { useAIAvailability } from '@strapi/admin/strapi-admin/ee';
+import '../../CTBSession/sessionContext.mjs';
+import { useCTBTracking } from '../../CTBSession/useCTBTracking.mjs';
+import { useDataManager } from '../../DataManager/useDataManager.mjs';
+import { FeedbackProvider } from '../FeedbackModal.mjs';
+import { useAIChat } from '../hooks/useAIFetch.mjs';
+import { useChatTitle } from '../hooks/useChatTitle.mjs';
+import { useLastSeenSchemas } from '../hooks/useLastSeenSchemas.mjs';
+import { transformCTBToChat } from '../lib/transforms/schemas/fromCTB.mjs';
+import { UploadProjectToChatProvider } from '../UploadCodeModal.mjs';
+import { UploadFigmaToChatProvider } from '../UploadFigmaModal.mjs';
+import { SchemaChatProvider } from './SchemaProvider.mjs';
+
+const ChatContext = /*#__PURE__*/ createContext(undefined);
+const generateRandomId = ()=>{
+    return Math.random().toString(36).substring(2, 15);
+};
+const BaseChatProvider = ({ children, defaultOpen = false })=>{
+    const [chatId, setChatId] = useState(generateRandomId());
+    const [isChatOpen, setIsChatOpen] = useState(defaultOpen);
+    const [openCount, setOpenCount] = useState(0);
+    const [input, setInput] = useState('');
+    // Files
+    const [attachments, setAttachments] = useState([]);
+    const { trackUsage } = useCTBTracking();
+    // DataManager
+    const { components, contentTypes } = useDataManager();
+    // Last user seen schemas
+    const { lastSeenSchemas } = useLastSeenSchemas();
+    // Schemas to be sent to the chat
+    const schemas = useMemo(()=>{
+        return [
+            ...Object.values(contentTypes).filter((schema)=>schema.status !== 'REMOVED')// Probably we should still keep this one, not sure
+            .filter((schema)=>schema.uid !== 'plugin::users-permissions.user').filter((schema)=>schema.visible).map(transformCTBToChat),
+            ...Object.values(components).filter((schema)=>schema.status !== 'REMOVED').map(transformCTBToChat)
+        ];
+    }, [
+        contentTypes,
+        components
+    ]);
+    const { id, messages, sendMessage: _sendMessage, status, stop, ...chat } = useAIChat({
+        id: chatId?.toString(),
+        experimental_throttle: 100
+    });
+    /* -------------------------------------------------------------------------------------------------
+   * AI SDK chat overrides
+   * -----------------------------------------------------------------------------------------------*/ // NOTE: body is using state variables, so they can not be passed as a prop in useChat
+    const sendMessage = async (message, options)=>{
+        if (status === 'streaming' || status === 'submitted') {
+            return;
+        }
+        return _sendMessage(message, {
+            ...options,
+            body: {
+                ...options?.body,
+                schemas,
+                metadata: {
+                    lastSeenSchemas: lastSeenSchemas.map((schema)=>schema.uid)
+                }
+            }
+        });
+    };
+    const handleSubmit = (e)=>{
+        e.preventDefault();
+        if (status === 'streaming' || status === 'submitted') {
+            return;
+        }
+        const readyAttachments = attachments.filter((a)=>a.status !== 'loading');
+        if (input.trim().length === 0 && attachments.length === 0) {
+            return;
+        }
+        const files = readyAttachments.map((attachment)=>({
+                type: 'file',
+                filename: attachment.filename,
+                mediaType: attachment.mediaType,
+                url: attachment.url
+            }));
+        sendMessage({
+            text: input,
+            files
+        });
+        setInput('');
+        setAttachments([]);
+    };
+    /* -------------------------------------------------------------------------------------------------
+   * Chat title
+   * -----------------------------------------------------------------------------------------------*/ const { title, generateTitle, resetTitle } = useChatTitle({
+        chatId: id,
+        messages
+    });
+    // Automatically generate title when we have at least 1 message (user query)
+    useEffect(()=>{
+        if (messages.length >= 1 && !title) {
+            generateTitle();
+        }
+    }, [
+        messages.length,
+        title,
+        generateTitle
+    ]);
+    useEffect(()=>{
+        if (status === 'error') {
+            trackUsage('didAnswerMessage', {
+                successful: false
+            });
+        } else if (status === 'ready' && messages.length > 0 && messages[messages.length - 1]?.role === 'assistant') {
+            trackUsage('didAnswerMessage', {
+                successful: true
+            });
+        }
+    }, [
+        status,
+        messages,
+        trackUsage
+    ]);
+    const isChatAvailable = useAIAvailability();
+    return /*#__PURE__*/ jsx(ChatContext.Provider, {
+        value: {
+            isChatEnabled: isChatAvailable,
+            id,
+            status,
+            stop,
+            sendMessage,
+            ...chat,
+            messages,
+            handleSubmit,
+            input,
+            setInput,
+            handleInputChange: (e)=>setInput(e.target.value),
+            reset: ()=>{
+                stop();
+                setChatId(generateRandomId());
+                trackUsage('didStartNewChat');
+                resetTitle();
+            },
+            schemas,
+            // Chat
+            title,
+            isChatOpen,
+            openChat: ()=>{
+                setIsChatOpen(true);
+                // if this is the first open, it's a new chat
+                if (openCount === 0) {
+                    trackUsage('didStartNewChat');
+                }
+                setOpenCount((prev)=>prev + 1);
+            },
+            closeChat: ()=>setIsChatOpen(false),
+            // Attachments
+            attachments,
+            setAttachments
+        },
+        children: children
+    });
+};
+const ChatProvider = ({ children, defaultOpen = false })=>{
+    return /*#__PURE__*/ jsx(BaseChatProvider, {
+        defaultOpen: defaultOpen,
+        children: /*#__PURE__*/ jsx(SchemaChatProvider, {
+            children: /*#__PURE__*/ jsx(UploadProjectToChatProvider, {
+                children: /*#__PURE__*/ jsx(UploadFigmaToChatProvider, {
+                    children: /*#__PURE__*/ jsx(FeedbackProvider, {
+                        children: children
+                    })
+                })
+            })
+        })
+    });
+};
+function useStrapiChat() {
+    const context = useContext(ChatContext);
+    if (process.env.NODE_ENV === 'development') {
+        // In development, provide a fallback to prevent crashes during hot reloading
+        // This allows the app to continue working during HMR updates
+        if (context === undefined) {
+            console.warn('useStrapiChat was called outside of ChatProvider. This is only allowed in development mode.');
+            return {};
+        }
+    } else if (context === undefined) {
+        throw new Error('useStrapiChat must be used within a ChatProvider');
+    }
+    return context;
+}
+
+export { BaseChatProvider, ChatProvider, useStrapiChat };
+//# sourceMappingURL=ChatProvider.mjs.map