func (this *SingleMover) CollectUnexportedObjs() { this.unexportedObjs = make(map[*ast.Object]bool) for node, obj := range this.moveNodes { //printer.Fprint(os.Stdout, token.NewFileSet(), node) //fmt.Printf("\n%v %T\n", obj, node) if !unicode.IsUpper(utf8.NewString(obj.Name).At(0)) { this.unexportedObjs[obj] = true } if ts, ok := node.(*ast.TypeSpec); ok { if st, ok := ts.Type.(*ast.StructType); ok { for _, field := range st.Fields.List { for _, name := range field.Names { if !name.IsExported() { obj, _ := types.ExprType(name, LocalImporter) this.unexportedObjs[obj] = true } } } } } } return }
func (this *FieldDeclFinder) Visit(node ast.Node) ast.Visitor { if this.Obj != nil { return nil } switch n := node.(type) { case *ast.BlockStmt: return nil case *ast.TypeSpec: if n.Name.Name == this.typename { if st, ok := n.Type.(*ast.StructType); ok { for _, field := range st.Fields.List { for _, nid := range field.Names { if nid.Name == this.newname { this.NameExists = true return nil } if nid.Name == this.oldname { nid.Name = this.newname this.Obj, _ = types.ExprType(nid, LocalImporter) this.Updated = true this.Name = nid return nil } } } } return this } return nil } return this }
func (this DepthWalker) Visit(node ast.Node) ast.Visitor { if node == nil { return this + 1 } buffer := "" for i := 0; i < int(this); i++ { buffer += " " } fmt.Printf("%sPos: %d %s\n", buffer, node.Pos(), AllSources.Position(node.Pos())) fmt.Printf("%sEnd: %d %s\n", buffer, node.End(), AllSources.Position(node.End())) fmt.Printf("%s%T\n", buffer, node) fmt.Printf("%s%v\n", buffer, node) if e, ok := node.(ast.Expr); ok { obj, typ := types.ExprType(e, LocalImporter) fmt.Printf("%s%v\n", buffer, obj) fmt.Printf("%s%v\n", buffer, typ) } fmt.Println() switch n := node.(type) { } return this + 1 }
func (this *RenameWalker) Visit(node ast.Node) ast.Visitor { switch n := node.(type) { case *ast.Ident: obj, _ := types.ExprType(n, LocalImporter) if obj == this.Obj { this.Updated = true n.Name = this.NewName } case *ast.SelectorExpr: obj, _ := types.ExprType(n, LocalImporter) if obj == this.Obj { this.Updated = true n.Sel.Name = this.NewName } } return this }
func (this *DeclFinder) Visit(node ast.Node) ast.Visitor { if this.Obj != nil { return nil } switch n := node.(type) { case *ast.BlockStmt: return nil case *ast.ValueSpec: for _, name := range n.Names { if name.Name == this.newname { this.NameExists = true return nil } if name.Name == this.oldname { //this.Name = name this.Obj, _ = types.ExprType(name, LocalImporter) this.Node = node return nil } } return nil case *ast.FuncDecl: if n.Name.Name == this.newname { this.NameExists = true } if n.Name.Name == this.oldname { //this.Name = n.Name this.Node = node this.Obj, _ = types.ExprType(n.Name, LocalImporter) } return nil case *ast.TypeSpec: if n.Name.Name == this.newname { this.NameExists = true } if n.Name.Name == this.oldname { //this.Name = n.Name this.Node = node this.Obj, _ = types.ExprType(n.Name, LocalImporter) } return nil } return this }
func (this *PkgChanger) Visit(node ast.Node) ast.Visitor { if this.Renamed { return nil } if node == nil { return this } switch n := node.(type) { case *ast.ImportSpec: if string(n.Path.Value) == QuotePath(this.path) { if n.Name != nil { this.Renamed = true return nil } if n.Name == nil && this.newname != this.pkgname { n.Name = &ast.Ident{ Name: this.newname, NamePos: n.Pos(), } this.Updated = true } } case *ast.Ident: if n.Name == this.oldname { obj, typ := types.ExprType(n, LocalImporter) if obj == nil { n.Name = this.newname this.Updated = true } if obj != nil { if typ.Kind == ast.Pkg && obj.Name == this.oldname { n.Name = this.newname this.Updated = true } } } } return this }
func (this *SingleMover) UpdateOther() (err os.Error) { for _, path := range ImportedBy[QuotePath(this.oldpath)] { opkg := LocalImporter(path) for fpath, file := range opkg.Files { rw := ReferenceWalker{ UnexportedObjs: make(map[*ast.Object]bool), MoveObjs: this.moveObjs, SkipNodes: make(map[ast.Node]*ast.Object), SkipNodeParents: make(map[ast.Node]ast.Node), GoodReferenceParents: make(map[ast.Node]ast.Node), BadReferences: &[]ast.Node{}, } ast.Walk(&rw, file) if len(rw.GoodReferenceParents) == 0 { continue } newpkgname := GetUniqueIdent([]*ast.File{file}, this.pkg.Name) //construct the import nis := &ast.ImportSpec{ Name: &ast.Ident{Name: newpkgname}, Path: &ast.BasicLit{ Kind: token.STRING, Value: QuotePath(this.newpath), }, } ngd := &ast.GenDecl{ Tok: token.IMPORT, Specs: []ast.Spec{nis}, } file.Decls = append([]ast.Decl{ngd}, file.Decls...) for node, parent := range rw.GoodReferenceParents { getSel := func(sel *ast.SelectorExpr) *ast.SelectorExpr { obj, _ := types.ExprType(sel.X, LocalImporter) if obj.Kind == ast.Pkg { return &ast.SelectorExpr{ X: &ast.Ident{ Name: newpkgname, NamePos: sel.X.Pos(), }, Sel: sel.Sel, } } return sel } switch p := parent.(type) { case *ast.CallExpr: if sel, ok := node.(*ast.SelectorExpr); ok { p.Fun = getSel(sel) } else { return MakeErr("CallExpr w/ unexpected type %T\n", node) } case *ast.AssignStmt: for i, x := range p.Lhs { if x == node { if sel, ok := x.(*ast.SelectorExpr); ok { p.Lhs[i] = getSel(sel) } } } for i, x := range p.Rhs { if x == node { if sel, ok := x.(*ast.SelectorExpr); ok { p.Rhs[i] = getSel(sel) } } } case *ast.ValueSpec: if node == p.Type { if sel, ok := p.Type.(*ast.SelectorExpr); ok { p.Type = getSel(sel) } } for i, x := range p.Values { if x == node { if sel, ok := x.(*ast.SelectorExpr); ok { p.Values[i] = getSel(sel) } } } case *ast.StarExpr: if p.X == node { if sel, ok := p.X.(*ast.SelectorExpr); ok { p.X = getSel(sel) } } default: printer.Fprint(os.Stdout, AllSources, parent) return MakeErr("Unexpected remote parent %T\n", parent) } } //now that we've renamed some references, do we still need to import oldpath? oc := ObjChecker{ Objs: this.remainingObjs, } ast.Walk(&oc, file) if !oc.Found { ast.Walk(&ImportRemover{nil, this.oldpath}, file) } err = RewriteSource(fpath, file) if err != nil { return } } } return }
func (this *SingleMover) CreateNewSource() (err os.Error) { liw := make(ListImportWalker) for n := range this.moveNodes { ast.Walk(liw, n) } finalImports := make(map[*ast.ImportSpec]bool) for obj, is := range liw { if _, ok := this.moveObjs[obj]; !ok { finalImports[is] = true } } newfile := &ast.File{ Name: &ast.Ident{Name: this.pkg.Name}, } if len(finalImports) != 0 { for is := range finalImports { gdl := &ast.GenDecl{ Tok: token.IMPORT, Specs: []ast.Spec{is}, } newfile.Decls = append(newfile.Decls, gdl) } } var sortedNodes NodeSorter for mn := range this.moveNodes { sortedNodes = append(sortedNodes, mn) } sort.Sort(sortedNodes) for _, mn := range sortedNodes { switch m := mn.(type) { case ast.Decl: newfile.Decls = append(newfile.Decls, m) case *ast.TypeSpec: gdl := &ast.GenDecl{ Tok: token.TYPE, Specs: []ast.Spec{m}, } newfile.Decls = append(newfile.Decls, gdl) } } npf := ExprParentFinder{ ExprParents: make(map[ast.Expr]ast.Node), } for n := range this.moveNodes { ast.Walk(&npf, n) } var pkgfiles []*ast.File for _, pkgfile := range this.pkg.Files { pkgfiles = append(pkgfiles, pkgfile) } oldPkgNewName := GetUniqueIdent(pkgfiles, this.pkg.Name) needOldImport := false this.referenceBack = false for expr, parent := range npf.ExprParents { obj, _ := types.ExprType(expr, LocalImporter) if _, ok := this.moveObjs[obj]; ok { continue } if _, ok := this.allObjs[obj]; !ok { continue } if !unicode.IsUpper(utf8.NewString(obj.Name).At(0)) { position := AllSources.Position(expr.Pos()) fmt.Printf("At %v ", position) printer.Fprint(os.Stdout, token.NewFileSet(), expr) fmt.Println() err = MakeErr("Can't move code that references unexported objects") return } needOldImport = true this.referenceBack = true getSel := func(idn *ast.Ident) *ast.SelectorExpr { return &ast.SelectorExpr{ X: &ast.Ident{ Name: oldPkgNewName, NamePos: idn.NamePos, }, Sel: idn, } } switch p := parent.(type) { case *ast.CallExpr: if idn, ok := expr.(*ast.Ident); ok { p.Fun = getSel(idn) } else { err = MakeErr("CallExpr w/ unexpected type %T\n", expr) return } case *ast.AssignStmt: for i, x := range p.Lhs { if x == expr { if idn, ok := x.(*ast.Ident); ok { p.Lhs[i] = getSel(idn) } } } for i, x := range p.Rhs { if x == expr { if idn, ok := x.(*ast.Ident); ok { p.Rhs[i] = getSel(idn) } } } default: err = MakeErr("Unexpected parent %T\n", parent) return } } if needOldImport { is := &ast.ImportSpec{ Name: &ast.Ident{Name: oldPkgNewName}, Path: &ast.BasicLit{Value: QuotePath(this.oldpath)}, } gdl := &ast.GenDecl{ Tok: token.IMPORT, Specs: []ast.Spec{is}, } newfile.Decls = append([]ast.Decl{gdl}, newfile.Decls...) } err = os.MkdirAll(this.newpath, 0755) if err != nil { return } newSourcePath := filepath.Join(this.newpath, this.pkg.Name+".go") containedComments := make(CommentCollector) for node := range this.moveNodes { ast.Walk(containedComments, node) } for _, file := range this.pkg.Files { for i := len(file.Comments) - 1; i >= 0; i-- { cg := file.Comments[i] add := func() { newfile.Comments = append([]*ast.CommentGroup{cg}, newfile.Comments...) file.Comments[i] = file.Comments[len(file.Comments)-1] file.Comments = file.Comments[:len(file.Comments)-1] } if containedComments[cg] { add() } else { for node := range this.moveNodes { if node.Pos() <= cg.Pos() && node.End() >= cg.End() { add() break } } } } } err = NewSource(newSourcePath, newfile) if err != nil { return } return }