diff --git a/format/format.go b/format/format.go
index 91a0b3f..f1a307c 100644
--- a/format/format.go
+++ b/format/format.go
@@ -1,15 +1,22 @@
package format
import (
+ "bytes"
+ "encoding/binary"
"io/ioutil"
+ "log"
+ "os"
+ "path/filepath"
+ "runtime"
"strconv"
"strings"
"github.com/fengxxc/wechatmp2markdown/parse"
+ "github.com/fengxxc/wechatmp2markdown/util"
)
// Format format article
-func Format(article parse.Article) string {
+func Format(article parse.Article) (string, map[string][]byte) {
var result string
var titleMdStr string = formatTitle(article.Title)
result += titleMdStr
@@ -17,15 +24,72 @@ func Format(article parse.Article) string {
result += metaMdStr
var tagsMdStr string = formatTags(article.Tags)
result += tagsMdStr
- var content string = formatContent(article.Content, 0)
+ var saveImageBytes map[string][]byte
+ content, saveImageBytes := formatContent(article.Content, 0)
result += content
- return result
+ return result, saveImageBytes
}
// FormatAndSave fomat article and save to local file
func FormatAndSave(article parse.Article, filePath string) error {
- var result string = Format(article)
- return ioutil.WriteFile(filePath, []byte(result), 0644)
+ // basrPath := filepath.Join(filePath, )
+ var basePath string
+ var fileName string
+ var isWin bool = runtime.GOOS == "windows"
+ var separator string
+ if isWin {
+ separator = "\\"
+ } else {
+ separator = "/"
+ }
+ if filePath == "" {
+ filePath = "." + separator
+ }
+ if strings.HasPrefix(filePath, "./") || strings.HasPrefix(filePath, ".\\") {
+ wd, _ := os.Getwd()
+ filePath = strings.Replace(filePath, ".", wd, 1)
+ }
+ if strings.HasSuffix(filePath, ".md") {
+ // basePath = filePath[:len(filePath)-len(".md")]
+ basePath = filePath[:strings.LastIndex(filePath, separator)]
+ fileName = filePath
+ } else {
+ title := strings.TrimSpace(article.Title.Val.(string))
+ // title := "thisistitle"
+ basePath = filepath.Join(filePath, title)
+ fileName = filepath.Join(basePath, title+".md")
+ }
+
+ // make basePath dir if not exists
+ if _, err := os.Stat(basePath); err != nil {
+ if err := os.MkdirAll(basePath, 0644); err != nil {
+ panic(err)
+ }
+ }
+
+ var saveImageBytes map[string][]byte
+ result, saveImageBytes := Format(article)
+ if len(saveImageBytes) > 0 {
+ for imgTitle := range saveImageBytes {
+ // save to local
+ imgfileName := filepath.Join(basePath, imgTitle)
+ /* if err := ioutil.WriteFile(imgfileName, saveImageBytes[imgTitle], 0644); err != nil {
+ log.Fatalf("can not save image file: %s\n err: %v", imgfileName, err)
+ continue
+ } */
+ f, err := os.Create(imgfileName)
+ if err != nil {
+ // log.Fatalf("can not save image file: %s", imgTitle)
+ log.Fatalf("can not save image file: %s\n err: %v", imgfileName, err)
+ continue
+ }
+ defer f.Close()
+ buf := new(bytes.Buffer)
+ binary.Write(buf, binary.LittleEndian, saveImageBytes[imgTitle])
+ f.Write(buf.Bytes())
+ }
+ }
+ return ioutil.WriteFile(fileName, []byte(result), 0644)
}
func formatTitle(piece parse.Piece) string {
@@ -45,11 +109,13 @@ func formatTags(tags string) string {
return tags + " \n" // TODO
}
-func formatContent(pieces []parse.Piece, depth int) string {
+func formatContent(pieces []parse.Piece, depth int) (string, map[string][]byte) {
var contentMdStr string
var base64Imgs []string
+ var saveImageBytes map[string][]byte = make(map[string][]byte)
for _, piece := range pieces {
var pieceMdStr string
+ var patchSaveImageBytes map[string][]byte
switch piece.Type {
case parse.HEADER:
pieceMdStr = formatTitle(piece)
@@ -64,6 +130,17 @@ func formatContent(pieces []parse.Piece, depth int) string {
case parse.BOLD_ITALIC_TEXT:
pieceMdStr = "***" + piece.Val.(string) + "***"
case parse.IMAGE:
+ if piece.Val == nil {
+ pieceMdStr = formatImageInline(piece)
+ } else {
+ // will save to local
+ src := piece.Attrs["src"]
+ imgExt := util.ParseImageExtFromSrc(src)
+ var hashName string = util.MD5(piece.Val.([]byte)) + "." + imgExt
+ saveImageBytes[hashName] = piece.Val.([]byte)
+ pieceMdStr = formatImageFileReferInline(piece.Attrs["alt"], hashName)
+ }
+ case parse.IMAGE_BASE64:
pieceMdStr = formatImageRefer(piece, len(base64Imgs))
base64Imgs = append(base64Imgs, piece.Val.(string))
case parse.TABLE:
@@ -73,36 +150,38 @@ func formatContent(pieces []parse.Piece, depth int) string {
case parse.CODE_BLOCK:
pieceMdStr = formatCodeBlock(piece)
case parse.BLOCK_QUOTES:
- pieceMdStr = formatBlockQuote(piece, depth)
+ pieceMdStr, patchSaveImageBytes = formatBlockQuote(piece, depth)
case parse.O_LIST:
- pieceMdStr = formatList(piece, depth)
+ pieceMdStr, patchSaveImageBytes = formatList(piece, depth)
case parse.U_LIST:
- pieceMdStr = formatList(piece, depth)
+ pieceMdStr, patchSaveImageBytes = formatList(piece, depth)
case parse.HR:
// TODO
case parse.BR:
pieceMdStr = " \n"
}
contentMdStr += pieceMdStr
+ util.MergeMap(saveImageBytes, patchSaveImageBytes)
}
for i := 0; i < len(base64Imgs); i++ {
contentMdStr += "\n[" + strconv.Itoa(i) + "]:" + "data:image/png;base64," + base64Imgs[i]
}
- return contentMdStr
+ return contentMdStr, saveImageBytes
}
-func formatBlockQuote(piece parse.Piece, depth int) string {
+func formatBlockQuote(piece parse.Piece, depth int) (string, map[string][]byte) {
var bqMdString string
var prefix string = ">"
for i := 0; i < depth; i++ {
prefix += ">"
}
prefix += " "
- bqMdString = prefix + formatContent(piece.Val.([]parse.Piece), depth+1) + " \n"
- return bqMdString
+ var saveImageBytes map[string][]byte
+ bqMdString, saveImageBytes = formatContent(piece.Val.([]parse.Piece), depth+1)
+ return prefix + bqMdString + " \n", saveImageBytes
}
-func formatList(li parse.Piece, depth int) string {
+func formatList(li parse.Piece, depth int) (string, map[string][]byte) {
var listMdString string
var prefix string
for j := 0; j < depth; j++ {
@@ -113,8 +192,9 @@ func formatList(li parse.Piece, depth int) string {
} else if li.Type == parse.O_LIST {
prefix += strconv.Itoa(1) + ". " // 写死成1也大丈夫，markdown会自动累加序号
}
- listMdString = prefix + formatContent(li.Val.([]parse.Piece), depth+1) + " \n"
- return listMdString
+ var saveImageBytes map[string][]byte
+ listMdString, saveImageBytes = formatContent(li.Val.([]parse.Piece), depth+1)
+ return prefix + listMdString + " \n", saveImageBytes
}
func formatCodeBlock(piece parse.Piece) string {
@@ -128,11 +208,22 @@ func formatCodeBlock(piece parse.Piece) string {
return codeMdStr
}
-func formatImageInline(piece parse.Piece, index int) string {
- // return "![" + piece.Attrs["alt"] + "](" + piece.Attrs["src"] + " \"" + piece.Attrs["title"] + "\")"
+// 图片地址为本身src
+func formatImageInline(piece parse.Piece) string {
+ return "![" + piece.Attrs["alt"] + "](" + piece.Attrs["src"] + " \"" + piece.Attrs["title"] + "\")"
+}
+
+// 图片地址为本地引用
+func formatImageFileReferInline(alt string, refName string) string {
+ return "![" + alt + "](" + refName + ")"
+}
+
+// 图片转成base64并插在原地
+func formatImageBase64Inline(piece parse.Piece) string {
return "![" + piece.Attrs["alt"] + "](data:image/png;base64," + piece.Val.(string) + ")"
}
+// 图片地址为markdown内引用（用于base64）
func formatImageRefer(piece parse.Piece, index int) string {
return "![" + piece.Attrs["alt"] + "][" + strconv.Itoa(index) + "]"
}
diff --git a/main.go b/main.go
index b221de1..325d951 100644
--- a/main.go
+++ b/main.go
@@ -3,6 +3,7 @@ package main
import (
"fmt"
"os"
+ "strings"
"github.com/fengxxc/wechatmp2markdown/format"
"github.com/fengxxc/wechatmp2markdown/parse"
@@ -29,10 +30,35 @@ func main() {
return
}
+ // --image=base64 -ib 保存图片，base64格式，在md文件中（默认为此选项）
+ // --image=url -iu 只保留图片链接
+ // --image=save -is 保存图片，最终输出到文件夹
+ // --save=zip -sz 最终打包输出到zip
+ imageArgValue := "base64"
+ if args[3] != "" {
+ if strings.HasPrefix(args[3], "--image=") {
+ imageArgValue = args[3][len("--image="):]
+ } else if strings.HasPrefix(args[3], "-i") {
+ imageArgVal := args[3][len("-i"):]
+ switch imageArgVal {
+ case "u":
+ imageArgValue = "url"
+ case "s":
+ imageArgValue = "save"
+ case "b":
+ fallthrough
+ default:
+ imageArgValue = "base64"
+ }
+ }
+ }
+
+ var imagePolicy parse.ImagePolicy = parse.ImageArgValue2ImagePolicy(imageArgValue)
+
// cli pattern
url := args1
filename := args2
fmt.Printf("url: %s, filename: %s\n", url, filename)
- var articleStruct parse.Article = parse.ParseFromURL(url)
+ var articleStruct parse.Article = parse.ParseFromURL(url, imagePolicy)
format.FormatAndSave(articleStruct, filename)
}
diff --git a/parse/model.go b/parse/model.go
index d561ca7..5c07dc4 100644
--- a/parse/model.go
+++ b/parse/model.go
@@ -31,12 +31,13 @@ const (
ITALIC_TEXT // 4 斜体文字
BOLD_ITALIC_TEXT // 5 粗斜体
IMAGE // 6 图片
- TABLE // 7 表格
- CODE_INLINE // 8 代码 内联
- CODE_BLOCK // 9 代码 块
- BLOCK_QUOTES // 10 引用
- O_LIST // 11 有序列表
- U_LIST // 12 无序列表
- HR // 13 分隔线
- BR // 14 换行
+ IMAGE_BASE64 // 7 图片 base64
+ TABLE // 8 表格
+ CODE_INLINE // 9 代码 内联
+ CODE_BLOCK // 10 代码 块
+ BLOCK_QUOTES // 11 引用
+ O_LIST // 12 有序列表
+ U_LIST // 13 无序列表
+ HR // 14 分隔线
+ BR // 15 换行
)
diff --git a/parse/parse.go b/parse/parse.go
index c69db8e..95ded59 100644
--- a/parse/parse.go
+++ b/parse/parse.go
@@ -16,7 +16,7 @@ import (
"github.com/PuerkitoBio/goquery"
)
-func parseSection(s *goquery.Selection) []Piece {
+func parseSection(s *goquery.Selection, imagePolicy ImagePolicy) []Piece {
var pieces []Piece
pieces = append(pieces, Piece{BR, nil, nil})
s.Contents().Each(func(i int, sc *goquery.Selection) {
@@ -28,21 +28,31 @@ func parseSection(s *goquery.Selection) []Piece {
attr["src"], _ = sc.Attr("data-src")
attr["alt"], _ = sc.Attr("alt")
attr["title"], _ = sc.Attr("title")
- base64Image := img2base64(fetchImgFile(attr["src"]))
- pieces = append(pieces, Piece{IMAGE, base64Image, attr}, Piece{BR, nil, nil})
+ switch imagePolicy {
+ case IMAGE_POLICY_URL:
+ pieces = append(pieces, Piece{IMAGE, nil, attr}, Piece{BR, nil, nil})
+ case IMAGE_POLICY_SAVE:
+ image := fetchImgFile(attr["src"])
+ pieces = append(pieces, Piece{IMAGE, image, attr}, Piece{BR, nil, nil})
+ case IMAGE_POLICY_BASE64:
+ fallthrough
+ default:
+ base64Image := img2base64(fetchImgFile(attr["src"]))
+ pieces = append(pieces, Piece{IMAGE_BASE64, base64Image, attr}, Piece{BR, nil, nil})
+ }
} else if sc.Is("ol") {
- pieces = append(pieces, parseList(sc, O_LIST)...)
+ pieces = append(pieces, parseList(sc, O_LIST, imagePolicy)...)
} else if sc.Is("ul") {
- pieces = append(pieces, parseList(sc, U_LIST)...)
+ pieces = append(pieces, parseList(sc, U_LIST, imagePolicy)...)
} else if sc.Is("pre") || sc.Is("section.code-snippet__fix") {
// 代码块
pieces = append(pieces, parsePre(sc)...)
} else if sc.Is("p") || sc.Is("section") || sc.Is("span") {
- pieces = append(pieces, parseSection(sc)...)
+ pieces = append(pieces, parseSection(sc, imagePolicy)...)
} else if sc.Is("h1") || sc.Is("h2") || sc.Is("h3") || sc.Is("h4") || sc.Is("h5") || sc.Is("h6") {
pieces = append(pieces, parseHeader(sc)...)
} else if sc.Is("blockquote") {
- pieces = append(pieces, parseBlockQuote(sc)...)
+ pieces = append(pieces, parseBlockQuote(sc, imagePolicy)...)
} else if sc.Is("strong") {
pieces = append(pieces, parseStrong(sc)...)
} else {
@@ -83,19 +93,19 @@ func parsePre(s *goquery.Selection) []Piece {
return []Piece{p, {BR, nil, nil}}
}
-func parseList(s *goquery.Selection, ptype PieceType) []Piece {
+func parseList(s *goquery.Selection, ptype PieceType, imagePolicy ImagePolicy) []Piece {
var list []Piece
s.Find("li").Each(func(i int, sc *goquery.Selection) {
- list = append(list, Piece{ptype, parseSection(sc), nil})
+ list = append(list, Piece{ptype, parseSection(sc, imagePolicy), nil})
})
list = append(list, Piece{BR, nil, nil})
return list
}
-func parseBlockQuote(s *goquery.Selection) []Piece {
+func parseBlockQuote(s *goquery.Selection, imagePolicy ImagePolicy) []Piece {
var bq []Piece
s.Contents().Each(func(i int, sc *goquery.Selection) {
- bq = append(bq, Piece{BLOCK_QUOTES, parseSection(sc), nil})
+ bq = append(bq, Piece{BLOCK_QUOTES, parseSection(sc, imagePolicy), nil})
})
bq = append(bq, Piece{BR, nil, nil})
return bq
@@ -124,7 +134,7 @@ func parseMeta(s *goquery.Selection) []string {
return res
}
-func ParseFromReader(r io.Reader) Article {
+func ParseFromReader(r io.Reader, imagePolicy ImagePolicy) Article {
var article Article
doc, err := goquery.NewDocumentFromReader(r)
if err != nil {
@@ -144,11 +154,10 @@ func ParseFromReader(r io.Reader) Article {
// 从js中找到发布时间
re, _ := regexp.Compile("var ct = \"([0-9]+)\"")
findstrs := re.FindStringSubmatch(doc.Find("script").Text())
- if findstrs != nil && len(findstrs) > 1 {
+ if len(findstrs) > 1 {
var createTime string = findstrs[1]
timestamp, _ := strconv.Atoi(createTime)
time := time.Unix(int64(timestamp), 0)
- // fmt.Println(time)
article.Meta = append(article.Meta, time.Format("2006-01-02 15:04"))
}
@@ -162,17 +171,17 @@ func ParseFromReader(r io.Reader) Article {
// p[style="line-height: 1.5em;"] => 项目列表（有序/无序）
// section[style=".*text-align:center"]>img => 居中段落（图片）
content := mainContent.Find("#js_content")
- pieces := parseSection(content)
+ pieces := parseSection(content, imagePolicy)
article.Content = pieces
return article
}
-func ParseFromHTMLString(s string) Article {
- return ParseFromReader(strings.NewReader(s))
+func ParseFromHTMLString(s string, imagePolicy ImagePolicy) Article {
+ return ParseFromReader(strings.NewReader(s), imagePolicy)
}
-func ParseFromHTMLFile(filepath string) Article {
+func ParseFromHTMLFile(filepath string, imagePolicy ImagePolicy) Article {
file, err := os.Open(filepath)
if err != nil {
panic(err)
@@ -182,10 +191,10 @@ func ParseFromHTMLFile(filepath string) Article {
if err2 != nil {
panic(err)
}
- return ParseFromReader(bytes.NewReader(content))
+ return ParseFromReader(bytes.NewReader(content), imagePolicy)
}
-func ParseFromURL(url string) Article {
+func ParseFromURL(url string, imagePolicy ImagePolicy) Article {
res, err := http.Get(url)
if err != nil {
panic(err)
@@ -194,7 +203,7 @@ func ParseFromURL(url string) Article {
if res.StatusCode != 200 {
log.Fatalf("get from url %s error: %d %s", url, res.StatusCode, res.Status)
}
- return ParseFromReader(res.Body)
+ return ParseFromReader(res.Body, imagePolicy)
}
func removeBrAndBlank(s string) string {
@@ -230,3 +239,26 @@ func fetchImgFile(url string) []byte {
func img2base64(content []byte) string {
return base64.StdEncoding.EncodeToString(content)
}
+
+type ImagePolicy int32
+
+const (
+ IMAGE_POLICY_URL ImagePolicy = iota
+ IMAGE_POLICY_SAVE
+ IMAGE_POLICY_BASE64
+)
+
+func ImageArgValue2ImagePolicy(val string) ImagePolicy {
+ var imagePolicy ImagePolicy
+ switch val {
+ case "url":
+ imagePolicy = IMAGE_POLICY_URL
+ case "save":
+ imagePolicy = IMAGE_POLICY_SAVE
+ case "base64":
+ fallthrough
+ default:
+ imagePolicy = IMAGE_POLICY_BASE64
+ }
+ return imagePolicy
+}
diff --git a/server/server.go b/server/server.go
index 63ba743..7b01e18 100644
--- a/server/server.go
+++ b/server/server.go
@@ -4,26 +4,43 @@ import (
"fmt"
"log"
"net/http"
+ "regexp"
+ "strings"
"github.com/fengxxc/wechatmp2markdown/format"
"github.com/fengxxc/wechatmp2markdown/parse"
+ "github.com/fengxxc/wechatmp2markdown/util"
)
func Start(addr string) {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
- wechatmpURL := r.FormValue("url")
+ rawQuery := r.URL.RawQuery
+ paramsMap := parseParams(rawQuery)
+
+ // url param
+ wechatmpURL := paramsMap["url"]
+ fmt.Printf("accept url: %s\n", wechatmpURL)
+ imageArgValue := paramsMap["image"]
+ fmt.Printf(" image: %s\n", imageArgValue)
+ imagePolicy := parse.ImageArgValue2ImagePolicy(imageArgValue)
+
if wechatmpURL == "" {
w.WriteHeader(http.StatusBadRequest)
- w.Write([]byte("param 'url' must not be empty. please put in a wechatmp URL and try again."))
+ w.Write([]byte(defHTML))
return
}
w.Header().Set("Content-Type", "application/octet-stream")
- var articleStruct parse.Article = parse.ParseFromURL(wechatmpURL)
+ var articleStruct parse.Article = parse.ParseFromURL(wechatmpURL, imagePolicy)
title := articleStruct.Title.Val.(string)
- w.Header().Set("Content-Disposition", "attachment; filename="+title+".md")
- var mdString string = format.Format(articleStruct)
- w.Write([]byte(mdString))
- return
+ mdString, saveImageBytes := format.Format(articleStruct)
+ if len(saveImageBytes) > 0 {
+ w.Header().Set("Content-Disposition", "attachment; filename="+title+".zip")
+ saveImageBytes[title] = []byte(mdString)
+ util.HttpDownloadZip(w, saveImageBytes)
+ } else {
+ w.Header().Set("Content-Disposition", "attachment; filename="+title+".md")
+ w.Write([]byte(mdString))
+ }
})
fmt.Printf("wechatmp2markdown server listening on %s\n", addr)
@@ -31,3 +48,52 @@ func Start(addr string) {
log.Fatal(err)
}
}
+
+var defHTML string = `
+
+
+
+
+
+ wechatmp2markdown
+
+
+ wechatmp2markdown
+
+ -
+ param 'url' is required. please put in a wechatmp URL and try again.
+
+ -
+ param 'image' is optional, value include: 'url' / 'save' / 'base64'(default)
+
+ -
+ example: http://localhost:8964/?url=https://mp.weixin.qq.com/s?__biz=aaaa==&mid=1111&idx=2&sn=bbbb&chksm=cccc&scene=123&image=save
+
+
+
+
+`
+
+func parseParams(rawQuery string) map[string]string {
+ result := make(map[string]string)
+ reg := regexp.MustCompile(`(&?image=)([a-z]+)`)
+ matcheImage := reg.FindStringSubmatch(rawQuery)
+ var urlParamFull string = rawQuery
+ if len(matcheImage) > 1 {
+ // 有image参数
+ imageParamFull := matcheImage[0]
+ urlParamFull = strings.Replace(rawQuery, imageParamFull, "", 1)
+
+ if len(matcheImage) > 2 {
+ imageParamVal := matcheImage[2]
+ result["image"] = imageParamVal
+ }
+ }
+ regUrl := regexp.MustCompile(`(&?url=)(.+)`)
+ matcheUrl := regUrl.FindStringSubmatch(urlParamFull)
+ if len(matcheUrl) > 2 {
+ urlParamVal := matcheUrl[2]
+ result["url"] = urlParamVal
+ }
+ return result
+}
diff --git a/test/test1.go b/test/test1.go
index 92f513b..e6949e4 100644
--- a/test/test1.go
+++ b/test/test1.go
@@ -9,12 +9,12 @@ import (
)
func Test1() {
- var articleStruct parse.Article = parse.ParseFromHTMLFile("./test/test1.html")
+ var articleStruct parse.Article = parse.ParseFromHTMLFile("./test/test1.html", parse.IMAGE_POLICY_BASE64)
fmt.Println("-------------------test1.html parse-------------------")
fmt.Printf("%+v\n", articleStruct)
fmt.Println("-------------------test1.html format-------------------")
- var mdString string = format.Format(articleStruct)
+ mdString, _ := format.Format(articleStruct)
fmt.Print(mdString)
ioutil.WriteFile("./test/test1_target.md", []byte(mdString), 0644)
}
diff --git a/test/test2.go b/test/test2.go
index be7652b..d753fac 100644
--- a/test/test2.go
+++ b/test/test2.go
@@ -9,12 +9,13 @@ import (
)
func Test2() {
- var articleStruct parse.Article = parse.ParseFromURL("https://mp.weixin.qq.com/s?__biz=MzIzOTU0NTQ0MA==&mid=2247506315&idx=1&sn=1546be4ecece176f669da4eed7076ee2&chksm=e92ae484de5d6d92d93cd68b927fa91e2935a75c9aafc02f294237653ca8a342e8982cabbc1d&cur_album_id=1391790902901014528&scene=189#wechat_redirect")
+ // var articleStruct parse.Article = parse.ParseFromURL("https://mp.weixin.qq.com/s?__biz=MzIzOTU0NTQ0MA==&mid=2247506315&idx=1&sn=1546be4ecece176f669da4eed7076ee2&chksm=e92ae484de5d6d92d93cd68b927fa91e2935a75c9aafc02f294237653ca8a342e8982cabbc1d&cur_album_id=1391790902901014528&scene=189#wechat_redirect")
+ var articleStruct parse.Article = parse.ParseFromURL("https://mp.weixin.qq.com/s?__biz=MzU0OTE4MzYzMw==&mid=2247525863&idx=2&sn=d759f98b62f61f3a8312da4ee426c287&chksm=fbb1ec19ccc6650f40c0ef67b47163040c33f9dfe3d6f05bf28d4d823b6f847c09fea046b2eb&scene=132#wechat_redirect", parse.IMAGE_POLICY_BASE64)
byteArry, _ := json.MarshalIndent(articleStruct, "", " ")
// fmt.Println(string(byteArry))
ioutil.WriteFile("./test/test2_target.json", byteArry, 0644)
- var mdString string = format.Format((articleStruct))
+ mdString, _ := format.Format((articleStruct))
ioutil.WriteFile("./test/test2_target.md", []byte(mdString), 0644)
}
diff --git a/util/util.go b/util/util.go
new file mode 100644
index 0000000..be172b5
--- /dev/null
+++ b/util/util.go
@@ -0,0 +1,77 @@
+package util
+
+import (
+ "archive/zip"
+ "bytes"
+ "crypto/md5"
+ "encoding/hex"
+ "io"
+ "log"
+ "net/http"
+ "os"
+ "regexp"
+)
+
+func MergeMap(m1 map[string][]byte, m2 map[string][]byte) {
+ for k, v := range m2 {
+ m1[k] = v
+ }
+}
+
+func Zip(zipFileName string, files map[string][]byte) {
+ f, err := os.Create(zipFileName)
+ if err != nil {
+ log.Fatal(err)
+ return
+ }
+ defer f.Close()
+
+ zipWriter := zip.NewWriter(f)
+ for name, file := range files {
+ zw, err := zipWriter.Create(name)
+ if err != nil {
+ log.Fatal(err)
+ }
+ if _, err := io.Copy(zw, bytes.NewReader(file)); err != nil {
+ log.Fatal(err)
+ }
+ }
+ zipWriter.Close()
+}
+
+func HttpDownloadZip(w http.ResponseWriter, files map[string][]byte) {
+ zipWriter := zip.NewWriter(w)
+ for name, file := range files {
+ zw, err := zipWriter.Create(name)
+ if err != nil {
+ log.Fatal(err)
+ }
+ if _, err := io.Copy(zw, bytes.NewReader(file)); err != nil {
+ log.Fatal(err)
+ }
+ }
+ zipWriter.Close()
+}
+
+func MD5(content []byte) string {
+ hash := md5.New()
+ hash.Write(content)
+ md5Bytes := hash.Sum(nil)
+ return hex.EncodeToString(md5Bytes)
+}
+
+// 从图片src中解析出图片的扩展名
+func ParseImageExtFromSrc(src string) string {
+ reg := regexp.MustCompile(`(wx_fmt=)([a-zA-Z]+)(&?)`)
+ matches := reg.FindStringSubmatch(src)
+ if len(matches) < 3 {
+ return ""
+ }
+ return matches[2]
+}
+
+// 判断路径是否存在
+func PathIsExists(path string) (os.FileInfo, bool) {
+ f, err := os.Stat(path)
+ return f, err == nil || os.IsExist(err)
+}