resolves #201 resolve relative PlantUML includes from the diagram directory

src/asciidoctor-kroki.js

@@ -77,7 +77,7 @@ const processKroki = (processor, parent, attrs, diagramType, diagramText, contex
       if (plantUmlInclude) {
         diagramText = `!include ${plantUmlInclude}\n${diagramText}`
       }
-      diagramText = require('./preprocess.js').preprocessPlantUML(diagramText, context, doc.getBaseDir())
+      diagramText = require('./preprocess.js').preprocessPlantUML(diagramText, context, diagramDir)
     }
   }
   const blockId = attrs.id

src/preprocess.js

@@ -83,13 +83,13 @@ function removePlantUmlTags (diagramText) {
 /**
  * @param {string} diagramText
  * @param {any} context
- * @param {string} baseDir - base directory
+ * @param {string} diagramDir - diagram base directory
  * @returns {string}
  */
-module.exports.preprocessPlantUML = function (diagramText, context, baseDir = '.') {
+module.exports.preprocessPlantUML = function (diagramText, context, diagramDir = '') {
   const includeOnce = []
   const includeStack = []
-  diagramText = preprocessPlantUmlIncludes(diagramText, baseDir, includeOnce, includeStack, context.vfs)
+  diagramText = preprocessPlantUmlIncludes(diagramText, diagramDir, includeOnce, includeStack, context.vfs)
   return removePlantUmlTags(diagramText)
 }
 

test/fixtures/docs/diagrams/hello.puml

@@ -0,0 +1,3 @@
+!include ./style.puml
+
+Bob->Alice: Hello

test/fixtures/docs/diagrams/style.puml

@@ -0,0 +1 @@
+skinparam monochrome true

test/fixtures/docs/hello.adoc

@@ -0,0 +1 @@
+plantuml::diagrams/hello.puml[]

test/fixtures/plantuml/hello.puml

@@ -0,0 +1,2 @@
+!include ./style-general.iuml
+Bob->Alice: Hello

test/preprocess.spec.js

@@ -13,6 +13,9 @@ const url = require('url')
 
 chai.use(dirtyChai)
 
+const asciidoctorKroki = require('../src/asciidoctor-kroki.js')
+const asciidoctor = require('@asciidoctor/core')()
+
 const { preprocessVegaLite } = require('../src/preprocess.js')
 
 describe('Vega-Lite preprocessing', () => {
@@ -520,4 +523,12 @@ skinparam BackgroundColor black
       here -> there`
     )
   })
+
+  it('should resolve PlantUML includes from the diagram directory', () => {
+    const registry = asciidoctor.Extensions.create()
+    asciidoctorKroki.register(registry)
+    const file = path.join(__dirname, 'fixtures', 'docs', 'hello.adoc')
+    const html = asciidoctor.convertFile(file, { safe: 'safe', extension_registry: registry, to_file: false })
+    expect(html).to.contain('https://kroki.io/plantuml/svg/eNorzs7MK0gsSsxVyM3Py0_OKMrPTVUoKSpN5eJyyk_StXPMyUxOtVLwSM3JyQcAc1EPvA==')
+  })
 })

test/test.spec.js

@@ -98,6 +98,21 @@ alice -> bob
       expect(html).to.contain(`https://kroki.io/plantuml/svg/${encodeText(diagramText)}`)
       expect(html).to.contain('<div class="imageblock sequence kroki-format-svg kroki">')
     }).timeout(5000)
+    it('should convert a PlantUML diagram and resolve include relative to diagram directory', () => {
+      const file = ospath.join(__dirname, 'fixtures', 'plantuml', 'hello.puml')
+      const diagramText = fs.readFileSync(file, 'utf8')
+        .replace(/^!include (.*)\r?\n/m, fs.readFileSync(ospath.join(__dirname, 'fixtures', 'plantuml', 'style-general.iuml'), 'utf8') + '\n')
+      const input = `plantuml::${file}[svg,role=sequence]`
+      const registry = asciidoctor.Extensions.create()
+      asciidoctorKroki.register(registry)
+      const html = asciidoctor.convert(input, {
+        safe: 'safe',
+        extension_registry: registry,
+        base_dir: ospath.join(__dirname, 'fixtures')
+      })
+      expect(html).to.contain(`https://kroki.io/plantuml/svg/${encodeText(diagramText)}`)
+      expect(html).to.contain('<div class="imageblock sequence kroki-format-svg kroki">')
+    }).timeout(5000)
     it('should convert a diagram with a relative path to an image', () => {
       const input = `
 :imagesdir: .asciidoctor/kroki