145 lines
3.9 KiB
Go
145 lines
3.9 KiB
Go
package utils
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"net/url"
|
||
"os"
|
||
"path/filepath"
|
||
"unicode"
|
||
|
||
"github.com/phpdave11/gofpdf"
|
||
)
|
||
|
||
// cleanTextForPDF 清理文本，移除PDF不支持的字符（如emoji）
|
||
// gofpdf库不支持某些特殊字符
|
||
func cleanTextForPDF(text string) string {
|
||
var result []rune
|
||
for _, r := range text {
|
||
// 保留基本多文种平面（BMP）内的字符（码点 <= 0xFFFF）
|
||
// 这样可以保留中文、英文、数字等常用字符，但过滤掉emoji等特殊字符
|
||
if r <= 0xFFFF && (unicode.IsPrint(r) || unicode.IsSpace(r)) {
|
||
result = append(result, r)
|
||
}
|
||
}
|
||
return string(result)
|
||
}
|
||
|
||
// loadChineseFont 加载中文字体
|
||
func loadChineseFont(pdf *gofpdf.Fpdf, fontPath string) error {
|
||
var fontData []byte
|
||
var err error
|
||
|
||
// 如果提供了本地字体路径，优先使用本地字体
|
||
if fontPath == "" {
|
||
return errors.New("字体文件路径不能为空")
|
||
}
|
||
|
||
fontData, err = os.ReadFile(fontPath)
|
||
if err != nil {
|
||
return fmt.Errorf("读取字体文件失败: %v", err)
|
||
}
|
||
// 使用本地字体文件
|
||
pdf.AddUTF8FontFromBytes("Chinese", "", fontData)
|
||
return nil
|
||
}
|
||
|
||
// GeneratePDF 生成PDF文件
|
||
func GeneratePDF(text, imageURL, outputPath, fontPath string) error {
|
||
// 创建PDF实例，P=纵向，mm=毫米单位，A4=页面大小
|
||
pdf := gofpdf.New("P", "mm", "A4", "")
|
||
|
||
// 加载中文字体
|
||
err := loadChineseFont(pdf, fontPath)
|
||
if err != nil {
|
||
return fmt.Errorf("加载中文字体失败: %v", err)
|
||
}
|
||
|
||
// 添加新页面
|
||
pdf.AddPage()
|
||
|
||
// 设置字体，使用中文字体，12号字体
|
||
pdf.SetFont("Chinese", "", 12)
|
||
|
||
// 设置页面边距（左、上、右）
|
||
pdf.SetMargins(20, 10, 20)
|
||
|
||
// 设置当前位置（x, y），从左上角开始
|
||
pdf.SetXY(20, 10)
|
||
|
||
// 清理文本，移除PDF不支持的字符（如emoji）
|
||
cleanedText := cleanTextForPDF(text)
|
||
|
||
// 添加文本内容
|
||
// 使用MultiCell方法处理多行文本，支持自动换行
|
||
// 参数：宽度、行高、文本内容、边框、对齐方式、是否填充
|
||
// A4页面宽度210mm，减去左右边距40mm，可用宽度170mm
|
||
textWidth := 170.0
|
||
lineHeight := 7.0
|
||
pdf.MultiCell(textWidth, lineHeight, cleanedText, "", "L", false)
|
||
|
||
// 添加一些间距
|
||
pdf.Ln(5)
|
||
|
||
// 解析URL获取文件扩展名
|
||
u, err := url.Parse(imageURL)
|
||
if err != nil {
|
||
return fmt.Errorf("图片链接解析错误: %v", err)
|
||
}
|
||
fileExt := filepath.Ext(u.Path)
|
||
// 如果没有扩展名，默认使用.jpg
|
||
if fileExt == "" {
|
||
fileExt = ".jpg"
|
||
}
|
||
|
||
// 下载图片
|
||
resp, err := http.Get(imageURL)
|
||
if err != nil {
|
||
return fmt.Errorf("下载图片失败: %v", err)
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 读取图片数据
|
||
imageData, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
return fmt.Errorf("读取图片数据失败: %v", err)
|
||
}
|
||
|
||
// 将图片数据保存到临时文件（gofpdf需要文件路径）
|
||
tmpFile, err := os.CreateTemp("", "pdf_image_*"+fileExt)
|
||
if err != nil {
|
||
return fmt.Errorf("创建临时文件失败: %v", err)
|
||
}
|
||
defer os.Remove(tmpFile.Name()) // 使用完后删除临时文件
|
||
defer tmpFile.Close()
|
||
|
||
// 写入图片数据到临时文件
|
||
_, err = tmpFile.Write(imageData)
|
||
if err != nil {
|
||
return fmt.Errorf("写入临时文件失败: %v", err)
|
||
}
|
||
tmpFile.Close()
|
||
|
||
// A4纵向页面宽度210mm，减去左右边距40mm，可用宽度170mm
|
||
// 图片宽度设为可用宽度的70%
|
||
imageWidth := textWidth * 0.7
|
||
// 计算居中位置：页面宽度210mm，图片居中
|
||
imageX := (210.0 - imageWidth) / 2
|
||
currentY := pdf.GetY()
|
||
|
||
// 添加图片
|
||
// ImageOptions参数：图片路径、x坐标、y坐标、宽度、高度、是否流式布局、选项、链接
|
||
// 高度设为0表示按比例自动计算
|
||
pdf.ImageOptions(tmpFile.Name(), imageX, currentY, imageWidth, 0, false, gofpdf.ImageOptions{}, 0, "")
|
||
|
||
// 生成并保存PDF文件
|
||
err = pdf.OutputFileAndClose(outputPath)
|
||
if err != nil {
|
||
return fmt.Errorf("生成PDF失败: %v", err)
|
||
}
|
||
|
||
return nil
|
||
}
|