1 files changed, 71 insertions, 32 deletions
diff --git a/script/provider/markdown.lua b/script/provider/markdown.lua
index 5e564a40..7b94157e 100644
--- a/script/provider/markdown.lua
+++ b/script/provider/markdown.lua
@@ -1,3 +1,4 @@
+---@class markdown
 local mt = {}
 mt.__index = mt
 mt.__name = 'markdown'
@@ -8,50 +9,88 @@ function mt:__tostring()
     return self:string()
 end
 
-local function checkSplitLine(self)
-    if not self._splitLine then
-        return
-    end
-    self._splitLine = nil
-    if #self == 0 then
-        return
-    end
-
-    self[#self+1] = '\n---'
-end
-
+---@param language string
+---@param text string|markdown
 function mt:add(language, text)
     if not text then
         return
     end
-    text = tostring(text)
-    if #text == 0 then
-        return
-    end
-
-    checkSplitLine(self)
-
-    if language == 'md' then
-        if self._last == 'md' then
-            self[#self+1] = ''
-        end
-        self[#self+1] = text
+    if type(text) == 'table' then
+        self[#self+1] = {
+            type     = 'markdown',
+            markdown = text,
+        }
     else
-        if #self > 0 then
-            self[#self+1] = ''
+        text = tostring(text)
+        if #text == 0 then
+            return
         end
-        self[#self+1] = ('```%s\n%s\n```'):format(language, text)
+        self[#self+1] = {
+            type     = 'text',
+            language = language,
+            text     = text,
+        }
     end
+end
 
-    self._last = language
+function mt:splitLine()
+    self[#self+1] = {
+        type = 'splitline',
+    }
 end
 
 function mt:string()
-    return table.concat(self, '\n')
-end
+    local lines = {}
+    local language = 'md'
 
-function mt:splitLine()
-    self._splitLine = true
+    local function concat(markdown)
+        for _, obj in ipairs(markdown) do
+            if obj.type == 'splitline' then
+                if language ~= 'md' then
+                    lines[#lines+1] = '```'
+                    language = 'md'
+                end
+                if #lines > 0
+                and lines[#lines] ~= '---' then
+                    lines[#lines+1] = ''
+                    lines[#lines+1] = '---'
+                end
+            elseif obj.type == 'markdown' then
+                concat(obj.markdown)
+            else
+                if obj.language == language then
+                    lines[#lines+1] = obj.text
+                else
+                    if language ~= 'md' then
+                        lines[#lines+1] = '```'
+                    end
+                    if #lines > 0 then
+                        lines[#lines+1] = ''
+                    end
+                    if obj.language ~= 'md' then
+                        lines[#lines+1] = '```' .. obj.language
+                    end
+                    lines[#lines+1] = obj.text
+                    language = obj.language
+                end
+            end
+        end
+    end
+
+    concat(self)
+    if language ~= 'md' then
+        lines[#lines+1] = '```'
+    end
+    while true do
+        if lines[#lines] == '---'
+        or lines[#lines] == '' then
+            lines[#lines] = nil
+        else
+            break
+        end
+    end
+
+    return table.concat(lines, '\n')
 end
 
 return function ()