Computing total price

Author: moul

pkg/cli/x_billing.go

@@ -6,6 +6,7 @@ package cli
 
 import (
 	"fmt"
+	"math/big"
 	"text/tabwriter"
 	"time"
 
@@ -61,6 +62,9 @@ func runBilling(cmd *Command, rawArgs []string) error {
 	if err != nil {
 		return err
 	}
+
+	totalMonthPrice := new(big.Rat)
+
 	for _, server := range *servers {
 		if server.State != "running" {
 			continue
@@ -73,8 +77,12 @@ func runBilling(cmd *Command, rawArgs []string) error {
 		usage := pricing.NewUsageByPath("/compute/c1/run")
 		usage.SetStartEnd(modificationTime, time.Now().UTC())
 
+		totalMonthPrice = totalMonthPrice.Add(totalMonthPrice, usage.Total())
+
 		fmt.Fprintf(w, "server/%s\t%s\t%s\t%s\n", shortID, shortName, shortModificationDate, usage.TotalString())
 	}
 
+	fmt.Fprintf(w, "TOTAL\t\t\t%s\n", pricing.PriceString(totalMonthPrice, "EUR"))
+
 	return nil
 }

pkg/pricing/usage.go

@@ -1,11 +1,8 @@
 package pricing
 
 import (
-	"fmt"
 	"math/big"
 	"time"
-
-	"github.com/dustin/go-humanize"
 )
 
 type Usage struct {
@@ -82,7 +79,5 @@ func (u *Usage) Total() *big.Rat {
 }
 
 func (u *Usage) TotalString() string {
-	total := u.Total()
-	floatVal, _ := total.Float64()
-	return fmt.Sprintf("%s %s", humanize.Ftoa(floatVal), u.PricingObject.Currency)
+	return PriceString(u.Total(), u.PricingObject.Currency)
 }

pkg/pricing/utils.go

@@ -1,6 +1,11 @@
 package pricing
 
-import "math/big"
+import (
+	"fmt"
+	"math/big"
+
+	"github.com/scaleway/scaleway-cli/vendor/github.com/dustin/go-humanize"
+)
 
 var (
 	intZero = big.NewInt(0)
@@ -35,3 +40,8 @@ func ratMin(x, y *big.Rat) *big.Rat {
 	}
 	return x
 }
+
+func PriceString(price *big.Rat, currency string) string {
+	floatVal, _ := price.Float64()
+	return fmt.Sprintf("%s %s", humanize.Ftoa(floatVal), currency)
+}