mirror of https://github.com/mindoc-org/mindoc.git
80 lines
1.4 KiB
Go
80 lines
1.4 KiB
Go
package utils
|
||
|
||
import (
|
||
"os/exec"
|
||
"bufio"
|
||
"io"
|
||
"io/ioutil"
|
||
"errors"
|
||
|
||
"github.com/axgle/mahonia"
|
||
"github.com/astaxie/beego"
|
||
)
|
||
|
||
// 使用 wkhtmltopdf 是实现 html 转 pdf.
|
||
// 中文说明：http://www.jianshu.com/p/4d65857ffe5e#
|
||
func ConverterHtmlToPdf(uri []string,path string) (error) {
|
||
|
||
exe := beego.AppConfig.String("wkhtmltopdf")
|
||
|
||
if exe == "" {
|
||
return errors.New("wkhtmltopdf not exist.")
|
||
}
|
||
|
||
params := []string{"/C",exe,"--margin-bottom","25"}
|
||
|
||
params = append(params,uri...)
|
||
params = append(params,path)
|
||
|
||
cmd := exec.Command("cmd",params...)
|
||
|
||
stdout, err := cmd.StdoutPipe()
|
||
|
||
if err != nil {
|
||
return errors.New("StdoutPipe: " + err.Error())
|
||
}
|
||
stderr, err := cmd.StderrPipe()
|
||
if err != nil {
|
||
|
||
return errors.New("StderrPipe: " + err.Error())
|
||
}
|
||
|
||
if err := cmd.Start(); err != nil {
|
||
|
||
return errors.New("Start: "+ err.Error())
|
||
}
|
||
|
||
reader := bufio.NewReader(stdout)
|
||
enc := mahonia.NewDecoder("gbk")
|
||
|
||
//实时循环读取输出流中的一行内容
|
||
for {
|
||
line ,err2 := reader.ReadString('\n')
|
||
|
||
if err2 != nil || io.EOF == err2 {
|
||
break
|
||
}
|
||
|
||
beego.Info(enc.ConvertString(line))
|
||
}
|
||
|
||
bytesErr, err := ioutil.ReadAll(stderr)
|
||
|
||
if err == nil {
|
||
beego.Info(enc.ConvertString(string(bytesErr)))
|
||
}else{
|
||
beego.Error("Error: Stderr => " + err.Error())
|
||
return err
|
||
}
|
||
|
||
if err := cmd.Wait(); err != nil {
|
||
|
||
beego.Error("Error: ", err.Error())
|
||
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
|
||
}
|