完善分类

helper/tree/tree.go

@@ -51,6 +51,15 @@ func (n *Node[T, K]) OrderByLoop(fn func(T, int), orderBy func(T, T) bool) {
 }
 
 func Root[T any, K comparable](a []T, top K, fn func(T) (child, parent K)) *Node[T, K] {
+	m := root(a, top, fn)
+	return m[top]
+}
+
+func Roots[T any, K comparable](a []T, top K, fn func(T) (child, parent K)) map[K]*Node[T, K] {
+	return root(a, top, fn)
+}
+
+func root[T any, K comparable](a []T, top K, fn func(T) (child, parent K)) map[K]*Node[T, K] {
 	m := make(map[K]*Node[T, K])
 	m[top] = &Node[T, K]{Children: new([]Node[T, K])}
 	for _, t := range a {
@@ -64,5 +73,19 @@ func Root[T any, K comparable](a []T, top K, fn func(T) (child, parent K)) *Node
 		}
 		*parent.Children = append(*parent.Children, node)
 	}
-	return m[top]
+	return m
+}
+
+func Ancestor[T any, K comparable](root map[K]*Node[T, K], top K, child *Node[T, K]) *Node[T, K] {
+	a := child
+	for {
+		if a.Parent == top {
+			return a
+		}
+		aa, ok := root[a.Parent]
+		if !ok {
+			return a
+		}
+		a = aa
+	}
 }

internal/theme/wp/components/widget/category.go

@@ -82,6 +82,7 @@ func categoryUL(h *wp.Handle, args map[string]string, conf map[any]any, categori
 	s := str.NewBuilder()
 	s.WriteString("<ul>\n")
 	isCount := conf["count"].(int64)
+	if conf["hierarchical"].(int64) == 0 {
 		for _, category := range categories {
 			count := ""
 			if isCount != 0 {
@@ -92,10 +93,68 @@ func categoryUL(h *wp.Handle, args map[string]string, conf map[any]any, categori
 	</li>
 `, category.Terms.TermId, category.Name, category.Name, count)
 		}
+	} else {
+
+		m := tree.Roots(categories, 0, func(cate models.TermsMy) (child, parent uint64) {
+			return cate.TermTaxonomyId, cate.Parent
+		})
+		cate := &tree.Node[models.TermsMy, uint64]{Data: models.TermsMy{}}
+		if h.Scene() == constraints.Category {
+			cat := h.C.Param("category")
+			i, ca := slice.SearchFirst(categories, func(my models.TermsMy) bool {
+				return cat == my.Name
+			})
+			if i > 0 {
+				cate = m[ca.TermTaxonomyId]
+			}
+		}
+
+		r := m[0]
+		categoryLi(r, cate, tree.Ancestor(m, 0, cate), isCount, s)
+	}
+
 	s.WriteString("</ul>")
 	return s.String()
 }
 
+func categoryLi(root *tree.Node[models.TermsMy, uint64], cate, roots *tree.Node[models.TermsMy, uint64], isCount int64, s *str.Builder) {
+	for _, child := range *root.Children {
+		category := child.Data
+		count := ""
+		if isCount != 0 {
+			count = fmt.Sprintf("(%d)", category.Count)
+		}
+		var class []string
+
+		if len(*child.Children) > 0 && cate.Data.TermTaxonomyId > 0 {
+			if category.TermTaxonomyId == cate.Parent {
+				class = append(class, "current-cat-parent")
+			}
+
+			if cate.Parent > 0 && category.TermTaxonomyId == roots.Data.TermTaxonomyId {
+				class = append(class, "current-cat-ancestor")
+			}
+		}
+		aria := ""
+		if category.TermTaxonomyId == cate.Data.TermTaxonomyId {
+			class = append(class, "current-cat")
+			aria = `aria-current="page"`
+		}
+		s.Sprintf(`	<li class="cat-item cat-item-%d %s">
+		<a %s href="/p/category/%s">%s %s</a>
+	</li>
+`, category.Terms.TermId, strings.Join(class, " "), aria, category.Name, category.Name, count)
+
+		if len(*child.Children) > 0 {
+			s.WriteString(`<ul class="children">
+`)
+			categoryLi(&child, cate, roots, isCount, s)
+			s.WriteString(`</ul>`)
+		}
+	}
+
+}
+
 var categoryDropdownJs = `/* <![CDATA[ */
 (function() {
 	var dropdown = document.getElementById( "%s" );