wp-go/internal/theme/hook.go
2023-03-01 13:17:12 +08:00

25 lines
508 B
Go

package theme
import (
"github.com/fthvgb1/wp-go/internal/theme/twentyfifteen"
"github.com/fthvgb1/wp-go/internal/theme/wp"
)
var themeMap = map[string]func(*wp.Handle){}
func addThemeHookFunc(name string, fn func(handle *wp.Handle)) {
if _, ok := themeMap[name]; ok {
panic("exists same name theme")
}
themeMap[name] = fn
}
func Hook(themeName string, handle *wp.Handle) {
fn, ok := themeMap[themeName]
if ok && fn != nil {
fn(handle)
return
}
themeMap[twentyfifteen.ThemeName](handle)
}