go-zero/tools/goctl/pkg/parser/api/ast/importstatement.go

package ast

import "github.com/zeromicro/go-zero/tools/goctl/pkg/parser/api/token"

// ImportStmt represents an import statement.
type ImportStmt interface {
	Stmt
	importNode()
}

// ImportLiteralStmt represents an import literal statement.
type ImportLiteralStmt struct {
	// Import is the import token.
	Import *TokenNode
	// Value is the import value.
	Value *TokenNode
}

func (i *ImportLiteralStmt) HasHeadCommentGroup() bool {
	return i.Import.HasHeadCommentGroup()
}

func (i *ImportLiteralStmt) HasLeadingCommentGroup() bool {
	return i.Value.HasLeadingCommentGroup()
}

func (i *ImportLiteralStmt) CommentGroup() (head, leading CommentGroup) {
	return i.Import.HeadCommentGroup, i.Value.LeadingCommentGroup
}

func (i *ImportLiteralStmt) Format(prefix ...string) (result string) {
	if i.Value.IsZeroString() {
		return ""
	}
	w := NewBufferWriter()
	importNode := transferTokenNode(i.Import, ignoreLeadingComment(), withTokenNodePrefix(prefix...))
	w.Write(withNode(importNode, i.Value), withMode(ModeExpectInSameLine))
	return w.String()
}

func (i *ImportLiteralStmt) End() token.Position {
	return i.Value.End()
}

func (i *ImportLiteralStmt) importNode() {}

func (i *ImportLiteralStmt) Pos() token.Position {
	return i.Import.Pos()
}

func (i *ImportLiteralStmt) stmtNode() {}

type ImportGroupStmt struct {
	// Import is the import token.
	Import *TokenNode
	// LParen is the left parenthesis token.
	LParen *TokenNode
	// Values is the import values.
	Values []*TokenNode
	// RParen is the right parenthesis token.
	RParen *TokenNode
}

func (i *ImportGroupStmt) HasHeadCommentGroup() bool {
	return i.Import.HasHeadCommentGroup()
}

func (i *ImportGroupStmt) HasLeadingCommentGroup() bool {
	return i.RParen.HasLeadingCommentGroup()
}

func (i *ImportGroupStmt) CommentGroup() (head, leading CommentGroup) {
	return i.Import.HeadCommentGroup, i.RParen.LeadingCommentGroup
}

func (i *ImportGroupStmt) Format(prefix ...string) string {
	var textList []string
	for _, v := range i.Values {
		if v.IsZeroString() {
			continue
		}
		textList = append(textList, v.Format(Indent))
	}
	if len(textList) == 0 {
		return ""
	}

	importNode := transferTokenNode(i.Import, ignoreLeadingComment(), withTokenNodePrefix(prefix...))
	w := NewBufferWriter()
	w.Write(withNode(importNode, i.LParen), expectSameLine())
	w.NewLine()
	for _, v := range i.Values {
		node := transferTokenNode(v, withTokenNodePrefix(peekOne(prefix)+Indent))
		w.Write(withNode(node), expectSameLine())
		w.NewLine()
	}
	w.Write(withNode(transferTokenNode(i.RParen, withTokenNodePrefix(prefix...))))
	return w.String()
}

func (i *ImportGroupStmt) End() token.Position {
	return i.RParen.End()
}

func (i *ImportGroupStmt) importNode() {}

func (i *ImportGroupStmt) Pos() token.Position {
	return i.Import.Pos()
}

func (i *ImportGroupStmt) stmtNode() {}