decorator

Documentation

Index

• func Decorate(fset *token.FileSet, n ast.Node) (dst.Node, error)
• func DecorateFile(fset *token.FileSet, f *ast.File) (*dst.File, error)
• func Fprint(w io.Writer, f *dst.File) error
• func Parse(src interface{}) (*dst.File, error)
• func ParseDir(fset *token.FileSet, dir string, filter func(os.FileInfo) bool, ...) (map[string]*dst.Package, error)
• func ParseFile(fset *token.FileSet, filename string, src interface{}, mode parser.Mode) (*dst.File, error)
• func Print(f *dst.File) error
• func Restore(file *dst.File) (*token.FileSet, *ast.File, error)
• type AstMap
• type Decorator
  • func New(fset *token.FileSet) *Decorator
  • func NewWithImports(pkg *packages.Package) *Decorator
  • func (d *Decorator) DecorateFile(f *ast.File) (*dst.File, error)
  • func (d *Decorator) DecorateNode(n ast.Node) (dst.Node, error)
  • func (d *Decorator) Parse(src interface{}) (*dst.File, error)
  • func (d *Decorator) ParseDir(dir string, filter func(os.FileInfo) bool, mode parser.Mode) (map[string]*dst.Package, error)
  • func (d *Decorator) ParseFile(filename string, src interface{}, mode parser.Mode) (*dst.File, error)
• type DstMap
• type FileRestorer
  • func (fr *FileRestorer) Restore() (*ast.File, error)
• type Map
• type Package
  • func Load(cfg *packages.Config, patterns ...string) ([]*Package, error)
  • func (p *Package) Save() error
• type Restorer
  • func NewRestorer() *Restorer
  • func NewRestorerWithImports(path, dir string) *Restorer
  • func (pr *Restorer) FileRestorer(name string, file *dst.File) *FileRestorer
  • func (pr *Restorer) Fprint(w io.Writer, f *dst.File) error
  • func (pr *Restorer) Print(f *dst.File) error
  • func (pr *Restorer) RestoreFile(name string, file *dst.File) (*ast.File, error)

Functions

func Decorate

func Decorate(fset *token.FileSet, n ast.Node) (dst.Node, error)

Decorate decorates an ast.Node and returns a dst.Node.

func DecorateFile

func DecorateFile(fset *token.FileSet, f *ast.File) (*dst.File, error)

Decorate decorates a *ast.File and returns a *dst.File.

func Fprint

func Fprint(w io.Writer, f *dst.File) error

Fprint uses format.Node to print a *dst.File to a writer

func Parse

func Parse(src interface{}) (*dst.File, error)

Parse uses parser.ParseFile to parse and decorate a Go source file. The src parameter should be string, []byte, or io.Reader.

func ParseDir

func ParseDir(fset *token.FileSet, dir string, filter func(os.FileInfo) bool, mode parser.Mode) (map[string]*dst.Package, error)

ParseDir uses parser.ParseDir to parse and decorate a directory containing Go source. The ParseComments flag is added to mode if it doesn't exist.

func ParseFile

func ParseFile(fset *token.FileSet, filename string, src interface{}, mode parser.Mode) (*dst.File, error)

ParseFile uses parser.ParseFile to parse and decorate a Go source file. The ParseComments flag is added to mode if it doesn't exist.

func Print

func Print(f *dst.File) error

Print uses format.Node to print a *dst.File to stdout

func Restore

func Restore(file *dst.File) (*token.FileSet, *ast.File, error)

Restore restores a *dst.File to a *token.FileSet and a *ast.File

Types

type AstMap

type AstMap struct {
	Nodes   map[dst.Node]ast.Node       // Mapping from dst to ast Nodes
	Objects map[*dst.Object]*ast.Object // Mapping from dst to ast Objects
	Scopes  map[*dst.Scope]*ast.Scope   // Mapping from dst to ast Scopes
}

AstMap holds a record of the mapping from dst to ast nodes, objects and scopes.

type Decorator

