Printing billing usage for servers

Author: moul

pkg/cli/x_billing.go

@@ -4,7 +4,16 @@
 
 package cli
 
-import "fmt"
+import (
+	"fmt"
+	"text/tabwriter"
+	"time"
+
+	"github.com/scaleway/scaleway-cli/pkg/commands"
+	"github.com/scaleway/scaleway-cli/pkg/pricing"
+	"github.com/scaleway/scaleway-cli/pkg/utils"
+	"github.com/scaleway/scaleway-cli/vendor/github.com/docker/docker/pkg/units"
+)
 
 var cmdBilling = &Command{
 	Exec:        runBilling,
@@ -16,20 +25,56 @@ var cmdBilling = &Command{
 
 func init() {
 	cmdBilling.Flag.BoolVar(&billingHelp, []string{"h", "-help"}, false, "Print usage")
+	cmdBilling.Flag.BoolVar(&billingNoTrunc, []string{"-no-trunc"}, false, "Don't truncate output")
+}
+
+// BillingArgs are flags for the `RunBilling` function
+type BillingArgs struct {
+	NoTrunc bool
 }
 
 // Flags
-var billingHelp bool // -h, --help flag
+var billingHelp bool    // -h, --help flag
+var billingNoTrunc bool // --no-trunc flag
 
-func runBilling(cmd *Command, args []string) error {
+func runBilling(cmd *Command, rawArgs []string) error {
 	if billingHelp {
 		return cmd.PrintUsage()
 	}
-	if len(args) > 0 {
+	if len(rawArgs) > 0 {
 		return cmd.PrintShortUsage()
 	}
 
-	fmt.Println("BILLING")
+	// cli parsing
+	args := commands.PsArgs{
+		NoTrunc: billingNoTrunc,
+	}
+	ctx := cmd.GetContext(rawArgs)
+
+	// table
+	w := tabwriter.NewWriter(ctx.Stdout, 20, 1, 3, ' ', 0)
+	defer w.Flush()
+	fmt.Fprintf(w, "ID\tNAME\tSTARTED\tMONTH PRICE\n")
+
+	// servers
+	servers, err := cmd.API.GetServers(true, 0)
+	if err != nil {
+		return err
+	}
+	for _, server := range *servers {
+		if server.State != "running" {
+			continue
+		}
+		shortID := utils.TruncIf(server.Identifier, 8, !args.NoTrunc)
+		shortName := utils.TruncIf(utils.Wordify(server.Name), 25, !args.NoTrunc)
+		modificationTime, _ := time.Parse("2006-01-02T15:04:05.000000+00:00", server.ModificationDate)
+		modificationAgo := time.Now().UTC().Sub(modificationTime)
+		shortModificationDate := units.HumanDuration(modificationAgo)
+		usage := pricing.NewUsageByPath("/compute/c1/run")
+		usage.SetStartEnd(modificationTime, time.Now().UTC())
+
+		fmt.Fprintf(w, "server/%s\t%s\t%s\t%s\n", shortID, shortName, shortModificationDate, usage.TotalString())
+	}
 
 	return nil
 }

pkg/pricing/usage.go

@@ -1,8 +1,11 @@
 package pricing
 
 import (
+	"fmt"
 	"math/big"
 	"time"
+
+	"github.com/dustin/go-humanize"
 )
 
 type Usage struct {
@@ -77,3 +80,9 @@ func (u *Usage) Total() *big.Rat {
 	total := new(big.Rat).Mul(u.BillableQuantity(), u.PricingObject.UnitPrice)
 	return ratMin(total, u.PricingObject.UnitPriceCap)
 }
+
+func (u *Usage) TotalString() string {
+	total := u.Total()
+	floatVal, _ := total.Float64()
+	return fmt.Sprintf("%s %s", humanize.Ftoa(floatVal), u.PricingObject.Currency)
+}