Change max body size using envvars when rendering diagramas (#1761)

yuzutech · Jul 22, 2024 · 21c486c · 21c486c
1 parent 04d7d2e
commit 21c486c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/diagrams.net/src/index.js b/diagrams.net/src/index.js
@@ -19,7 +19,7 @@ import { SyntaxError, TimeoutError, Worker } from './worker.js'
       const url = new URL(req.url, 'http://localhost') // create a URL object. The base is not important here
       const outputType = url.pathname.match(/\/(png|svg)$/)?.[1]
       if (outputType) {
-        const diagramSource = await micro.text(req, { limit: '1mb', encoding: 'utf8' })
+        const diagramSource = await micro.text(req, { limit: (process.env.KROKI_MAX_BODY_SIZE ?? '1mb'), encoding: 'utf8' })
         if (diagramSource) {
           try {
             const isPng = outputType === 'png'

diff --git a/excalidraw/src/index.js b/excalidraw/src/index.js
@@ -15,7 +15,7 @@ import { create } from './browser-instance.js'
     micro.serve(async (req, res) => {
       // TODO: add a /_status route (return excalidraw version)
       // TODO: read the diagram source as plain text
-      const diagramSource = await micro.text(req, { limit: '1mb', encoding: 'utf8' })
+      const diagramSource = await micro.text(req, { limit: (process.env.KROKI_MAX_BODY_SIZE ?? '1mb'), encoding: 'utf8' })
       if (diagramSource) {
         try {
           const svg = await worker.convert(new Task(diagramSource))

diff --git a/mermaid/src/index.js b/mermaid/src/index.js
@@ -27,7 +27,7 @@ import { create } from './browser-instance.js'
       const url = new URL(req.url, 'http://localhost') // create a URL object. The base is not important here
       const outputType = url.pathname.match(/\/(png|svg)$/)?.[1]
       if (outputType) {
-        const diagramSource = await micro.text(req, { limit: '1mb', encoding: 'utf8' })
+        const diagramSource = await micro.text(req, { limit: (process.env.KROKI_MAX_BODY_SIZE ?? '1mb'), encoding: 'utf8' })
         if (diagramSource) {
           try {
             const isPng = outputType === 'png'