goctl added

tools/goctl/api/parser/util.go

@@ -0,0 +1,103 @@
+package parser
+
+import (
+	"bufio"
+	"errors"
+	"regexp"
+	"strings"
+
+	"zero/tools/goctl/api/spec"
+)
+
+const (
+	// struct匹配
+	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")
+)
+
+var 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
+}