package wp import ( "encoding/json" "fmt" "github.com/fthvgb1/wp-go/app/cmd/reload" "github.com/fthvgb1/wp-go/app/pkg/cache" "github.com/fthvgb1/wp-go/app/pkg/logs" "github.com/fthvgb1/wp-go/app/pkg/models" "github.com/fthvgb1/wp-go/app/wpconfig" "github.com/fthvgb1/wp-go/helper/slice" str "github.com/fthvgb1/wp-go/helper/strings" "github.com/fthvgb1/wp-go/model" "regexp" ) func (h *Handle) DisplayHeaderText() bool { return h.themeMods.ThemeSupport.CustomHeader.HeaderText && "blank" != h.themeMods.HeaderTextcolor } func (h *Handle) GetCustomHeaderImg() (r models.PostThumbnail, isRand bool) { var err error img := reload.GetAnyValBys("headerImages", h.theme, func(theme string) []models.PostThumbnail { hs, er := h.GetHeaderImages(h.theme) if er != nil { err = er return nil } return hs }) if err != nil { logs.Error(err, "获取页眉背景图失败") return } hs := slice.Copy(img) if len(hs) < 1 { return } if len(hs) > 1 { isRand = true } r, _ = slice.RandPop(&hs) return } type VideoPlay struct { Pause string `json:"pause,omitempty"` Play string `json:"play,omitempty"` PauseSpeak string `json:"pauseSpeak,omitempty"` PlaySpeak string `json:"playSpeak,omitempty"` } type VideoSetting struct { MimeType string `json:"mimeType,omitempty"` PosterUrl string `json:"posterUrl,omitempty"` VideoUrl string `json:"videoUrl,omitempty"` Width int `json:"width,omitempty"` Height int `json:"height,omitempty"` MinWidth int `json:"minWidth,omitempty"` MinHeight int `json:"minHeight,omitempty"` L10n VideoPlay `json:"l10n"` } var videoReg = regexp.MustCompile(`^https?://(?:www\.)?(?:youtube\.com/watch|youtu\.be/)`) func GetVideoSetting(h *Handle, u string) (string, error) { img, _ := h.GetCustomHeaderImg() v := VideoSetting{ MimeType: GetMimeType(u), PosterUrl: img.Path, VideoUrl: u, Width: img.Width, Height: img.Height, MinWidth: 900, MinHeight: 500, L10n: VideoPlay{ Pause: "暂停", Play: "播放", PauseSpeak: "视频已暂停", PlaySpeak: "视频正在播放", }, } if is := videoReg.FindString(u); is != "" { v.MimeType = "video/x-youtube" } _ = h.DoActionFilter("videoSetting", "", &v) s, err := json.Marshal(v) if err != nil { return "", err } setting := fmt.Sprintf(`var %s = %s`, "_wpCustomHeaderSettings", string(s)) script := str.Join(`\n") return script, nil } func CustomVideo(h *Handle, scene ...string) (ok bool) { mod, err := wpconfig.GetThemeMods(h.theme) if err != nil { logs.Error(err, "getThemeMods fail", h.theme) return } if !mod.ThemeSupport.CustomHeader.Video || (mod.HeaderVideo < 1 && mod.ExternalHeaderVideo == "") { return } u := "" if mod.HeaderVideo > 0 { post, err := cache.GetPostById(h.C, uint64(mod.HeaderVideo)) if err != nil { logs.Error(err, "get headerVideo fail", mod.HeaderVideo) return } u = post.Metas["_wp_attached_file"].(string) u = str.Join("/wp-content/uploads/", u) } else { u = mod.ExternalHeaderVideo } hs, err := GetVideoSetting(h, u) if err != nil { logs.Error(err, "get headerVideo fail", mod.HeaderVideo) return } scripts := []string{ "/wp-includes/js/dist/vendor/wp-polyfill-inert.min.js", "/wp-includes/js/dist/vendor/regenerator-runtime.min.js", "/wp-includes/js/dist/vendor/wp-polyfill.min.js", "/wp-includes/js/dist/dom-ready.min.js", "/wp-includes/js/dist/hooks.min.js", "/wp-includes/js/dist/i18n.min.js", "/wp-includes/js/dist/a11y.min.js", "/wp-includes/js/wp-custom-header.min.js", } scripts = slice.Map(scripts, func(t string) string { return fmt.Sprintf(` `, t, str.Replaces(t, []string{ "/wp-includes/js/dist/vendor/", "/wp-includes/js/dist/", "/wp-includes/js/", ".min.js", ".js", "wp-", "", })) }) var tr = ` ` c := []Components[string]{ NewComponent("wp-a11y-js-translations", tr, true, 10.0065, nil), NewComponent("VideoSetting", hs, true, 10.0064, nil), NewComponent("header-script", scripts[len(scripts)-1], true, 10.0063, nil), } for _, s := range scene { h.PushGroupFooterScript(s, "wp-custom-header", 10.0066, scripts[0:len(scripts)-2]...) h.PushFooterScript(s, c...) } ok = true return } func (h *Handle) GetHeaderImages(theme string) (r []models.PostThumbnail, err error) { meta, err := wpconfig.GetThemeMods(theme) if err != nil || meta.HeaderImage == "" { return } if "random-uploaded-image" != meta.HeaderImage { m, er := cache.GetPostById(h.C, uint64(meta.HeaderImagData.AttachmentId)) if er != nil { err = er return } m.Thumbnail = thumb(m, theme) r = []models.PostThumbnail{m.Thumbnail} return } headers, er := model.Finds[models.Posts](h.C, model.Conditions( model.Where(model.SqlBuilder{ {"post_type", "attachment"}, {"post_status", "inherit"}, {"meta_value", theme}, {"meta_key", "_wp_attachment_is_custom_header"}, }), model.Fields("a.ID"), model.Group("a.ID"), model.Join(model.SqlBuilder{ {" a", "left join", "wp_postmeta b", "a.ID=b.post_id"}, }), )) if er != nil { err = er return } if len(headers) > 0 { posts, er := cache.GetPostsByIds(h.C, slice.Map(headers, func(t models.Posts) uint64 { return t.Id })) if er != nil { err = er return } r = slice.Map(posts, func(m models.Posts) models.PostThumbnail { return thumb(m, theme) }) } return } func thumb(m models.Posts, theme string) models.PostThumbnail { m.Thumbnail = wpconfig.Thumbnail(m.AttachmentMetadata, "full", "", "thumbnail", "post-thumbnail", fmt.Sprintf("%s-thumbnail-avatar", theme)) m.Thumbnail.Width = m.AttachmentMetadata.Width m.Thumbnail.Height = m.AttachmentMetadata.Height if m.Thumbnail.Path != "" { if len(m.AttachmentMetadata.Sizes) > 0 { m.Thumbnail.Srcset = str.Join(m.Thumbnail.Path, " 2000w, ", m.Thumbnail.Srcset) } } return m.Thumbnail }