Skip to content

Commit c0ce8e5

Browse files
committed
可以编译文件了,可以生成html文件了
1 parent 6194380 commit c0ce8e5

File tree

1 file changed

+37
-18
lines changed

1 file changed

+37
-18
lines changed

Diff for: build.go

+37-18
Original file line numberDiff line numberDiff line change
@@ -6,32 +6,51 @@ import (
66
"io/ioutil"
77
"os"
88
"path/filepath"
9+
"strings"
910
)
1011

1112
//定义一个访问者结构体
1213
type Visitor struct{}
1314

14-
//定义其碰到目录的时候的行为
15-
func (v *Visitor) VisitDir(path string, f *os.FileInfo) bool {
16-
fmt.Println(path)
17-
return true
18-
}
19-
20-
//定义其碰到文件的时候的行为
21-
func (v *Visitor) VisitFile(path string, f *os.FileInfo) {
22-
fmt.Println(path)
23-
//@todo 如何读取md文件转化为html
24-
input, err := ioutil.ReadAll()
25-
output := blackfriday.MarkdownBasic(input)
15+
func (self *Visitor) visit(path string, f os.FileInfo, err error) error {
16+
if f == nil {
17+
return err
18+
}
19+
if f.IsDir() {
20+
return nil
21+
} else if (f.Mode() & os.ModeSymlink) > 0 {
22+
return nil
23+
} else {
24+
if strings.Contains(f.Name(), ".md") {
25+
fmt.Println(f)
26+
file, err := os.Open(f.Name())
27+
if err != nil {
28+
return err
29+
}
30+
input, _ := ioutil.ReadAll(file)
31+
output := blackfriday.MarkdownCommon(input)
32+
var out *os.File
33+
if out, err = os.Create(f.Name() + ".html"); err != nil {
34+
fmt.Fprintf(os.Stderr, "Error creating %s: %v", f.Name(), err)
35+
os.Exit(-1)
36+
}
37+
defer out.Close()
38+
if _, err = out.Write(output); err != nil {
39+
fmt.Fprintln(os.Stderr, "Error writing output:", err)
40+
os.Exit(-1)
41+
}
42+
}
43+
}
44+
return nil
2645
}
2746

2847
func main() {
2948
v := &Visitor{}
30-
errors := make(chan os.Error, 64) //错误消息使用64个缓存,可以随意
31-
filepath.Walk("./", v, errors)
32-
select {
33-
case err := <-errors:
34-
panic(err)
35-
default:
49+
err := filepath.Walk("./", func(path string, f os.FileInfo, err error) error {
50+
return v.visit(path, f, err)
51+
})
52+
53+
if err != nil {
54+
fmt.Printf("filepath.Walk() returned %v\n", err)
3655
}
3756
}

0 commit comments

Comments
 (0)