优化代码

helper/func.go

@@ -1,6 +1,7 @@
 package helper
 
 import (
+	"context"
 	"fmt"
 	str "github.com/fthvgb1/wp-go/helper/strings"
 	"net/url"
@@ -94,3 +95,15 @@ func ToBool[T comparable](t T) bool {
 	var vv T
 	return vv != t
 }
+
+func GetValFromContext[V, K any](ctx context.Context, k K, defaults V) V {
+	v := ctx.Value(k)
+	if v == nil {
+		return defaults
+	}
+	vv, ok := v.(V)
+	if !ok {
+		return defaults
+	}
+	return vv
+}

helper/func_test.go

@@ -1,6 +1,7 @@
 package helper
 
 import (
+	"context"
 	"fmt"
 	"reflect"
 	"testing"
@@ -235,3 +236,34 @@ func TestIsZeros(t *testing.T) {
 		}
 	})
 }
+
+func TestGetValFromContext(t *testing.T) {
+	type args[K any, V any] struct {
+		ctx      context.Context
+		k        K
+		defaults V
+	}
+	type testCase[K any, V any] struct {
+		name string
+		args args[K, V]
+		want V
+	}
+	tests := []testCase[string, int]{
+		{
+			name: "t1",
+			args: args[string, int]{
+				ctx:      context.WithValue(context.Background(), "kk", 1),
+				k:        "kk",
+				defaults: 0,
+			},
+			want: 1,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := GetValFromContext(tt.args.ctx, tt.args.k, tt.args.defaults); !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("GetValFromContext() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}

helper/html/html.go

@@ -193,19 +193,15 @@ func RenderedHtml(t *template.Template, data map[string]any) (r string, err erro
 	return
 }
 
-func BuildOptions[T any, K comparable](a []T, selected K, fn func(T) (K, any), attr ...string) string {
+func BuildOptions[T any, K comparable](a []T, selected K, fn func(T) (K, any, string)) string {
 	s := strings2.NewBuilder()
-	att := ""
-	if len(attr) > 0 {
-		att = strings.Join(attr, " ")
-	}
 	for _, t := range a {
-		k, v := fn(t)
+		k, v, attr := fn(t)
 		ss := ""
 		if k == selected {
 			ss = "selected"
 		}
-		s.Sprintf(`<option %s %s value="%v">%v</option>`, ss, att, v, k)
+		s.Sprintf(`<option %s %s value="%v">%v</option>`, ss, attr, v, k)
 	}
 	return s.String()
 }

internal/pkg/dao/common.go

@@ -2,6 +2,7 @@ package dao
 
 import (
 	"context"
+	"github.com/fthvgb1/wp-go/helper"
 	"github.com/fthvgb1/wp-go/internal/pkg/constraints"
 	"github.com/fthvgb1/wp-go/internal/pkg/models"
 	"github.com/fthvgb1/wp-go/internal/wpconfig"
@@ -32,11 +33,17 @@ func CategoriesAndTags(a ...any) (terms []models.TermsMy, err error) {
 			in = []any{"post_tag"}
 		}
 	}
-	terms, err = model.Finds[models.TermsMy](ctx, model.Conditions(
-		model.Where(model.SqlBuilder{
-			{"tt.count", ">", "0", "int"},
+	w := model.SqlBuilder{
 		{"tt.taxonomy", "in", ""},
-		}),
+	}
+	if helper.GetValFromContext(ctx, "onlyTop", false) {
+		w = append(w, []string{"tt.parent", "=", "0", "int"})
+	}
+	if !helper.GetValFromContext(ctx, "showCountZero", false) {
+		w = append(w, []string{"tt.count", ">", "0", "int"})
+	}
+	terms, err = model.Finds[models.TermsMy](ctx, model.Conditions(
+		model.Where(w),
 		model.Fields("t.term_id"),
 		model.Order(model.SqlBuilder{{"t.name", "asc"}}),
 		model.Join(model.SqlBuilder{

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

@@ -84,16 +84,27 @@ 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)
+	currentCate := models.TermsMy{}
+	if h.Scene() == constraints.Category {
+		cat := h.C.Param("category")
+		_, currentCate = slice.SearchFirst(categories, func(my models.TermsMy) bool {
+			return cat == my.Name
+		})
+	}
 	if conf["hierarchical"].(int64) == 0 {
 		for _, category := range categories {
 			count := ""
 			if isCount != 0 {
 				count = fmt.Sprintf("(%d)", category.Count)
 			}
-			s.Sprintf(`	<li class="cat-item cat-item-%d">
+			current := ""
+			if category.TermTaxonomyId == currentCate.TermTaxonomyId {
+				current = "current-cat"
+			}
+			s.Sprintf(`	<li class="cat-item cat-item-%d %s">
 		<a href="/p/category/%s">%s %s</a>
 	</li>
-`, category.Terms.TermId, category.Name, category.Name, count)
+`, category.Terms.TermId, current, category.Name, category.Name, count)
 		}
 	} else {
 
@@ -101,16 +112,9 @@ func categoryUL(h *wp.Handle, args map[string]string, conf map[any]any, categori
 			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]
+		if currentCate.TermTaxonomyId > 0 {
+			cate = m[currentCate.TermTaxonomyId]
 		}
-		}
-
 		r := m[0]
 		categoryLi(r, cate, tree.Ancestor(m, 0, cate), isCount, s)
 	}

model/query.go

@@ -2,6 +2,7 @@ package model
 
 import (
 	"context"
+	"github.com/fthvgb1/wp-go/helper"
 	"github.com/fthvgb1/wp-go/helper/number"
 	str "github.com/fthvgb1/wp-go/helper/strings"
 	"golang.org/x/exp/constraints"
@@ -64,20 +65,12 @@ func pagination[T Model](db dbQuery, ctx context.Context, q QueryCondition, page
 		return
 	}
 	q.Offset = offset
-	m := ctx.Value("handle=>toMap")
+	m := helper.GetValFromContext[*[]map[string]string](ctx, "handle=>toMap", nil)
 	if m == nil {
 		r, err = finds[T](db, ctx, q)
 		return
 	}
-	mm, ok := m.(*[]map[string]string)
-	if ok {
-		mx, er := findToStringMap[T](db, ctx, q)
-		if er != nil {
-			err = er
-			return
-		}
-		*mm = mx
-	}
+	*m, err = findToStringMap[T](db, ctx, q)
 	return
 }
 

model/sqxquery.go

@@ -3,6 +3,7 @@ package model
 import (
 	"context"
 	"fmt"
+	"github.com/fthvgb1/wp-go/helper"
 	"github.com/fthvgb1/wp-go/helper/slice"
 	str "github.com/fthvgb1/wp-go/helper/strings"
 	"github.com/jmoiron/sqlx"
@@ -35,11 +36,9 @@ func SetGet(db *SqlxQuery, fn func(context.Context, any, string, ...any) error)
 }
 
 func (r *SqlxQuery) Selects(ctx context.Context, dest any, sql string, params ...any) error {
-	v := ctx.Value("handle=>")
-	if v != nil {
-		vv, ok := v.(string)
-		if ok && vv != "" {
-			switch vv {
+	v := helper.GetValFromContext(ctx, "handle=>", "")
+	if v != "" {
+		switch v {
 		case "string":
 			return ToMapSlice(r.sqlx, dest.(*[]map[string]string), sql, params...)
 		case "scanner":
@@ -47,21 +46,18 @@ func (r *SqlxQuery) Selects(ctx context.Context, dest any, sql string, params ..
 			return Scanner[any](r.sqlx, dest, sql, params...)(fn.(func(any)))
 		}
 	}
-	}
+
 	return r.sqlx.Select(dest, sql, params...)
 }
 
 func (r *SqlxQuery) Gets(ctx context.Context, dest any, sql string, params ...any) error {
-	v := ctx.Value("handle=>")
-	if v != nil {
-		vv, ok := v.(string)
-		if ok && vv != "" {
-			switch vv {
+	v := helper.GetValFromContext(ctx, "handle=>", "")
+	if v != "" {
+		switch v {
 		case "string":
 			return GetToMap(r.sqlx, dest.(*map[string]string), sql, params...)
 		}
 	}
-	}
 	return r.sqlx.Get(dest, sql, params...)
 }