1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
package main
import (
"flag"
"io/ioutil"
"log"
"os"
"os/exec"
"path/filepath"
"regexp"
"gopkg.in/yaml.v2"
)
// Assumes we're getting a pageName ending in .page
func convertPage(newDir, pageName string) error {
yamlName := pageName[:len(pageName)-5] + ".yml"
headName := pageName[:len(pageName)-5] + ".head"
newPageName := pageName[:len(pageName)-5] + ".md"
hasHead := false
headBytes := []byte{}
if _, err := os.Stat(headName); err == nil {
hasHead = true
}
if hasHead {
var err error
headBytes, err = ioutil.ReadFile(headName)
if err != nil {
return err
}
}
y, err := yaml.Marshal(map[string]string{
"head": string(headBytes),
})
if err != nil {
return err
}
cmd := exec.Command("mkdir", "-p", filepath.Join(newDir, filepath.Dir(pageName)))
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
err = cmd.Run()
if err != nil {
return err
}
cmd = exec.Command("cp", "-v", pageName, filepath.Join(newDir, newPageName))
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
err = cmd.Run()
if err != nil {
return err
}
if len(headBytes) == 0 {
return nil
} else {
return ioutil.WriteFile(newDir+"/"+yamlName, y, 0644)
}
}
func main() {
fl := flag.NewFlagSet("Website converter", flag.ExitOnError)
origDirectory := fl.String("d", ".", "Directory to read from")
newDirectory := fl.String("D", ".", "Directory to write to")
_ = fl.Parse(os.Args[1:])
newDir, _ := filepath.Abs(*newDirectory)
origDir, _ := filepath.Abs(*origDirectory)
if err := os.Chdir(origDir); err != nil {
log.Fatal(err)
}
pageRe := regexp.MustCompile(`\.page$`)
_ = filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
if err != nil {
// Prevents panic
log.Println(err)
return err
}
if pageRe.MatchString(info.Name()) {
if err := convertPage(newDir, path); err != nil {
log.Println(err)
return err
}
}
return nil
})
}
|