type Decorator struct {
	Map
	Filenames map[*dst.File]string // Source file names
	Fset      *token.FileSet

	// If a Resolver is provided, it is used to resolve remote identifiers from *ast.Ident and
	// *ast.SelectorExpr nodes. Usually a remote identifier is a SelectorExpr qualified identifier,
	// but in the case of dot-imports they can be simply Ident nodes. During decoration, remote
	// identifiers are replaced with *dst.Ident with Path set to the path of imported package.
	Resolver resolver.IdentResolver
	// Local package path - required if Resolver is set.
	Path string
}

func New

func New(fset *token.FileSet) *Decorator

New returns a new decorator

func NewWithImports

func NewWithImports(pkg *packages.Package) *Decorator

NewWithImports returns a new package decorator with import management attributes set.

func (*Decorator) DecorateFile

func (d *Decorator) DecorateFile(f *ast.File) (*dst.File, error)

func (*Decorator) DecorateNode

func (d *Decorator) DecorateNode(n ast.Node) (dst.Node, error)

Decorate decorates an ast.Node and returns a dst.Node

func (*Decorator) Parse

func (d *Decorator) Parse(src interface{}) (*dst.File, error)

Parse uses parser.ParseFile to parse and decorate a Go source file. The src parameter should be string, []byte, or io.Reader.

func (*Decorator) ParseDir

func (d *Decorator) ParseDir(dir string, filter func(os.FileInfo) bool, mode parser.Mode) (map[string]*dst.Package, error)

ParseDir uses parser.ParseDir to parse and decorate a directory containing Go source. The ParseComments flag is added to mode if it doesn't exist.

func (*Decorator) ParseFile

func (d *Decorator) ParseFile(filename string, src interface{}, mode parser.Mode) (*dst.File, error)

ParseFile uses parser.ParseFile to parse and decorate a Go source file. The ParseComments flag is added to mode if it doesn't exist.

type DstMap

type DstMap struct {
	Nodes   map[ast.Node]dst.Node       // Mapping from ast to dst Nodes
	Objects map[*ast.Object]*dst.Object // Mapping from ast to dst Objects
	Scopes  map[*ast.Scope]*dst.Scope   // Mapping from ast to dst Scopes
}

DstMap holds a record of the mapping from ast to dst nodes, objects and scopes.

type FileRestorer

type FileRestorer struct {
	*Restorer
	Alias map[string]string // map of package path -> package alias for imports
	// contains filtered or unexported fields
}

func (*FileRestorer) Restore

func (fr *FileRestorer) Restore() (*ast.File, error)

type Map

type Map struct {
	Ast AstMap
	Dst DstMap
}

Map holds a record of the mapping between ast and dst nodes, objects and scopes.

type Package

type Package struct {
	*packages.Package
	Dir       string
	Decorator *Decorator
	Files     []*dst.File
}

func Load

func Load(cfg *packages.Config, patterns ...string) ([]*Package, error)

func (*Package) Save

func (p *Package) Save() error

type Restorer

type Restorer struct {
	Map
	Fset *token.FileSet // Fset is the *token.FileSet in use. Set this to use a pre-existing FileSet.

	// If a Resolver is provided, the names of all imported packages are resolved, and the imports
	// block is updated. All remote identifiers are updated (sometimes this involves changing
	// SelectorExpr.X.Name, or even swapping between Ident and SelectorExpr). To force specific
	// import alias names, use the FileRestorer.Alias map.
	Resolver resolver.PackageResolver
	// Local package path - required if Resolver is set.
	Path string
}

func NewRestorer

func NewRestorer() *Restorer

NewRestorer returns a restorer.

func NewRestorerWithImports

func NewRestorerWithImports(path, dir string) *Restorer

NewRestorerWithImports returns a restorer with import management attributes set.

func (*Restorer) FileRestorer

func (pr *Restorer) FileRestorer(name string, file *dst.File) *FileRestorer

func (*Restorer) Fprint

func (pr *Restorer) Fprint(w io.Writer, f *dst.File) error

Fprint uses format.Node to print a *dst.File to a writer

func (*Restorer) Print

func (pr *Restorer) Print(f *dst.File) error

Print uses format.Node to print a *dst.File to stdout

func (*Restorer) RestoreFile

func (pr *Restorer) RestoreFile(name string, file *dst.File) (*ast.File, error)

RestoreFile restores a *dst.File to an *ast.File