package util

import (
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"

	"github.com/spf13/pflag"
)

func HandleError(err error) {
	fmt.Fprintln(os.Stderr, err.Error())
	os.Exit(1)
}

// ParseSource returns the struct of parsed source options.
func ParseSource(cmdFlags *pflag.FlagSet) STDINPayload {
	var (
		inputFile    string
		sourceData   []byte
		stdinPayload STDINPayload
		err          error
	)

	// Set default values of our source object
	stdinPayload.Source.Insecure = false
	stdinPayload.Source.SkipSSLVerification = false
	stdinPayload.Source.Port = "443"
	stdinPayload.Version.Number = "0"

	inputFile, err = cmdFlags.GetString("file")

	if err != nil {
		HandleError(err)
	}

	// If a config
	if inputFile == "" {
		sourceData, err = parseStdin()
	} else {
		sourceData, err = ioutil.ReadFile(inputFile)
	}

	if err != nil {
		HandleError(err)
	}

	err = json.Unmarshal(sourceData, &stdinPayload)

	if err != nil {
		HandleError(err)
	}

	return stdinPayload
}

func parseStdin() ([]byte, error) {
	stat, _ := os.Stdin.Stat()

	var (
		bytes []byte
		err   error
	)

	if (stat.Mode() & os.ModeCharDevice) == 0 {
		bytes, err = ioutil.ReadAll(os.Stdin)
	} else {
		bytes, err = nil, errors.New("no data found in stdin")
	}

	return bytes, err
}

// CheckResponse deals with http responses and automatically crashes the app
// if needed
func CheckResponse(response *http.Response) {
	if response.StatusCode != http.StatusOK {
		bodyBytes, err := ioutil.ReadAll(response.Body)

		if err != nil {
			HandleError(err)
		}

		bodyString := string(bodyBytes)
		fmt.Fprintf(os.Stderr, "request url: %s\n", response.Request.URL)
		fmt.Fprintf(os.Stderr, "code: %d\n", response.StatusCode)
		fmt.Fprintln(os.Stderr, bodyString)
		os.Exit(1)
	}

}