You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
package parser
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/api/spec"
|
|
)
|
|
|
|
// struct匹配
|
|
const typeRegex = `(?m)(?m)(^ *type\s+[a-zA-Z][a-zA-Z0-9_-]+\s+(((struct)\s*?\{[\w\W]*?[^\{]\})|([a-zA-Z][a-zA-Z0-9_-]+)))|(^ *type\s*?\([\w\W]+\}\s*\))`
|
|
|
|
var (
|
|
emptyStrcut = errors.New("struct body not found")
|
|
emptyType spec.Type
|
|
)
|
|
|
|
func GetType(api *spec.ApiSpec, t string) spec.Type {
|
|
for _, tp := range api.Types {
|
|
if tp.Name == t {
|
|
return tp
|
|
}
|
|
}
|
|
|
|
return emptyType
|
|
}
|
|
|
|
func isLetterDigit(r rune) bool {
|
|
return (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') || ('0' <= r && r <= '9')
|
|
}
|
|
|
|
func isSpace(r rune) bool {
|
|
return r == ' ' || r == '\t'
|
|
}
|
|
|
|
func isNewline(r rune) bool {
|
|
return r == '\n' || r == '\r'
|
|
}
|
|
|
|
func read(r *bufio.Reader) (rune, error) {
|
|
ch, _, err := r.ReadRune()
|
|
return ch, err
|
|
}
|
|
|
|
func readLine(r *bufio.Reader) (string, error) {
|
|
line, _, err := r.ReadLine()
|
|
if err != nil {
|
|
return "", err
|
|
} else {
|
|
return string(line), nil
|
|
}
|
|
}
|
|
|
|
func skipSpaces(r *bufio.Reader) error {
|
|
for {
|
|
next, err := read(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !isSpace(next) {
|
|
return unread(r)
|
|
}
|
|
}
|
|
}
|
|
|
|
func unread(r *bufio.Reader) error {
|
|
return r.UnreadRune()
|
|
}
|
|
|
|
func MatchStruct(api string) (info, structBody, service string, err error) {
|
|
r := regexp.MustCompile(typeRegex)
|
|
indexes := r.FindAllStringIndex(api, -1)
|
|
if len(indexes) == 0 {
|
|
return "", "", "", emptyStrcut
|
|
}
|
|
startIndexes := indexes[0]
|
|
endIndexes := indexes[len(indexes)-1]
|
|
|
|
info = api[:startIndexes[0]]
|
|
structBody = api[startIndexes[0]:endIndexes[len(endIndexes)-1]]
|
|
service = api[endIndexes[len(endIndexes)-1]:]
|
|
|
|
firstIIndex := strings.Index(info, "i")
|
|
if firstIIndex > 0 {
|
|
info = info[firstIIndex:]
|
|
}
|
|
|
|
lastServiceRightBraceIndex := strings.LastIndex(service, "}") + 1
|
|
var firstServiceIndex int
|
|
for index, char := range service {
|
|
if !isSpace(char) && !isNewline(char) {
|
|
firstServiceIndex = index
|
|
break
|
|
}
|
|
}
|
|
service = service[firstServiceIndex:lastServiceRightBraceIndex]
|
|
return
|
|
}
|