package main

import (
	"flag"
	"fmt"
	"gopkg.in/ini.v1"
	"os"
)

// Config has all the configuration parsed from the command line.
type Config struct {
	TargetURL     string
	ProxyPort     string
	DashboardPort string
	MaxCaptures   int
}

// ReadConfig reads the arguments from the command line.
func ReadConfig() Config {
	defaultTargetURL := "https://jsonplaceholder.typicode.com"
	defaultProxyPort := "9000"
	defaultDashboardPort := "9001"
	defaultMaxCaptures := 16

	configFile := ".capture.ini"

	if _, err := os.Stat(configFile); err == nil {
		cfg, err := ini.Load(configFile)

		if err != nil {
			fmt.Printf("Fail to read file %s: %v", configFile, err)
			os.Exit(1)
		}

		section := cfg.Section("")

		defaultTargetURL = section.Key("url").MustString(defaultTargetURL)
		defaultProxyPort = section.Key("port").MustString(defaultProxyPort)
		defaultDashboardPort = section.Key("dashboard").MustString(defaultDashboardPort)
		defaultMaxCaptures = section.Key("captures").MustInt(defaultMaxCaptures)
	}

	targetURL := flag.String("url", defaultTargetURL, "Required. Set the url you want to proxy")
	proxyPort := flag.String("port", defaultProxyPort, "Set the proxy port")
	dashboardPort := flag.String("dashboard", defaultDashboardPort, "Set the dashboard port")
	maxCaptures := flag.Int("captures", defaultMaxCaptures, "Set how many captures to show in the dashboard")

	flag.Parse()

	return Config{
		TargetURL:     *targetURL,
		ProxyPort:     *proxyPort,
		MaxCaptures:   *maxCaptures,
		DashboardPort: *dashboardPort,
	}
}