131 lines
2.5 KiB
Go
131 lines
2.5 KiB
Go
// Copyright 2015 <chaishushan{AT}gmail.com>. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// +build ingore
|
|
|
|
//
|
|
// 给特殊模式标定的单词增加链接.
|
|
//
|
|
// Example:
|
|
// addlinks
|
|
// addlinks dir "\.go$"
|
|
//
|
|
// Help:
|
|
// addlinks -h
|
|
//
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
const MaxFileSize = 8 << 20 // 8MB
|
|
|
|
const usage = `
|
|
Usage: addlinks dir [nameFilter]
|
|
addlinks -h
|
|
|
|
Example:
|
|
addlinks
|
|
addlinks dir "\.go$"
|
|
|
|
Report bugs to <chaishushan{AT}gmail.com>.
|
|
`
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 || os.Args[1] == "-h" {
|
|
fmt.Fprintln(os.Stderr, usage[1:len(usage)-1])
|
|
os.Exit(0)
|
|
}
|
|
dir, nameFilter := os.Args[1], ".*"
|
|
if len(os.Args) > 2 {
|
|
nameFilter = os.Args[2]
|
|
}
|
|
|
|
total := 0
|
|
filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
log.Fatal("filepath.Walk: ", err)
|
|
return err
|
|
}
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
relpath, err := filepath.Rel(dir, path)
|
|
if err != nil {
|
|
log.Fatal("filepath.Rel: ", err)
|
|
return err
|
|
}
|
|
mathed, err := regexp.MatchString(nameFilter, relpath)
|
|
if err != nil {
|
|
log.Fatal("regexp.MatchString: ", err)
|
|
}
|
|
if mathed {
|
|
if changed := convertFile(path); changed {
|
|
fmt.Printf("%s\n", relpath)
|
|
total++
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
fmt.Printf("total %d\n", total)
|
|
}
|
|
|
|
func convertFile(path string) (changed bool) {
|
|
abspath, err := filepath.Abs(path)
|
|
if err != nil {
|
|
log.Fatal("convertFile: filepath.Abs:", err)
|
|
}
|
|
|
|
fi, err := os.Lstat(abspath)
|
|
if err != nil {
|
|
log.Fatal("convertFile: os.Lstat:", err)
|
|
}
|
|
if fi.Size() > MaxFileSize {
|
|
return false
|
|
}
|
|
|
|
oldData, err := ioutil.ReadFile(abspath)
|
|
if err != nil {
|
|
log.Fatal("convertFile: ioutil.ReadFile:", err)
|
|
}
|
|
if !utf8.Valid(oldData) {
|
|
return false
|
|
}
|
|
|
|
newData := append([]byte{}, oldData...)
|
|
for re, v := range _RegexpLinksTable {
|
|
newData = re.ReplaceAll(newData, []byte(v))
|
|
}
|
|
|
|
if string(newData) == string(oldData) {
|
|
return false
|
|
}
|
|
|
|
err = ioutil.WriteFile(abspath, newData, 0666)
|
|
if err != nil {
|
|
log.Fatal("convertFile: ioutil.WriteFile:", err)
|
|
}
|
|
return true
|
|
}
|
|
|
|
var _RegexpLinksTable = func() map[*regexp.Regexp]string {
|
|
m := make(map[*regexp.Regexp]string)
|
|
for k, v := range _LinkTable {
|
|
reKey := regexp.MustCompile(regexp.QuoteMeta(`[`+k+`]`) + `\(\S*\)`)
|
|
m[reKey] = fmt.Sprintf("[%s](%s)", k, v)
|
|
}
|
|
return m
|
|
}()
|
|
|
|
var _LinkTable = map[string]string{
|
|
"Alan A. A. Donovan": "https://github.com/adonovan",
|
|
}
|