scriptcat 命令

Author: CodFrm

cmd/scriptcat/exec.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"github.com/scriptscat/cloudcat/pkg/scriptcat"
+	"github.com/spf13/cobra"
+	"io/ioutil"
+)
+
+type execCmd struct {
+}
+
+func newExecCmd() *execCmd {
+	return &execCmd{}
+}
+
+func (e *execCmd) Commands() []*cobra.Command {
+	ret := &cobra.Command{
+		Use:   "exec [file] [flags]",
+		Short: "执行一个脚本猫脚本",
+		RunE:  e.exec,
+	}
+
+	return []*cobra.Command{ret}
+}
+
+func (e *execCmd) exec(cmd *cobra.Command, args []string) error {
+
+	sc, err := scriptcat.NewScriptCat()
+	if err != nil {
+		return err
+	}
+
+	script, err := ioutil.ReadFile(args[0])
+	if err != nil {
+		return err
+	}
+
+	return sc.Run(string(script))
+}

cmd/scriptcat/main.go

@@ -0,0 +1,21 @@
+package main
+
+import (
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+func main() {
+
+	rootCmd := &cobra.Command{
+		Use: "scriptcat",
+	}
+
+	execCmd := newExecCmd()
+
+	rootCmd.AddCommand(execCmd.Commands()...)
+
+	if err := rootCmd.Execute(); err != nil {
+		logrus.Fatalln(err)
+	}
+}

go.mod

@@ -2,4 +2,9 @@ module github.com/scriptscat/cloudcat
 
 go 1.15
 
-require rogchap.com/v8go v0.6.0
+require (
+	github.com/sirupsen/logrus v1.8.1
+	github.com/spf13/cobra v1.2.1
+	github.com/stretchr/testify v1.7.0
+	rogchap.com/v8go v0.6.0
+)

go.sum

@@ -1,4 +1,4 @@
-package scriptcat
+package executor
 
 import "rogchap.com/v8go"
 
@@ -7,19 +7,19 @@ type Context struct {
 	opts *Options
 }
 
-func NewContext(iso *Isolate, opt ...Option) (*Context, error) {
-	global, err := v8go.NewObjectTemplate(iso.iso)
+func NewContext(executor *Executor, opt ...Option) (*Context, error) {
+	global, err := v8go.NewObjectTemplate(executor.iso)
 	if err != nil {
 		return nil, err
 	}
 	options := &Options{
-		iso:    iso.iso,
+		iso:    executor.iso,
 		global: global,
 	}
 	for _, o := range opt {
 		o(options)
 	}
-	ctx, err := v8go.NewContext(iso.iso, options.global)
+	ctx, err := v8go.NewContext(executor.iso, options.global)
 	if err != nil {
 		return nil, err
 	}

pkg/scriptcat/context.go pkg/executor/context.gopkg/scriptcat/context.go renamed to pkg/executor/context.go

@@ -0,0 +1,19 @@
+package executor
+
+import "rogchap.com/v8go"
+
+type Executor struct {
+	iso *v8go.Isolate
+}
+
+func NewExecutor() (*Executor, error) {
+	iso, err := v8go.NewIsolate()
+	if err != nil {
+		return nil, err
+	}
+	return &Executor{iso: iso}, nil
+}
+
+func (c *Executor) Run(ctx *Context, source string) (*v8go.Value, error) {
+	return ctx.RunScript(source, "main.js")
+}

pkg/executor/executor.go

@@ -1,13 +1,13 @@
-package scriptcat
+package executor
 
 import (
 	"testing"
 )
 
 func TestIsolate(t *testing.T) {
-	iso, _ := NewIsolate()
+	iso, _ := NewExecutor()
 
-	ctx, _ := NewContext(iso, GmXmlHttpRequest())
+	ctx, _ := NewContext(iso, GmXmlHttpRequest(nil))
 
 	iso.Run(ctx, "GM_xmlhttpRequest({url:'https://bbs.tampermonkey.net.cn/'})")
 

pkg/scriptcat/isolate_test.go pkg/executor/executor_test.gopkg/scriptcat/isolate_test.go renamed to pkg/executor/executor_test.go

@@ -1,6 +1,8 @@
-package scriptcat
+package executor
 
 import (
+	"net/http"
+
 	"rogchap.com/v8go"
 )
 
@@ -15,7 +17,7 @@ func globalFunc(opts *Options, name string, callback v8go.FunctionCallback) {
 	}
 }
 
-func GmXmlHttpRequest() Option {
+func GmXmlHttpRequest(jar http.CookieJar) Option {
 	return func(opts *Options) {
 		globalFunc(opts, "GM_xmlhttpRequest", func(info *v8go.FunctionCallbackInfo) *v8go.Value {
 

pkg/scriptcat/gm.go pkg/executor/gm.gopkg/scriptcat/gm.go renamed to pkg/executor/gm.go

@@ -0,0 +1,11 @@
+package executor
+
+import "rogchap.com/v8go"
+
+type Options struct {
+	iso    *v8go.Isolate
+	global *v8go.ObjectTemplate
+	err    error
+}
+
+type Option func(opts *Options)