mirror of
https://github.com/aykhans/sarin.git
synced 2026-04-18 13:49:37 +00:00
Aykhan Shahsuvarov
2eebac68c9
First signal cancels the main context; second releases the terminal and exits 130, unblocking shutdown when captcha polls or user scripts ignore context. Also moves the bubbletea models and streamProgress out of sarin.go into a new tui.go.
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"go.aykhans.me/sarin/internal/config"
|
|
"go.aykhans.me/sarin/internal/sarin"
|
|
"go.aykhans.me/sarin/internal/types"
|
|
utilsErr "go.aykhans.me/utils/errors"
|
|
)
|
|
|
|
func main() {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
stopCtrl := sarin.NewStopController(cancel)
|
|
go listenForTermination(stopCtrl.Stop)
|
|
|
|
combinedConfig := config.ReadAllConfigs()
|
|
|
|
combinedConfig.SetDefaults()
|
|
|
|
if *combinedConfig.ShowConfig {
|
|
if !combinedConfig.Print() {
|
|
return
|
|
}
|
|
}
|
|
|
|
_ = utilsErr.MustHandle(combinedConfig.Validate(),
|
|
utilsErr.OnType(func(err types.FieldValidationErrors) error {
|
|
for _, fieldErr := range err.Errors {
|
|
if fieldErr.Value == "" {
|
|
fmt.Fprintln(os.Stderr,
|
|
config.StyleYellow.Render(fmt.Sprintf("[VALIDATION] Field '%s': ", fieldErr.Field))+fieldErr.Err.Error(),
|
|
)
|
|
} else {
|
|
fmt.Fprintln(os.Stderr,
|
|
config.StyleYellow.Render(fmt.Sprintf("[VALIDATION] Field '%s' (%s): ", fieldErr.Field, fieldErr.Value))+fieldErr.Err.Error(),
|
|
)
|
|
}
|
|
}
|
|
os.Exit(1)
|
|
return nil
|
|
}),
|
|
)
|
|
|
|
srn, err := sarin.NewSarin(
|
|
ctx,
|
|
combinedConfig.Methods, combinedConfig.URL, *combinedConfig.Timeout,
|
|
*combinedConfig.Concurrency, combinedConfig.Requests, combinedConfig.Duration,
|
|
*combinedConfig.Quiet, *combinedConfig.Insecure, combinedConfig.Params, combinedConfig.Headers,
|
|
combinedConfig.Cookies, combinedConfig.Bodies, combinedConfig.Proxies, combinedConfig.Values,
|
|
*combinedConfig.Output != config.ConfigOutputTypeNone,
|
|
*combinedConfig.DryRun,
|
|
combinedConfig.Lua, combinedConfig.Js,
|
|
)
|
|
_ = utilsErr.MustHandle(err,
|
|
utilsErr.OnType(func(err types.ProxyDialError) error {
|
|
fmt.Fprintln(os.Stderr, config.StyleRed.Render("[PROXY] ")+err.Error())
|
|
os.Exit(1)
|
|
return nil
|
|
}),
|
|
utilsErr.OnSentinel(types.ErrScriptEmpty, func(err error) error {
|
|
fmt.Fprintln(os.Stderr, config.StyleRed.Render("[SCRIPT] ")+err.Error())
|
|
os.Exit(1)
|
|
return nil
|
|
}),
|
|
utilsErr.OnType(func(err types.ScriptLoadError) error {
|
|
fmt.Fprintln(os.Stderr, config.StyleRed.Render("[SCRIPT] ")+err.Error())
|
|
os.Exit(1)
|
|
return nil
|
|
}),
|
|
)
|
|
|
|
srn.Start(ctx, stopCtrl)
|
|
|
|
switch *combinedConfig.Output {
|
|
case config.ConfigOutputTypeNone:
|
|
return
|
|
case config.ConfigOutputTypeJSON:
|
|
srn.GetResponses().PrintJSON()
|
|
case config.ConfigOutputTypeYAML:
|
|
srn.GetResponses().PrintYAML()
|
|
default:
|
|
srn.GetResponses().PrintTable()
|
|
}
|
|
}
|
|
|
|
func listenForTermination(stop func()) {
|
|
sigChan := make(chan os.Signal, 4)
|
|
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
|
|
for range sigChan {
|
|
stop()
|
|
}
|
|
}
|