capture/config.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,
	}
}