1 files changed, 99 insertions, 0 deletions
diff --git a/src/client/views/nodes/DataVizBox/DocCreatorMenu/Backend/TemplateManager.ts b/src/client/views/nodes/DataVizBox/DocCreatorMenu/Backend/TemplateManager.ts
new file mode 100644
index 000000000..6d63078a8
--- /dev/null
+++ b/src/client/views/nodes/DataVizBox/DocCreatorMenu/Backend/TemplateManager.ts
@@ -0,0 +1,99 @@
+import { action, makeAutoObservable } from 'mobx';
+import { Col } from '../DocCreatorMenu';
+import { FieldSettings } from '../TemplateFieldTypes/TemplateField';
+import { Template } from '../Template';
+import { NumListCast } from '../../../../../../fields/Doc';
+import { DataVizBox } from '../../DataVizBox';
+import { TemplateFieldType } from '../TemplateBackend';
+import { TemplateMenuGPTManager } from './TemplateMenuGPTManager';
+
+export type Conditional = {
+    field: string;
+    operator: '=' | '>' | '<';
+    condition: string;
+    target: 'self' | 'template';
+    attribute: string;
+    value: string;
+}
+
+export class TemplateManager {
+
+    templates: Template[] = [];
+
+    fieldConditions: Record<string, Conditional[]> = {};
+
+    constructor(templateSettings: FieldSettings[]) {
+        makeAutoObservable(this);
+        this.templates = this.initializeTemplates(templateSettings);
+    }
+
+    initializeTemplates = (templateSettings: FieldSettings[]) => templateSettings.map(settings => {
+        return new Template(settings, this.fieldConditions)});
+
+    getValidTemplates = (cols: Col[]) => this.templates.filter(template => template.isValidTemplate(cols));
+
+    addTemplate = (newTemplate: Template) => this.templates.push(newTemplate);
+
+    removeTemplate = (template: Template) => {
+        this.templates.splice(this.templates.indexOf(template), 1);
+        template.cleanup();
+    };
+
+    addFieldCondition = (fieldTitle: string, condition: Conditional) => {
+        if (this.fieldConditions[fieldTitle] === undefined) {
+            this.fieldConditions[fieldTitle] = [condition];
+        } else {
+            this.fieldConditions[fieldTitle].push(condition);
+        }
+    }
+
+    removeFieldCondition = (fieldTitle: string, condition: Conditional) => {
+        if (this.fieldConditions[fieldTitle]) {
+            this.fieldConditions[fieldTitle] = this.fieldConditions[fieldTitle].filter(cond => cond !== condition);
+        }
+    }
+
+    createDocsFromTemplate = action((dv: DataVizBox, template: Template, csvColumns: Col[], GPTManager: TemplateMenuGPTManager, debug: boolean = false) => {
+        const fields = Array.from(Object.keys(dv.records[0]));
+
+        const processContent = (content: { [title: string]: string }) => {
+            const templateCopy = template.cloneBase();
+
+            fields
+                .filter(title => title)
+                .forEach(title => {
+                    const field = templateCopy.getFieldByTitle(title);
+                    field && field.setContent(content[title], field.viewType);
+                });
+
+            const gptFunc = (type: TemplateFieldType) => (type === TemplateFieldType.VISUAL ? GPTManager.renderGPTImageCall : GPTManager.renderGPTTextCall);
+            const gptPromises = csvColumns
+                .filter(field => field.type !== TemplateFieldType.UNSET && field.AIGenerated)
+                .map(field => {
+                    const templateField = templateCopy.getFieldByTitle(field.title);
+                    if (templateField !== undefined) {
+                        return gptFunc(field.type)(templateCopy, field, templateField.getID);
+                    }
+                });
+
+            return Promise.all(gptPromises)
+        };
+
+        const rowContents = debug
+            ? [{}, {}, {}, {}]
+            : NumListCast(dv.layoutDoc.dataViz_selectedRows).map(row =>
+                  fields.reduce(
+                      (values, col) => {
+                          values[col] = dv.records[row][col];
+                          return values;
+                      },
+                      {} as { [title: string]: string }
+                  )
+              );
+        return Promise.all(rowContents.map(processContent)).then(
+            action(renderedDocs => {
+                return renderedDocs;
+            })
+        );
+    });
+}