mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
14abd422b2
Have the modules set their own titles If no title is specified, use this title, rather than default to name
146 lines
3.5 KiB
Go
146 lines
3.5 KiB
Go
package blockfolio
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/wtf"
|
|
)
|
|
|
|
type Widget struct {
|
|
wtf.TextWidget
|
|
|
|
app *tview.Application
|
|
device_token string
|
|
settings *Settings
|
|
}
|
|
|
|
func NewWidget(app *tview.Application, settings *Settings) *Widget {
|
|
widget := Widget{
|
|
TextWidget: wtf.NewTextWidget(app, settings.common, false),
|
|
|
|
app: app,
|
|
device_token: settings.deviceToken,
|
|
settings: settings,
|
|
}
|
|
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
func (widget *Widget) Refresh() {
|
|
positions, err := Fetch(widget.device_token)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
content := widget.contentFrom(positions)
|
|
|
|
widget.app.QueueUpdateDraw(func() {
|
|
widget.View.SetTitle(widget.CommonSettings.Title)
|
|
widget.View.SetText(content)
|
|
})
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
func (widget *Widget) contentFrom(positions *AllPositionsResponse) string {
|
|
res := ""
|
|
totalFiat := float32(0.0)
|
|
|
|
for i := 0; i < len(positions.PositionList); i++ {
|
|
colorForChange := widget.settings.colors.grows
|
|
|
|
if positions.PositionList[i].TwentyFourHourPercentChangeFiat <= 0 {
|
|
colorForChange = widget.settings.colors.drop
|
|
}
|
|
|
|
totalFiat += positions.PositionList[i].HoldingValueFiat
|
|
|
|
if widget.settings.displayHoldings {
|
|
res = res + fmt.Sprintf(
|
|
"[%s]%-6s - %5.2f ([%s]%.3fk [%s]%.2f%s)\n",
|
|
widget.settings.colors.name,
|
|
positions.PositionList[i].Coin,
|
|
positions.PositionList[i].Quantity,
|
|
colorForChange,
|
|
positions.PositionList[i].HoldingValueFiat/1000,
|
|
colorForChange,
|
|
positions.PositionList[i].TwentyFourHourPercentChangeFiat,
|
|
"%",
|
|
)
|
|
} else {
|
|
res = res + fmt.Sprintf(
|
|
"[%s]%-6s - %5.2f ([%s]%.2f%s)\n",
|
|
widget.settings.colors.name,
|
|
positions.PositionList[i].Coin,
|
|
positions.PositionList[i].Quantity,
|
|
colorForChange,
|
|
positions.PositionList[i].TwentyFourHourPercentChangeFiat,
|
|
"%",
|
|
)
|
|
}
|
|
}
|
|
|
|
if widget.settings.displayHoldings {
|
|
res = res + fmt.Sprintf("\n[%s]Total value: $%.3fk", "green", totalFiat/1000)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
//always the same
|
|
const magic = "edtopjhgn2345piuty89whqejfiobh89-2q453"
|
|
|
|
type Position struct {
|
|
Coin string `json:"coin"`
|
|
LastPriceFiat float32 `json:"lastPriceFiat"`
|
|
TwentyFourHourPercentChangeFiat float32 `json:"twentyFourHourPercentChangeFiat"`
|
|
Quantity float32 `json:"quantity"`
|
|
HoldingValueFiat float32 `json:"holdingValueFiat"`
|
|
}
|
|
|
|
type AllPositionsResponse struct {
|
|
PositionList []Position `json:"positionList"`
|
|
}
|
|
|
|
func MakeApiRequest(token string, method string) ([]byte, error) {
|
|
client := &http.Client{}
|
|
url := "https://api-v0.blockfolio.com/rest/" + method + "/" + token + "?use_alias=true&fiat_currency=USD"
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Add("magic", magic)
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return body, err
|
|
}
|
|
|
|
func GetAllPositions(token string) (*AllPositionsResponse, error) {
|
|
jsn, _ := MakeApiRequest(token, "get_all_positions")
|
|
var parsed AllPositionsResponse
|
|
|
|
err := json.Unmarshal(jsn, &parsed)
|
|
if err != nil {
|
|
log.Fatalf("Failed to parse json %v", err)
|
|
return nil, err
|
|
}
|
|
return &parsed, err
|
|
}
|
|
|
|
func Fetch(token string) (*AllPositionsResponse, error) {
|
|
return GetAllPositions(token)
|
|
}
|