wp-go/internal/wpconfig/options.go

53 lines
1.0 KiB
Go
Raw Permalink Normal View History

2022-11-17 08:29:39 +00:00
package wpconfig
2022-11-15 03:11:08 +00:00
import (
"context"
"github.com/fthvgb1/wp-go/internal/pkg/models"
"github.com/fthvgb1/wp-go/model"
"github.com/fthvgb1/wp-go/safety"
2023-01-19 06:12:59 +00:00
"strings"
2022-11-15 03:11:08 +00:00
)
var options safety.Map[string, string]
var ctx context.Context
2022-11-15 03:11:08 +00:00
func InitOptions() error {
options.Flush()
if ctx == nil {
ctx = context.Background()
}
ops, err := model.FindToStringMap[models.Options](ctx, model.Conditions(
model.Where(model.SqlBuilder{{"autoload", "yes"}}),
2023-02-21 11:49:13 +00:00
model.Fields("option_name k, option_value v"),
))
2022-11-15 03:11:08 +00:00
if err != nil {
return err
}
for _, option := range ops {
2023-02-21 11:49:13 +00:00
options.Store(option["k"], option["v"])
2022-11-15 03:11:08 +00:00
}
return nil
}
2023-01-19 06:12:59 +00:00
func GetOption(k string) string {
v, ok := options.Load(k)
if ok {
return v
}
vv, err := model.GetField[models.Options](ctx, "option_value", model.Conditions(model.Where(model.SqlBuilder{{"option_name", k}})))
options.Store(k, vv)
if err != nil {
return ""
}
return vv
}
2023-01-19 06:12:59 +00:00
func GetLang() string {
s, ok := options.Load("WPLANG")
2023-01-19 06:12:59 +00:00
if !ok {
s = "zh-CN"
}
return strings.Replace(s, "_", "-", 1)
}