Redmage/views/homeview/homeview_data.go

96 lines
2.1 KiB
Go

package homeview
import (
"fmt"
"slices"
"strings"
"time"
"github.com/tigorlazuardi/redmage/api"
"github.com/tigorlazuardi/redmage/models"
)
type Data struct {
SubredditsList api.ListSubredditsResult
RecentlyAddedImages RecentlyAddedImages
TotalImages int64
Error string
Now time.Time
SFW bool
}
type RecentlyAddedImages = []RecentlyAddedImage
type RecentlyAddedImage struct {
Device *models.Device
Subreddits []Subreddit
}
type Subreddit struct {
Subreddit *models.Subreddit
Images models.ImageSlice
}
func NewRecentlyAddedImages(images models.ImageSlice) RecentlyAddedImages {
r := make(RecentlyAddedImages, 0, len(images))
var count int
for _, image := range images {
if image.R.Device == nil || image.R.Subreddit == nil {
continue
}
var deviceFound bool
for i, ra := range r {
if ra.Device.ID == image.R.Device.ID {
deviceFound = true
var subredditFound bool
for j, subreddit := range r[i].Subreddits {
if subreddit.Subreddit.ID == image.R.Subreddit.ID {
subredditFound = true
r[i].Subreddits[j].Images = append(r[i].Subreddits[j].Images, image)
count++
}
}
if !subredditFound {
r[i].Subreddits = append(r[i].Subreddits, Subreddit{
Subreddit: image.R.Subreddit,
Images: models.ImageSlice{image},
})
count++
}
}
}
if !deviceFound {
count++
r = append(r, RecentlyAddedImage{
Device: image.R.Device,
Subreddits: []Subreddit{
{
Subreddit: image.R.Subreddit,
Images: models.ImageSlice{image},
},
},
})
}
}
for _, r := range r {
slices.SortFunc(r.Subreddits, func(left, right Subreddit) int {
leftName := strings.ToLower(left.Subreddit.Name)
rightName := strings.ToLower(right.Subreddit.Name)
return strings.Compare(leftName, rightName)
})
}
slices.SortFunc(r, func(left, right RecentlyAddedImage) int {
leftName := strings.ToLower(left.Device.Name)
rightName := strings.ToLower(right.Device.Name)
return strings.Compare(leftName, rightName)
})
fmt.Println("image count", count)
return r
}