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

package parser

import (
	"bufio"
	"errors"
	"strings"

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

var emptyType spec.Type

type ApiStruct struct {
	Info       string
	StructBody string
	Service    string
	Imports    string
}

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) (*ApiStruct, error) {
	var result ApiStruct
	scanner := bufio.NewScanner(strings.NewReader(api))
	var parseInfo = false
	var parseImport = false
	var parseType = false
	var parseSevice = false
	var segment string
	for scanner.Scan() {
		line := strings.TrimSpace(scanner.Text())

		if line == "@doc(" {
			parseInfo = true
		}
		if line == ")" && parseInfo {
			parseInfo = false
			result.Info = segment + ")"
			segment = ""
			continue
		}

		if strings.HasPrefix(line, "import") {
			parseImport = true
		}
		if parseImport && (strings.HasPrefix(line, "type") || strings.HasPrefix(line, "@server") ||
			strings.HasPrefix(line, "service")) {
			parseImport = false
			result.Imports = segment
			segment = line + "\n"
			continue
		}

		if strings.HasPrefix(line, "type") {
			parseType = true
		}
		if strings.HasPrefix(line, "@server") || strings.HasPrefix(line, "service") {
			if parseType {
				parseType = false
				result.StructBody = segment
				segment = line + "\n"
				continue
			}
			parseSevice = true
		}
		segment += scanner.Text() + "\n"
	}

	if !parseSevice {
		return nil, errors.New("no service defined")
	}
	result.Service = segment
	return &result, nil
}