gosora/cmd/hook_gen/main.go

// +build hookgen

package main // import "github.com/Azareal/Gosora/hook_gen"

import (
	"fmt"
	"log"
	"runtime/debug"
	"strings"

	h "github.com/Azareal/Gosora/cmd/common_hook_gen"
	c "github.com/Azareal/Gosora/common"
	_ "github.com/Azareal/Gosora/extend"
)

// TODO: Make sure all the errors in this file propagate upwards properly
func main() {
	// Capture panics instead of closing the window at a superhuman speed before the user can read the message on Windows
	defer func() {
		if r := recover(); r != nil {
			fmt.Println(r)
			debug.PrintStack()
		}
	}()

	hooks := make(map[string]int)
	for _, pl := range c.Plugins {
		if len(pl.Meta.Hooks) > 0 {
			for _, hook := range pl.Meta.Hooks {
				hooks[hook]++
			}
			continue
		}
		if pl.Init != nil {
			if e := pl.Init(pl); e != nil {
				log.Print("early plugin init err: ", e)
				return
			}
		}
		if pl.Hooks != nil {
			log.Print("Hooks not nil for ", pl.UName)
			for hook, _ := range pl.Hooks {
				hooks[hook] += 1
			}
		}
	}
	log.Printf("hooks: %+v\n", hooks)

	imports := []string{"net/http"}
	hookVars := h.HookVars{imports, nil}
	var params2sb strings.Builder
	add := func(name, params, ret, htype string, multiHook, skip bool, defaultRet, pure string) {
		first := true
		for _, param := range strings.Split(params, ",") {
			if !first {
				params2sb.WriteRune(',')
			}
			pspl := strings.Split(strings.ReplaceAll(strings.TrimSpace(param), "  ", " "), " ")
			params2sb.WriteString(pspl[0])
			first = false
		}
		hookVars.Hooks = append(hookVars.Hooks, h.Hook{name, params, params2sb.String(), ret, htype, hooks[name] > 0, multiHook, skip, defaultRet, pure})
		params2sb.Reset()
	}

	h.AddHooks(add)
	h.Write(hookVars)
	log.Println("Successfully generated the hooks")
}