wp-go/app/theme/hook.go

package theme

import (
	"github.com/fthvgb1/wp-go/app/theme/wp"
	str "github.com/fthvgb1/wp-go/helper/strings"
	"github.com/fthvgb1/wp-go/safety"
)

var themeMap = safety.NewMap[string, func(*wp.Handle)]()

func AddThemeHookFunc(name string, fn func(handle *wp.Handle)) {
	if _, ok := themeMap.Load(name); ok {
		panic("exists same name theme")
	}
	themeMap.Store(name, fn)
}

func IsThemeHookFuncExist(name string) bool {
	_, ok := themeMap.Load(name)
	return ok
}

func Hook(themeName string, h *wp.Handle) {
	fn, ok := themeMap.Load(themeName)
	if ok && fn != nil {
		fn(h)
		return
	}
	panic(str.Join("theme ", themeName, " don't exist"))
}