You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.8 KiB
Go
112 lines
2.8 KiB
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() {}
|