package main

import (
	"flag"
	"fmt"
	"net/url"
	"os"
	"strings"
	"time"
)

const (
	// LogLocationStdout indicates the stdout IO stream
	LogLocationStdout = "stdout"
	// LogLocationStderr indicates the stderr IO stream
	LogLocationStderr = "stderr"
	// LogLocationNull indicates the null IO stream (/dev/null)
	LogLocationNull = "null"
)

type config struct {
	// RFCCompliant determines whether `Cache-Control: no-cache` requests are honored. The ability to ignore `no-cache` is necessary to protect origin servers from DDOS attacks. In general, CDNs and caching proxies with the goal of origin protection should set RFCComplaint false. Cache with other goals (performance, load balancing, etc) should set RFCCompliant true.
	RFCCompliant       bool `json:"rfc_compliant"`
	dispersion         int
	loginDispersion    int
	retries            int
	waitForParents     bool
	logLevel           int
	toURL              url
	toCacheURL         url
	mode               string
	username           string
	password           string
	LogLocationError   string `json:"log_location_error"`
	LogLocationWarning string `json:"log_location_warning"`
	LogLocationInfo    string `json:"log_location_info"`
	LogLocationDebug   string `json:"log_location_debug"`
	LogLocationEvent   string `json:"log_location_event"`
}

func main() {

	parseCommand()

	hostname, error := os.Hostname()
	if error != nil {
		panic(error)
	}
	fmt.Println("Starting config check at", time.Now())

	shortHostname := strings.Split(hostname, ".")[0]

	fmt.Println("full hostname: ", hostname)
	fmt.Println("short hostname: ", shortHostname)
	fmt.Println(username, password, mode, urlString)
	fmt.Println(dispersion)

}

func parseCommand() {

	var dispersionPtr = flag.Int("dispersion", 300, "Dispersion range in seconds.")
	var loginDispersionPtr = flag.Int("login_dispersion", 0, "Login dispersion range in seconds.")
	var retriesPtr = flag.Int("retries", 3, "Retry connection attempts to TO URL <number> times.")
	var waitForParentsPtr = flag.Bool("wait_for_parents", true, "Do not update if a parent cache is still pending.")
	var logLevelPtr = flag.String("log_level", "WARN", "Log level: <ALL, TRACE, DEBUG, INFO, WARN, ERROR, FATAL, NONE> Default:WARN")
	flag.Parse()

	var logLevel int
	switch *logLevelPtr {
	case "ALL":
		logLevel = 7
	case "TRACE":
		logLevel = 6
	case "DEBUG":
		logLevel = 5
	case "INFO":
		logLevel = 4
	case "WARN":
		logLevel = 3
	case "ERROR":
		logLevel = 2
	case "FATAL":
		logLevel = 1
	case "NONE":
		logLevel = 0
	}

	args := flag.Args()

	if len(args) < 3 {
		usage()
	}

	mode = args[0]
	if mode != "syncds" && mode != "report" && mode != "badass" && mode != "revalidate" {
		usage()
	}

	toURL, e := url.Parse(args[1])
	if e != nil {
		usage()
	}

	if strings.Count(args[2], ":") == 1 {
		username := strings.Split(args[2], ":")[0]
		password := strings.Split(args[2], ":")[1]
	} else {
		usage()
	}

	var config = Config{
		dispersion:         *dispersionPtr,
		loginDispersion:    *loginDispersionPtr,
		retries:            *retriesPtr,
		waitForParents:     *waitForParentsPtr,
		logLevel:           logLevel,
		username:           username,
		password:           password,
		toURL:              toURL,
		mode:               mode,
		LogLocationError:   LogLocationStderr,
		LogLocationWarning: LogLocationStdout,
		LogLocationInfo:    LogLocationNull,
		LogLocationDebug:   LogLocationNull,
		LogLocationEvent:   LogLocationStdout,
	}
	return
}

func usage() {
	fmt.Println(`====-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-====

	Usage: traffic_ops_ort [optional flags] <mode> <log_level> <Traffic_Ops_URL> <Traffic_Ops_Login>


	<mode> = interactive - asks questions during config process.
	<mode> = report - prints config differences and exits.
	<mode> = badass - attempts to fix all config differences that it can.
	<mode> = syncds - syncs delivery services with what is configured in Traffic Ops.
	<mode> = revalidate - checks for updated revalidations in Traffic Ops and applies them.  Requires Traffic Ops 2.1.

	<Traffic_Ops_URL> = URL to Traffic Ops host. Example: https://trafficops.company.net

	<Traffic_Ops_Login> => Example: 'username:password'

	[optional flags]:
	--dispersion=<time> => wait a random number between 0 and <time> before starting. Default = 300.
	--login_dispersion=<time> => wait a random number between 0 and <time> before login. Default = 0.
	--retries=<number> => retry connection to Traffic Ops URL <number> times. Default = 3.
	--wait_for_parents=<0|1> => do not update if parent_pending = 1 in the update json. Default = 1, wait for parents.
	--log_level=<ALL, TRACE, DEBUG, INFO, WARN, ERROR, FATAL, NONE> => Logging level.  Default is WARN

	====-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-====`)
	os.Exit(0)
}