实现导入

converter/converter.go

@@ -542,3 +542,25 @@ func (this *Converter) convertToDocx() (err error) {
 	}
 	return cmd.Run()
 }
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+

converter/import.go

@@ -0,0 +1,118 @@
+package converter
+
+import (
+	"errors"
+	"github.com/lifei6671/mindoc/utils"
+	"os"
+	"path/filepath"
+	"crypto/md5"
+	"io"
+	"fmt"
+	"os/exec"
+	"github.com/lifei6671/mindoc/utils/ziptil"
+	"io/ioutil"
+	"encoding/xml"
+)
+
+type ResolveResult struct {
+
+}
+
+type XmlResult struct {
+	XMLName     xml.Name `xml:"ncx"`
+	Head XmlHead `xml:"head"`
+	NavMap XmlTocNavMap `xml:"navMap"`
+	Title string `xml:"docTitle>text"`
+}
+
+type XmlHead struct {
+	XMLName xml.Name `xml:"head"`
+	Meta []XmlMeta `xml:"meta"`
+}
+
+type XmlMeta struct {
+	XMLName xml.Name `xml:"meta"`
+	Content string `xml:"content,attr"`
+	Name string `xml:"name,attr"`
+
+}
+type XmlDocTitle struct {
+	Text string `xml:"text"`
+}
+type XmlTocNavMap struct {
+	XMLName xml.Name `xml:"navMap"`
+	NavPoint []XmlNavPoint `xml:"navPoint"`
+}
+
+type XmlNavPoint struct {
+	XMLName xml.Name `xml:"navPoint"`
+	Content XmlContent `xml:"content"`
+	NavLabel string `xml:"navLabel>text"`
+}
+
+type XmlContent struct {
+	XMLName xml.Name `xml:"content"`
+	Src string `xml:"src,attr"`
+}
+
+type XmlNavLabel struct {
+
+}
+
+func Resolve(p string) (ResolveResult,error) {
+	result := ResolveResult{
+
+	}
+
+	if !utils.FileExists(p) {
+		return  result,errors.New("文件不存在 " + p)
+	}
+
+	w := md5.New()
+	io.WriteString(w, p)   //将str写入到w中
+	md5str := fmt.Sprintf("%x", w.Sum(nil))  //w.Sum(nil)将w的hash转成[]byte格式
+
+	tempPath := filepath.Join(os.TempDir(),md5str)
+
+	os.MkdirAll(tempPath,0766)
+
+	epub := filepath.Join(tempPath , "book.epub")
+
+	args := []string{p,epub}
+
+	cmd := exec.Command(ebookConvert, args...)
+
+
+	if err := cmd.Run(); err != nil {
+		fmt.Println("执行转换命令失败：" + err.Error())
+		return result,err
+	}
+	fmt.Println(epub)
+
+	unzipPath := filepath.Join(tempPath,"output")
+
+	if err := ziptil.Unzip(epub, unzipPath); err != nil {
+		fmt.Println("解压缩失败：" + err.Error())
+		return result,err
+	}
+	xmlPath := filepath.Join(unzipPath,"toc.ncx")
+
+	data,err := ioutil.ReadFile(xmlPath);
+
+	if err != nil {
+		fmt.Println("toc.ncx 文件不存在：" + err.Error())
+		return  result,err
+	}
+	v := XmlResult{}
+
+	err = xml.Unmarshal([]byte(data), &v)
+
+	if err != nil {
+		fmt.Println("解析XML失败：" + err.Error())
+		return  result,err
+	}
+
+	fmt.Println(v)
+
+	return  result,nil
+}