From 03ecfd6980acb763a1bfc948990a1aaba3e58b48 Mon Sep 17 00:00:00 2001
From: NSEcho <xdaemonx@protonmail.ch>
Date: Sun, 17 Sep 2023 23:59:55 +0200
Subject: [PATCH] fix: add more session fields

---
 cmd/crash.go   | 2 ++
 cmd/fuzz.go    | 3 +++
 cmd/session.go | 3 +++
 3 files changed, 8 insertions(+)

diff --git a/cmd/crash.go b/cmd/crash.go
index dcd6268..2cfb278 100644
--- 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
index 13fca44..913c682 100644
--- 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
index 5df7a28..321b389 100644
--- 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 {