package pdf import ( "os" "path/filepath" "github.com/jung-kurt/gofpdf/v2" "go.uber.org/zap" ) // FontManager 字体管理器 type FontManager struct { logger *zap.Logger chineseFontName string chineseFontLoaded bool watermarkFontName string watermarkFontLoaded bool } // NewFontManager 创建字体管理器 func NewFontManager(logger *zap.Logger) *FontManager { return &FontManager{ logger: logger, chineseFontName: "ChineseFont", watermarkFontName: "WatermarkFont", } } // LoadChineseFont 加载中文字体到PDF func (fm *FontManager) LoadChineseFont(pdf *gofpdf.Fpdf) bool { if fm.chineseFontLoaded { return true } fontPaths := fm.getChineseFontPaths() if len(fontPaths) == 0 { // 字体文件不存在,使用系统默认字体,不记录警告 return false } // 尝试加载字体 for _, fontPath := range fontPaths { if fm.tryAddFont(pdf, fontPath, fm.chineseFontName) { fm.chineseFontLoaded = true return true } } // 无法加载字体,使用系统默认字体,不记录警告 return false } // LoadWatermarkFont 加载水印字体到PDF func (fm *FontManager) LoadWatermarkFont(pdf *gofpdf.Fpdf) bool { if fm.watermarkFontLoaded { return true } fontPaths := fm.getWatermarkFontPaths() if len(fontPaths) == 0 { return false } // 尝试加载字体 for _, fontPath := range fontPaths { if fm.tryAddFont(pdf, fontPath, fm.watermarkFontName) { fm.watermarkFontLoaded = true return true } } return false } // tryAddFont 尝试添加字体(统一处理中文字体和水印字体) func (fm *FontManager) tryAddFont(pdf *gofpdf.Fpdf, fontPath, fontName string) bool { defer func() { if r := recover(); r != nil { fm.logger.Error("添加字体时发生panic", zap.String("font_path", fontPath), zap.String("font_name", fontName), zap.Any("panic_value", r), ) } }() // 确保路径是绝对路径 absFontPath, err := filepath.Abs(fontPath) if err != nil { fm.logger.Warn("无法获取字体文件绝对路径", zap.String("font_path", fontPath), zap.Error(err), ) absFontPath = fontPath } fm.logger.Debug("尝试添加字体", zap.String("font_path", absFontPath), zap.String("font_name", fontName), ) // 检查文件是否存在 fileInfo, err := os.Stat(absFontPath) if err != nil { fm.logger.Debug("字体文件不存在", zap.String("font_path", absFontPath), zap.Error(err), ) return false } if !fileInfo.Mode().IsRegular() { fm.logger.Warn("字体路径不是普通文件", zap.String("font_path", absFontPath), ) return false } // 确保路径是绝对路径(gofpdf在Output时需要绝对路径) // 首先确保absFontPath是绝对路径 if !filepath.IsAbs(absFontPath) { fm.logger.Warn("字体路径不是绝对路径,重新转换", zap.String("original_path", absFontPath), ) if newAbsPath, err := filepath.Abs(absFontPath); err == nil { absFontPath = newAbsPath } } // 使用filepath.ToSlash统一路径分隔符(Linux下使用/) // 注意:ToSlash不会改变路径的绝对/相对性质,只统一分隔符 normalizedPath := filepath.ToSlash(absFontPath) // 在Linux下,绝对路径必须以/开头 // 如果normalizedPath不是以/开头,说明转换有问题 if len(normalizedPath) == 0 || normalizedPath[0] != '/' { fm.logger.Error("字体路径转换后不是绝对路径(不以/开头)", zap.String("abs_font_path", absFontPath), zap.String("normalized_path", normalizedPath), ) // 重新转换为绝对路径 if newAbsPath, err := filepath.Abs(absFontPath); err == nil { absFontPath = newAbsPath normalizedPath = filepath.ToSlash(newAbsPath) fm.logger.Info("重新转换后的路径", zap.String("new_normalized_path", normalizedPath), ) } } fm.logger.Debug("准备添加字体到gofpdf", zap.String("original_path", fontPath), zap.String("abs_path", absFontPath), zap.String("normalized_path", normalizedPath), zap.String("font_name", fontName), ) // gofpdf v2使用AddUTF8Font添加支持UTF-8的字体 // 注意:gofpdf在Output时可能会重新解析路径,必须确保路径格式正确 // 关键:确保路径是绝对路径且以/开头,并使用filepath.ToSlash统一分隔符 // 如果normalizedPath不是以/开头,说明路径有问题,需要重新处理 if len(normalizedPath) == 0 || normalizedPath[0] != '/' { fm.logger.Error("字体路径格式错误,无法添加到PDF", zap.String("normalized_path", normalizedPath), zap.String("font_name", fontName), ) return false } // 记录传递给gofpdf的实际路径 fm.logger.Info("添加字体到gofpdf", zap.String("font_path", normalizedPath), zap.String("font_name", fontName), zap.Bool("is_absolute", len(normalizedPath) > 0 && normalizedPath[0] == '/'), ) pdf.AddUTF8Font(fontName, "", normalizedPath) // 常规样式 pdf.AddUTF8Font(fontName, "B", normalizedPath) // 粗体样式 // 验证字体是否可用 // 注意:gofpdf可能在AddUTF8Font时不会立即加载字体,而是在Output时才加载 // 所以这里验证可能失败,但不一定代表字体无法使用 pdf.SetFont(fontName, "", 12) testWidth := pdf.GetStringWidth("测试") if testWidth == 0 { fm.logger.Warn("字体添加后验证失败(测试文本宽度为0),但会在Output时重新尝试", zap.String("font_path", normalizedPath), zap.String("font_name", fontName), ) // 注意:即使验证失败,也返回true,因为gofpdf在Output时才会真正加载字体文件 // 这里的验证可能不准确 } fm.logger.Info("字体已添加到PDF(将在Output时加载)", zap.String("font_path", normalizedPath), zap.String("font_name", fontName), zap.Float64("test_width", testWidth), ) return true } // getChineseFontPaths 获取中文字体路径列表(仅TTF格式) func (fm *FontManager) getChineseFontPaths() []string { // 按优先级排序的字体文件列表 fontNames := []string{ "simhei.ttf", // 黑体(默认) "simkai.ttf", // 楷体(备选) "simfang.ttf", // 仿宋(备选) } return fm.buildFontPaths(fontNames) } // getWatermarkFontPaths 获取水印字体路径列表(仅TTF格式) func (fm *FontManager) getWatermarkFontPaths() []string { // 水印字体文件名(尝试大小写变体) fontNames := []string{ "YunFengFeiYunTi-2.ttf", // 优先尝试大写版本 "yunfengfeiyunti-2.ttf", // 小写版本(兼容) } return fm.buildFontPaths(fontNames) } // buildFontPaths 构建字体文件路径列表(仅从resources/pdf/fonts加载) // 返回所有存在的字体文件的绝对路径 func (fm *FontManager) buildFontPaths(fontNames []string) []string { // 获取resources/pdf目录(已返回绝对路径) resourcesPDFDir := GetResourcesPDFDir() if resourcesPDFDir == "" { fm.logger.Error("无法获取resources/pdf目录路径") return []string{} } // 构建字体目录路径(resourcesPDFDir已经是绝对路径) fontsDir := filepath.Join(resourcesPDFDir, "fonts") fm.logger.Debug("查找字体文件", zap.String("resources_pdf_dir", resourcesPDFDir), zap.String("fonts_dir", fontsDir), zap.Strings("font_names", fontNames), ) // 构建字体文件路径列表(都是绝对路径) var fontPaths []string for _, fontName := range fontNames { fontPath := filepath.Join(fontsDir, fontName) // 确保是绝对路径 if absPath, err := filepath.Abs(fontPath); err == nil { fontPaths = append(fontPaths, absPath) } else { fm.logger.Warn("无法获取字体文件绝对路径", zap.String("font_path", fontPath), zap.Error(err), ) } } // 过滤出实际存在的字体文件 var existingFonts []string for _, fontPath := range fontPaths { if info, err := os.Stat(fontPath); err == nil && info.Mode().IsRegular() { existingFonts = append(existingFonts, fontPath) fm.logger.Debug("找到字体文件", zap.String("font_path", fontPath)) } else { fm.logger.Debug("字体文件不存在", zap.String("font_path", fontPath), zap.Error(err), ) } } if len(existingFonts) == 0 { fm.logger.Warn("未找到任何字体文件", zap.String("fonts_dir", fontsDir), zap.Strings("attempted_fonts", fontPaths), ) } else { fm.logger.Info("找到字体文件", zap.Int("count", len(existingFonts)), zap.Strings("font_paths", existingFonts), ) } return existingFonts } // SetFont 设置中文字体 func (fm *FontManager) SetFont(pdf *gofpdf.Fpdf, style string, size float64) { if fm.chineseFontLoaded { pdf.SetFont(fm.chineseFontName, style, size) } else { // 如果没有中文字体,使用Arial作为后备 pdf.SetFont("Arial", style, size) } } // 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) } } // IsChineseFontAvailable 检查中文字体是否可用 func (fm *FontManager) IsChineseFontAvailable() bool { return fm.chineseFontLoaded }