Add Surge MITM and scripts

script/modules/sgstore/module.go

@@ -0,0 +1,76 @@
+package sgstore
+
+import (
+	"context"
+
+	"github.com/sagernet/sing-box/adapter"
+	"github.com/sagernet/sing-box/script/jsc"
+	"github.com/sagernet/sing/service"
+
+	"github.com/dop251/goja"
+)
+
+type SurgePersistentStore struct {
+	vm        *goja.Runtime
+	cacheFile adapter.CacheFile
+	data      map[string]string
+	tag       string
+}
+
+func Enable(vm *goja.Runtime, ctx context.Context) {
+	object := vm.NewObject()
+	cacheFile := service.FromContext[adapter.CacheFile](ctx)
+	tag := vm.Get("$script").(*goja.Object).Get("name").String()
+	store := &SurgePersistentStore{
+		vm:        vm,
+		cacheFile: cacheFile,
+		tag:       tag,
+	}
+	if cacheFile == nil {
+		store.data = make(map[string]string)
+	}
+	object.Set("read", store.js_read)
+	object.Set("write", store.js_write)
+	vm.Set("$persistentStore", object)
+}
+
+func (s *SurgePersistentStore) js_read(call goja.FunctionCall) goja.Value {
+	if len(call.Arguments) > 1 {
+		panic(s.vm.NewTypeError("invalid arguments"))
+	}
+	key := jsc.AssertString(s.vm, call.Argument(0), "key", true)
+	if key == "" {
+		key = s.tag
+	}
+	var value string
+	if s.cacheFile != nil {
+		value = s.cacheFile.SurgePersistentStoreRead(key)
+	} else {
+		value = s.data[key]
+	}
+	if value == "" {
+		return goja.Null()
+	} else {
+		return s.vm.ToValue(value)
+	}
+}
+
+func (s *SurgePersistentStore) js_write(call goja.FunctionCall) goja.Value {
+	if len(call.Arguments) == 0 || len(call.Arguments) > 2 {
+		panic(s.vm.NewTypeError("invalid arguments"))
+	}
+	data := jsc.AssertString(s.vm, call.Argument(0), "data", true)
+	key := jsc.AssertString(s.vm, call.Argument(1), "key", true)
+	if key == "" {
+		key = s.tag
+	}
+	if s.cacheFile != nil {
+		err := s.cacheFile.SurgePersistentStoreWrite(key, data)
+		if err != nil {
+			panic(s.vm.NewGoError(err))
+		}
+	} else {
+		s.data[key] = data
+	}
+	return goja.Undefined()
+}