refactor extension setup; encapsulate extension in LocationService

1 files changed, 5 insertions, 3 deletions

diff --git a/src/Locations/FoldersLocationService.kt b/src/Locations/FoldersLocationService.kt
index 6f3954c4..4f56475c 100644
--- a/src/Locations/FoldersLocationService.kt
+++ b/src/Locations/FoldersLocationService.kt
@@ -2,10 +2,12 @@ package org.jetbrains.dokka
 
 import java.io.File
 
-public fun FoldersLocationService(root: String): FoldersLocationService = FoldersLocationService(File(root))
-public class FoldersLocationService(val root: File) : FileLocationService {
+public fun FoldersLocationService(root: String): FoldersLocationService = FoldersLocationService(File(root), "")
+public class FoldersLocationService(val root: File, val extension: String) : FileLocationService {
+    override fun withExtension(newExtension: String): LocationService = FoldersLocationService(root, newExtension)
+
     override fun location(node: DocumentationNode): FileLocation {
-        return FileLocation(File(root, relativePathToNode(node)))
+        return FileLocation(File(root, relativePathToNode(node)).appendExtension(extension))
     }
 }