mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
14e7619075
Have all instances take a function Update the remaining modules to take this into account Numerous smaller refactors to make some widgets work more or less the same
120 lines
3.0 KiB
Go
120 lines
3.0 KiB
Go
package ipapi
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
"text/template"
|
|
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/view"
|
|
)
|
|
|
|
// Widget widget struct
|
|
type Widget struct {
|
|
view.TextWidget
|
|
|
|
result string
|
|
settings *Settings
|
|
}
|
|
|
|
type ipinfo struct {
|
|
Query string `json:"query"`
|
|
ISP string `json:"isp"`
|
|
AS string `json:"as"`
|
|
City string `json:"city"`
|
|
Region string `json:"region"`
|
|
Country string `json:"country"`
|
|
CountryCode string `json:"countryCode"`
|
|
Latitude float64 `json:"lat"`
|
|
Longitude float64 `json:"lon"`
|
|
PostalCode string `json:"zip"`
|
|
Organization string `json:"org"`
|
|
Timezone string `json:"timezone"`
|
|
}
|
|
|
|
// NewWidget constructor
|
|
func NewWidget(app *tview.Application, settings *Settings) *Widget {
|
|
widget := Widget{
|
|
TextWidget: view.NewTextWidget(app, settings.common, false),
|
|
|
|
settings: settings,
|
|
}
|
|
|
|
widget.View.SetWrap(false)
|
|
|
|
return &widget
|
|
}
|
|
|
|
// Refresh refresh the module
|
|
func (widget *Widget) Refresh() {
|
|
widget.ipinfo()
|
|
|
|
widget.Redraw(func() (string, string, bool) { return widget.CommonSettings().Title, widget.result, false })
|
|
}
|
|
|
|
//this method reads the config and calls ipinfo for ip information
|
|
func (widget *Widget) ipinfo() {
|
|
client := &http.Client{}
|
|
req, err := http.NewRequest("GET", "http://ip-api.com/json", nil)
|
|
if err != nil {
|
|
widget.result = err.Error()
|
|
return
|
|
}
|
|
req.Header.Set("User-Agent", "curl")
|
|
response, err := client.Do(req)
|
|
if err != nil {
|
|
widget.result = err.Error()
|
|
return
|
|
}
|
|
defer response.Body.Close()
|
|
var info ipinfo
|
|
err = json.NewDecoder(response.Body).Decode(&info)
|
|
if err != nil {
|
|
widget.result = err.Error()
|
|
return
|
|
}
|
|
|
|
widget.setResult(&info)
|
|
}
|
|
|
|
func (widget *Widget) setResult(info *ipinfo) {
|
|
resultTemplate, _ := template.New("ipinfo_result").Parse(
|
|
formatableText("IP Address", "Ip") +
|
|
formatableText("ISP", "ISP") +
|
|
formatableText("AS", "AS") +
|
|
formatableText("City", "City") +
|
|
formatableText("Region", "Region") +
|
|
formatableText("Country", "Country") +
|
|
formatableText("Coordinates", "Coordinates") +
|
|
formatableText("Postal Code", "PostalCode") +
|
|
formatableText("Organization", "Organization") +
|
|
formatableText("Timezone", "Timezone"),
|
|
)
|
|
|
|
resultBuffer := new(bytes.Buffer)
|
|
|
|
resultTemplate.Execute(resultBuffer, map[string]string{
|
|
"nameColor": widget.settings.colors.name,
|
|
"valueColor": widget.settings.colors.value,
|
|
"Ip": info.Query,
|
|
"ISP": info.ISP,
|
|
"AS": info.AS,
|
|
"City": info.City,
|
|
"Region": info.Region,
|
|
"Country": info.Country,
|
|
"Coordinates": strconv.FormatFloat(info.Latitude, 'f', 6, 64) + "," + strconv.FormatFloat(info.Longitude, 'f', 6, 64),
|
|
"PostalCode": info.PostalCode,
|
|
"Organization": info.Organization,
|
|
"Timezone": info.Timezone,
|
|
})
|
|
|
|
widget.result = resultBuffer.String()
|
|
}
|
|
|
|
func formatableText(key, value string) string {
|
|
return fmt.Sprintf(" [{{.nameColor}}]%s: [{{.valueColor}}]{{.%s}}\n", key, value)
|
|
}
|