mirror of
https://github.com/aykhans/dodo.git
synced 2025-04-16 09:53:12 +00:00
Aykhan Shahsuvarov
00f0bcb2de
- Moved readers to the config package - Added an option to read remote config files - Moved the validation package to the config package and removed the validator dependency - Moved the customerrors package to the config package - Replaced fatih/color with jedib0t/go-pretty/v6/text - Removed proxy check functionality - Added param, header, cookie, body, and proxy flags to the CLI - Allowed multiple values for the same key in params, headers, and cookies
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/aykhans/dodo/config"
|
|
"github.com/aykhans/dodo/requests"
|
|
"github.com/aykhans/dodo/types"
|
|
"github.com/aykhans/dodo/utils"
|
|
"github.com/jedib0t/go-pretty/v6/text"
|
|
)
|
|
|
|
func main() {
|
|
conf := config.NewConfig()
|
|
configFile, err := conf.ReadCLI()
|
|
if err != nil {
|
|
utils.PrintErrAndExit(err)
|
|
}
|
|
|
|
if configFile.String() != "" {
|
|
tempConf := config.NewConfig()
|
|
if err := tempConf.ReadFile(configFile); err != nil {
|
|
utils.PrintErrAndExit(err)
|
|
}
|
|
tempConf.MergeConfig(conf)
|
|
conf = tempConf
|
|
}
|
|
conf.SetDefaults()
|
|
|
|
if errs := conf.Validate(); len(errs) > 0 {
|
|
utils.PrintErrAndExit(errors.Join(errs...))
|
|
}
|
|
|
|
requestConf := config.NewRequestConfig(conf)
|
|
requestConf.Print()
|
|
|
|
if !requestConf.Yes {
|
|
response := config.CLIYesOrNoReader("Do you want to continue?", false)
|
|
if !response {
|
|
utils.PrintAndExit("Exiting...\n")
|
|
}
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
go listenForTermination(func() { cancel() })
|
|
|
|
responses, err := requests.Run(ctx, requestConf)
|
|
if err != nil {
|
|
if err == types.ErrInterrupt {
|
|
fmt.Println(text.FgYellow.Sprint(err.Error()))
|
|
return
|
|
}
|
|
utils.PrintErrAndExit(err)
|
|
}
|
|
|
|
responses.Print()
|
|
}
|
|
|
|
func listenForTermination(do func()) {
|
|
sigChan := make(chan os.Signal, 1)
|
|
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
|
|
<-sigChan
|
|
do()
|
|
}
|