TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2025-05-03 23:02:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

compiler: make ForEachFile accept a package

Browse source 
Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
This commit is contained in:
Evgenii Stratonikov 2020-07-28 18:40:41 +03:00
parent ac040a6f22
commit b8d7e93459
3 changed files with 37 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

5
pkg/compiler/compiler.go
View file

@ -6,6 +6,7 @@ import (
"fmt"
"go/ast"
"go/parser"
"go/types"
"io"
"io/ioutil"
"os"
@ -44,11 +45,11 @@ type buildInfo struct {
}
// ForEachFile executes fn on each file used in current program.
func (c *codegen) ForEachFile(fn func(*ast.File)) {
func (c *codegen) ForEachFile(fn func(*ast.File, *types.Package)) {
for _, pkg := range c.buildInfo.program.AllPackages {
c.typeInfo = &pkg.Info
for _, f := range pkg.Files {
fn(f)
fn(f, pkg.Pkg)
}
}
}