wp-go/app/plugins/wphandle/handle.go

package wphandle

import (
	"errors"
	"github.com/fthvgb1/wp-go/app/pkg/config"
	"github.com/fthvgb1/wp-go/app/pkg/logs"
	"github.com/fthvgb1/wp-go/app/plugins/wphandle/apply"
	"github.com/fthvgb1/wp-go/app/plugins/wphandle/enlightjs"
	"github.com/fthvgb1/wp-go/app/plugins/wphandle/hiddenlogin"
	"github.com/fthvgb1/wp-go/app/theme/wp"
	str "github.com/fthvgb1/wp-go/helper/strings"
	"github.com/fthvgb1/wp-go/safety"
	"path/filepath"
	"plugin"
)

var plugins = func() *safety.Map[string, func(*wp.Handle)] {
	m := safety.NewMap[string, func(*wp.Handle)]()
	m.Store("Enlightjs", enlightjs.EnlighterJS)
	m.Store("HiddenLogin", hiddenlogin.HiddenLogin)
	return m
}()

func RegisterPlugin(name string, fn func(*wp.Handle)) {
	plugins.Store(name, fn)
}

func UsePlugins(h *wp.Handle, calls ...string) {
	calls = append(calls, config.GetConfig().Plugins...)
	for _, call := range calls {
		call = str.FirstUpper(call)
		if fn, ok := plugins.Load(call); ok {
			fn(h)
		}
	}
}

func LoadPlugins() {
	dirPath := config.GetConfig().PluginPath
	if dirPath == "" {
		return
	}
	glob, err := filepath.Glob(filepath.Join(dirPath, "*.so"))
	if err != nil {
		logs.Error(err, "读取插件目录错误", dirPath)
		return
	}
	for _, entry := range glob {
		p, err := plugin.Open(entry)
		if err != nil {
			logs.Error(err, "读取插件错误", entry)
			continue
		}
		name := filepath.Ext(entry)
		name = filepath.Base(entry[0 : len(entry)-len(name)])
		name = str.FirstUpper(name)
		pl, err := p.Lookup(name)
		if err != nil {
			logs.Error(err, "插件lookup错误", entry)
			continue
		}
		plu, ok := pl.(func(*wp.Handle))
		if !ok {
			logs.Error(errors.New("switch func(*wp.Handle) fail"), "插件转换错误", entry)
			continue
		}
		RegisterPlugin(name, plu)
	}
	apply.SetFn(func(h *wp.Handle) {
		UsePlugins(h)
	})
}