fix: add more session fields

NSEcho · Sep 17, 2023 · 03ecfd6 · 03ecfd6
1 parent feab57d
commit 03ecfd6
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/cmd/crash.go b/cmd/crash.go
@@ -73,6 +73,8 @@ var crashCmd = &cobra.Command{
 
 		time.Sleep(2 * time.Second)
 
+		_ = script.ExportsCall("setup", s.Method, s.UIApp, s.Delegate, s.Scene)
+
 		_ = script.ExportsCall("fuzz", s.Method, string(input))
 
 		return nil

diff --git a/cmd/fuzz.go b/cmd/fuzz.go
@@ -145,8 +145,11 @@ var fuzzCmd = &cobra.Command{
 			s := Session{
 				App:      app,
 				Base:     base,
+				Delegate: delegate,
 				Function: fn,
 				Method:   method,
+				Scene:    scene,
+				UIApp:    uiapp,
 			}
 			if err := s.WriteToFile(); err != nil {
 				l.Errorf("Error writing session file: %v", err)

diff --git a/cmd/session.go b/cmd/session.go
@@ -26,8 +26,11 @@ func NewSession(sessionFile string) (*Session, error) {
 type Session struct {
 	App      string `yaml:"app"`
 	Base     string `yaml:"base"`
+	Delegate string `yaml:"delegate"`
 	Function string `yaml:"fn"`
 	Method   string `yaml:"method"`
+	Scene    string `yaml:"scene"`
+	UIApp    string `yaml:"uiapp"`
 }
 
 func (s *Session) WriteToFile() error {