yao/sui/core/script.go

package core

import (
	"fmt"
	"strings"
	"time"

	"github.com/google/uuid"
	jsoniter "github.com/json-iterator/go"
	"github.com/yaoapp/gou/application"
	v8 "github.com/yaoapp/gou/runtime/v8"
	"github.com/yaoapp/gou/runtime/v8/bridge"
)

// Scripts loaded scripts
var Scripts = map[string]*Script{}

const (
	saveScript uint8 = iota
	removeScript
)

// Script the script
type Script struct {
	*v8.Script
}

type scriptData struct {
	file   string
	script *Script
	cmd    uint8
}

var chScript = make(chan *scriptData, 1)

func init() {
	go scriptWriter()
}

func scriptWriter() {
	for {
		select {
		case data := <-chScript:
			switch data.cmd {
			case saveScript:
				Scripts[data.file] = data.script
			case removeScript:
				delete(Scripts, data.file)
			}
		}
	}
}

// LoadScript load the script
func LoadScript(file string) (*Script, error) {

	if script, has := Scripts[file]; has {
		return script, nil
	}

	base := strings.TrimSuffix(file, ".sui")
	file = base + ".ts"
	if exist, _ := application.App.Exists(file); !exist {
		file = base + ".js"
	}

	if exist, _ := application.App.Exists(file); !exist {
		return nil, nil
	}

	source, err := application.App.Read(file)
	if err != nil {
		return nil, err
	}

	v8script, err := v8.MakeScript(source, file, 5*time.Second)
	if err != nil {
		return nil, err
	}

	script := &Script{Script: v8script}
	chScript <- &scriptData{file, script, saveScript}
	return script, nil
}

// Call the script method
// This will be refactored to improve the performance
func (script *Script) Call(r *Request, method string, args ...any) (interface{}, error) {
	ctx, err := script.NewContext(r.Sid, nil)
	if err != nil {
		return nil, err
	}
	defer ctx.Close()

	res, err := ctx.Call(method, args...)
	if err != nil {
		return nil, err
	}
	return res, nil
}

// ConstantsToString get the constants from the script
func (script *Script) ConstantsToString() (string, error) {
	constants, err := script.Constants()
	if err != nil {
		return "", err
	}
	raw, err := jsoniter.MarshalToString(constants)
	if err != nil {
		return "", err
	}
	return raw, nil
}

// Constants  get the constants from the script
// This will be refactored to improve the performance
func (script *Script) Constants() (map[string]interface{}, error) {
	uuid := uuid.New().String()
	ctx, err := script.NewContext(uuid, nil)
	if err != nil {
		return nil, err
	}
	defer ctx.Close()

	global := ctx.Global()
	if global == nil {
		return nil, fmt.Errorf("global is nil")
	}

	if !global.Has("__sui_constants") {
		return nil, nil
	}

	res, err := global.Get("__sui_constants")
	if err != nil {
		return nil, err
	}
	defer res.Release()

	goValues, err := bridge.GoValue(res, ctx.Context)
	if err != nil {
		return nil, err
	}

	if constants, ok := goValues.(map[string]interface{}); ok {
		return constants, nil
	}

	return nil, fmt.Errorf("constants is %v should be Record<string, any>", goValues)
}

// Helpers get the helpers from the script
// This will be refactored to improve the performance
func (script *Script) Helpers() ([]string, error) {
	uuid := uuid.New().String()
	ctx, err := script.NewContext(uuid, nil)
	if err != nil {
		return nil, err
	}
	defer ctx.Close()

	global := ctx.Global()
	if global == nil {
		return nil, fmt.Errorf("global is nil")
	}

	if !global.Has("__sui_helpers") {
		return nil, nil
	}

	res, err := global.Get("__sui_helpers")
	if err != nil {
		return nil, err
	}
	defer res.Release()

	goValues, err := bridge.GoValue(res, ctx.Context)
	if err != nil {
		return nil, err
	}

	if helpers, ok := goValues.([]interface{}); ok {
		methods := []string{}
		for _, key := range helpers {
			methods = append(methods, fmt.Sprintf("%v", key))
		}
		return methods, nil
	}

	return nil, fmt.Errorf("helpers is %v should be []string", goValues)
}