Add metering command

Signed-off-by: Alex Ellis (OpenFaaS Ltd) <[email protected]>

Author: alexellis

README.md

@@ -108,6 +108,17 @@ actuated-cli logs \
     server1
 ```
 
+## View the metering snapshot from a VM
+
+Use the `--id` flag to specify the runner ID.
+
+```bash
+actuated-cli metering \
+    --owner actuated-samples \
+    --id ea5c285282620927689d90af3cfa3be2d5e2d004 \
+    server1
+```
+
 ## Check the logs of the actuated agent service
 
 Show the logs of the actuated agent binary running on your server.

cmd/metering.go

@@ -0,0 +1,86 @@
+package cmd
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"strings"
+
+	"github.com/self-actuated/actuated-cli/pkg"
+	"github.com/spf13/cobra"
+)
+
+func makeMetering() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "metering",
+		Short: "Fetch metering from a VM",
+		Long:  `Fetch the metering snapshot from a specific VM.`,
+
+		Example: `# Get the metering snapshot from a specific VM using its hostname as the --id
+actuated-cli metering --owner=OWNER --id=ID HOST
+`,
+	}
+
+	cmd.RunE = runMeteringE
+
+	cmd.Flags().StringP("owner", "o", "", "List logs owned by this user")
+	cmd.Flags().String("id", "", "ID variable for a specific runner VM hostname")
+
+	return cmd
+}
+
+func runMeteringE(cmd *cobra.Command, args []string) error {
+	if len(args) < 1 {
+		return fmt.Errorf("specify the host as an argument")
+	}
+	host := strings.TrimSpace(args[0])
+
+	pat, err := getPat(cmd)
+	if err != nil {
+		return err
+	}
+
+	staff, err := cmd.Flags().GetBool("staff")
+	if err != nil {
+		return err
+	}
+
+	owner, err := cmd.Flags().GetString("owner")
+	if err != nil {
+		return err
+	}
+
+	id, err := cmd.Flags().GetString("id")
+	if err != nil {
+		return err
+	}
+
+	if len(host) == 0 {
+		return fmt.Errorf("host is required")
+	}
+
+	if len(owner) == 0 {
+		return fmt.Errorf("owner is required")
+	}
+
+	if len(pat) == 0 {
+		return fmt.Errorf("pat is required")
+	}
+
+	c := pkg.NewClient(http.DefaultClient, os.Getenv("ACTUATED_URL"))
+
+	res, status, err := c.GetMetering(pat, owner, host, id, staff)
+
+	if err != nil {
+		return err
+	}
+
+	if status != http.StatusAccepted {
+		return fmt.Errorf("unexpected status code: %d, body: %s", status, res)
+	}
+
+	fmt.Println(res)
+
+	return nil
+
+}

pkg/client.go

@@ -1,6 +1,8 @@
 package pkg
 
 import (
+	"bytes"
+	"encoding/json"
 	"fmt"
 	"io"
 	"log"
@@ -292,6 +294,67 @@ func (c *Client) GetLogs(patStr, owner, host, id string, age time.Duration, staf
 	return string(body), res.StatusCode, nil
 }
 
+func (c *Client) GetMetering(patStr, owner, host, id string, staff bool) (string, int, error) {
+
+	u, _ := url.Parse(c.baseURL)
+	u.Path = "/api/v1/metering"
+
+	q := u.Query()
+	q.Set("owner", owner)
+	q.Set("host", host)
+
+	if len(id) > 0 {
+		q.Set("id", id)
+	} else {
+		return "", http.StatusBadRequest, fmt.Errorf("id is required")
+	}
+
+	if staff {
+		q.Set("staff", "1")
+	}
+
+	u.RawQuery = q.Encode()
+
+	req, err := http.NewRequest(http.MethodGet, u.String(), nil)
+	if err != nil {
+		return "", http.StatusBadRequest, err
+	}
+
+	req.Header.Set("Authorization", "Bearer "+patStr)
+
+	if os.Getenv("DEBUG") == "1" {
+		sanitised := http.Header{}
+		for k, v := range req.Header {
+
+			if k == "Authorization" {
+				v = []string{"redacted"}
+			}
+			sanitised[k] = v
+		}
+
+		fmt.Printf("URL %s\nHeaders: %v\n", u.String(), sanitised)
+	}
+
+	res, err := c.httpClient.Do(req)
+	if err != nil {
+		return "", http.StatusBadRequest, err
+	}
+
+	var body []byte
+	if res.Body != nil {
+		defer res.Body.Close()
+		body, _ = io.ReadAll(res.Body)
+	}
+
+	var prettyJSON bytes.Buffer
+
+	if err = json.Indent(&prettyJSON, []byte(body), "", "  "); err != nil {
+		return "", http.StatusBadRequest, err
+	}
+
+	return prettyJSON.String(), res.StatusCode, nil
+}
+
 func (c *Client) GetAgentLogs(patStr, owner, host string, age time.Duration, staff bool) (string, int, error) {
 
 	mins := int(age.Minutes())