diff --git a/format/format.go b/format/format.go index 1393803..0e885a9 100644 --- a/format/format.go +++ b/format/format.go @@ -1,6 +1,7 @@ package format import ( + "io/ioutil" "strconv" "github.com/fengxxc/wechatmp2markdown/parse" @@ -19,6 +20,11 @@ func Format(article parse.Article) string { return result } +func FormatAndSave(article parse.Article, filePath string) { + var result string = Format(article) + ioutil.WriteFile(filePath, []byte(result), 0644) +} + func formatTitle(piece parse.Piece) string { var prefix string level, _ := strconv.Atoi(piece.Attrs["level"]) diff --git a/main.go b/main.go index 9a1eb29..a92b3f1 100644 --- a/main.go +++ b/main.go @@ -1,8 +1,23 @@ package main -import "github.com/fengxxc/wechatmp2markdown/test" +import ( + "fmt" + "os" + + "github.com/fengxxc/wechatmp2markdown/format" + "github.com/fengxxc/wechatmp2markdown/parse" +) func main() { // test.Test1() - test.Test2() + // test.Test2() + args := os.Args + if len(args) <= 2 { + panic("not enough args") + } + url := args[1] + filename := args[2] + fmt.Printf("url: %s, filename: %s\n", url, filename) + var articleStruct parse.Article = parse.ParseFromURL(url) + format.FormatAndSave(articleStruct, filename) }