summaryrefslogtreecommitdiff
path: root/src/cmd/fix/httputil.go
blob: 86c42e1602d06eca98fb31664bd9014b16f83291 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
// Copyright 2011 The Go Authors.  All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package main

import "go/ast"

func init() {
	register(httputilFix)
}

var httputilFix = fix{
	"httputil",
	"2011-11-18",
	httputil,
	`Move some functions in http package into httputil package.

http://codereview.appspot.com/5336049
`,
}

var httputilFuncs = []string{
	"DumpRequest",
	"DumpRequestOut",
	"DumpResponse",
	"NewChunkedReader",
	"NewChunkedWriter",
	"NewClientConn",
	"NewProxyClientConn",
	"NewServerConn",
	"NewSingleHostReverseProxy",
}

func httputil(f *ast.File) bool {
	if imports(f, "net/http/httputil") {
		return false
	}

	fixed := false

	walk(f, func(n interface{}) {
		// Rename package name.
		if expr, ok := n.(ast.Expr); ok {
			for _, s := range httputilFuncs {
				if isPkgDot(expr, "http", s) {
					if !fixed {
						addImport(f, "net/http/httputil")
						fixed = true
					}
					expr.(*ast.SelectorExpr).X.(*ast.Ident).Name = "httputil"
				}
			}
		}
	})

	// Remove the net/http import if no longer needed.
	if fixed && !usesImport(f, "net/http") {
		deleteImport(f, "net/http")
	}

	return fixed
}