go-gin-api/internal/web/controller/generator_handler/func_handlerexecute.go

package generator_handler

import (
	"bytes"
	"fmt"
	"os/exec"
	"runtime"

	"github.com/xinliangnote/go-gin-api/internal/pkg/core"
)

type handlerExecuteRequest struct {
	Name string `form:"name"`
}

const handlergenSh = "./scripts/handlergen.sh"

func (h *handler) HandlerExecute() core.HandlerFunc {
	return func(c core.Context) {
		req := new(handlerExecuteRequest)
		if err := c.ShouldBindPostForm(req); err != nil {
			c.Payload("参数传递有误")
			return
		}

		shellPath := fmt.Sprintf("%s %s", handlergenSh, req.Name)

		// runtime.GOOS = linux or darwin
		command := exec.Command("/bin/bash", "-c", shellPath)

		if runtime.GOOS == "windows" {
			command = exec.Command("cmd", "/C", shellPath)
		}

		var stderr bytes.Buffer
		command.Stderr = &stderr

		output, err := command.Output()
		if err != nil {
			c.Payload(stderr.String())
			return
		}

		c.Payload(string(output))
	}
}