From 89bb8e384d39dd1872b6b02786ed321c301277c6 Mon Sep 17 00:00:00 2001
From: "Alex Ellis (OpenFaaS Ltd)" <alexellis2@gmail.com>
Date: Sat, 3 Feb 2024 11:28:03 +0100
Subject: [PATCH] Add increases command to CLI

Signed-off-by: Alex Ellis (OpenFaaS Ltd) <alexellis2@gmail.com>
---
 cmd/increases.go | 88 ++++++++++++++++++++++++++++++++++++++++++++++++
 cmd/root.go      |  1 +
 pkg/client.go    | 55 ++++++++++++++++++++++++++++++
 3 files changed, 144 insertions(+)
 create mode 100644 cmd/increases.go

diff --git a/cmd/increases.go b/cmd/increases.go
new file mode 100644
index 0000000..7bdd6ec
--- /dev/null
+++ b/cmd/increases.go
@@ -0,0 +1,88 @@
+package cmd
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/self-actuated/actuated-cli/pkg"
+	"github.com/spf13/cobra"
+)
+
+func makeIncreases() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "increases",
+		Short: "Get job increases for an organisation",
+	}
+
+	cmd.RunE = runIncreasesE
+
+	cmd.Flags().Bool("json", false, "Request output in JSON format")
+	cmd.Flags().Bool("staff", false, "Request staff increases")
+	cmd.Flags().Int("days", 30, "The number of days to look back for increases")
+
+	return cmd
+}
+
+func runIncreasesE(cmd *cobra.Command, args []string) error {
+
+	var owner string
+	if len(args) == 1 {
+		owner = strings.TrimSpace(args[0])
+	}
+
+	pat, err := getPat(cmd)
+	if err != nil {
+		return err
+	}
+
+	staff, err := cmd.Flags().GetBool("staff")
+	if err != nil {
+		return err
+	}
+
+	requestJson, err := cmd.Flags().GetBool("json")
+	if err != nil {
+		return err
+	}
+
+	if len(pat) == 0 {
+		return fmt.Errorf("pat is required")
+	}
+
+	c := pkg.NewClient(http.DefaultClient, os.Getenv("ACTUATED_URL"))
+	days, err := cmd.Flags().GetInt("days")
+	if err != nil {
+		return err
+	}
+
+	startDate := time.Now().Add(-1 * time.Duration(days) * 24 * time.Hour)
+
+	res, status, err := c.GetBuildIncreases(pat, owner, startDate, staff, requestJson)
+	if err != nil {
+		return err
+	}
+
+	if status != http.StatusOK {
+		return fmt.Errorf("unexpected status code: %d, message: %s", status, string(res))
+	}
+
+	if requestJson {
+
+		var prettyJSON bytes.Buffer
+		err := json.Indent(&prettyJSON, []byte(res), "", "  ")
+		if err != nil {
+			return err
+		}
+		res = prettyJSON.String()
+	}
+
+	fmt.Println(res)
+
+	return nil
+
+}
diff --git a/cmd/root.go b/cmd/root.go
index dd9b077..69823cb 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -47,6 +47,7 @@ https://github.com/self-actuated/actuated-cli
 	root.AddCommand(makeAgentLogs())
 	root.AddCommand(makeLogs())
 	root.AddCommand(makeUpgrade())
+	root.AddCommand(makeIncreases())
 
 	root.AddCommand(makeSSH())
 
diff --git a/pkg/client.go b/pkg/client.go
index d0a95f1..7573b27 100644
--- a/pkg/client.go
+++ b/pkg/client.go
@@ -76,6 +76,61 @@ func (c *Client) ListJobs(patStr string, owner string, staff bool, json bool) (s
 	return string(body), res.StatusCode, nil
 }
 
+func (c *Client) GetBuildIncreases(patStr string, owner string, startDate time.Time, staff bool, json bool) (string, int, error) {
+
+	u, _ := url.Parse(c.baseURL)
+	u.Path = "/api/v1/job-increases"
+	q := u.Query()
+
+	if staff {
+		q.Set("staff", "1")
+	}
+
+	if len(owner) > 0 {
+		q.Set("owner", owner)
+	}
+	q.Add("startDate", startDate.Format("2006-01-02"))
+	log.Printf("Date: %s", startDate.Format("2006-01-02"))
+	u.RawQuery = q.Encode()
+
+	req, err := http.NewRequest(http.MethodGet, u.String(), nil)
+	if err != nil {
+		return "", http.StatusBadRequest, err
+	}
+
+	if json {
+		req.Header.Set("Accept", "application/json")
+	}
+
+	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.StatusServiceUnavailable, err
+	}
+
+	var body []byte
+	if res.Body != nil {
+		defer res.Body.Close()
+		body, _ = io.ReadAll(res.Body)
+	}
+
+	return string(body), res.StatusCode, nil
+}
+
 func (c *Client) ListRunners(patStr string, owner string, staff, images, json bool) (string, int, error) {
 
 	u, _ := url.Parse(c.baseURL)