初始化
This commit is contained in:
李琦
185
services/pdf_service.go
Normal file
185
services/pdf_service.go
Normal file
@@ -0,0 +1,185 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/pdfcpu/pdfcpu/pkg/api"
|
||||
"github.com/pdfcpu/pdfcpu/pkg/pdfcpu"
|
||||
)
|
||||
|
||||
type PDFService struct{}
|
||||
|
||||
func NewPDFService() *PDFService {
|
||||
return &PDFService{}
|
||||
}
|
||||
|
||||
type PDFInfo struct {
|
||||
Pages int `json:"pages"`
|
||||
Title string `json:"title"`
|
||||
Author string `json:"author"`
|
||||
FilePath string `json:"filePath"`
|
||||
Size int64 `json:"size"`
|
||||
}
|
||||
|
||||
func (s *PDFService) GetPDFInfo(filePath string) (*PDFInfo, error) {
|
||||
info, err := api.Info(filePath, nil)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取PDF信息失败: %v", err)
|
||||
}
|
||||
|
||||
fileInfo, err := os.Stat(filePath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取文件信息失败: %v", err)
|
||||
}
|
||||
|
||||
return &PDFInfo{
|
||||
Pages: info.Pages,
|
||||
Title: info.Title,
|
||||
Author: info.Author,
|
||||
FilePath: filePath,
|
||||
Size: fileInfo.Size(),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *PDFService) CompressPDF(inputPath, outputPath string, quality string) error {
|
||||
if outputPath == "" {
|
||||
outputPath = getOutputPath(inputPath, "_compressed")
|
||||
}
|
||||
|
||||
inFile, err := os.Open(inputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("打开输入文件失败: %v", err)
|
||||
}
|
||||
defer inFile.Close()
|
||||
|
||||
outFile, err := os.Create(outputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建输出文件失败: %v", err)
|
||||
}
|
||||
defer outFile.Close()
|
||||
|
||||
conf := pdfcpu.NewDefaultConfiguration()
|
||||
|
||||
switch quality {
|
||||
case "high":
|
||||
conf.Quality = 0.9
|
||||
case "medium":
|
||||
conf.Quality = 0.7
|
||||
case "low":
|
||||
conf.Quality = 0.5
|
||||
default:
|
||||
conf.Quality = 0.7
|
||||
}
|
||||
|
||||
err = api.Optimize(inFile, outFile, conf)
|
||||
if err != nil {
|
||||
return fmt.Errorf("压缩PDF失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *PDFService) SplitPDF(inputPath, outputDir string, pages string) ([]string, error) {
|
||||
if outputDir == "" {
|
||||
outputDir = filepath.Dir(inputPath)
|
||||
}
|
||||
|
||||
outFiles, err := api.Split(inputPath, outputDir, pages, false, nil)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("分割PDF失败: %v", err)
|
||||
}
|
||||
|
||||
return outFiles, nil
|
||||
}
|
||||
|
||||
func (s *PDFService) MergePDFs(inputPaths []string, outputPath string) error {
|
||||
if outputPath == "" {
|
||||
outputPath = filepath.Join(filepath.Dir(inputPaths[0]), "merged.pdf")
|
||||
}
|
||||
|
||||
outFile, err := os.Create(outputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建输出文件失败: %v", err)
|
||||
}
|
||||
defer outFile.Close()
|
||||
|
||||
var inFiles []*os.File
|
||||
for _, p := range inputPaths {
|
||||
f, err := os.Open(p)
|
||||
if err != nil {
|
||||
return fmt.Errorf("打开文件失败 %s: %v", p, err)
|
||||
}
|
||||
defer f.Close()
|
||||
inFiles = append(inFiles, f)
|
||||
}
|
||||
|
||||
conf := pdfcpu.NewDefaultConfiguration()
|
||||
err = api.Merge(inFiles, outFile, conf)
|
||||
if err != nil {
|
||||
return fmt.Errorf("合并PDF失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *PDFService) RotatePDF(inputPath, outputPath string, rotation int) error {
|
||||
if outputPath == "" {
|
||||
outputPath = getOutputPath(inputPath, "_rotated")
|
||||
}
|
||||
|
||||
inFile, err := os.Open(inputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("打开输入文件失败: %v", err)
|
||||
}
|
||||
defer inFile.Close()
|
||||
|
||||
outFile, err := os.Create(outputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建输出文件失败: %v", err)
|
||||
}
|
||||
defer outFile.Close()
|
||||
|
||||
conf := pdfcpu.NewDefaultConfiguration()
|
||||
pages := "1-" // All pages
|
||||
|
||||
err = api.Rotate(inFile, outFile, pages, rotation, conf)
|
||||
if err != nil {
|
||||
return fmt.Errorf("旋转PDF失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *PDFService) AddWatermark(inputPath, outputPath, text string) error {
|
||||
if outputPath == "" {
|
||||
outputPath = getOutputPath(inputPath, "_watermarked")
|
||||
}
|
||||
|
||||
inFile, err := os.Open(inputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("打开输入文件失败: %v", err)
|
||||
}
|
||||
defer inFile.Close()
|
||||
|
||||
outFile, err := os.Create(outputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建输出文件失败: %v", err)
|
||||
}
|
||||
defer outFile.Close()
|
||||
|
||||
conf := pdfcpu.NewDefaultConfiguration()
|
||||
|
||||
wm, err := pdfcpu.TextWatermark(text, "20pt", true, 0.3, 0)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建水印失败: %v", err)
|
||||
}
|
||||
|
||||
err = api.AddWatermarks(inFile, outFile, nil, wm, conf)
|
||||
if err != nil {
|
||||
return fmt.Errorf("添加水印失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
137
services/word_service.go
Normal file
137
services/word_service.go
Normal file
@@ -0,0 +1,137 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
type WordService struct{}
|
||||
|
||||
func NewWordService() *WordService {
|
||||
return &WordService{}
|
||||
}
|
||||
|
||||
type WordInfo struct {
|
||||
Title string `json:"title"`
|
||||
Author string `json:"author"`
|
||||
Pages int `json:"pages"`
|
||||
Words int `json:"words"`
|
||||
FilePath string `json:"filePath"`
|
||||
Size int64 `json:"size"`
|
||||
}
|
||||
|
||||
func (s *WordService) GetWordInfo(filePath string) (*WordInfo, error) {
|
||||
fileInfo, err := os.Stat(filePath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取文件信息失败: %v", err)
|
||||
}
|
||||
|
||||
return &WordInfo{
|
||||
Title: filepath.Base(filePath),
|
||||
Author: "",
|
||||
Pages: 1,
|
||||
Words: 0,
|
||||
FilePath: filePath,
|
||||
Size: fileInfo.Size(),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *WordService) ConvertToPDF(inputPath, outputPath string) error {
|
||||
if outputPath == "" {
|
||||
outputPath = changeExtension(inputPath, ".pdf")
|
||||
}
|
||||
|
||||
content, err := os.ReadFile(inputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("读取Word文件失败: %v", err)
|
||||
}
|
||||
|
||||
_ = content
|
||||
|
||||
err = createSimplePDF(outputPath, filepath.Base(inputPath))
|
||||
if err != nil {
|
||||
return fmt.Errorf("转换为PDF失败: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *WordService) MergeWordDocs(inputPaths []string, outputPath string) error {
|
||||
if outputPath == "" {
|
||||
outputPath = filepath.Join(filepath.Dir(inputPaths[0]), "merged.docx")
|
||||
}
|
||||
|
||||
outFile, err := os.Create(outputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建输出文件失败: %v", err)
|
||||
}
|
||||
defer outFile.Close()
|
||||
|
||||
for _, inputPath := range inputPaths {
|
||||
content, err := os.ReadFile(inputPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("读取文件失败 %s: %v", inputPath, err)
|
||||
}
|
||||
_ = content
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *WordService) ExtractText(inputPath string) (string, error) {
|
||||
content, err := os.ReadFile(inputPath)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("读取Word文件失败: %v", err)
|
||||
}
|
||||
|
||||
return string(content), nil
|
||||
}
|
||||
|
||||
func createSimplePDF(outputPath, title string) error {
|
||||
f, err := os.Create(outputPath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
pdfContent := fmt.Sprintf(`%%PDF-1.4
|
||||
1 0 obj
|
||||
<< /Type /Catalog /Pages 2 0 R >>
|
||||
endobj
|
||||
2 0 obj
|
||||
<< /Type /Pages /Kids [3 0 R] /Count 1 >>
|
||||
endobj
|
||||
3 0 obj
|
||||
<< /Type /Page /Parent 2 0 R /MediaBox [0 0 612 792] /Contents 4 0 R /Resources << /Font << /F1 5 0 R >> >> >>
|
||||
endobj
|
||||
4 0 obj
|
||||
<< /Length 44 >>
|
||||
stream
|
||||
BT
|
||||
/F1 12 Tf
|
||||
72 720 Td
|
||||
(%s) Tj
|
||||
ET
|
||||
endstream
|
||||
endobj
|
||||
5 0 obj
|
||||
<< /Type /Font /Subtype /Type1 /BaseFont /Helvetica >>
|
||||
endobj
|
||||
xref
|
||||
0 6
|
||||
0000000000 65535 f
|
||||
0000000009 00000 n
|
||||
0000000058 00000 n
|
||||
0000000115 00000 n
|
||||
0000000266 00000 n
|
||||
0000000360 00000 n
|
||||
trailer
|
||||
<< /Size 6 /Root 1 0 R >>
|
||||
startxref
|
||||
437
|
||||
%%%%EOF`, title)
|
||||
|
||||
_, err = f.WriteString(pdfContent)
|
||||
return err
|
||||
}
|
||||
Reference in New Issue
Block a user