78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
|
package qyglreport
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/base64"
|
|||
|
|
"html/template"
|
|||
|
|
"mime"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"sync"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
defaultBannerRel = "resources/qygl-media/banner.jpg"
|
|||
|
|
defaultFooterRel = "resources/qygl-media/footer.png"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
loadOnce sync.Once
|
|||
|
|
bannerURL template.URL
|
|||
|
|
footerURL template.URL
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func loadPosterAssets() {
|
|||
|
|
bannerURL = dataURLFromFile(resolveMediaFile([]string{
|
|||
|
|
defaultBannerRel,
|
|||
|
|
"resources/qygl-media/banner.png",
|
|||
|
|
}), "/reports/qygl/media/banner.jpg")
|
|||
|
|
footerURL = dataURLFromFile(defaultFooterRel, "/reports/qygl/media/footer.png")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func resolveMediaFile(candidates []string) string {
|
|||
|
|
for _, p := range candidates {
|
|||
|
|
if _, err := os.Stat(p); err == nil {
|
|||
|
|
return p
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if len(candidates) > 0 {
|
|||
|
|
return candidates[0]
|
|||
|
|
}
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func dataURLFromFile(path string, httpFallback string) template.URL {
|
|||
|
|
if path == "" {
|
|||
|
|
return template.URL(httpFallback)
|
|||
|
|
}
|
|||
|
|
data, err := os.ReadFile(path)
|
|||
|
|
if err != nil {
|
|||
|
|
return template.URL(httpFallback)
|
|||
|
|
}
|
|||
|
|
ext := filepath.Ext(path)
|
|||
|
|
mimeType := mime.TypeByExtension(ext)
|
|||
|
|
if mimeType == "" {
|
|||
|
|
switch ext {
|
|||
|
|
case ".jpg", ".jpeg":
|
|||
|
|
mimeType = "image/jpeg"
|
|||
|
|
case ".png":
|
|||
|
|
mimeType = "image/png"
|
|||
|
|
default:
|
|||
|
|
mimeType = "application/octet-stream"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
encoded := base64.StdEncoding.EncodeToString(data)
|
|||
|
|
return template.URL("data:" + mimeType + ";base64," + encoded)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PageTemplateData 企业报告 HTML 模板数据（海报内嵌 data URL，PDF 生成不依赖外链）
|
|||
|
|
func PageTemplateData(reportJSON template.JS) gin.H {
|
|||
|
|
loadOnce.Do(loadPosterAssets)
|
|||
|
|
return gin.H{
|
|||
|
|
"ReportJSON": reportJSON,
|
|||
|
|
"BannerPosterURL": bannerURL,
|
|||
|
|
"FooterPosterURL": footerURL,
|
|||
|
|
}
|
|||
|
|
}
|