@apart-tech/intelligence-core 1.21.0 → 1.21.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.d.ts +9 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -1
- package/dist/services/__tests__/node-service.test.js +3 -3
- package/dist/services/__tests__/node-service.test.js.map +1 -1
- package/dist/services/agent-cli-reference.d.ts +1 -1
- package/dist/services/agent-cli-reference.d.ts.map +1 -1
- package/dist/services/agent-cli-reference.js +2 -2
- package/dist/services/chat-capture-service.d.ts.map +1 -1
- package/dist/services/chat-capture-service.js +1 -0
- package/dist/services/chat-capture-service.js.map +1 -1
- package/dist/services/chat-tool-definitions.d.ts +178 -0
- package/dist/services/chat-tool-definitions.d.ts.map +1 -0
- package/dist/services/chat-tool-definitions.js +120 -0
- package/dist/services/chat-tool-definitions.js.map +1 -0
- package/dist/services/chat-tool-definitions.test.d.ts +2 -0
- package/dist/services/chat-tool-definitions.test.d.ts.map +1 -0
- package/dist/services/chat-tool-definitions.test.js +150 -0
- package/dist/services/chat-tool-definitions.test.js.map +1 -0
- package/dist/services/document-post-check-service.d.ts +47 -0
- package/dist/services/document-post-check-service.d.ts.map +1 -0
- package/dist/services/document-post-check-service.js +130 -0
- package/dist/services/document-post-check-service.js.map +1 -0
- package/dist/services/document-post-check-service.test.d.ts +2 -0
- package/dist/services/document-post-check-service.test.d.ts.map +1 -0
- package/dist/services/document-post-check-service.test.js +150 -0
- package/dist/services/document-post-check-service.test.js.map +1 -0
- package/dist/services/document-render-service.d.ts +27 -0
- package/dist/services/document-render-service.d.ts.map +1 -0
- package/dist/services/document-render-service.js +186 -0
- package/dist/services/document-render-service.js.map +1 -0
- package/dist/services/document-render-service.test.d.ts +2 -0
- package/dist/services/document-render-service.test.d.ts.map +1 -0
- package/dist/services/document-render-service.test.js +144 -0
- package/dist/services/document-render-service.test.js.map +1 -0
- package/dist/services/document-service.d.ts +4 -1
- package/dist/services/document-service.d.ts.map +1 -1
- package/dist/services/document-service.js +24 -1
- package/dist/services/document-service.js.map +1 -1
- package/dist/services/document-tool-handler.d.ts +44 -0
- package/dist/services/document-tool-handler.d.ts.map +1 -0
- package/dist/services/document-tool-handler.js +179 -0
- package/dist/services/document-tool-handler.js.map +1 -0
- package/dist/services/document-tool-handler.test.d.ts +2 -0
- package/dist/services/document-tool-handler.test.d.ts.map +1 -0
- package/dist/services/document-tool-handler.test.js +149 -0
- package/dist/services/document-tool-handler.test.js.map +1 -0
- package/dist/services/node-service.js +2 -2
- package/dist/services/node-service.js.map +1 -1
- package/dist/services/quota-service.d.ts +2 -0
- package/dist/services/quota-service.d.ts.map +1 -1
- package/dist/services/quota-service.js +17 -1
- package/dist/services/quota-service.js.map +1 -1
- package/dist/services/usage-service.d.ts +21 -1
- package/dist/services/usage-service.d.ts.map +1 -1
- package/dist/services/usage-service.js +37 -1
- package/dist/services/usage-service.js.map +1 -1
- package/dist/types/index.d.ts +5 -0
- package/dist/types/index.d.ts.map +1 -1
- package/package.json +11 -1
- package/prisma/schema.prisma +24 -18
|
@@ -0,0 +1,186 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Stateless document rendering service.
|
|
3
|
+
*
|
|
4
|
+
* Converts structured LLM tool-call arguments into file buffers.
|
|
5
|
+
* Each method returns { buffer, mimeType, extension, fileName }.
|
|
6
|
+
*/
|
|
7
|
+
import ExcelJS from "exceljs";
|
|
8
|
+
// eslint-disable-next-line @typescript-eslint/no-require-imports
|
|
9
|
+
import PptxGenJSDefault from "pptxgenjs";
|
|
10
|
+
import Papa from "papaparse";
|
|
11
|
+
import MarkdownIt from "markdown-it";
|
|
12
|
+
const md = new MarkdownIt();
|
|
13
|
+
export class DocumentRenderService {
|
|
14
|
+
/**
|
|
15
|
+
* Render an Excel spreadsheet (.xlsx) from structured data.
|
|
16
|
+
*/
|
|
17
|
+
async renderSpreadsheet(args) {
|
|
18
|
+
const workbook = new ExcelJS.Workbook();
|
|
19
|
+
for (const sheet of args.sheets) {
|
|
20
|
+
const ws = workbook.addWorksheet(sheet.name);
|
|
21
|
+
// Set columns with headers and optional widths
|
|
22
|
+
ws.columns = sheet.columns.map((col) => ({
|
|
23
|
+
header: col.header,
|
|
24
|
+
key: col.header,
|
|
25
|
+
width: col.width ?? 15,
|
|
26
|
+
}));
|
|
27
|
+
// Add data rows
|
|
28
|
+
for (const row of sheet.rows) {
|
|
29
|
+
ws.addRow(row);
|
|
30
|
+
}
|
|
31
|
+
// Apply formulas
|
|
32
|
+
if (sheet.formulas) {
|
|
33
|
+
for (const f of sheet.formulas) {
|
|
34
|
+
const cell = ws.getCell(f.cell);
|
|
35
|
+
cell.value = { formula: f.formula };
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
const arrayBuffer = await workbook.xlsx.writeBuffer();
|
|
40
|
+
const buffer = Buffer.from(arrayBuffer);
|
|
41
|
+
return {
|
|
42
|
+
buffer,
|
|
43
|
+
mimeType: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|
44
|
+
extension: "xlsx",
|
|
45
|
+
fileName: `${args.fileName}.xlsx`,
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
/**
|
|
49
|
+
* Render a PowerPoint presentation (.pptx) from slide definitions.
|
|
50
|
+
*/
|
|
51
|
+
async renderPresentation(args) {
|
|
52
|
+
// Handle both ESM and CJS module resolution
|
|
53
|
+
const PptxGenJS = PptxGenJSDefault.default ?? PptxGenJSDefault;
|
|
54
|
+
const pptx = new PptxGenJS();
|
|
55
|
+
for (const slideData of args.slides) {
|
|
56
|
+
const slide = pptx.addSlide();
|
|
57
|
+
switch (slideData.layout) {
|
|
58
|
+
case "title": {
|
|
59
|
+
if (slideData.title) {
|
|
60
|
+
slide.addText(slideData.title, {
|
|
61
|
+
x: 0.5,
|
|
62
|
+
y: 1.5,
|
|
63
|
+
w: 9,
|
|
64
|
+
h: 1.5,
|
|
65
|
+
fontSize: 36,
|
|
66
|
+
bold: true,
|
|
67
|
+
align: "center",
|
|
68
|
+
color: "363636",
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
if (slideData.subtitle) {
|
|
72
|
+
slide.addText(slideData.subtitle, {
|
|
73
|
+
x: 0.5,
|
|
74
|
+
y: 3.2,
|
|
75
|
+
w: 9,
|
|
76
|
+
h: 1,
|
|
77
|
+
fontSize: 20,
|
|
78
|
+
align: "center",
|
|
79
|
+
color: "666666",
|
|
80
|
+
});
|
|
81
|
+
}
|
|
82
|
+
break;
|
|
83
|
+
}
|
|
84
|
+
case "content": {
|
|
85
|
+
if (slideData.title) {
|
|
86
|
+
slide.addText(slideData.title, {
|
|
87
|
+
x: 0.5,
|
|
88
|
+
y: 0.3,
|
|
89
|
+
w: 9,
|
|
90
|
+
h: 0.8,
|
|
91
|
+
fontSize: 24,
|
|
92
|
+
bold: true,
|
|
93
|
+
color: "363636",
|
|
94
|
+
});
|
|
95
|
+
}
|
|
96
|
+
if (slideData.body?.length) {
|
|
97
|
+
slide.addText(slideData.body.map((line) => ({
|
|
98
|
+
text: line,
|
|
99
|
+
options: { bullet: true, breakLine: true },
|
|
100
|
+
})), { x: 0.5, y: 1.3, w: 9, h: 4, fontSize: 16, color: "404040", valign: "top" });
|
|
101
|
+
}
|
|
102
|
+
break;
|
|
103
|
+
}
|
|
104
|
+
case "two-column": {
|
|
105
|
+
if (slideData.title) {
|
|
106
|
+
slide.addText(slideData.title, {
|
|
107
|
+
x: 0.5,
|
|
108
|
+
y: 0.3,
|
|
109
|
+
w: 9,
|
|
110
|
+
h: 0.8,
|
|
111
|
+
fontSize: 24,
|
|
112
|
+
bold: true,
|
|
113
|
+
color: "363636",
|
|
114
|
+
});
|
|
115
|
+
}
|
|
116
|
+
if (slideData.leftColumn?.length) {
|
|
117
|
+
slide.addText(slideData.leftColumn.map((line) => ({
|
|
118
|
+
text: line,
|
|
119
|
+
options: { bullet: true, breakLine: true },
|
|
120
|
+
})), { x: 0.5, y: 1.3, w: 4.2, h: 4, fontSize: 14, color: "404040", valign: "top" });
|
|
121
|
+
}
|
|
122
|
+
if (slideData.rightColumn?.length) {
|
|
123
|
+
slide.addText(slideData.rightColumn.map((line) => ({
|
|
124
|
+
text: line,
|
|
125
|
+
options: { bullet: true, breakLine: true },
|
|
126
|
+
})), { x: 5.3, y: 1.3, w: 4.2, h: 4, fontSize: 14, color: "404040", valign: "top" });
|
|
127
|
+
}
|
|
128
|
+
break;
|
|
129
|
+
}
|
|
130
|
+
case "blank":
|
|
131
|
+
// Empty slide — intentionally left blank
|
|
132
|
+
break;
|
|
133
|
+
}
|
|
134
|
+
if (slideData.notes) {
|
|
135
|
+
slide.addNotes(slideData.notes);
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
const output = await pptx.write({ outputType: "nodebuffer" });
|
|
139
|
+
const buffer = Buffer.from(output);
|
|
140
|
+
return {
|
|
141
|
+
buffer,
|
|
142
|
+
mimeType: "application/vnd.openxmlformats-officedocument.presentationml.presentation",
|
|
143
|
+
extension: "pptx",
|
|
144
|
+
fileName: `${args.fileName}.pptx`,
|
|
145
|
+
};
|
|
146
|
+
}
|
|
147
|
+
/**
|
|
148
|
+
* Render a Word document (.docx) from markdown content.
|
|
149
|
+
* Converts markdown → HTML → DOCX via @turbodocx/html-to-docx.
|
|
150
|
+
*/
|
|
151
|
+
async renderDocument(args) {
|
|
152
|
+
const html = md.render(args.content);
|
|
153
|
+
// Dynamic import since html-to-docx may not have proper ESM exports
|
|
154
|
+
const { default: htmlToDocx } = await import("@turbodocx/html-to-docx");
|
|
155
|
+
const buffer = await htmlToDocx(html, null, {
|
|
156
|
+
table: { row: { cantSplit: true } },
|
|
157
|
+
footer: true,
|
|
158
|
+
pageNumber: true,
|
|
159
|
+
});
|
|
160
|
+
return {
|
|
161
|
+
buffer: Buffer.from(buffer),
|
|
162
|
+
mimeType: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
|
|
163
|
+
extension: "docx",
|
|
164
|
+
fileName: `${args.fileName}.docx`,
|
|
165
|
+
};
|
|
166
|
+
}
|
|
167
|
+
/**
|
|
168
|
+
* Render a CSV file from structured data.
|
|
169
|
+
*/
|
|
170
|
+
async renderCsv(args) {
|
|
171
|
+
const csv = Papa.unparse({
|
|
172
|
+
fields: args.headers,
|
|
173
|
+
data: args.rows,
|
|
174
|
+
}, {
|
|
175
|
+
delimiter: args.delimiter,
|
|
176
|
+
});
|
|
177
|
+
const buffer = Buffer.from(csv, "utf-8");
|
|
178
|
+
return {
|
|
179
|
+
buffer,
|
|
180
|
+
mimeType: "text/csv",
|
|
181
|
+
extension: "csv",
|
|
182
|
+
fileName: `${args.fileName}.csv`,
|
|
183
|
+
};
|
|
184
|
+
}
|
|
185
|
+
}
|
|
186
|
+
//# sourceMappingURL=document-render-service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-render-service.js","sourceRoot":"","sources":["../../src/services/document-render-service.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,iEAAiE;AACjE,OAAO,gBAAgB,MAAM,WAAW,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,UAAU,MAAM,aAAa,CAAC;AAgBrC,MAAM,EAAE,GAAG,IAAI,UAAU,EAAE,CAAC;AAE5B,MAAM,OAAO,qBAAqB;IAChC;;OAEG;IACH,KAAK,CAAC,iBAAiB,CAAC,IAA2B;QACjD,MAAM,QAAQ,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QAExC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,MAAM,EAAE,GAAG,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAE7C,+CAA+C;YAC/C,EAAE,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACvC,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,GAAG,EAAE,GAAG,CAAC,MAAM;gBACf,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE;aACvB,CAAC,CAAC,CAAC;YAEJ,gBAAgB;YAChB,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC7B,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjB,CAAC;YAED,iBAAiB;YACjB,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACnB,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;oBAChC,IAAI,CAAC,KAAK,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAA8B,CAAC;gBAClE,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAExC,OAAO;YACL,MAAM;YACN,QAAQ,EAAE,mEAAmE;YAC7E,SAAS,EAAE,MAAM;YACjB,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,OAAO;SAClC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,IAA4B;QACnD,4CAA4C;QAC5C,MAAM,SAAS,GAAI,gBAAwB,CAAC,OAAO,IAAI,gBAAgB,CAAC;QACxE,MAAM,IAAI,GAAG,IAAI,SAAS,EAAE,CAAC;QAE7B,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAE9B,QAAQ,SAAS,CAAC,MAAM,EAAE,CAAC;gBACzB,KAAK,OAAO,CAAC,CAAC,CAAC;oBACb,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;wBACpB,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,EAAE;4BAC7B,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,CAAC;4BACJ,CAAC,EAAE,GAAG;4BACN,QAAQ,EAAE,EAAE;4BACZ,IAAI,EAAE,IAAI;4BACV,KAAK,EAAE,QAAQ;4BACf,KAAK,EAAE,QAAQ;yBAChB,CAAC,CAAC;oBACL,CAAC;oBACD,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;wBACvB,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,EAAE;4BAChC,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,CAAC;4BACJ,CAAC,EAAE,CAAC;4BACJ,QAAQ,EAAE,EAAE;4BACZ,KAAK,EAAE,QAAQ;4BACf,KAAK,EAAE,QAAQ;yBAChB,CAAC,CAAC;oBACL,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,SAAS,CAAC,CAAC,CAAC;oBACf,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;wBACpB,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,EAAE;4BAC7B,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,CAAC;4BACJ,CAAC,EAAE,GAAG;4BACN,QAAQ,EAAE,EAAE;4BACZ,IAAI,EAAE,IAAI;4BACV,KAAK,EAAE,QAAQ;yBAChB,CAAC,CAAC;oBACL,CAAC;oBACD,IAAI,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;wBAC3B,KAAK,CAAC,OAAO,CACX,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;4BAC5B,IAAI,EAAE,IAAI;4BACV,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;yBAC3C,CAAC,CAAC,EACH,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAC7E,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,YAAY,CAAC,CAAC,CAAC;oBAClB,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;wBACpB,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,EAAE;4BAC7B,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,GAAG;4BACN,CAAC,EAAE,CAAC;4BACJ,CAAC,EAAE,GAAG;4BACN,QAAQ,EAAE,EAAE;4BACZ,IAAI,EAAE,IAAI;4BACV,KAAK,EAAE,QAAQ;yBAChB,CAAC,CAAC;oBACL,CAAC;oBACD,IAAI,SAAS,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;wBACjC,KAAK,CAAC,OAAO,CACX,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;4BAClC,IAAI,EAAE,IAAI;4BACV,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;yBAC3C,CAAC,CAAC,EACH,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAC/E,CAAC;oBACJ,CAAC;oBACD,IAAI,SAAS,CAAC,WAAW,EAAE,MAAM,EAAE,CAAC;wBAClC,KAAK,CAAC,OAAO,CACX,SAAS,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;4BACnC,IAAI,EAAE,IAAI;4BACV,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;yBAC3C,CAAC,CAAC,EACH,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAC/E,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,OAAO;oBACV,yCAAyC;oBACzC,MAAM;YACV,CAAC;YAED,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;gBACpB,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAqB,CAAC,CAAC;QAElD,OAAO;YACL,MAAM;YACN,QAAQ,EAAE,2EAA2E;YACrF,SAAS,EAAE,MAAM;YACjB,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,OAAO;SAClC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc,CAAC,IAAwB;QAC3C,MAAM,IAAI,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrC,oEAAoE;QACpE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE;YAC1C,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE;YACnC,MAAM,EAAE,IAAI;YACZ,UAAU,EAAE,IAAI;SACjB,CAAW,CAAC;QAEb,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;YAC3B,QAAQ,EAAE,yEAAyE;YACnF,SAAS,EAAE,MAAM;YACjB,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,OAAO;SAClC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,IAAmB;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC;YACvB,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,EAAE;YACD,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAEzC,OAAO;YACL,MAAM;YACN,QAAQ,EAAE,UAAU;YACpB,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,MAAM;SACjC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-render-service.test.d.ts","sourceRoot":"","sources":["../../src/services/document-render-service.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
import { describe, it, expect } from "vitest";
|
|
2
|
+
import { DocumentRenderService } from "./document-render-service.js";
|
|
3
|
+
const svc = new DocumentRenderService();
|
|
4
|
+
describe("DocumentRenderService", () => {
|
|
5
|
+
describe("renderSpreadsheet", () => {
|
|
6
|
+
it("produces a non-empty xlsx buffer with correct metadata", async () => {
|
|
7
|
+
const result = await svc.renderSpreadsheet({
|
|
8
|
+
fileName: "test",
|
|
9
|
+
sheets: [
|
|
10
|
+
{
|
|
11
|
+
name: "Sheet1",
|
|
12
|
+
columns: [{ header: "Name" }, { header: "Value", width: 20 }],
|
|
13
|
+
rows: [
|
|
14
|
+
["Alice", 100],
|
|
15
|
+
["Bob", 200],
|
|
16
|
+
],
|
|
17
|
+
},
|
|
18
|
+
],
|
|
19
|
+
});
|
|
20
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
21
|
+
expect(result.mimeType).toBe("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
|
22
|
+
expect(result.extension).toBe("xlsx");
|
|
23
|
+
expect(result.fileName).toBe("test.xlsx");
|
|
24
|
+
});
|
|
25
|
+
it("writes formulas correctly", async () => {
|
|
26
|
+
const result = await svc.renderSpreadsheet({
|
|
27
|
+
fileName: "formulas",
|
|
28
|
+
sheets: [
|
|
29
|
+
{
|
|
30
|
+
name: "Data",
|
|
31
|
+
columns: [{ header: "A" }],
|
|
32
|
+
rows: [[10], [20]],
|
|
33
|
+
formulas: [{ cell: "A3", formula: "=SUM(A1:A2)" }],
|
|
34
|
+
},
|
|
35
|
+
],
|
|
36
|
+
});
|
|
37
|
+
// The buffer is valid xlsx — we check it's non-empty
|
|
38
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
39
|
+
expect(result.fileName).toBe("formulas.xlsx");
|
|
40
|
+
});
|
|
41
|
+
it("handles multiple sheets", async () => {
|
|
42
|
+
const result = await svc.renderSpreadsheet({
|
|
43
|
+
fileName: "multi",
|
|
44
|
+
sheets: [
|
|
45
|
+
{ name: "Sheet1", columns: [{ header: "A" }], rows: [["x"]] },
|
|
46
|
+
{ name: "Sheet2", columns: [{ header: "B" }], rows: [["y"]] },
|
|
47
|
+
],
|
|
48
|
+
});
|
|
49
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
50
|
+
});
|
|
51
|
+
});
|
|
52
|
+
describe("renderPresentation", () => {
|
|
53
|
+
it("produces a non-empty pptx buffer", async () => {
|
|
54
|
+
const result = await svc.renderPresentation({
|
|
55
|
+
fileName: "slides",
|
|
56
|
+
slides: [
|
|
57
|
+
{ layout: "title", title: "Hello", subtitle: "World" },
|
|
58
|
+
{ layout: "content", title: "Details", body: ["Point 1", "Point 2"] },
|
|
59
|
+
],
|
|
60
|
+
});
|
|
61
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
62
|
+
expect(result.mimeType).toBe("application/vnd.openxmlformats-officedocument.presentationml.presentation");
|
|
63
|
+
expect(result.extension).toBe("pptx");
|
|
64
|
+
expect(result.fileName).toBe("slides.pptx");
|
|
65
|
+
});
|
|
66
|
+
it("handles two-column layout", async () => {
|
|
67
|
+
const result = await svc.renderPresentation({
|
|
68
|
+
fileName: "cols",
|
|
69
|
+
slides: [
|
|
70
|
+
{
|
|
71
|
+
layout: "two-column",
|
|
72
|
+
title: "Comparison",
|
|
73
|
+
leftColumn: ["Pro 1", "Pro 2"],
|
|
74
|
+
rightColumn: ["Con 1"],
|
|
75
|
+
},
|
|
76
|
+
],
|
|
77
|
+
});
|
|
78
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
79
|
+
});
|
|
80
|
+
it("handles blank layout", async () => {
|
|
81
|
+
const result = await svc.renderPresentation({
|
|
82
|
+
fileName: "blank",
|
|
83
|
+
slides: [{ layout: "blank" }],
|
|
84
|
+
});
|
|
85
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
86
|
+
});
|
|
87
|
+
it("includes speaker notes", async () => {
|
|
88
|
+
const result = await svc.renderPresentation({
|
|
89
|
+
fileName: "notes",
|
|
90
|
+
slides: [{ layout: "content", title: "Topic", body: ["Body"], notes: "Say this" }],
|
|
91
|
+
});
|
|
92
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
93
|
+
});
|
|
94
|
+
});
|
|
95
|
+
describe("renderDocument", () => {
|
|
96
|
+
it("produces a non-empty docx buffer from markdown", async () => {
|
|
97
|
+
const result = await svc.renderDocument({
|
|
98
|
+
fileName: "readme",
|
|
99
|
+
content: "# Hello\n\nThis is a **test** document.\n\n- Item 1\n- Item 2\n",
|
|
100
|
+
});
|
|
101
|
+
expect(result.buffer.length).toBeGreaterThan(0);
|
|
102
|
+
expect(result.mimeType).toBe("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
|
|
103
|
+
expect(result.extension).toBe("docx");
|
|
104
|
+
expect(result.fileName).toBe("readme.docx");
|
|
105
|
+
});
|
|
106
|
+
});
|
|
107
|
+
describe("renderCsv", () => {
|
|
108
|
+
it("produces correct CSV with default delimiter", async () => {
|
|
109
|
+
const result = await svc.renderCsv({
|
|
110
|
+
fileName: "data",
|
|
111
|
+
headers: ["Name", "Age"],
|
|
112
|
+
rows: [["Alice", 30], ["Bob", 25]],
|
|
113
|
+
delimiter: ",",
|
|
114
|
+
});
|
|
115
|
+
const text = result.buffer.toString("utf-8");
|
|
116
|
+
expect(text).toContain("Name,Age");
|
|
117
|
+
expect(text).toContain("Alice,30");
|
|
118
|
+
expect(result.mimeType).toBe("text/csv");
|
|
119
|
+
expect(result.extension).toBe("csv");
|
|
120
|
+
expect(result.fileName).toBe("data.csv");
|
|
121
|
+
});
|
|
122
|
+
it("uses semicolon delimiter when specified", async () => {
|
|
123
|
+
const result = await svc.renderCsv({
|
|
124
|
+
fileName: "euro",
|
|
125
|
+
headers: ["A", "B"],
|
|
126
|
+
rows: [["x", "y"]],
|
|
127
|
+
delimiter: ";",
|
|
128
|
+
});
|
|
129
|
+
const text = result.buffer.toString("utf-8");
|
|
130
|
+
expect(text).toContain("A;B");
|
|
131
|
+
});
|
|
132
|
+
it("uses tab delimiter when specified", async () => {
|
|
133
|
+
const result = await svc.renderCsv({
|
|
134
|
+
fileName: "tsv",
|
|
135
|
+
headers: ["Col1", "Col2"],
|
|
136
|
+
rows: [["a", "b"]],
|
|
137
|
+
delimiter: "\t",
|
|
138
|
+
});
|
|
139
|
+
const text = result.buffer.toString("utf-8");
|
|
140
|
+
expect(text).toContain("Col1\tCol2");
|
|
141
|
+
});
|
|
142
|
+
});
|
|
143
|
+
});
|
|
144
|
+
//# sourceMappingURL=document-render-service.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-render-service.test.js","sourceRoot":"","sources":["../../src/services/document-render-service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAErE,MAAM,GAAG,GAAG,IAAI,qBAAqB,EAAE,CAAC;AAExC,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,iBAAiB,CAAC;gBACzC,QAAQ,EAAE,MAAM;gBAChB,MAAM,EAAE;oBACN;wBACE,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;wBAC7D,IAAI,EAAE;4BACJ,CAAC,OAAO,EAAE,GAAG,CAAC;4BACd,CAAC,KAAK,EAAE,GAAG,CAAC;yBACb;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,mEAAmE,CAAC,CAAC;YAClG,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,iBAAiB,CAAC;gBACzC,QAAQ,EAAE,UAAU;gBACpB,MAAM,EAAE;oBACN;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;wBAC1B,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;wBAClB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;qBACnD;iBACF;aACF,CAAC,CAAC;YAEH,qDAAqD;YACrD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,iBAAiB,CAAC;gBACzC,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE;oBACN,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE;oBAC7D,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE;iBAC9D;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC;gBAC1C,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE;oBACN,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE;oBACtD,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE;iBACtE;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,2EAA2E,CAAC,CAAC;YAC1G,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC;gBAC1C,QAAQ,EAAE,MAAM;gBAChB,MAAM,EAAE;oBACN;wBACE,MAAM,EAAE,YAAY;wBACpB,KAAK,EAAE,YAAY;wBACnB,UAAU,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC;wBAC9B,WAAW,EAAE,CAAC,OAAO,CAAC;qBACvB;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YACpC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC;gBAC1C,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;aAC9B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC;gBAC1C,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;aACnF,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC;gBACtC,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,iEAAiE;aAC3E,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,yEAAyE,CAAC,CAAC;YACxG,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC;gBACjC,QAAQ,EAAE,MAAM;gBAChB,OAAO,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC;gBACxB,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAClC,SAAS,EAAE,GAAG;aACf,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC;gBACjC,QAAQ,EAAE,MAAM;gBAChB,OAAO,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;gBACnB,IAAI,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;gBAClB,SAAS,EAAE,GAAG;aACf,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC;gBACjC,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;gBACzB,IAAI,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;gBAClB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -14,6 +14,7 @@ import type { TagService } from "./tag-service.js";
|
|
|
14
14
|
import type { PiiEncryptionService } from "./pii-encryption-service.js";
|
|
15
15
|
import type { PiiEncryptionOptions } from "./pii-encryption-service.js";
|
|
16
16
|
import type { UsageService } from "./usage-service.js";
|
|
17
|
+
import type { ContentPolicyService } from "./content-policy-service.js";
|
|
17
18
|
/**
|
|
18
19
|
* Minimal GCS interface so core doesn't depend on @google-cloud/storage.
|
|
19
20
|
* The API package passes the real `Storage` instance which satisfies this.
|
|
@@ -24,6 +25,7 @@ export interface GcsFile {
|
|
|
24
25
|
getSignedUrl(options: any): Promise<[string, ...any[]]>;
|
|
25
26
|
exists(): Promise<[boolean, ...any[]]>;
|
|
26
27
|
download(): Promise<[Buffer, ...any[]]>;
|
|
28
|
+
save(data: Buffer, options?: any): Promise<void>;
|
|
27
29
|
}
|
|
28
30
|
export interface GcsBucket {
|
|
29
31
|
file(name: string): GcsFile;
|
|
@@ -52,8 +54,9 @@ export declare class DocumentService {
|
|
|
52
54
|
private piiEncryption?;
|
|
53
55
|
private piiOptions?;
|
|
54
56
|
private usageService?;
|
|
57
|
+
private contentPolicyService?;
|
|
55
58
|
private tenantCtx;
|
|
56
|
-
constructor(db: PrismaClient, storage: GcsStorage, bucketName: string, ocrService: OcrService, nodeService: NodeService, chunkService: ChunkService, tagService: TagService, piiEncryption?: PiiEncryptionService | undefined, piiOptions?: PiiEncryptionOptions | undefined, tenantCtx?: TenantContext, usageService?: UsageService | undefined);
|
|
59
|
+
constructor(db: PrismaClient, storage: GcsStorage, bucketName: string, ocrService: OcrService, nodeService: NodeService, chunkService: ChunkService, tagService: TagService, piiEncryption?: PiiEncryptionService | undefined, piiOptions?: PiiEncryptionOptions | undefined, tenantCtx?: TenantContext, usageService?: UsageService | undefined, contentPolicyService?: ContentPolicyService | undefined);
|
|
57
60
|
/**
|
|
58
61
|
* Create Document records and generate signed PUT URLs for direct upload.
|
|
59
62
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"document-service.d.ts","sourceRoot":"","sources":["../../src/services/document-service.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAG7D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAErD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;
|
|
1
|
+
{"version":3,"file":"document-service.d.ts","sourceRoot":"","sources":["../../src/services/document-service.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAG7D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAErD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAExE;;;;;GAKG;AACH,MAAM,WAAW,OAAO;IACtB,YAAY,CAAC,OAAO,EAAE,GAAG,GAAG,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC;IACxD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC;IACvC,QAAQ,IAAI,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC;IACxC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAClD;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC;CAC7B;AAED,MAAM,WAAW,UAAU;IACzB,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;CACjC;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAYD,qBAAa,eAAe;IAIxB,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,YAAY;IACpB,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,aAAa,CAAC;IACtB,OAAO,CAAC,UAAU,CAAC;IAEnB,OAAO,CAAC,YAAY,CAAC;IACrB,OAAO,CAAC,oBAAoB,CAAC;IAd/B,OAAO,CAAC,SAAS,CAAgB;gBAGvB,EAAE,EAAE,YAAY,EAChB,OAAO,EAAE,UAAU,EACnB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,UAAU,EACtB,WAAW,EAAE,WAAW,EACxB,YAAY,EAAE,YAAY,EAC1B,UAAU,EAAE,UAAU,EACtB,aAAa,CAAC,EAAE,oBAAoB,YAAA,EACpC,UAAU,CAAC,EAAE,oBAAoB,YAAA,EACzC,SAAS,CAAC,EAAE,aAAa,EACjB,YAAY,CAAC,EAAE,YAAY,YAAA,EAC3B,oBAAoB,CAAC,EAAE,oBAAoB,YAAA;IAKrD;;OAEG;IACG,gBAAgB,CACpB,KAAK,EAAE,qBAAqB,EAAE,EAC9B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,oBAAoB,EAAE,CAAC;IA6ClC;;;;;OAKG;IACG,gBAAgB,CACpB,WAAW,EAAE,MAAM,EAAE,GACpB,OAAO,CAAC;QAAE,SAAS,EAAE;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAAC,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAyC/E;;OAEG;IACG,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgKxD;;OAEG;IACG,aAAa,CAAC,IAAI,CAAC,EAAE;QACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAiBvB;;OAEG;IACG,cAAc,CAClB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC;QACT,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CA0BH"}
|
|
@@ -20,8 +20,9 @@ export class DocumentService {
|
|
|
20
20
|
piiEncryption;
|
|
21
21
|
piiOptions;
|
|
22
22
|
usageService;
|
|
23
|
+
contentPolicyService;
|
|
23
24
|
tenantCtx;
|
|
24
|
-
constructor(db, storage, bucketName, ocrService, nodeService, chunkService, tagService, piiEncryption, piiOptions, tenantCtx, usageService) {
|
|
25
|
+
constructor(db, storage, bucketName, ocrService, nodeService, chunkService, tagService, piiEncryption, piiOptions, tenantCtx, usageService, contentPolicyService) {
|
|
25
26
|
this.db = db;
|
|
26
27
|
this.storage = storage;
|
|
27
28
|
this.bucketName = bucketName;
|
|
@@ -32,6 +33,7 @@ export class DocumentService {
|
|
|
32
33
|
this.piiEncryption = piiEncryption;
|
|
33
34
|
this.piiOptions = piiOptions;
|
|
34
35
|
this.usageService = usageService;
|
|
36
|
+
this.contentPolicyService = contentPolicyService;
|
|
35
37
|
this.tenantCtx = tenantCtx ?? { organizationId: SINGLE_TENANT_ORG_ID };
|
|
36
38
|
}
|
|
37
39
|
/**
|
|
@@ -168,6 +170,27 @@ export class DocumentService {
|
|
|
168
170
|
if (this.usageService) {
|
|
169
171
|
await this.usageService.incrementOcrPages(this.tenantCtx.organizationId, pageCount);
|
|
170
172
|
}
|
|
173
|
+
// 5c. Content policy evaluation on OCR-extracted text
|
|
174
|
+
if (this.contentPolicyService) {
|
|
175
|
+
try {
|
|
176
|
+
const policyResult = await this.contentPolicyService.evaluate(ocrResult.text);
|
|
177
|
+
if (!policyResult.allowed) {
|
|
178
|
+
const block = policyResult.violations.find((v) => v.action === "block");
|
|
179
|
+
await this.db.document.update({
|
|
180
|
+
where: { id: doc.id },
|
|
181
|
+
data: {
|
|
182
|
+
status: "failed",
|
|
183
|
+
errorMessage: `Blocked by content policy: ${block?.policyName ?? "unknown"}`,
|
|
184
|
+
},
|
|
185
|
+
});
|
|
186
|
+
return;
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
catch (err) {
|
|
190
|
+
console.error(`[DocumentService] Content policy evaluation failed for ${doc.id}:`, err);
|
|
191
|
+
// Non-fatal — proceed without blocking
|
|
192
|
+
}
|
|
193
|
+
}
|
|
171
194
|
// 6. Create node with extracted text
|
|
172
195
|
const node = await this.nodeService.create({
|
|
173
196
|
type: "document",
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"document-service.js","sourceRoot":"","sources":["../../src/services/document-service.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"document-service.js","sourceRoot":"","sources":["../../src/services/document-service.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AA2CvD,MAAM,oBAAoB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAC1D,MAAM,WAAW,GAA2B;IAC1C,iBAAiB,EAAE,KAAK;IACxB,yEAAyE,EAAE,MAAM;IACjF,2EAA2E,EAAE,MAAM;IACnF,WAAW,EAAE,KAAK;IAClB,YAAY,EAAE,MAAM;IACpB,YAAY,EAAE,MAAM;CACrB,CAAC;AAEF,MAAM,OAAO,eAAe;IAIhB;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IAEA;IACA;IAdF,SAAS,CAAgB;IAEjC,YACU,EAAgB,EAChB,OAAmB,EACnB,UAAkB,EAClB,UAAsB,EACtB,WAAwB,EACxB,YAA0B,EAC1B,UAAsB,EACtB,aAAoC,EACpC,UAAiC,EACzC,SAAyB,EACjB,YAA2B,EAC3B,oBAA2C;QAX3C,OAAE,GAAF,EAAE,CAAc;QAChB,YAAO,GAAP,OAAO,CAAY;QACnB,eAAU,GAAV,UAAU,CAAQ;QAClB,eAAU,GAAV,UAAU,CAAY;QACtB,gBAAW,GAAX,WAAW,CAAa;QACxB,iBAAY,GAAZ,YAAY,CAAc;QAC1B,eAAU,GAAV,UAAU,CAAY;QACtB,kBAAa,GAAb,aAAa,CAAuB;QACpC,eAAU,GAAV,UAAU,CAAuB;QAEjC,iBAAY,GAAZ,YAAY,CAAe;QAC3B,yBAAoB,GAApB,oBAAoB,CAAuB;QAEnD,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,EAAE,cAAc,EAAE,oBAAoB,EAAE,CAAC;IACzE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,KAA8B,EAC9B,SAAiB;QAEjB,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC;YAChD,6CAA6C;YAC7C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACxC,IAAI,EAAE;oBACJ,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,aAAa,EAAE,IAAI,CAAC,SAAS;oBAC7B,WAAW,EAAE,EAAE,EAAE,oBAAoB;oBACrC,MAAM,EAAE,WAAW;oBACnB,SAAS;oBACT,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc;iBAC9C;aACF,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,IAAI,GAAG,CAAC,EAAE,aAAa,GAAG,EAAE,CAAC;YAEjF,oCAAoC;YACpC,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE,EAAE,WAAW,EAAE;aACtB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACpD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACzC,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC;gBAC7C,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,oBAAoB;gBAC1C,WAAW,EAAE,IAAI,CAAC,QAAQ;aAC3B,CAAC,CAAC;YAEH,OAAO,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,SAAS;gBACT,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,gBAAgB,CACpB,WAAqB;QAErB,OAAO,CAAC,IAAI,CACV,iFAAiF,WAAW,CAAC,MAAM,cAAc;YACjH,iFAAiF,CAClF,CAAC;QACF,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAqC,EAAE,CAAC;QAEtD,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAC3C,KAAK,EAAE,EAAE,EAAE,EAAE;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;gBAC1C,SAAS;YACX,CAAC;YAED,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,GAAG,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;gBAC9D,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,kBAAkB,GAAG,CAAC,MAAM,yCAAyC,CAAC,CAAC;gBACnG,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;gBAClD,SAAS;YACX,CAAC;YAED,2BAA2B;YAC3B,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE;aAC/B,CAAC,CAAC;YAEH,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;YAEpD,yCAAyC;YACzC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACzC,OAAO,CAAC,KAAK,CAAC,kCAAkC,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,UAAkB;QACtC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC3C,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,YAAY,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,CAAC;YACH,+BAA+B;YAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACpD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,0BAA0B,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;YAC/D,CAAC;YAED,kCAAkC;YAClC,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC9C,MAAM,WAAW,GAAG,UAAU,CAAC,QAAQ,CAAC;iBACrC,MAAM,CAAC,UAAU,CAAC;iBAClB,MAAM,CAAC,KAAK,CAAC,CAAC;YAEjB,wCAAwC;YACxC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAChD,KAAK,EAAE;oBACL,WAAW;oBACX,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE;oBACnB,MAAM,EAAE,OAAO;iBAChB;aACF,CAAC,CAAC;YAEH,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;oBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;oBACrB,IAAI,EAAE;wBACJ,WAAW;wBACX,MAAM,EAAE,OAAO;wBACf,MAAM,EAAE,QAAQ,CAAC,MAAM;qBACxB;iBACF,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,sCAAsC;YACtC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC;gBAC3C,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,oBAAoB;aAC3C,CAAC,CAAC;YAEH,SAAS;YACT,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAE7D,kCAAkC;YAClC,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;YAEzD,2BAA2B;YAC3B,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,MAAM,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;YACtF,CAAC;YAED,sDAAsD;YACtD,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;gBAC9B,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC9E,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;wBAC1B,MAAM,KAAK,GAAG,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC;wBACxE,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;4BAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;4BACrB,IAAI,EAAE;gCACJ,MAAM,EAAE,QAAQ;gCAChB,YAAY,EAAE,8BAA8B,KAAK,EAAE,UAAU,IAAI,SAAS,EAAE;6BAC7E;yBACF,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,0DAA0D,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;oBACxF,uCAAuC;gBACzC,CAAC;YACH,CAAC;YAED,qCAAqC;YACrC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACzC,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,GAAG,CAAC,QAAQ;gBACnB,OAAO,EAAE,SAAS,CAAC,IAAI;gBACvB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE;oBACR,gBAAgB,EAAE,GAAG,CAAC,EAAE;oBACxB,QAAQ,EAAE,SAAS,CAAC,KAAK;oBACzB,QAAQ,EAAE,GAAG,CAAC,QAAQ;iBACvB;aACF,CAAC,CAAC;YAEH,oBAAoB;YACpB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;YAC/E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;gBAC9D,oCAAoC;YACtC,CAAC;YAED,2BAA2B;YAC3B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAC7B,IAAI,CAAC,EAAE,EACP;oBACE;wBACE,OAAO,EAAE,iBAAiB;wBAC1B,SAAS,EAAE,GAAG,CAAC,EAAE;wBACjB,KAAK,EAAE;4BACL,QAAQ,EAAE,GAAG,CAAC,QAAQ;4BACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ;4BACtB,WAAW,EAAE,GAAG,CAAC,WAAW;yBAC7B;qBACF;oBACD;wBACE,OAAO,EAAE,oBAAoB;wBAC7B,SAAS,EAAE,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,SAAS;qBAClD;iBACF,EACD,kBAAkB,CACnB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;gBAC7D,oCAAoC;YACtC,CAAC;YAED,4BAA4B;YAC5B,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE;oBACJ,WAAW;oBACX,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,SAAS,CAAC,KAAK;oBACzB,WAAW,EAAE,SAAS,CAAC,QAAQ;oBAC/B,YAAY,EAAE,SAAS,CAAC,SAAS;oBACjC,SAAS;oBACT,MAAM,EAAE,IAAI,CAAC,EAAE;iBAChB;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,iBAAiB;YACjB,MAAM,YAAY,GAChB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnD,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,QAAQ;oBAChB,YAAY;iBACb;aACF,CAAC,CAAC;YACH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,IAInB;QACC,MAAM,KAAK,GAA4B,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;YACtB,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC;QAC9B,CAAC;QACD,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,EAAE,WAAW,EAAE,CAAC;YACtB,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACvC,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC/B,KAAK;YACL,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAClB,UAAkB;QAOlB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC3C,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,YAAY,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAE7C,MAAM,CAAC,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC;YAC/C,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,oBAAoB;YAC1C,mBAAmB,EAAE,yBAAyB,GAAG,CAAC,QAAQ,GAAG;SAC9D,CAAC,CAAC;QAEH,OAAO;YACL,WAAW;YACX,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,SAAS,EAAE,oBAAoB,GAAG,IAAI;SACvC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Orchestrates document creation tool calls: policy check → PII check →
|
|
3
|
+
* render → GCS upload → Document record creation → signed download URL.
|
|
4
|
+
*/
|
|
5
|
+
import type { PrismaClient } from "@prisma/client";
|
|
6
|
+
import type { TenantContext } from "../db/tenant.js";
|
|
7
|
+
import type { GcsStorage } from "./document-service.js";
|
|
8
|
+
import type { DocumentRenderService } from "./document-render-service.js";
|
|
9
|
+
import type { ContentPolicyService } from "./content-policy-service.js";
|
|
10
|
+
import type { PiiDetectorService } from "./pii-detector-service.js";
|
|
11
|
+
export interface DocumentToolInput {
|
|
12
|
+
toolName: string;
|
|
13
|
+
args: unknown;
|
|
14
|
+
sessionId: string;
|
|
15
|
+
messageId?: string;
|
|
16
|
+
model?: string;
|
|
17
|
+
userId: string;
|
|
18
|
+
}
|
|
19
|
+
export interface DocumentToolResult {
|
|
20
|
+
success: boolean;
|
|
21
|
+
documentId?: string;
|
|
22
|
+
downloadUrl?: string;
|
|
23
|
+
fileName?: string;
|
|
24
|
+
mimeType?: string;
|
|
25
|
+
policyViolation?: boolean;
|
|
26
|
+
policyMessage?: string;
|
|
27
|
+
piiWarning?: boolean;
|
|
28
|
+
error?: string;
|
|
29
|
+
}
|
|
30
|
+
export declare class DocumentToolHandler {
|
|
31
|
+
private renderService;
|
|
32
|
+
private contentPolicyService;
|
|
33
|
+
private piiDetector;
|
|
34
|
+
private storage;
|
|
35
|
+
private bucketName;
|
|
36
|
+
private tenantCtx;
|
|
37
|
+
private db;
|
|
38
|
+
private piiOutputPolicy;
|
|
39
|
+
constructor(renderService: DocumentRenderService, contentPolicyService: ContentPolicyService | undefined, piiDetector: PiiDetectorService | undefined, storage: GcsStorage, bucketName: string, tenantCtx: TenantContext, db: PrismaClient, piiOutputPolicy?: "warn" | "block" | "strip");
|
|
40
|
+
handleToolCall(input: DocumentToolInput): Promise<DocumentToolResult>;
|
|
41
|
+
private validateArgs;
|
|
42
|
+
private render;
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=document-tool-handler.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"document-tool-handler.d.ts","sourceRoot":"","sources":["../../src/services/document-tool-handler.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAEnD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,KAAK,EAAE,qBAAqB,EAAgB,MAAM,8BAA8B,CAAC;AACxF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AASpE,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAID,qBAAa,mBAAmB;IAE5B,OAAO,CAAC,aAAa;IACrB,OAAO,CAAC,oBAAoB;IAC5B,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,eAAe;gBAPf,aAAa,EAAE,qBAAqB,EACpC,oBAAoB,EAAE,oBAAoB,GAAG,SAAS,EACtD,WAAW,EAAE,kBAAkB,GAAG,SAAS,EAC3C,OAAO,EAAE,UAAU,EACnB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,aAAa,EACxB,EAAE,EAAE,YAAY,EAChB,eAAe,GAAE,MAAM,GAAG,OAAO,GAAG,OAAgB;IAGxD,cAAc,CAAC,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAmH3E,OAAO,CAAC,YAAY;YA8BN,MAAM;CAcrB"}
|