Documentation
¶
Overview ¶
Package astutil contains common utilities for working with the Go AST.
Index ¶
- func AddImport(fset *token.FileSet, f *ast.File, ipath string) (added bool)
- func AddNamedImport(fset *token.FileSet, f *ast.File, name, ipath string) (added bool)
- func Apply(root ast.Node, pre, post ApplyFunc) (result ast.Node)
- func DeleteImport(fset *token.FileSet, f *ast.File, path string) (deleted bool)
- func DeleteNamedImport(fset *token.FileSet, f *ast.File, name, path string) (deleted bool)
- func Imports(fset *token.FileSet, f *ast.File) [][]*ast.ImportSpec
- func NodeDescription(n ast.Node) string
- func PathEnclosingInterval(root *ast.File, start, end token.Pos) (path []ast.Node, exact bool)
- func RewriteImport(fset *token.FileSet, f *ast.File, oldPath, newPath string) (rewrote bool)
- func Unparen(e ast.Expr) ast.Expr
- func UsesImport(f *ast.File, path string) (used bool)
- type ApplyFunc
- type Cursor
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func AddNamedImport ¶
AddNamedImport adds the import path to the file f, if absent. If name is not empty, it is used to rename the import.
For example, calling
AddNamedImport(fset, f, "pathpkg", "path")
adds
import pathpkg "path"
func Apply ¶
Apply traverses a syntax tree recursively, starting with root, and calling pre and post for each node as described below. Apply returns the syntax tree, possibly modified.
If pre is not nil, it is called for each node before the node's children are traversed (pre-order). If pre returns false, no children are traversed, and post is not called for that node.
If post is not nil, and a prior call of pre didn't return false, post is called for each node after its children are traversed (post-order). If post returns false, traversal is terminated and Apply returns immediately.
Only fields that refer to AST nodes are considered children; i.e., token.Pos, Scopes, Objects, and fields of basic types (strings, etc.) are ignored.
Children are traversed in the order in which they appear in the respective node's struct definition. A package's files are traversed in the filenames' alphabetical order.
func DeleteImport ¶
DeleteImport deletes the import path from the file f, if present.
func DeleteNamedImport ¶
DeleteNamedImport deletes the import with the given name and path from the file f, if present.
func NodeDescription ¶
NodeDescription returns a description of the concrete type of n suitable for a user interface.
TODO(adonovan): in some cases (e.g. Field, FieldList, Ident, StarExpr) we could be much more specific given the path to the AST root. Perhaps we should do that.
func PathEnclosingInterval ¶
PathEnclosingInterval returns the node that encloses the source interval [start, end), and all its ancestors up to the AST root.
The definition of "enclosing" used by this function considers additional whitespace abutting a node to be enclosed by it. In this example:
z := x + y // add them
<-A->
<----B----->
the ast.BinaryExpr(+) node is considered to enclose interval B even though its [Pos()..End()) is actually only interval A. This behaviour makes user interfaces more tolerant of imperfect input.
This function treats tokens as nodes, though they are not included in the result. e.g. PathEnclosingInterval("+") returns the enclosing ast.BinaryExpr("x + y").
If start==end, the 1-char interval following start is used instead.
The 'exact' result is true if the interval contains only path[0] and perhaps some adjacent whitespace. It is false if the interval overlaps multiple children of path[0], or if it contains only interior whitespace of path[0]. In this example:
z := x + y // add them
<--C--> <---E-->
^
D
intervals C, D and E are inexact. C is contained by the z-assignment statement, because it spans three of its children (:=, x, +). So too is the 1-char interval D, because it contains only interior whitespace of the assignment. E is considered interior whitespace of the BlockStmt containing the assignment.
Precondition: [start, end) both lie within the same file as root. TODO(adonovan): return (nil, false) in this case and remove precond. Requires FileSet; see loader.tokenFileContainsPos.
Postcondition: path is never nil; it always contains at least 'root'.
func RewriteImport ¶
RewriteImport rewrites any import of path oldPath to path newPath.
Types ¶
type ApplyFunc ¶
An ApplyFunc is invoked by Apply for each node n, even if n is nil, before and/or after the node's children, using a Cursor describing the current node and providing operations on it.
The return value of ApplyFunc controls the syntax tree traversal. See Apply for details.
type Cursor ¶
type Cursor struct {
// contains filtered or unexported fields
}
A Cursor describes a node encountered during Apply. Information about the node and its parent is available from the Node, Parent, Name, and Index methods.
If p is a variable of type and value of the current parent node c.Parent(), and f is the field identifier with name c.Name(), the following invariants hold:
p.f == c.Node() if c.Index() < 0 p.f[c.Index()] == c.Node() if c.Index() >= 0
The methods Replace, Delete, InsertBefore, and InsertAfter can be used to change the AST without disrupting Apply.
func (*Cursor) Delete ¶
func (c *Cursor) Delete()
Delete deletes the current Node from its containing slice. If the current Node is not part of a slice, Delete panics. As a special case, if the current node is a package file, Delete removes it from the package's Files map.
func (*Cursor) Index ¶
Index reports the index >= 0 of the current Node in the slice of Nodes that contains it, or a value < 0 if the current Node is not part of a slice. The index of the current node changes if InsertBefore is called while processing the current node.
func (*Cursor) InsertAfter ¶
InsertAfter inserts n after the current Node in its containing slice. If the current Node is not part of a slice, InsertAfter panics. Apply does not walk n.
func (*Cursor) InsertBefore ¶
InsertBefore inserts n before the current Node in its containing slice. If the current Node is not part of a slice, InsertBefore panics. Apply will not walk n.
func (*Cursor) Name ¶
Name returns the name of the parent Node field that contains the current Node. If the parent is a *ast.Package and the current Node is a *ast.File, Name returns the filename for the current Node.
Source Files