fix: sub, sup tags highlight

Author: kawre

lua/leetcode-ui/group/tag/init.lua

@@ -59,14 +59,14 @@ function Tag.normalize(text)
     return norm
 end
 
-function Tag:add_indent(item)
+function Tag:add_indent(item, text)
     if item.class and item.class.name == "LeetLine" then
-        table.insert(item._texts, 1, Indent("\t", "leetcode_indent"))
+        table.insert(item._texts, 1, Indent(text or "\t", "leetcode_indent"))
         return
     end
 
     for _, c in ipairs(item:contents()) do
-        self:add_indent(c)
+        self:add_indent(c, text)
     end
 end
 
@@ -122,7 +122,7 @@ function Tag:get_el_data(node)
     return { tag = tag, attrs = attrs }
 end
 
-function Tag:parse_helper() --
+function Tag:parse_node() --
     ---@param child TSNode
     for child in self.node:iter_children() do
         local ntype = child:type()
@@ -212,7 +212,7 @@ function Tag:init(text, opts, node, tags) --
 
     Tag.super.init(self, {}, opts)
 
-    self:parse_helper()
+    self:parse_node()
 end
 
 ---@type fun(text: string, opts: lc.ui.opts, node: TSNode, tags: lc.ui.Tag[]): lc.ui.Tag

lua/leetcode-ui/group/tag/li.lua

@@ -29,6 +29,7 @@ function Li:init(text, opts, node, tags)
     end
 
     local parent = self.tags[r]
+
     if parent then
         if parent.order then
             self.pre = parent.order .. ". "

lua/leetcode-ui/group/tag/pre.lua

@@ -10,11 +10,8 @@ local Pre = Tag:extend("LeetTagPre")
 function Pre:contents()
     local items = Pre.super.contents(self)
 
-    for _, value in ipairs(items) do
-        for _, line in ipairs(value:contents()) do
-            local indent = Line():append("\t▎\t", "leetcode_indent")
-            table.insert(line._texts, 1, indent)
-        end
+    for _, item in ipairs(items) do
+        self:add_indent(item, "\t▎\t")
     end
 
     return items

lua/leetcode-ui/group/tag/sub.lua

@@ -5,14 +5,10 @@ local log = require("leetcode.logger")
 ---@class lc.ui.Tag.sub : lc.ui.Tag
 local Sub = Tag:extend("LeetTagSub")
 
-function Sub:contents()
-    local Group = require("leetcode-ui.group")
-    local grp = Group()
+function Sub:parse_node()
+    self:append("_")
 
-    grp:append("_", "leetcode_alt")
-    grp:append(self:content(), "Number")
-
-    return grp:contents()
+    Sub.super.parse_node(self)
 end
 
 ---@type fun(): lc.ui.Tag.sub

lua/leetcode-ui/group/tag/sup.lua

@@ -1,18 +1,15 @@
 local Tag = require("leetcode-ui.group.tag")
+local Line = require("leetcode-ui.line")
 
 local log = require("leetcode.logger")
 
 ---@class lc.ui.Tag.sup : lc.ui.Tag
 local Sup = Tag:extend("LeetTagSup")
 
-function Sup:contents()
-    local Group = require("leetcode-ui.group")
-    local grp = Group()
+function Sup:parse_node()
+    self:append("^")
 
-    grp:append("^", "leetcode_alt")
-    grp:append(self:content(), "Number")
-
-    return grp:contents()
+    Sup.super.parse_node(self)
 end
 
 ---@type fun(): lc.ui.Tag.sup

lua/leetcode/theme/init.lua

@@ -33,8 +33,8 @@ local highlights = {
     -- ol = "",
     -- li = "",
     -- font = "",
-    -- sup = "",
-    -- sub = "",
+    sup = "header",
+    sub = "header",
     -- small = "",
     -- div = "",
 }