This commit is contained in:
abs3nt 2023-02-28 18:37:33 -08:00
parent 08a0b2c078
commit af3ceec223

View File

@ -12,10 +12,16 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
type Tokens []struct { type Search struct {
ID string `json:"id"` Coins []struct {
Symbol string `json:"symbol"` ID string `json:"id"`
Name string `json:"name"` Name string `json:"name"`
APISymbol string `json:"api_symbol"`
Symbol string `json:"symbol"`
MarketCapRank int `json:"market_cap_rank"`
Thumb string `json:"thumb"`
Large string `json:"large"`
} `json:"coins"`
} }
type Price map[string]USD type Price map[string]USD
@ -26,8 +32,7 @@ type USD struct {
var ( var (
price_url = "https://api.coingecko.com/api/v3/simple/price?" price_url = "https://api.coingecko.com/api/v3/simple/price?"
vs_url = "https://api.coingecko.com/api/v3/simple/supported_vs_currencies" search_url = "https://api.coingecko.com/api/v3/search?query="
tokens_url = "https://api.coingecko.com/api/v3/coins/list"
) )
var amount float64 var amount float64
@ -37,6 +42,13 @@ var cryptoCmd = &cobra.Command{
Use: "crypto", Use: "crypto",
Short: "get price of token", Short: "get price of token",
Long: `get price of token`, Long: `get price of token`,
Args: cobra.MatchAll(cobra.RangeArgs(1, 2), cobra.OnlyValidArgs),
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
if len(args) >= 2 {
return nil, cobra.ShellCompDirectiveNoFileComp
}
return getTokens(toComplete), cobra.ShellCompDirectiveNoFileComp
},
RunE: func(cmd *cobra.Command, args []string) error { RunE: func(cmd *cobra.Command, args []string) error {
price, err := usdPrice(args[0]) price, err := usdPrice(args[0])
if err != nil { if err != nil {
@ -55,6 +67,31 @@ var cryptoCmd = &cobra.Command{
}, },
} }
func getTokens(in string) []string {
if in == "" {
return []string{"ethereum", "bitcoin", "tether", "binancecoin", "usd-coin", "ripple", "cardano", "dogecoin", "matic-network"}
}
resp, err := http.Get(search_url + in)
if err != nil {
return []string{err.Error()}
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
return []string{err.Error()}
}
var tokens Search
err = json.Unmarshal(body, &tokens)
if err != nil {
return []string{err.Error()}
}
out := []string{}
for _, token := range tokens.Coins {
out = append(out, token.ID)
}
return out
}
func usdPrice(token string) (float64, error) { func usdPrice(token string) (float64, error) {
resp, err := http.Get(price_url + fmt.Sprintf("ids=%s&vs_currencies=usd", token)) resp, err := http.Get(price_url + fmt.Sprintf("ids=%s&vs_currencies=usd", token))
if err != nil { if err != nil {