mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
124 lines
3.0 KiB
Go
124 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),
|
|
|
|
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 func() { _ = 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)
|
|
|
|
err := 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,
|
|
})
|
|
|
|
if err != nil {
|
|
widget.result = err.Error()
|
|
}
|
|
|
|
widget.result = resultBuffer.String()
|
|
}
|
|
|
|
func formatableText(key, value string) string {
|
|
return fmt.Sprintf(" [{{.nameColor}}]%s: [{{.valueColor}}]{{.%s}}\n", key, value)
|
|
}
|