package ipinfo

import (
	"encoding/json"
	"fmt"
	"net/http"
	"text/template"

	"bytes"

	"github.com/olebedev/config"
	"github.com/senorprogrammer/wtf/wtf"
)

// Config is a pointer to the global config object
var Config *config.Config

type Widget struct {
	wtf.TextWidget
	result string
	colors struct {
		name, value string
	}
}

type ipinfo struct {
	Ip           string `json:"ip"`
	Hostname     string `json:"hostname"`
	City         string `json:"city"`
	Region       string `json:"region"`
	Country      string `json:"country"`
	Coordinates  string `json:"loc"`
	PostalCode   string `json:"postal"`
	Organization string `json:"org"`
}

func NewWidget() *Widget {
	widget := Widget{
		TextWidget: wtf.NewTextWidget(" IPInfo ", "ipinfo", false),
	}

	widget.View.SetWrap(true)

	widget.config()

	return &widget
}

func (widget *Widget) Refresh() {
	if widget.Disabled() {
		return
	}

	widget.UpdateRefreshedAt()
	widget.ipinfo()
	widget.View.Clear()

	widget.View.SetText(widget.result)
}

//this method reads the config and calls ipinfo for ip information
func (widget *Widget) ipinfo() {
	client := &http.Client{}
	req, err := http.NewRequest("GET", "https://ipinfo.io/", nil)
	if err != nil {
		widget.result = fmt.Sprintf("%s", err.Error())
		return
	}
	req.Header.Set("User-Agent", "curl")
	response, err := client.Do(req)
	if err != nil {
		widget.result = fmt.Sprintf("%s", err.Error())
		return
	}
	defer response.Body.Close()
	if err != nil {
		widget.result = fmt.Sprintf("%s", err.Error())
		return
	}
	var info ipinfo
	err = json.NewDecoder(response.Body).Decode(&info)
	if err != nil {
		widget.result = fmt.Sprintf("%s", err.Error())
		return
	}

	widget.setResult(&info)
}

// read module configs
func (widget *Widget) config() {
	nameColor, valueColor := Config.UString("wtf.mods.ipinfo.colors.name", "red"), Config.UString("wtf.mods.ipinfo.colors.value", "white")
	widget.colors.name = nameColor
	widget.colors.value = valueColor
}

func (widget *Widget) setResult(info *ipinfo) {
	resultTemplate, _ := template.New("ipinfo_result").Parse(
		"[{{.nameColor}}]IP Address: [{{.valueColor}}]{{.Ip}}\n" +
			"[{{.nameColor}}]Hostname: [{{.valueColor}}]{{.Hostname}}\n" +
			"[{{.nameColor}}]City: [{{.valueColor}}]{{.City}}\n" +
			"[{{.nameColor}}]Region: [{{.valueColor}}]{{.Region}}\n" +
			"[{{.nameColor}}]Country: [{{.valueColor}}]{{.Country}}\n" +
			"[{{.nameColor}}]Coordinates: [{{.valueColor}}]{{.Coordinates}}\n" +
			"[{{.nameColor}}]Postal Code: [{{.valueColor}}]{{.PostalCode}}\n" +
			"[{{.nameColor}}]Organization: [{{.valueColor}}]{{.Organization}}\n",
	)

	resultBuffer := new(bytes.Buffer)

	resultTemplate.Execute(resultBuffer, map[string]string{
		"nameColor":    widget.colors.name,
		"valueColor":   widget.colors.value,
		"Ip":           info.Ip,
		"Hostname":     info.Hostname,
		"City":         info.City,
		"Region":       info.Region,
		"Country":      info.Country,
		"Coordinates":  info.Coordinates,
		"PostalCode":   info.PostalCode,
		"Organization": info.Organization,
	})

	widget.result = resultBuffer.String()
}