diff --git a/internal/shared/pdf/font_manager.go b/internal/shared/pdf/font_manager.go index 05a80b9..b86af92 100644 --- a/internal/shared/pdf/font_manager.go +++ b/internal/shared/pdf/font_manager.go @@ -2,6 +2,7 @@ package pdf import ( "os" + "path/filepath" "runtime" "github.com/jung-kurt/gofpdf/v2" @@ -12,157 +13,140 @@ import ( type FontManager struct { logger *zap.Logger chineseFontName string - chineseFontPath string chineseFontLoaded bool watermarkFontName string - watermarkFontPath string watermarkFontLoaded bool + baseDir string // 缓存基础目录，避免重复计算 } // NewFontManager 创建字体管理器 func NewFontManager(logger *zap.Logger) *FontManager { + // 获取当前文件所在目录 + _, filename, _, _ := runtime.Caller(0) + baseDir := filepath.Dir(filename) + return &FontManager{ logger: logger, chineseFontName: "ChineseFont", watermarkFontName: "WatermarkFont", + baseDir: baseDir, } } -// LoadChineseFont 加载中文字体到PDF（只使用黑体） +// LoadChineseFont 加载中文字体到PDF func (fm *FontManager) LoadChineseFont(pdf *gofpdf.Fpdf) bool { if fm.chineseFontLoaded { return true } - fontPaths := fm.getHeiFontPaths() // 只获取黑体路径 + fontPaths := fm.getChineseFontPaths() if len(fontPaths) == 0 { - fm.logger.Warn("未找到黑体字体路径，PDF中的中文可能显示为空白") + fm.logger.Warn("未找到中文字体文件，PDF中的中文可能显示为空白") return false } - // 尝试添加黑体字体 + // 尝试加载字体 for _, fontPath := range fontPaths { - if fm.tryAddFont(pdf, fontPath) { - fm.chineseFontPath = fontPath + if fm.tryAddFont(pdf, fontPath, fm.chineseFontName) { fm.chineseFontLoaded = true - fm.logger.Info("成功添加黑体字体（常规和粗体）", zap.String("font_path", fontPath)) + fm.logger.Info("成功加载中文字体", zap.String("font_path", fontPath)) return true } } - fm.logger.Warn("未找到可用的黑体字体文件，PDF中的中文可能显示为空白") + fm.logger.Warn("所有中文字体文件都无法加载，PDF中的中文可能显示为空白") return false } -// LoadWatermarkFont 加载水印字体到PDF（使用宋体或其他非黑体字体） +// LoadWatermarkFont 加载水印字体到PDF func (fm *FontManager) LoadWatermarkFont(pdf *gofpdf.Fpdf) bool { if fm.watermarkFontLoaded { return true } - fontPaths := fm.getWatermarkFontPaths() // 获取水印字体路径（宋体等） + fontPaths := fm.getWatermarkFontPaths() if len(fontPaths) == 0 { - // 如果找不到水印字体，使用主字体（黑体） - fm.logger.Warn("未找到水印字体，将使用主字体（黑体）") + fm.logger.Warn("未找到水印字体文件，将使用主字体") return false } - // 尝试添加水印字体 + // 尝试加载字体 for _, fontPath := range fontPaths { - if fm.tryAddWatermarkFont(pdf, fontPath) { - fm.watermarkFontPath = fontPath + if fm.tryAddFont(pdf, fontPath, fm.watermarkFontName) { fm.watermarkFontLoaded = true - fm.logger.Info("成功添加水印字体", zap.String("font_path", fontPath)) + fm.logger.Info("成功加载水印字体", zap.String("font_path", fontPath)) return true } } - fm.logger.Warn("未找到可用的水印字体文件，将使用主字体（黑体）") + fm.logger.Warn("所有水印字体文件都无法加载，将使用主字体") return false } -// tryAddFont 尝试添加字体（捕获panic） -func (fm *FontManager) tryAddFont(pdf *gofpdf.Fpdf, fontPath string) bool { +// tryAddFont 尝试添加字体（统一处理中文字体和水印字体） +func (fm *FontManager) tryAddFont(pdf *gofpdf.Fpdf, fontPath, fontName string) bool { defer func() { if r := recover(); r != nil { - fm.logger.Error("添加字体时发生panic", zap.Any("panic", r), zap.String("font_path", fontPath)) + fm.logger.Error("添加字体时发生panic", + zap.Any("panic", r), + zap.String("font_path", fontPath), + zap.String("font_name", fontName)) } }() + // 检查文件是否存在 + if _, err := os.Stat(fontPath); err != nil { + return false + } + // gofpdf v2使用AddUTF8Font添加支持UTF-8的字体 - pdf.AddUTF8Font(fm.chineseFontName, "", fontPath) // 常规样式 - pdf.AddUTF8Font(fm.chineseFontName, "B", fontPath) // 粗体样式 + pdf.AddUTF8Font(fontName, "", fontPath) // 常规样式 + pdf.AddUTF8Font(fontName, "B", fontPath) // 粗体样式 + + // 验证字体是否可用 + pdf.SetFont(fontName, "", 12) + testWidth := pdf.GetStringWidth("测试") + if testWidth == 0 { + fm.logger.Debug("字体加载后无法使用", + zap.String("font_path", fontPath), + zap.String("font_name", fontName), + zap.Float64("test_width", testWidth)) + return false + } + return true } -// tryAddWatermarkFont 尝试添加水印字体（捕获panic） -func (fm *FontManager) tryAddWatermarkFont(pdf *gofpdf.Fpdf, fontPath string) bool { - defer func() { - if r := recover(); r != nil { - fm.logger.Error("添加水印字体时发生panic", zap.Any("panic", r), zap.String("font_path", fontPath)) - } - }() - - // gofpdf v2使用AddUTF8Font添加支持UTF-8的字体 - pdf.AddUTF8Font(fm.watermarkFontName, "", fontPath) // 常规样式 - pdf.AddUTF8Font(fm.watermarkFontName, "B", fontPath) // 粗体样式 - return true -} - -// getHeiFontPaths 获取黑体字体路径（支持跨平台，只返回黑体） -func (fm *FontManager) getHeiFontPaths() []string { - var fontPaths []string - - // Windows系统 - if runtime.GOOS == "windows" { - fontPaths = []string{ - `C:\Windows\Fonts\simhei.ttf`, // 黑体（优先） - } - } else if runtime.GOOS == "linux" { - // Linux系统黑体字体路径 - fontPaths = []string{ - "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc", // 文泉驿正黑（黑体） - "/usr/share/fonts/truetype/wqy/wqy-microhei.ttc", // 文泉驿微米黑（黑体） - } - } else if runtime.GOOS == "darwin" { - // macOS系统黑体字体路径 - fontPaths = []string{ - "/Library/Fonts/Microsoft/SimHei.ttf", // 黑体 - "/System/Library/Fonts/STHeiti Light.ttc", // 黑体 - } +// getChineseFontPaths 获取中文字体路径列表（仅TTF格式） +func (fm *FontManager) getChineseFontPaths() []string { + // 按优先级排序的字体文件列表 + fontNames := []string{ + "simhei.ttf", // 黑体（默认） + "simkai.ttf", // 楷体（备选） + "simfang.ttf", // 仿宋（备选） } - // 过滤出实际存在的字体文件 - var existingFonts []string - for _, fontPath := range fontPaths { - if _, err := os.Stat(fontPath); err == nil { - existingFonts = append(existingFonts, fontPath) - } - } - - return existingFonts + return fm.buildFontPaths(fontNames) } -// getWatermarkFontPaths 获取水印字体路径（支持跨平台，使用宋体或其他非黑体字体） +// getWatermarkFontPaths 获取水印字体路径列表（仅TTF格式） func (fm *FontManager) getWatermarkFontPaths() []string { + // 水印字体文件名（支持大小写变体） + fontNames := []string{ + "yunfengfeiyunti-2.ttf", + "YunFengFeiYunTi-2.ttf", // 大写版本（兼容） + } + + return fm.buildFontPaths(fontNames) +} + +// buildFontPaths 构建字体文件路径列表 +func (fm *FontManager) buildFontPaths(fontNames []string) []string { var fontPaths []string - // Windows系统 - if runtime.GOOS == "windows" { - fontPaths = []string{ - `C:\Windows\Fonts\simsun.ttf`, // 宋体（优先用于水印） - `C:\Windows\Fonts\simkai.ttf`, // 楷体（备选） - } - } else if runtime.GOOS == "linux" { - // Linux系统宋体字体路径 - fontPaths = []string{ - "/usr/share/fonts/truetype/arphic/uming.ttc", // 文鼎PL UMing（宋体） - } - } else if runtime.GOOS == "darwin" { - // macOS系统宋体字体路径 - fontPaths = []string{ - "/System/Library/Fonts/STSong.ttc", // 宋体 - } + // 使用 pdf/fonts/ 目录 + for _, fontName := range fontNames { + fontPaths = append(fontPaths, filepath.Join(fm.baseDir, "fonts", fontName)) } // 过滤出实际存在的字体文件 @@ -176,22 +160,22 @@ func (fm *FontManager) getWatermarkFontPaths() []string { return existingFonts } -// SetFont 设置字体（使用黑体） +// SetFont 设置中文字体 func (fm *FontManager) SetFont(pdf *gofpdf.Fpdf, style string, size float64) { if fm.chineseFontLoaded { pdf.SetFont(fm.chineseFontName, style, size) } else { - // 如果没有黑体字体，使用Arial作为后备 + // 如果没有中文字体，使用Arial作为后备 pdf.SetFont("Arial", style, size) } } -// SetWatermarkFont 设置水印字体（使用宋体或其他非黑体字体） +// SetWatermarkFont 设置水印字体 func (fm *FontManager) SetWatermarkFont(pdf *gofpdf.Fpdf, style string, size float64) { if fm.watermarkFontLoaded { pdf.SetFont(fm.watermarkFontName, style, size) } else { - // 如果水印字体不可用，使用主字体（黑体）作为后备 + // 如果水印字体不可用，使用主字体作为后备 fm.SetFont(pdf, style, size) } } @@ -200,18 +184,3 @@ func (fm *FontManager) SetWatermarkFont(pdf *gofpdf.Fpdf, style string, size flo func (fm *FontManager) IsChineseFontAvailable() bool { return fm.chineseFontLoaded } - -// GetChineseFontName 获取中文字体名称 -func (fm *FontManager) GetChineseFontName() string { - return fm.chineseFontName -} - -// GetWatermarkFontName 获取水印字体名称 -func (fm *FontManager) GetWatermarkFontName() string { - return fm.watermarkFontName -} - -// IsWatermarkFontAvailable 检查水印字体是否可用 -func (fm *FontManager) IsWatermarkFontAvailable() bool { - return fm.watermarkFontLoaded -} diff --git a/internal/shared/pdf/fonts/YunFengFeiYunTi-2.ttf b/internal/shared/pdf/fonts/YunFengFeiYunTi-2.ttf new file mode 100644 index 0000000..d7935d0 Binary files /dev/null and b/internal/shared/pdf/fonts/YunFengFeiYunTi-2.ttf differ diff --git a/internal/shared/pdf/fonts/msyh.ttc b/internal/shared/pdf/fonts/msyh.ttc new file mode 100644 index 0000000..ea174b2 Binary files /dev/null and b/internal/shared/pdf/fonts/msyh.ttc differ diff --git a/internal/shared/pdf/fonts/simfang.ttf b/internal/shared/pdf/fonts/simfang.ttf new file mode 100644 index 0000000..e344231 Binary files /dev/null and b/internal/shared/pdf/fonts/simfang.ttf differ diff --git a/internal/shared/pdf/fonts/simhei.ttf b/internal/shared/pdf/fonts/simhei.ttf new file mode 100644 index 0000000..3326815 Binary files /dev/null and b/internal/shared/pdf/fonts/simhei.ttf differ diff --git a/internal/shared/pdf/fonts/simkai.ttf b/internal/shared/pdf/fonts/simkai.ttf new file mode 100644 index 0000000..d785e05 Binary files /dev/null and b/internal/shared/pdf/fonts/simkai.ttf differ