calculate the total volume and USD prices

price.go

@@ -2,7 +2,11 @@ package vwap
 
 import (
 	"encoding/json"
+	"fmt"
+	"math"
+	"math/big"
 	"net/http"
+	"strconv"
 )
 
 const priceEndpoint = "http://dev.api.gnoswap.io/v1/tokens/prices"
@@ -55,3 +59,48 @@ func fetchTokenPrices(endpoint string) ([]TokenPrice, error) {
 
 	return apiResponse.Data, nil
 }
+
+// calculateVolume calculates the total volume in the USD for each token.
+func calculateVolume(prices []TokenPrice) map[string]float64 {
+	volumeByToken := make(map[string]float64)
+
+	for _, price := range prices {
+		volume, err := strconv.ParseFloat(price.VolumeUSD24h, 64)
+		if err != nil {
+			fmt.Printf("failed to parse volume for token %s: %v\n", price.Path, err)
+		}
+
+		volumeByToken[price.Path] = volume
+	}
+
+	return volumeByToken
+}
+
+// calculateTokenUSDPrices calculates the USD price based on a base token (wugnot) price and token ratios.
+func calculateTokenUSDPrices(tokenData *TokenData, baseTokenPrice float64) map[string]float64 {
+	tokenPrices := make(map[string]float64)
+	baseRatio := new(big.Float)
+
+	// fund the base token ratio
+	for _, token := range tokenData.TokenRatio {
+		if token.TokenName == string(wugnot) {
+			ratio, _ := new(big.Float).SetString(token.Ratio)
+			baseRatio.Quo(ratio, big.NewFloat(math.Pow(2, 96)))
+			break
+		}
+	}
+
+	// calculate token prices based on the base token price and ratios.
+	for _, token := range tokenData.TokenRatio {
+		if token.TokenName != string(wugnot) {
+			ratio, _ := new(big.Float).SetString(token.Ratio)
+			tokenRatio := new(big.Float).Quo(ratio, big.NewFloat(math.Pow(2, 96)))
+			tokenPrice := new(big.Float).Quo(baseRatio, tokenRatio)
+
+			price, _ := tokenPrice.Float64()
+			tokenPrices[token.TokenName] = price * baseTokenPrice
+		}
+	}
+
+	return tokenPrices
+}

price_test.go

@@ -121,3 +121,69 @@ func TestFetchTokenPricesLive(t *testing.T) {
 		}
 	}
 }
+
+func TestCalculateVolume(t *testing.T) {
+	t.Parallel()
+	mockTokenPrices := []TokenPrice{
+		{
+			Path:         "token1",
+			VolumeUSD24h: "1000.50",
+		},
+		{
+			Path:         "token2",
+			VolumeUSD24h: "2500.75",
+		},
+		{
+			Path:         "token3",
+			VolumeUSD24h: "500.25",
+		},
+	}
+
+	volumes := calculateVolume(mockTokenPrices)
+
+	assert.Equal(t, 3, len(volumes))
+	assert.InDelta(t, 1000.50, volumes["token1"], 0.001)
+	assert.InDelta(t, 2500.75, volumes["token2"], 0.001)
+	assert.InDelta(t, 500.25, volumes["token3"], 0.001)
+}
+
+func TestCalculateTokenPrices(t *testing.T) {
+	t.Parallel()
+	mockTokenData := &TokenData{
+		Status: struct {
+			Height    int `json:"height"`
+			Timestamp int `json:"timestamp"`
+		}{
+			Height:    61946,
+			Timestamp: 1715064552,
+		},
+		TokenRatio: []struct {
+			TokenName string `json:"token"`
+			Ratio     string `json:"ratio"`
+		}{
+			{
+				TokenName: "gno.land/r/demo/wugnot",
+				Ratio:     "79228162514264337593543950336",
+			},
+			{
+				TokenName: "gno.land/r/demo/qux",
+				Ratio:     "60536002769587966558221762891",
+			},
+			{
+				TokenName: "gno.land/r/demo/foo",
+				Ratio:     "121074204438706762251182081654",
+			},
+			{
+				TokenName: "gno.land/r/demo/gns",
+				Ratio:     "236174327852992866806677676716",
+			},
+		},
+	}
+
+	baseTokenPrice := 1.0
+	tokenPrices := calculateTokenUSDPrices(mockTokenData, baseTokenPrice)
+
+	assert.InDelta(t, 1.3087775685458196, tokenPrices["gno.land/r/demo/qux"], 1e-5)
+	assert.InDelta(t, 0.6543768995349725, tokenPrices["gno.land/r/demo/foo"], 1e-5)
+	assert.InDelta(t, 0.3354647528142011, tokenPrices["gno.land/r/demo/gns"], 1e-5)
+}