Add config as language file

This patch will change several behaviour while generating the ebook:

- `build.go` now accepts two environment variables: TMP, WORKDIR

    * TMP:

Temporary directory path.
The directory stores intermediate files like '*.html' or metadata.txt.

    * WORKDIR:

Script base directory.

- `config` as language file

config file will be `source`d to change its ebook metadata.

- `genepub.sh` is integrated to `build.sh` and `build.go`

See `FixHeader` and `RemoveFooterLink` functions.

ebook/build.go

@@ -11,48 +11,98 @@ import (
 )
 
 // 定义一个访问者结构体
-type Visitor struct {}
+type Visitor struct{}
 
-func (self *Visitor) visit(path string, f os.FileInfo, err error) error {
-	if f == nil {
-		return err
-	}
-	if f.IsDir() {
-		return nil
-	} else if (f.Mode() & os.ModeSymlink) > 0 {
-		return nil
-	} else {
-		if strings.HasSuffix(f.Name(), ".md") {
-			fmt.Println(f)
-			file, err := os.Open(f.Name())
-			if err != nil {
-				return err
-			}
-			input, _ := ioutil.ReadAll(file)
-			input = regexp.MustCompile("\\[(.*?)\\]\\(<?(.*?)\\.md>?\\)").ReplaceAll(input, []byte("[$1](<$2.html>)"))
-			var out *os.File
-			if out, err = os.Create(strings.Replace(f.Name(), ".md", ".html", -1)); err != nil {
-				fmt.Fprintf(os.Stderr, "Error creating %s: %v", f.Name(), err)
-				os.Exit(-1)
-			}
-			defer out.Close()
-			md := md2min.New("none")
-			err = md.Parse(input, out)
-			if err != nil {
-				fmt.Fprintln(os.Stderr, "Parsing Error", err)
-				os.Exit(-1)
-			}
+func (self *Visitor) md2html(arg map[string]string) error {
+	from := arg["from"]
+	to := arg["to"]
+	err := filepath.Walk(from+"/", func(path string, f os.FileInfo, err error) error {
+		if f == nil {
+			return err
 		}
+		if f.IsDir() {
+			return nil
+		}
+		if (f.Mode() & os.ModeSymlink) > 0 {
+			return nil
+		}
+		if !strings.HasSuffix(f.Name(), ".md") {
+			return nil
+		}
+
+		file, err := os.Open(path)
+		if err != nil {
+			return err
+		}
+
+		input_byte, _ := ioutil.ReadAll(file)
+		input := string(input_byte)
+		input = regexp.MustCompile(`\[(.*?)\]\(<?(.*?)\.md>?\)`).ReplaceAllString(input, "[$1](<$2.html>)")
+
+		if f.Name() == "README.md" {
+			input = regexp.MustCompile(`https:\/\/github\.com\/astaxie\/build-web-application-with-golang\/blob\/master\/`).ReplaceAllString(input, "")
+		}
+
+		// 以#开头的行，在#后增加空格
+		// 以#开头的行, 删除多余的空格
+		input = FixHeader(input)
+
+		// 删除页面链接
+		input = RemoveFooterLink(input)
+
+		var out *os.File
+		filename := strings.Replace(f.Name(), ".md", ".html", -1)
+		fmt.Println(to + "/" + filename)
+		if out, err = os.Create(to + "/" + filename); err != nil {
+			fmt.Fprintf(os.Stderr, "Error creating %s: %v", f.Name(), err)
+			os.Exit(-1)
+		}
+		defer out.Close()
+		md := md2min.New("none")
+		err = md.Parse([]byte(input), out)
+		if err != nil {
+			fmt.Fprintln(os.Stderr, "Parsing Error", err)
+			os.Exit(-1)
+		}
+
+		return nil
+	})
+	return err
+}
+
+func FixHeader(input string) string {
+	re_header := regexp.MustCompile(`(?m)^#.+$`)
+	re_sub := regexp.MustCompile(`^(#+)\s*(.+)$`)
+	fixer := func(header string) string {
+		s := re_sub.FindStringSubmatch(header)
+		return s[1] + " " + s[2]
 	}
-	return nil
+	return re_header.ReplaceAllStringFunc(input, fixer)
+}
+
+func RemoveFooterLink(input string) string {
+	re_footer := regexp.MustCompile(`(?m)^#{2,} links.*?\n(.+\n)*`)
+	return re_footer.ReplaceAllString(input, "")
 }
 
 func main() {
-	v := &Visitor{}
-	err := filepath.Walk("./", func(path string, f os.FileInfo, err error) error {
-		return v.visit(path, f, err)
-	})
+	tmp := os.Getenv("TMP")
+	if tmp == "" {
+		tmp = "."
+	}
 
+	workdir := os.Getenv("WORKDIR")
+	if workdir == "" {
+		workdir = "."
+	}
+
+	arg := map[string]string{
+		"from": workdir,
+		"to":   tmp,
+	}
+
+	v := &Visitor{}
+	err := v.md2html(arg)
 	if err != nil {
 		fmt.Printf("filepath.Walk() returned %v\n", err)
 	}