wp-go/internal/wpconfig/options.go

56 lines
1.1 KiB
Go
Raw 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()
}
2023-01-12 12:42:16 +00:00
ops, err := model.SimpleFind[models.Options](ctx, model.SqlBuilder{{"autoload", "yes"}}, "option_name, option_value")
2022-11-15 03:11:08 +00:00
if err != nil {
return err
}
if len(ops) == 0 {
2023-01-12 12:42:16 +00:00
ops, err = model.SimpleFind[models.Options](ctx, nil, "option_name, option_value")
2022-11-15 03:11:08 +00:00
if err != nil {
return err
}
}
for _, option := range ops {
options.Store(option.OptionName, option.OptionValue)
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, string](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)
}