tools: Index | Files

package astutil

import ""

Package astutil contains common utilities for working with the Go AST.


Package Files

enclosing.go imports.go rewrite.go util.go

func AddImport Uses

func AddImport(fset *token.FileSet, f *ast.File, ipath string) (added bool)

AddImport adds the import path to the file f, if absent.

func AddNamedImport Uses

func AddNamedImport(fset *token.FileSet, f *ast.File, name, ipath string) (added bool)

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")


import pathpkg "path"

func Apply Uses

func Apply(root ast.Node, pre, post ApplyFunc) (result ast.Node)

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 Uses

func DeleteImport(fset *token.FileSet, f *ast.File, path string) (deleted bool)

DeleteImport deletes the import path from the file f, if present.

func DeleteNamedImport Uses

func DeleteNamedImport(fset *token.FileSet, f *ast.File, name, path string) (deleted bool)

DeleteNamedImport deletes the import with the given name and path from the file f, if present.

func Imports Uses

func Imports(fset *token.FileSet, f *ast.File) [][]*ast.ImportSpec

Imports returns the file imports grouped by paragraph.

func NodeDescription Uses

func NodeDescription(n ast.Node) string

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 Uses

func PathEnclosingInterval(root *ast.File, start, end token.Pos) (path []ast.Node, exact bool)

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

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-->

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 Uses

func RewriteImport(fset *token.FileSet, f *ast.File, oldPath, newPath string) (rewrote bool)

RewriteImport rewrites any import of path oldPath to path newPath.

func Unparen Uses

func Unparen(e ast.Expr) ast.Expr

Unparen returns e with any enclosing parentheses stripped.

func UsesImport Uses

func UsesImport(f *ast.File, path string) (used bool)

UsesImport reports whether a given import is used.

type ApplyFunc Uses

type ApplyFunc func(*Cursor) bool

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 Uses

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 Uses

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 Uses

func (c *Cursor) Index() int

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 Uses

func (c *Cursor) InsertAfter(n ast.Node)

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 Uses

func (c *Cursor) InsertBefore(n ast.Node)

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 Uses

func (c *Cursor) Name() string

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.

func (*Cursor) Node Uses

func (c *Cursor) Node() ast.Node

Node returns the current Node.

func (*Cursor) Parent Uses

func (c *Cursor) Parent() ast.Node

Parent returns the parent of the current Node.

func (*Cursor) Replace Uses

func (c *Cursor) Replace(n ast.Node)

Replace replaces the current Node with n. The replacement node is not walked by Apply.

Package astutil imports 7 packages (graph) and is imported by 101 packages. Updated 2017-12-12. Refresh now. Tools for package owners.