go-zero/tools/goctl/api/parser/util.go

package parser

import (
	"bufio"
	"errors"
	"regexp"
	"strings"

	"github.com/tal-tech/go-zero/tools/goctl/api/spec"
)

// struct match
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
}