mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
155 lines
3.3 KiB
Go
155 lines
3.3 KiB
Go
package price
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/wtfutil/wtf/wtf"
|
|
)
|
|
|
|
var baseURL = "https://min-api.cryptocompare.com/data/price"
|
|
var ok = true
|
|
|
|
// Widget define wtf widget to register widget later
|
|
type Widget struct {
|
|
*list
|
|
|
|
Result string
|
|
|
|
RefreshInterval int
|
|
}
|
|
|
|
// NewWidget Make new instance of widget
|
|
func NewWidget() *Widget {
|
|
widget := Widget{}
|
|
|
|
widget.setList()
|
|
|
|
return &widget
|
|
}
|
|
|
|
func (widget *Widget) setList() {
|
|
currenciesMap, _ := wtf.Config.Map("wtf.mods.cryptolive.currencies")
|
|
|
|
widget.list = &list{}
|
|
|
|
for currency := range currenciesMap {
|
|
displayName, _ := wtf.Config.String("wtf.mods.cryptolive.currencies." + currency + ".displayName")
|
|
toList := getToList(currency)
|
|
widget.list.addItem(currency, displayName, toList)
|
|
}
|
|
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
// Refresh & update after interval time
|
|
func (widget *Widget) Refresh(wg *sync.WaitGroup) {
|
|
if len(widget.list.items) == 0 {
|
|
return
|
|
}
|
|
|
|
widget.updateCurrencies()
|
|
|
|
if !ok {
|
|
widget.Result = fmt.Sprint("Please check your internet connection!")
|
|
return
|
|
}
|
|
widget.display()
|
|
wg.Done()
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) display() {
|
|
str := ""
|
|
var (
|
|
fromNameColor = wtf.Config.UString("wtf.mods.cryptolive.colors.from.name", "coral")
|
|
fromDisplayNameColor = wtf.Config.UString("wtf.mods.cryptolive.colors.from.displayName", "grey")
|
|
toNameColor = wtf.Config.UString("wtf.mods.cryptolive.colors.to.name", "white")
|
|
toPriceColor = wtf.Config.UString("wtf.mods.cryptolive.colors.to.price", "green")
|
|
)
|
|
for _, item := range widget.list.items {
|
|
str += fmt.Sprintf(" [%s]%s[%s] (%s)\n", fromNameColor, item.displayName, fromDisplayNameColor, item.name)
|
|
for _, toItem := range item.to {
|
|
str += fmt.Sprintf("\t[%s]%s: [%s]%f\n", toNameColor, toItem.name, toPriceColor, toItem.price)
|
|
}
|
|
str += "\n"
|
|
}
|
|
|
|
widget.Result = fmt.Sprintf("\n%s", str)
|
|
}
|
|
|
|
func getToList(fromName string) []*toCurrency {
|
|
toNames, _ := wtf.Config.List("wtf.mods.cryptolive.currencies." + fromName + ".to")
|
|
|
|
var toList []*toCurrency
|
|
|
|
for _, to := range toNames {
|
|
toList = append(toList, &toCurrency{
|
|
name: to.(string),
|
|
price: 0,
|
|
})
|
|
}
|
|
|
|
return toList
|
|
}
|
|
|
|
func (widget *Widget) updateCurrencies() {
|
|
defer func() {
|
|
recover()
|
|
}()
|
|
for _, fromCurrency := range widget.list.items {
|
|
|
|
var (
|
|
client http.Client
|
|
jsonResponse cResponse
|
|
)
|
|
|
|
client = http.Client{
|
|
Timeout: time.Duration(5 * time.Second),
|
|
}
|
|
|
|
request := makeRequest(fromCurrency)
|
|
response, err := client.Do(request)
|
|
|
|
if err != nil {
|
|
ok = false
|
|
} else {
|
|
ok = true
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
|
|
_ = json.NewDecoder(response.Body).Decode(&jsonResponse)
|
|
|
|
setPrices(&jsonResponse, fromCurrency)
|
|
}
|
|
|
|
}
|
|
|
|
func makeRequest(currency *fromCurrency) *http.Request {
|
|
fsym := currency.name
|
|
tsyms := ""
|
|
for _, to := range currency.to {
|
|
tsyms += fmt.Sprintf("%s,", to.name)
|
|
}
|
|
|
|
url := fmt.Sprintf("%s?fsym=%s&tsyms=%s", baseURL, fsym, tsyms)
|
|
request, err := http.NewRequest("GET", url, nil)
|
|
|
|
if err != nil {
|
|
}
|
|
|
|
return request
|
|
}
|
|
|
|
func setPrices(response *cResponse, currencry *fromCurrency) {
|
|
for idx, toCurrency := range currencry.to {
|
|
currencry.to[idx].price = (*response)[toCurrency.name]
|
|
}
|
|
}
|