71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
"github.com/fthvgb1/wp-go/helper/slice"
|
|
"github.com/fthvgb1/wp-go/internal/pkg/logs"
|
|
"github.com/fthvgb1/wp-go/internal/pkg/models"
|
|
"github.com/fthvgb1/wp-go/internal/plugins"
|
|
"github.com/fthvgb1/wp-go/model"
|
|
"strconv"
|
|
)
|
|
|
|
func GetPostMetaByPostIds(args ...any) (r map[uint64]map[string]any, err error) {
|
|
r = make(map[uint64]map[string]any)
|
|
ctx := args[0].(context.Context)
|
|
ids := args[1].([]uint64)
|
|
rr, err := model.Find[models.Postmeta](ctx, model.SqlBuilder{
|
|
{"post_id", "in", ""},
|
|
}, "*", "", nil, nil, nil, 0, slice.ToAnySlice(ids))
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, postmeta := range rr {
|
|
if _, ok := r[postmeta.PostId]; !ok {
|
|
r[postmeta.PostId] = make(map[string]any)
|
|
}
|
|
if postmeta.MetaKey == "_wp_attachment_metadata" {
|
|
metadata, err := models.AttachmentMetadata(postmeta.MetaValue)
|
|
if err != nil {
|
|
logs.ErrPrintln(err, "解析postmeta失败", postmeta.MetaId, postmeta.MetaValue)
|
|
continue
|
|
}
|
|
r[postmeta.PostId][postmeta.MetaKey] = metadata
|
|
|
|
} else {
|
|
r[postmeta.PostId][postmeta.MetaKey] = postmeta.MetaValue
|
|
}
|
|
|
|
}
|
|
return
|
|
}
|
|
|
|
func ToPostThumb(c context.Context, meta map[string]any, host string) (r models.PostThumbnail) {
|
|
if meta != nil {
|
|
m, ok := meta["_thumbnail_id"]
|
|
if !ok {
|
|
return
|
|
}
|
|
id, err := strconv.ParseUint(m.(string), 10, 64)
|
|
if err != nil {
|
|
return
|
|
}
|
|
mx, err := GetPostMetaByPostIds(c, []uint64{id})
|
|
if err != nil || mx == nil {
|
|
return
|
|
}
|
|
mm, ok := mx[id]
|
|
if !ok || mm == nil {
|
|
return
|
|
}
|
|
x, ok := mm["_wp_attachment_metadata"]
|
|
if ok {
|
|
metadata, ok := x.(models.WpAttachmentMetadata)
|
|
if ok {
|
|
r = plugins.Thumbnail(metadata, "post-thumbnail", host, "thumbnail")
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|