func main() {
flag.Parse()
args := flag.Args()
if len(args) == 0 || len(args) > 3 {
PrintUsage()
os.Exit(1)
}
switch args[0] {
default:
PrintUsage()
os.Exit(1)
case "config":
cnf, err := gor.ReadConfig(".")
if err != nil {
log.Fatal(err)
}
log.Println("RuhohSpec: ", cnf["RuhohSpec"])
buf, err := json.MarshalIndent(cnf, "", " ")
if err != nil {
log.Fatal(err)
}
log.Println("global config\n", string(buf))
case "new":
if len(args) == 1 {
log.Fatalln(os.Args[0], "new", "<dir>")
}
CmdInit(args[1])
case "posts":
gor.ListPosts()
case "payload":
payload, err := gor.BuildPlayload("./")
if err != nil {
log.Fatal(err)
}
buf, err := json.MarshalIndent(payload, "", " ")
if err != nil {
log.Fatal(err)
}
log.Println(string(buf))
case "compile":
err := gor.Compile()
if err != nil {
log.Fatal(err)
}
case "post":
if len(args) == 1 {
log.Fatal("gor post <title>")
} else if len(args) == 2 {
gor.CreateNewPost(args[1])
} else {
gor.CreateNewPostWithImgs(args[1], args[2])
}
case "http":
log.Println("Listen at ", *http_addr)
log.Println(http.ListenAndServe(*http_addr, http.FileServer(http.Dir("compiled"))))
case "pprof":
f, _ := os.OpenFile("gor.pprof", os.O_WRONLY|os.O_TRUNC|os.O_CREATE, os.ModePerm)
defer f.Close()
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
for i := 0; i < 100; i++ {
err := gor.Compile()
if err != nil {
log.Fatal(err)
}
}
case ".update.zip.go":
d, _ := ioutil.ReadFile("gor-content.zip")
_zip, _ := os.OpenFile("zip.go", os.O_WRONLY|os.O_TRUNC|os.O_CREATE, os.ModePerm)
header := `package main
const INIT_ZIP="`
_zip.Write([]byte(header))
encoder := base64.NewEncoder(base64.StdEncoding, _zip)
encoder.Write(d)
encoder.Close()
_zip.Write([]byte(`"`))
_zip.Sync()
_zip.Close()
}
}
func main() {
flag.Parse()
args = flag.Args()
if len(args) == 0 || len(args) > 4 {
PrintUsage()
os.Exit(1)
}
switch args[0] {
default:
PrintUsage()
os.Exit(1)
case "config":
cnf, err := gor.ReadConfig(".")
if err != nil {
log.Fatal(err)
}
log.Println("RuhohSpec: ", cnf["RuhohSpec"])
buf, err := json.MarshalIndent(cnf, "", " ")
if err != nil {
log.Fatal(err)
}
log.Println("global config\n", string(buf))
case "new":
fallthrough
case "init":
if len(args) == 1 {
log.Fatalln(os.Args[0], "new", "<dir>")
}
CmdInit(args[1])
case "posts":
gor.ListPosts()
case "payload":
payload, err := gor.BuildPayload("./")
if err != nil {
log.Fatal(err)
}
buf, err := json.MarshalIndent(payload, "", " ")
if err != nil {
log.Fatal(err)
}
log.Println(string(buf))
case "compile":
fallthrough
case "build":
fallthrough
case "c":
_compile()
case "post":
if len(args) == 1 {
log.Fatal("gor post <title>")
} else if len(args) == 2 {
gor.CreateNewPost(args[1])
} else {
gor.CreateNewPostWithImgs(args[1], args[2])
}
case "addimg":
if len(args) == 3 {
gor.AddImgs(args[1], args[2], "")
} else if len(args) == 4 {
gor.AddImgs(args[1], args[2], args[3])
} else {
log.Fatal("gor post <title> <dir> or <date>")
}
case "http":
_http()
case "preview":
gor.HTML_EXT = _watch_js
_compile()
go _http()
_watch()
case "pprof":
f, _ := os.OpenFile("gor.pprof", os.O_WRONLY|os.O_TRUNC|os.O_CREATE, os.ModePerm)
defer f.Close()
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
for i := 0; i < 100; i++ {
_compile()
}
case ".update.zip.go":
d, _ := ioutil.ReadFile("gor-content.zip")
_zip, _ := os.OpenFile("zip.go", os.O_WRONLY|os.O_TRUNC|os.O_CREATE, os.ModePerm)
header := `package main
const INIT_ZIP="`
_zip.Write([]byte(header))
encoder := base64.NewEncoder(base64.StdEncoding, _zip)
encoder.Write(d)
encoder.Close()
_zip.Write([]byte(`"`))
_zip.Sync()
_zip.Close()
}
}