types.go

package gogencli

import (
	"go/ast"
	"strings"
)

func (g *Generator) IsGoGenerateLine(node ast.Node) bool {
	if node == nil || g.root == nil || g.root.TokenFile == nil {
		return false
	}

	switch n := node.(type) {
	case *ast.CommentGroup:
		if n == nil {
			return false
		}
		for _, c := range n.List {
			if g.IsGoGenerateLine(c) {
				return true
			}
		}
	case *ast.Comment:
		if n == nil {
			return false
		}
		if !n.Pos().IsValid() {
			return false
		}

		// check if the comment is a //go:generate comment
		text := n.Text
		if !strings.HasPrefix(text, "//go:generate") {
			return false
		}
		// check if the comment is on the same line as $GOLINE
		var line int
		for l := range g.root.TokenFile.Lines() {
			if g.root.TokenFile.LineStart(l+1) > n.Pos() {
				break
			}
			// $GOLINE env var is 1-based.
			line = l + 1
		}
		if line != g.goLine {
			return false
		}

		return true
	}
	return false
}