package api import ( "context" "encoding/json" "strconv" "strings" "time" "github.com/stephenafamo/bob" "github.com/stephenafamo/bob/dialect/sqlite" "github.com/stephenafamo/bob/dialect/sqlite/dialect" "github.com/stephenafamo/bob/dialect/sqlite/sm" "github.com/tigorlazuardi/redmage/models" "github.com/tigorlazuardi/redmage/pkg/errs" ) type ImageListParams struct { Q string SFW bool OrderBy string Sort string Offset int64 Limit int64 Device int32 Subreddit int32 CreatedAt time.Time } func (ilp *ImageListParams) FillFromQuery(query Queryable) { ilp.Q = query.Get("q") ilp.SFW, _ = strconv.ParseBool(query.Get("sfw")) ilp.OrderBy = query.Get("order_by") ilp.Sort = strings.ToLower(query.Get("sort")) ilp.Offset, _ = strconv.ParseInt(query.Get("offset"), 10, 64) ilp.Limit, _ = strconv.ParseInt(query.Get("limit"), 10, 64) if ilp.Limit > 100 { ilp.Limit = 100 } if ilp.Limit < 1 { ilp.Limit = 25 } device, _ := strconv.ParseInt(query.Get("device"), 10, 32) ilp.Device = int32(device) subreddit, _ := strconv.ParseInt(query.Get("subreddit"), 10, 32) ilp.Subreddit = int32(subreddit) createdAtint, _ := strconv.ParseInt(query.Get("created_at"), 10, 64) if createdAtint > 0 { ilp.CreatedAt = time.Unix(createdAtint, 0) } } func (ilp ImageListParams) CountQuery() (expr []bob.Mod[*dialect.SelectQuery]) { if ilp.Q != "" { arg := sqlite.Arg("%" + ilp.Q + "%") expr = append(expr, sm.Where( models.ImageColumns.Title.Like(arg). Or(models.ImageColumns.Poster.Like(arg)). Or(models.ImageColumns.ImageRelativePath.Like(arg)), ), ) } if ilp.SFW { expr = append(expr, models.SelectWhere.Images.NSFW.EQ(0)) } if ilp.Device > 0 { expr = append(expr, models.SelectWhere.Images.DeviceID.EQ(ilp.Device)) } if ilp.Subreddit > 0 { expr = append(expr, models.SelectWhere.Images.SubredditID.EQ(ilp.Subreddit)) } if !ilp.CreatedAt.IsZero() { expr = append(expr, models.SelectWhere.Images.CreatedAt.GTE(ilp.CreatedAt.Format(time.RFC3339))) } return expr } func (ilp ImageListParams) Query() (expr []bob.Mod[*dialect.SelectQuery]) { expr = append(expr, ilp.CountQuery()...) if ilp.Limit > 0 { expr = append(expr, sm.Limit(ilp.Limit)) } if ilp.Offset > 0 { expr = append(expr, sm.Offset(ilp.Offset)) } if ilp.OrderBy != "" { order := sm.OrderBy(sqlite.Quote(ilp.OrderBy)) if ilp.Sort == "desc" { expr = append(expr, order.Desc()) } else { expr = append(expr, order.Asc()) } } else { expr = append(expr, sm.OrderBy(models.ImageColumns.CreatedAt).Desc()) } return expr } type ImageListResult struct { Total int64 Images models.ImageSlice } func (im ImageListResult) MarshalJSON() ([]byte, error) { type I struct { *models.Image Device *models.Device `json:"device,omitempty"` Subreddit *models.Subreddit `json:"subreddit,omitempty"` } type A struct { Total int64 `json:"total"` Images []I `json:"images"` } a := A{Total: im.Total} a.Images = make([]I, len(im.Images)) for i := 0; i < len(a.Images); i++ { a.Images[i].Image = im.Images[i] a.Images[i].Device = im.Images[i].R.Device a.Images[i].Subreddit = im.Images[i].R.Subreddit } return json.Marshal(a) } func (api *API) ImagesList(ctx context.Context, params ImageListParams) (result ImageListResult, err error) { ctx, span := tracer.Start(ctx, "*API.ImagesList") defer span.End() result.Images, err = models.Images.Query(ctx, api.db, params.Query()...).All() if err != nil { return result, errs.Wrapw(err, "failed to query for images", "params", params) } result.Total, err = models.Images.Query(ctx, api.db, params.CountQuery()...).Count() if err != nil { return result, errs.Wrapw(err, "failed to query for images", "params", params) } return result, err } func (api *API) ImagesListWithDevicesAndSubreddits(ctx context.Context, params ImageListParams) (result ImageListResult, err error) { ctx, span := tracer.Start(ctx, "*API.ImagesListWithDevicesAndSubreddits") defer span.End() result, err = api.ImagesList(ctx, params) if err != nil { return result, err } if err := result.Images.LoadImageDevice(ctx, api.db); err != nil { return result, errs.Wrapw(err, "failed to load image devices", "params", params) } if err := result.Images.LoadImageSubreddit(ctx, api.db); err != nil { return result, errs.Wrapw(err, "failed to load image subreddits", "params", params) } return result, err }