47 lines
1.7 KiB
Go
47 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
)
|
|
|
|
// Config has all the configuration parsed from the command line
|
|
type Config struct {
|
|
TargetURL string `json:"targetURL"`
|
|
ProxyPort string `json:"proxyPort"`
|
|
MaxCaptures int `json:"maxCaptures"`
|
|
Dashboard string `json:"dashboard"`
|
|
DashboardPath string `json:"dashboardPath"`
|
|
DashboardConnPath string `json:"dashboardConnPath"`
|
|
DashboardClearPath string `json:"dashboardClearPath"`
|
|
DashboardRetryPath string `json:"dashboardRetryPath"`
|
|
DashboardItemInfoPath string `json:"dashboardItemInfoPath"`
|
|
}
|
|
|
|
// ReadConfig reads the arguments from the command line
|
|
func ReadConfig() Config {
|
|
targetURL := flag.String("url", "https://jsonplaceholder.typicode.com", "Required. Set the base url you want to capture")
|
|
proxyPort := flag.String("port", "9000", "Set the proxy port")
|
|
dashboard := flag.String("dashboard", "dashboard", "Set the dashboard name")
|
|
maxCaptures := flag.Int("max-captures", 16, "Set the max number of captures to show in the dashboard")
|
|
flag.Parse()
|
|
|
|
dashboardPath := fmt.Sprintf("/%s/", *dashboard)
|
|
dashboardConnPath := fmt.Sprintf("/%s/conn/", *dashboard)
|
|
dashboardClearPath := fmt.Sprintf("/%s/clear/", *dashboard)
|
|
dashboardRetryPath := fmt.Sprintf("/%s/retry/", *dashboard)
|
|
dashboardItemInfoPath := fmt.Sprintf("/%s/items/", *dashboard)
|
|
|
|
return Config{
|
|
TargetURL: *targetURL,
|
|
ProxyPort: *proxyPort,
|
|
MaxCaptures: *maxCaptures,
|
|
Dashboard: *dashboard,
|
|
DashboardPath: dashboardPath,
|
|
DashboardConnPath: dashboardConnPath,
|
|
DashboardClearPath: dashboardClearPath,
|
|
DashboardRetryPath: dashboardRetryPath,
|
|
DashboardItemInfoPath: dashboardItemInfoPath,
|
|
}
|
|
}
|