github.com/yanyiwu/go@v0.0.0-20150106053140-03d6637dbb7f/src/cmd/fix/fix.go (about) 1 // Copyright 2011 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 package main 6 7 import ( 8 "fmt" 9 "go/ast" 10 "go/parser" 11 "go/token" 12 "os" 13 "path" 14 "reflect" 15 "strconv" 16 "strings" 17 ) 18 19 type fix struct { 20 name string 21 date string // date that fix was introduced, in YYYY-MM-DD format 22 f func(*ast.File) bool 23 desc string 24 } 25 26 // main runs sort.Sort(byName(fixes)) before printing list of fixes. 27 type byName []fix 28 29 func (f byName) Len() int { return len(f) } 30 func (f byName) Swap(i, j int) { f[i], f[j] = f[j], f[i] } 31 func (f byName) Less(i, j int) bool { return f[i].name < f[j].name } 32 33 // main runs sort.Sort(byDate(fixes)) before applying fixes. 34 type byDate []fix 35 36 func (f byDate) Len() int { return len(f) } 37 func (f byDate) Swap(i, j int) { f[i], f[j] = f[j], f[i] } 38 func (f byDate) Less(i, j int) bool { return f[i].date < f[j].date } 39 40 var fixes []fix 41 42 func register(f fix) { 43 fixes = append(fixes, f) 44 } 45 46 // walk traverses the AST x, calling visit(y) for each node y in the tree but 47 // also with a pointer to each ast.Expr, ast.Stmt, and *ast.BlockStmt, 48 // in a bottom-up traversal. 49 func walk(x interface{}, visit func(interface{})) { 50 walkBeforeAfter(x, nop, visit) 51 } 52 53 func nop(interface{}) {} 54 55 // walkBeforeAfter is like walk but calls before(x) before traversing 56 // x's children and after(x) afterward. 57 func walkBeforeAfter(x interface{}, before, after func(interface{})) { 58 before(x) 59 60 switch n := x.(type) { 61 default: 62 panic(fmt.Errorf("unexpected type %T in walkBeforeAfter", x)) 63 64 case nil: 65 66 // pointers to interfaces 67 case *ast.Decl: 68 walkBeforeAfter(*n, before, after) 69 case *ast.Expr: 70 walkBeforeAfter(*n, before, after) 71 case *ast.Spec: 72 walkBeforeAfter(*n, before, after) 73 case *ast.Stmt: 74 walkBeforeAfter(*n, before, after) 75 76 // pointers to struct pointers 77 case **ast.BlockStmt: 78 walkBeforeAfter(*n, before, after) 79 case **ast.CallExpr: 80 walkBeforeAfter(*n, before, after) 81 case **ast.FieldList: 82 walkBeforeAfter(*n, before, after) 83 case **ast.FuncType: 84 walkBeforeAfter(*n, before, after) 85 case **ast.Ident: 86 walkBeforeAfter(*n, before, after) 87 case **ast.BasicLit: 88 walkBeforeAfter(*n, before, after) 89 90 // pointers to slices 91 case *[]ast.Decl: 92 walkBeforeAfter(*n, before, after) 93 case *[]ast.Expr: 94 walkBeforeAfter(*n, before, after) 95 case *[]*ast.File: 96 walkBeforeAfter(*n, before, after) 97 case *[]*ast.Ident: 98 walkBeforeAfter(*n, before, after) 99 case *[]ast.Spec: 100 walkBeforeAfter(*n, before, after) 101 case *[]ast.Stmt: 102 walkBeforeAfter(*n, before, after) 103 104 // These are ordered and grouped to match ../../go/ast/ast.go 105 case *ast.Field: 106 walkBeforeAfter(&n.Names, before, after) 107 walkBeforeAfter(&n.Type, before, after) 108 walkBeforeAfter(&n.Tag, before, after) 109 case *ast.FieldList: 110 for _, field := range n.List { 111 walkBeforeAfter(field, before, after) 112 } 113 case *ast.BadExpr: 114 case *ast.Ident: 115 case *ast.Ellipsis: 116 walkBeforeAfter(&n.Elt, before, after) 117 case *ast.BasicLit: 118 case *ast.FuncLit: 119 walkBeforeAfter(&n.Type, before, after) 120 walkBeforeAfter(&n.Body, before, after) 121 case *ast.CompositeLit: 122 walkBeforeAfter(&n.Type, before, after) 123 walkBeforeAfter(&n.Elts, before, after) 124 case *ast.ParenExpr: 125 walkBeforeAfter(&n.X, before, after) 126 case *ast.SelectorExpr: 127 walkBeforeAfter(&n.X, before, after) 128 case *ast.IndexExpr: 129 walkBeforeAfter(&n.X, before, after) 130 walkBeforeAfter(&n.Index, before, after) 131 case *ast.SliceExpr: 132 walkBeforeAfter(&n.X, before, after) 133 if n.Low != nil { 134 walkBeforeAfter(&n.Low, before, after) 135 } 136 if n.High != nil { 137 walkBeforeAfter(&n.High, before, after) 138 } 139 case *ast.TypeAssertExpr: 140 walkBeforeAfter(&n.X, before, after) 141 walkBeforeAfter(&n.Type, before, after) 142 case *ast.CallExpr: 143 walkBeforeAfter(&n.Fun, before, after) 144 walkBeforeAfter(&n.Args, before, after) 145 case *ast.StarExpr: 146 walkBeforeAfter(&n.X, before, after) 147 case *ast.UnaryExpr: 148 walkBeforeAfter(&n.X, before, after) 149 case *ast.BinaryExpr: 150 walkBeforeAfter(&n.X, before, after) 151 walkBeforeAfter(&n.Y, before, after) 152 case *ast.KeyValueExpr: 153 walkBeforeAfter(&n.Key, before, after) 154 walkBeforeAfter(&n.Value, before, after) 155 156 case *ast.ArrayType: 157 walkBeforeAfter(&n.Len, before, after) 158 walkBeforeAfter(&n.Elt, before, after) 159 case *ast.StructType: 160 walkBeforeAfter(&n.Fields, before, after) 161 case *ast.FuncType: 162 walkBeforeAfter(&n.Params, before, after) 163 if n.Results != nil { 164 walkBeforeAfter(&n.Results, before, after) 165 } 166 case *ast.InterfaceType: 167 walkBeforeAfter(&n.Methods, before, after) 168 case *ast.MapType: 169 walkBeforeAfter(&n.Key, before, after) 170 walkBeforeAfter(&n.Value, before, after) 171 case *ast.ChanType: 172 walkBeforeAfter(&n.Value, before, after) 173 174 case *ast.BadStmt: 175 case *ast.DeclStmt: 176 walkBeforeAfter(&n.Decl, before, after) 177 case *ast.EmptyStmt: 178 case *ast.LabeledStmt: 179 walkBeforeAfter(&n.Stmt, before, after) 180 case *ast.ExprStmt: 181 walkBeforeAfter(&n.X, before, after) 182 case *ast.SendStmt: 183 walkBeforeAfter(&n.Chan, before, after) 184 walkBeforeAfter(&n.Value, before, after) 185 case *ast.IncDecStmt: 186 walkBeforeAfter(&n.X, before, after) 187 case *ast.AssignStmt: 188 walkBeforeAfter(&n.Lhs, before, after) 189 walkBeforeAfter(&n.Rhs, before, after) 190 case *ast.GoStmt: 191 walkBeforeAfter(&n.Call, before, after) 192 case *ast.DeferStmt: 193 walkBeforeAfter(&n.Call, before, after) 194 case *ast.ReturnStmt: 195 walkBeforeAfter(&n.Results, before, after) 196 case *ast.BranchStmt: 197 case *ast.BlockStmt: 198 walkBeforeAfter(&n.List, before, after) 199 case *ast.IfStmt: 200 walkBeforeAfter(&n.Init, before, after) 201 walkBeforeAfter(&n.Cond, before, after) 202 walkBeforeAfter(&n.Body, before, after) 203 walkBeforeAfter(&n.Else, before, after) 204 case *ast.CaseClause: 205 walkBeforeAfter(&n.List, before, after) 206 walkBeforeAfter(&n.Body, before, after) 207 case *ast.SwitchStmt: 208 walkBeforeAfter(&n.Init, before, after) 209 walkBeforeAfter(&n.Tag, before, after) 210 walkBeforeAfter(&n.Body, before, after) 211 case *ast.TypeSwitchStmt: 212 walkBeforeAfter(&n.Init, before, after) 213 walkBeforeAfter(&n.Assign, before, after) 214 walkBeforeAfter(&n.Body, before, after) 215 case *ast.CommClause: 216 walkBeforeAfter(&n.Comm, before, after) 217 walkBeforeAfter(&n.Body, before, after) 218 case *ast.SelectStmt: 219 walkBeforeAfter(&n.Body, before, after) 220 case *ast.ForStmt: 221 walkBeforeAfter(&n.Init, before, after) 222 walkBeforeAfter(&n.Cond, before, after) 223 walkBeforeAfter(&n.Post, before, after) 224 walkBeforeAfter(&n.Body, before, after) 225 case *ast.RangeStmt: 226 walkBeforeAfter(&n.Key, before, after) 227 walkBeforeAfter(&n.Value, before, after) 228 walkBeforeAfter(&n.X, before, after) 229 walkBeforeAfter(&n.Body, before, after) 230 231 case *ast.ImportSpec: 232 case *ast.ValueSpec: 233 walkBeforeAfter(&n.Type, before, after) 234 walkBeforeAfter(&n.Values, before, after) 235 walkBeforeAfter(&n.Names, before, after) 236 case *ast.TypeSpec: 237 walkBeforeAfter(&n.Type, before, after) 238 239 case *ast.BadDecl: 240 case *ast.GenDecl: 241 walkBeforeAfter(&n.Specs, before, after) 242 case *ast.FuncDecl: 243 if n.Recv != nil { 244 walkBeforeAfter(&n.Recv, before, after) 245 } 246 walkBeforeAfter(&n.Type, before, after) 247 if n.Body != nil { 248 walkBeforeAfter(&n.Body, before, after) 249 } 250 251 case *ast.File: 252 walkBeforeAfter(&n.Decls, before, after) 253 254 case *ast.Package: 255 walkBeforeAfter(&n.Files, before, after) 256 257 case []*ast.File: 258 for i := range n { 259 walkBeforeAfter(&n[i], before, after) 260 } 261 case []ast.Decl: 262 for i := range n { 263 walkBeforeAfter(&n[i], before, after) 264 } 265 case []ast.Expr: 266 for i := range n { 267 walkBeforeAfter(&n[i], before, after) 268 } 269 case []*ast.Ident: 270 for i := range n { 271 walkBeforeAfter(&n[i], before, after) 272 } 273 case []ast.Stmt: 274 for i := range n { 275 walkBeforeAfter(&n[i], before, after) 276 } 277 case []ast.Spec: 278 for i := range n { 279 walkBeforeAfter(&n[i], before, after) 280 } 281 } 282 after(x) 283 } 284 285 // imports returns true if f imports path. 286 func imports(f *ast.File, path string) bool { 287 return importSpec(f, path) != nil 288 } 289 290 // importSpec returns the import spec if f imports path, 291 // or nil otherwise. 292 func importSpec(f *ast.File, path string) *ast.ImportSpec { 293 for _, s := range f.Imports { 294 if importPath(s) == path { 295 return s 296 } 297 } 298 return nil 299 } 300 301 // importPath returns the unquoted import path of s, 302 // or "" if the path is not properly quoted. 303 func importPath(s *ast.ImportSpec) string { 304 t, err := strconv.Unquote(s.Path.Value) 305 if err == nil { 306 return t 307 } 308 return "" 309 } 310 311 // declImports reports whether gen contains an import of path. 312 func declImports(gen *ast.GenDecl, path string) bool { 313 if gen.Tok != token.IMPORT { 314 return false 315 } 316 for _, spec := range gen.Specs { 317 impspec := spec.(*ast.ImportSpec) 318 if importPath(impspec) == path { 319 return true 320 } 321 } 322 return false 323 } 324 325 // isPkgDot returns true if t is the expression "pkg.name" 326 // where pkg is an imported identifier. 327 func isPkgDot(t ast.Expr, pkg, name string) bool { 328 sel, ok := t.(*ast.SelectorExpr) 329 return ok && isTopName(sel.X, pkg) && sel.Sel.String() == name 330 } 331 332 // isPtrPkgDot returns true if f is the expression "*pkg.name" 333 // where pkg is an imported identifier. 334 func isPtrPkgDot(t ast.Expr, pkg, name string) bool { 335 ptr, ok := t.(*ast.StarExpr) 336 return ok && isPkgDot(ptr.X, pkg, name) 337 } 338 339 // isTopName returns true if n is a top-level unresolved identifier with the given name. 340 func isTopName(n ast.Expr, name string) bool { 341 id, ok := n.(*ast.Ident) 342 return ok && id.Name == name && id.Obj == nil 343 } 344 345 // isName returns true if n is an identifier with the given name. 346 func isName(n ast.Expr, name string) bool { 347 id, ok := n.(*ast.Ident) 348 return ok && id.String() == name 349 } 350 351 // isCall returns true if t is a call to pkg.name. 352 func isCall(t ast.Expr, pkg, name string) bool { 353 call, ok := t.(*ast.CallExpr) 354 return ok && isPkgDot(call.Fun, pkg, name) 355 } 356 357 // If n is an *ast.Ident, isIdent returns it; otherwise isIdent returns nil. 358 func isIdent(n interface{}) *ast.Ident { 359 id, _ := n.(*ast.Ident) 360 return id 361 } 362 363 // refersTo returns true if n is a reference to the same object as x. 364 func refersTo(n ast.Node, x *ast.Ident) bool { 365 id, ok := n.(*ast.Ident) 366 // The test of id.Name == x.Name handles top-level unresolved 367 // identifiers, which all have Obj == nil. 368 return ok && id.Obj == x.Obj && id.Name == x.Name 369 } 370 371 // isBlank returns true if n is the blank identifier. 372 func isBlank(n ast.Expr) bool { 373 return isName(n, "_") 374 } 375 376 // isEmptyString returns true if n is an empty string literal. 377 func isEmptyString(n ast.Expr) bool { 378 lit, ok := n.(*ast.BasicLit) 379 return ok && lit.Kind == token.STRING && len(lit.Value) == 2 380 } 381 382 func warn(pos token.Pos, msg string, args ...interface{}) { 383 if pos.IsValid() { 384 msg = "%s: " + msg 385 arg1 := []interface{}{fset.Position(pos).String()} 386 args = append(arg1, args...) 387 } 388 fmt.Fprintf(os.Stderr, msg+"\n", args...) 389 } 390 391 // countUses returns the number of uses of the identifier x in scope. 392 func countUses(x *ast.Ident, scope []ast.Stmt) int { 393 count := 0 394 ff := func(n interface{}) { 395 if n, ok := n.(ast.Node); ok && refersTo(n, x) { 396 count++ 397 } 398 } 399 for _, n := range scope { 400 walk(n, ff) 401 } 402 return count 403 } 404 405 // rewriteUses replaces all uses of the identifier x and !x in scope 406 // with f(x.Pos()) and fnot(x.Pos()). 407 func rewriteUses(x *ast.Ident, f, fnot func(token.Pos) ast.Expr, scope []ast.Stmt) { 408 var lastF ast.Expr 409 ff := func(n interface{}) { 410 ptr, ok := n.(*ast.Expr) 411 if !ok { 412 return 413 } 414 nn := *ptr 415 416 // The child node was just walked and possibly replaced. 417 // If it was replaced and this is a negation, replace with fnot(p). 418 not, ok := nn.(*ast.UnaryExpr) 419 if ok && not.Op == token.NOT && not.X == lastF { 420 *ptr = fnot(nn.Pos()) 421 return 422 } 423 if refersTo(nn, x) { 424 lastF = f(nn.Pos()) 425 *ptr = lastF 426 } 427 } 428 for _, n := range scope { 429 walk(n, ff) 430 } 431 } 432 433 // assignsTo returns true if any of the code in scope assigns to or takes the address of x. 434 func assignsTo(x *ast.Ident, scope []ast.Stmt) bool { 435 assigned := false 436 ff := func(n interface{}) { 437 if assigned { 438 return 439 } 440 switch n := n.(type) { 441 case *ast.UnaryExpr: 442 // use of &x 443 if n.Op == token.AND && refersTo(n.X, x) { 444 assigned = true 445 return 446 } 447 case *ast.AssignStmt: 448 for _, l := range n.Lhs { 449 if refersTo(l, x) { 450 assigned = true 451 return 452 } 453 } 454 } 455 } 456 for _, n := range scope { 457 if assigned { 458 break 459 } 460 walk(n, ff) 461 } 462 return assigned 463 } 464 465 // newPkgDot returns an ast.Expr referring to "pkg.name" at position pos. 466 func newPkgDot(pos token.Pos, pkg, name string) ast.Expr { 467 return &ast.SelectorExpr{ 468 X: &ast.Ident{ 469 NamePos: pos, 470 Name: pkg, 471 }, 472 Sel: &ast.Ident{ 473 NamePos: pos, 474 Name: name, 475 }, 476 } 477 } 478 479 // renameTop renames all references to the top-level name old. 480 // It returns true if it makes any changes. 481 func renameTop(f *ast.File, old, new string) bool { 482 var fixed bool 483 484 // Rename any conflicting imports 485 // (assuming package name is last element of path). 486 for _, s := range f.Imports { 487 if s.Name != nil { 488 if s.Name.Name == old { 489 s.Name.Name = new 490 fixed = true 491 } 492 } else { 493 _, thisName := path.Split(importPath(s)) 494 if thisName == old { 495 s.Name = ast.NewIdent(new) 496 fixed = true 497 } 498 } 499 } 500 501 // Rename any top-level declarations. 502 for _, d := range f.Decls { 503 switch d := d.(type) { 504 case *ast.FuncDecl: 505 if d.Recv == nil && d.Name.Name == old { 506 d.Name.Name = new 507 d.Name.Obj.Name = new 508 fixed = true 509 } 510 case *ast.GenDecl: 511 for _, s := range d.Specs { 512 switch s := s.(type) { 513 case *ast.TypeSpec: 514 if s.Name.Name == old { 515 s.Name.Name = new 516 s.Name.Obj.Name = new 517 fixed = true 518 } 519 case *ast.ValueSpec: 520 for _, n := range s.Names { 521 if n.Name == old { 522 n.Name = new 523 n.Obj.Name = new 524 fixed = true 525 } 526 } 527 } 528 } 529 } 530 } 531 532 // Rename top-level old to new, both unresolved names 533 // (probably defined in another file) and names that resolve 534 // to a declaration we renamed. 535 walk(f, func(n interface{}) { 536 id, ok := n.(*ast.Ident) 537 if ok && isTopName(id, old) { 538 id.Name = new 539 fixed = true 540 } 541 if ok && id.Obj != nil && id.Name == old && id.Obj.Name == new { 542 id.Name = id.Obj.Name 543 fixed = true 544 } 545 }) 546 547 return fixed 548 } 549 550 // matchLen returns the length of the longest prefix shared by x and y. 551 func matchLen(x, y string) int { 552 i := 0 553 for i < len(x) && i < len(y) && x[i] == y[i] { 554 i++ 555 } 556 return i 557 } 558 559 // addImport adds the import path to the file f, if absent. 560 func addImport(f *ast.File, ipath string) (added bool) { 561 if imports(f, ipath) { 562 return false 563 } 564 565 // Determine name of import. 566 // Assume added imports follow convention of using last element. 567 _, name := path.Split(ipath) 568 569 // Rename any conflicting top-level references from name to name_. 570 renameTop(f, name, name+"_") 571 572 newImport := &ast.ImportSpec{ 573 Path: &ast.BasicLit{ 574 Kind: token.STRING, 575 Value: strconv.Quote(ipath), 576 }, 577 } 578 579 // Find an import decl to add to. 580 var ( 581 bestMatch = -1 582 lastImport = -1 583 impDecl *ast.GenDecl 584 impIndex = -1 585 ) 586 for i, decl := range f.Decls { 587 gen, ok := decl.(*ast.GenDecl) 588 if ok && gen.Tok == token.IMPORT { 589 lastImport = i 590 // Do not add to import "C", to avoid disrupting the 591 // association with its doc comment, breaking cgo. 592 if declImports(gen, "C") { 593 continue 594 } 595 596 // Compute longest shared prefix with imports in this block. 597 for j, spec := range gen.Specs { 598 impspec := spec.(*ast.ImportSpec) 599 n := matchLen(importPath(impspec), ipath) 600 if n > bestMatch { 601 bestMatch = n 602 impDecl = gen 603 impIndex = j 604 } 605 } 606 } 607 } 608 609 // If no import decl found, add one after the last import. 610 if impDecl == nil { 611 impDecl = &ast.GenDecl{ 612 Tok: token.IMPORT, 613 } 614 f.Decls = append(f.Decls, nil) 615 copy(f.Decls[lastImport+2:], f.Decls[lastImport+1:]) 616 f.Decls[lastImport+1] = impDecl 617 } 618 619 // Ensure the import decl has parentheses, if needed. 620 if len(impDecl.Specs) > 0 && !impDecl.Lparen.IsValid() { 621 impDecl.Lparen = impDecl.Pos() 622 } 623 624 insertAt := impIndex + 1 625 if insertAt == 0 { 626 insertAt = len(impDecl.Specs) 627 } 628 impDecl.Specs = append(impDecl.Specs, nil) 629 copy(impDecl.Specs[insertAt+1:], impDecl.Specs[insertAt:]) 630 impDecl.Specs[insertAt] = newImport 631 if insertAt > 0 { 632 // Assign same position as the previous import, 633 // so that the sorter sees it as being in the same block. 634 prev := impDecl.Specs[insertAt-1] 635 newImport.Path.ValuePos = prev.Pos() 636 newImport.EndPos = prev.Pos() 637 } 638 639 f.Imports = append(f.Imports, newImport) 640 return true 641 } 642 643 // deleteImport deletes the import path from the file f, if present. 644 func deleteImport(f *ast.File, path string) (deleted bool) { 645 oldImport := importSpec(f, path) 646 647 // Find the import node that imports path, if any. 648 for i, decl := range f.Decls { 649 gen, ok := decl.(*ast.GenDecl) 650 if !ok || gen.Tok != token.IMPORT { 651 continue 652 } 653 for j, spec := range gen.Specs { 654 impspec := spec.(*ast.ImportSpec) 655 if oldImport != impspec { 656 continue 657 } 658 659 // We found an import spec that imports path. 660 // Delete it. 661 deleted = true 662 copy(gen.Specs[j:], gen.Specs[j+1:]) 663 gen.Specs = gen.Specs[:len(gen.Specs)-1] 664 665 // If this was the last import spec in this decl, 666 // delete the decl, too. 667 if len(gen.Specs) == 0 { 668 copy(f.Decls[i:], f.Decls[i+1:]) 669 f.Decls = f.Decls[:len(f.Decls)-1] 670 } else if len(gen.Specs) == 1 { 671 gen.Lparen = token.NoPos // drop parens 672 } 673 if j > 0 { 674 // We deleted an entry but now there will be 675 // a blank line-sized hole where the import was. 676 // Close the hole by making the previous 677 // import appear to "end" where this one did. 678 gen.Specs[j-1].(*ast.ImportSpec).EndPos = impspec.End() 679 } 680 break 681 } 682 } 683 684 // Delete it from f.Imports. 685 for i, imp := range f.Imports { 686 if imp == oldImport { 687 copy(f.Imports[i:], f.Imports[i+1:]) 688 f.Imports = f.Imports[:len(f.Imports)-1] 689 break 690 } 691 } 692 693 return 694 } 695 696 // rewriteImport rewrites any import of path oldPath to path newPath. 697 func rewriteImport(f *ast.File, oldPath, newPath string) (rewrote bool) { 698 for _, imp := range f.Imports { 699 if importPath(imp) == oldPath { 700 rewrote = true 701 // record old End, because the default is to compute 702 // it using the length of imp.Path.Value. 703 imp.EndPos = imp.End() 704 imp.Path.Value = strconv.Quote(newPath) 705 } 706 } 707 return 708 } 709 710 func usesImport(f *ast.File, path string) (used bool) { 711 spec := importSpec(f, path) 712 if spec == nil { 713 return 714 } 715 716 name := spec.Name.String() 717 switch name { 718 case "<nil>": 719 // If the package name is not explicitly specified, 720 // make an educated guess. This is not guaranteed to be correct. 721 lastSlash := strings.LastIndex(path, "/") 722 if lastSlash == -1 { 723 name = path 724 } else { 725 name = path[lastSlash+1:] 726 } 727 case "_", ".": 728 // Not sure if this import is used - err on the side of caution. 729 return true 730 } 731 732 walk(f, func(n interface{}) { 733 sel, ok := n.(*ast.SelectorExpr) 734 if ok && isTopName(sel.X, name) { 735 used = true 736 } 737 }) 738 739 return 740 } 741 742 func expr(s string) ast.Expr { 743 x, err := parser.ParseExpr(s) 744 if err != nil { 745 panic("parsing " + s + ": " + err.Error()) 746 } 747 // Remove position information to avoid spurious newlines. 748 killPos(reflect.ValueOf(x)) 749 return x 750 } 751 752 var posType = reflect.TypeOf(token.Pos(0)) 753 754 func killPos(v reflect.Value) { 755 switch v.Kind() { 756 case reflect.Ptr, reflect.Interface: 757 if !v.IsNil() { 758 killPos(v.Elem()) 759 } 760 case reflect.Slice: 761 n := v.Len() 762 for i := 0; i < n; i++ { 763 killPos(v.Index(i)) 764 } 765 case reflect.Struct: 766 n := v.NumField() 767 for i := 0; i < n; i++ { 768 f := v.Field(i) 769 if f.Type() == posType { 770 f.SetInt(0) 771 continue 772 } 773 killPos(f) 774 } 775 } 776 } 777 778 // A Rename describes a single renaming. 779 type rename struct { 780 OldImport string // only apply rename if this import is present 781 NewImport string // add this import during rewrite 782 Old string // old name: p.T or *p.T 783 New string // new name: p.T or *p.T 784 } 785 786 func renameFix(tab []rename) func(*ast.File) bool { 787 return func(f *ast.File) bool { 788 return renameFixTab(f, tab) 789 } 790 } 791 792 func parseName(s string) (ptr bool, pkg, nam string) { 793 i := strings.Index(s, ".") 794 if i < 0 { 795 panic("parseName: invalid name " + s) 796 } 797 if strings.HasPrefix(s, "*") { 798 ptr = true 799 s = s[1:] 800 i-- 801 } 802 pkg = s[:i] 803 nam = s[i+1:] 804 return 805 } 806 807 func renameFixTab(f *ast.File, tab []rename) bool { 808 fixed := false 809 added := map[string]bool{} 810 check := map[string]bool{} 811 for _, t := range tab { 812 if !imports(f, t.OldImport) { 813 continue 814 } 815 optr, opkg, onam := parseName(t.Old) 816 walk(f, func(n interface{}) { 817 np, ok := n.(*ast.Expr) 818 if !ok { 819 return 820 } 821 x := *np 822 if optr { 823 p, ok := x.(*ast.StarExpr) 824 if !ok { 825 return 826 } 827 x = p.X 828 } 829 if !isPkgDot(x, opkg, onam) { 830 return 831 } 832 if t.NewImport != "" && !added[t.NewImport] { 833 addImport(f, t.NewImport) 834 added[t.NewImport] = true 835 } 836 *np = expr(t.New) 837 check[t.OldImport] = true 838 fixed = true 839 }) 840 } 841 842 for ipath := range check { 843 if !usesImport(f, ipath) { 844 deleteImport(f, ipath) 845 } 846 } 847 return fixed 848 }