Sillyv2/core/grpc_plugins.go

package core

import (
	"crypto/sha1"
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"path/filepath"
	"strings"

	"github.com/cdle/sillyplus/core/common"
	"github.com/cdle/sillyplus/core/storage"
	"github.com/cdle/sillyplus/utils"
	"github.com/dop251/goja"
	"github.com/fsnotify/fsnotify"
	"github.com/google/uuid"
	cron "github.com/robfig/cron/v3"
)

func init() {
	go initNodePlugins()
}

func initNodePlugins() {
	root := utils.ExecPath + "/plugins"
	plugins := []string{root}
	os.Mkdir(root, 0755)
	filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if !strings.HasPrefix(path, "/Users/a1-6/Code/sillyplus/plugins/") {
			return nil
		}
		files := strings.Split(strings.Replace(path, root+"/", "", 1), "/")
		// var plugin_dir = false
		// var plugin_index = false
		switch len(files) {
		case 1:
			// plugin_dir = true
			if info.IsDir() {
				plugins = append(plugins, path)
			}
		case 2:
			if (files[1] == "main.ts" || files[1] == "main.js") && !info.IsDir() {
				AddNodePlugin(path, files[0])
			}
		}
		return nil
	})
	watcher, err := fsnotify.NewWatcher()
	if err != nil {
		fmt.Println("创建监视器失败：", err)
		return
	}
	defer watcher.Close()

	// 要监控的文件夹路径

	for _, dir := range plugins {
		err = watcher.Add(dir)
		if err != nil {
			fmt.Println("添加监视目录失败：", err)
			return
		}
	}

	for {
		select {
		case event, ok := <-watcher.Events:
			if !ok {
				return
			}
			// fmt.Println(event.Name, "op", event.Op.String())
			files := strings.Split(strings.Replace(event.Name, root+"/", "", 1), "/")
			var plugin_dir = false
			var plugin_index = false
			var plugin_name = ""
			switch len(files) {
			case 1:
				plugin_dir = true
				fmt.Println("目录事件")
				plugin_name = files[0]
			case 2:
				if files[1] == "main.ts" || files[1] == "main.js" {
					plugin_index = true
					fmt.Println("入口文件事件")
				}
				plugin_name = files[0]
			}
			switch event.Op.String() {
			case "CREATE":
				if plugin_dir {
					info, err := os.Stat(event.Name)
					// fmt.Println(err)
					if err == nil && info.IsDir() {
						watcher.Add(event.Name)
						// fmt.Println("增加插件目录", event.Name)
					} else {
						// fmt.Println("非插件目录", event.Name)
					}
				} else if plugin_index {
					// fmt.Println("增加插件", event.Name)
					AddNodePlugin(event.Name, plugin_name)
				}
			case "REMOVE", "RENAME", "REMOVE|RENAME":
				if plugin_dir {
					watcher.Remove(event.Name)
					// fmt.Println("移除插件目录", event.Name)
					// fmt.Println("移除插件", plugin_name)
					RemNodePlugin(plugin_name)
				} else if plugin_index {
					// fmt.Println("移除插件", plugin_name)
					RemNodePlugin(plugin_name)
				}
			case "WRITE":
				if plugin_index {
					RemNodePlugin(plugin_name)
					AddNodePlugin(event.Name, plugin_name)
					// fmt.Println("变更插件", event.Name, plugin_name)
				}
			}
		case err, ok := <-watcher.Errors:
			if !ok {
				return
			}
			fmt.Println("错误：", err)
		}
	}
}

func RemNodePlugin(name string) error {
	pluginLock.Lock()
	defer pluginLock.Unlock()
	key := nameUuid(name)
	fmt.Println("rem", key, name)
	for i := range Functions {
		if Functions[i].UUID == key {
			fmt.Println("pl", key)
			DestroyAdapterByUUID(key)
			Functions[i].Running = false
			if len(Functions[i].CronIds) != 0 {
				for _, id := range Functions[i].CronIds {
					CRON.Remove(cron.EntryID(id))
				}
			}
			Functions = append(Functions[:i], Functions[i+1:]...)
			CancelPluginCrons(key)
			CancelPluginWebs(key)
			CancelPluginlistening(key)
			CancelHttpListen(key)
			remStatic(key)
			storage.DisableHandle(key)
			break
		}
	}
	return nil
}

func nameUuid(name string) string {
	hash := sha1.Sum([]byte(name))
	return uuid.NewSHA1(uuid.Nil, hash[:]).String()
}

func AddNodePlugin(path, name string) error {
	pluginLock.Lock()
	defer pluginLock.Unlock()
	file, err := os.Open(path)
	if err != nil {
		return err
	}
	data, err := ioutil.ReadAll(file)
	if err != nil {
		return err
	}
	uuid := nameUuid(name)
	// fmt.Println("add,", uuid, name)
	f, cbs := pluginParse(string(data), uuid)
	f.Suffix = ".ts"
	f.Type = "typescript"
	f.Handle = func(s common.Sender, f func(vm *goja.Runtime)) interface{} {
		cmd := exec.Command("node", "/Users/a1-6/.nvm/versions/node/v18.16.1/lib/node_modules/ts-node/dist/bin.js", path)
		cmd.Env = append(cmd.Env, "SENDER_ID="+s.GetID())
		// 执行命令，并捕获其输出
		output, err := cmd.Output()
		if err != nil {
			fmt.Println("命令执行失败：", err)
			return nil
		}
		// 输出命令的输出结果
		fmt.Println(string(output))
		return nil
	}
	for _, cb := range cbs {
		cb()
	}
	AddCommand([]*common.Function{f})
	return nil
}