decorator

Documentation

Index

• func Decorate(fset *token.FileSet, n ast.Node) dst.Node
• func DecorateFile(fset *token.FileSet, f *ast.File) *dst.File
• 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)
• 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
  • func (d *Decorator) DecorateNode(n ast.Node) dst.Node
  • 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
• 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

Functions

func Decorate

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

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

func DecorateFile

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

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)

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
}

type Decorator

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

	Path string // local package path, used to ensure the local path is not set in idents

	// If a Resolver is provided, it is used to resolve Ident nodes. During decoration, remote
	// identifiers (e.g. usually part of a qualified identifier SelectorExpr, but sometimes on
	// their own for dot-imported packages) are updated with the path of the package they are
	// imported from.
	Resolver resolver.IdentResolver
	// contains filtered or unexported fields
}

func New

func New(fset *token.FileSet) *Decorator

New returns a new package 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

func (*Decorator) DecorateNode

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

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
}

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

type Map

type Map struct {
	Ast AstMap
	Dst DstMap
}

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
}

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

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