Code re-organization

report/command.go

@@ -0,0 +1,75 @@
+package report
+
+import (
+	"context"
+	"fmt"
+	"github.com/charmbracelet/lipgloss"
+	"github.com/charmbracelet/lipgloss/table"
+	"github.com/siteworxpro/img-proxy-url-generator/aws"
+	"github.com/siteworxpro/img-proxy-url-generator/config"
+	"github.com/siteworxpro/img-proxy-url-generator/generator"
+	"github.com/siteworxpro/img-proxy-url-generator/redis"
+	"sort"
+	"strconv"
+	"time"
+)
+
+const lastAccessKey = "imgproxy:%s:last_access"
+const requestsKey = "imgproxy:%s:requests"
+
+var rowStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("#FFFFFF"))
+var headerStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("#FFFFFF")).Bold(true)
+
+func Handle(config *config.Config) error {
+	a := aws.NewClient(config)
+	r, err := redis.New(config)
+	ig, err := generator.NewGenerator(config)
+	if err != nil {
+		return err
+	}
+
+	var continuationToken *string
+	list, err := a.ListBucketContents(continuationToken)
+	if err != nil {
+		return err
+	}
+
+	var rows [][]string
+	for list.StartAfter != "" {
+		for _, image := range list.Images {
+			dlUrl, err := ig.GenerateUrl(image.S3Path, []string{}, generator.DEF)
+			if err != nil {
+				return err
+			}
+			lastAccessedS, err := r.GetClient().Get(context.Background(), fmt.Sprintf(lastAccessKey, image.S3Path)).Result()
+			lastAccessedI, _ := strconv.ParseInt(lastAccessedS, 10, 64)
+			lastAccessed := time.Unix(lastAccessedI, 0)
+			requestsCount, err := r.GetClient().Get(context.Background(), fmt.Sprintf(requestsKey, image.S3Path)).Result()
+
+			rows = append(rows, []string{image.S3Path, requestsCount, lastAccessed.Format(time.DateTime), dlUrl})
+		}
+
+		continuationToken = &list.StartAfter
+		list, err = a.ListBucketContents(continuationToken)
+	}
+
+	// sort by last accessed
+	sort.Slice(rows, func(i, j int) bool {
+		return rows[i][2] > rows[j][2]
+	})
+
+	t := table.New().StyleFunc(func(row int, col int) lipgloss.Style {
+		switch {
+		case row == 0:
+			return headerStyle
+		default:
+			return rowStyle
+		}
+	}).
+		Headers("Image", "Times Accessed", "Last Accessed", "URL").
+		Rows(rows...)
+
+	fmt.Println(t)
+
+	return nil
+}