mirror of
https://github.com/fumiama/gozel.git
synced 2026-06-12 12:30:33 +08:00
init: generator framework
This commit is contained in:
49
gen/scan.go
Normal file
49
gen/scan.go
Normal file
@@ -0,0 +1,49 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"os"
|
||||
"path"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var infhcache = make(map[string]string, 1024)
|
||||
|
||||
func infh(name string) string {
|
||||
if s, ok := infhcache[name]; ok {
|
||||
return s
|
||||
}
|
||||
s := " [" + name + "]"
|
||||
infhcache[name] = s
|
||||
return s
|
||||
}
|
||||
|
||||
func scanheader(name string, scan *bufio.Scanner) {
|
||||
ln := 0
|
||||
var regionfile *os.File
|
||||
for scan.Scan() {
|
||||
ln++
|
||||
t := scan.Text()
|
||||
switch {
|
||||
case strings.HasPrefix(t, "#pragma region "):
|
||||
region := strings.TrimSpace(t[15:])
|
||||
if region == "" {
|
||||
panic(fmt.Sprintf("%s L%d: unexpected empty region", name, ln))
|
||||
}
|
||||
fmt.Println(infh(name), "scanning region", region)
|
||||
f, err := os.Create(path.Join(name, region+".go"))
|
||||
if err != nil {
|
||||
panic(fmt.Sprintf("%s L%d: cannot create region %s, err: %v", name, ln, region, err))
|
||||
}
|
||||
f.WriteString("// Code generated by gen. DO NOT EDIT.\n\npackage ")
|
||||
f.WriteString(name)
|
||||
f.WriteString("\n")
|
||||
regionfile = f
|
||||
case strings.HasPrefix(t, "#pragma endregion"):
|
||||
fmt.Println(infh(name), "close region", regionfile.Name())
|
||||
_ = regionfile.Close()
|
||||
regionfile = nil
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user